Contents

# frozen_string_literal: true

require 'fileutils'

module AtCoderFriends
  # emits source skeletons and sample input/output(s)
  # of a problem to the specified directory.
  class Emitter
    include PathUtil

    def initialize(ctx)
      @src_dir = ctx.path
      @smp_dir = smp_dir(@src_dir)
    end

    def emit(pbm)
      pbm.smps.each { |smp| emit_sample(pbm, smp) }
      pbm.srcs.each { |src| emit_source(pbm, src) }
    end

    def emit_sample(pbm, smp)
      makedirs_unless @smp_dir
      smp_file = format(
        '%<q>s_%<n>03d.%<ext>s', q: pbm.q, n: smp.no, ext: smp.ext
      )
      smp_path = File.join(@smp_dir, smp_file)
      File.write(smp_path, smp.txt)
      puts smp_file
    end

    def emit_source(pbm, src)
      makedirs_unless @src_dir
      src_file = format('%<q>s.%<ext>s', q: pbm.q, ext: src.ext)
      src_path = File.join(@src_dir, src_file)
      File.write(src_path, src.txt)
      puts src_file
    end

    def makedirs_unless(dir)
      FileUtils.makedirs(dir) unless Dir.exist?(dir)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
at_coder_friends-0.5.0	lib/at_coder_friends/emitter.rb