at_coder_friends 0.3.1

data/lib/at_coder_friends/judge_test_runner.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  # run test cases for the specified program with actual input/output.
+  class JudgeTestRunner < TestRunner
+    include PathUtil
+
+    def initialize(path)
+      super(path)
+      @cases_dir = cases_dir(@dir)
+      @smp_dir = smp_dir(@dir)
+    end
+
+    def judge_all
+      puts "***** judge_all #{@prg} *****"
+      Dir["#{@cases_dir}/#{@q}/in/*.txt"].sort.each do |infile|
+        id = File.basename(infile, '.txt')
+        judge(id)
+      end
+    end
+
+    def judge_one(id)
+      puts "***** judge_one #{@prg} *****"
+      judge(id)
+    end
+
+    def judge(id)
+      infile = "#{@cases_dir}/#{@q}/in/#{id}.txt"
+      outfile = "#{@smp_dir}/#{@q}_#{id}.out"
+      expfile = "#{@cases_dir}/#{@q}/out/#{id}.txt"
+      run_test(id, infile, outfile, expfile)
+    end
+  end
+end

data/lib/at_coder_friends/path_util.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  # Common methods and behaviors for dealing with paths.
+  module PathUtil
+    module_function
+
+    SMP_DIR = 'data'
+    CASES_DIR = 'cases'
+
+    def contest_name(path)
+      path = File.expand_path(path)
+      dir = File.file?(path) ? File.dirname(path) : path
+      File.basename(dir).delete('#').downcase
+    end
+
+    def split_prg_path(path)
+      path = File.expand_path(path)
+      dir, prg = File.split(path)
+      base, ext = prg.split('.')
+      q = base.split('_')[0]
+      [path, dir, prg, base, ext, q]
+    end
+
+    def smp_dir(dir)
+      File.join(dir, SMP_DIR)
+    end
+
+    def cases_dir(dir)
+      File.join(dir, CASES_DIR)
+    end
+  end
+end

data/lib/at_coder_friends/problem.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  DataSample = Struct.new(:no, :ext, :txt) do
+    def initialize(no, ext, txt)
+      no = no.tr('０-９', '0-9').to_i
+      txt = txt.lstrip.gsub("\r\n", "\n")
+      super(no, ext, txt)
+    end
+  end
+
+  InputDef = Struct.new(:container, :item, :names, :size) do
+    def initialize(container, item, names, size = [])
+      super(container, item, names, size)
+    end
+  end
+
+  SourceSample = Struct.new(:ext, :txt)
+
+  # holds problem information
+  class Problem
+    attr_reader :q, :fmt, :smps, :srcs
+    attr_accessor :html, :desc, :defs
+
+    def initialize(q)
+      @q = q
+      @html = ''
+      @desc = ''
+      @fmt = ''
+      @smps = []
+      @defs = []
+      @srcs = []
+      yield self if block_given?
+    end
+
+    def fmt=(f)
+      @fmt = f.lstrip.gsub("\r\n", "\n")
+    end
+
+    def add_smp(no, ext, txt)
+      @smps << DataSample.new(no, ext, txt)
+    end
+
+    def add_src(ext, txt)
+      @srcs << SourceSample.new(ext, txt)
+    end
+  end
+end

data/lib/at_coder_friends/ruby_generator.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  # generates C++ source code from definition
+  class RubyGenerator
+    TEMPLATE = <<~TEXT
+      ### DCLS ###
+
+      puts ans
+    TEXT
+
+    def process(pbm)
+      src = generate(pbm.defs)
+      pbm.add_src(:rb, src)
+    end
+
+    def generate(defs)
+      dcls = gen_decls(defs).join("\n")
+      TEMPLATE.sub('### DCLS ###', dcls)
+    end
+
+    def gen_decls(defs)
+      defs.map { |inpdef| gen_decl(inpdef) }.flatten
+    end
+
+    def gen_decl(inpdef)
+      case inpdef.container
+      when :single
+        gen_single_decl(inpdef)
+      when :harray
+        gen_harray_decl(inpdef)
+      when :varray
+        if inpdef.names.size == 1
+          gen_varray_1_decl(inpdef)
+        else
+          gen_varray_n_decl(inpdef)
+        end
+      when :matrix
+        gen_matrix_decl(inpdef)
+      end
+    end
+
+    def gen_single_decl(inpdef)
+      names = inpdef.names
+      dcl = names.join(', ')
+      expr = gen_expr(inpdef.item, names.size > 1)
+      "#{dcl} = #{expr}"
+    end
+
+    def gen_harray_decl(inpdef)
+      v = inpdef.names[0]
+      dcl = "#{v}s"
+      expr = gen_expr(inpdef.item, true)
+      "#{dcl} = #{expr}"
+    end
+
+    def gen_varray_1_decl(inpdef)
+      v = inpdef.names[0]
+      sz = inpdef.size[0]
+      dcl = "#{v}s"
+      expr = gen_expr(inpdef.item, false)
+      "#{dcl} = Array.new(#{sz}) { #{expr} }"
+    end
+
+    def gen_varray_n_decl(inpdef)
+      names = inpdef.names
+      sz = inpdef.size[0]
+      dcl = names.map { |v| "#{v}s[i]" }.join(', ')
+      expr = gen_expr(inpdef.item, true)
+      ret = []
+      ret += names.map { |v| "#{v}s = Array.new(#{sz})" }
+      ret << "#{sz}.times do |i|"
+      ret << "  #{dcl} = #{expr}"
+      ret << 'end'
+      ret
+    end
+
+    def gen_matrix_decl(inpdef)
+      v = inpdef.names[0]
+      sz = inpdef.size[0]
+      decl = "#{v}ss"
+      expr = gen_expr(inpdef.item, true)
+      "#{decl} = Array.new(#{sz}) { #{expr} }"
+    end
+
+    def gen_expr(item, split)
+      case item
+      when :number
+        split ? 'gets.split.map(&:to_i)' : 'gets.to_i'
+      when :string
+        split ? 'gets.chomp.split' : 'gets.chomp'
+      when :char
+        'gets.chomp'
+      end
+    end
+  end
+end

data/lib/at_coder_friends/sample_test_runner.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  # run test cases for the specified program with sample input/output.
+  class SampleTestRunner < TestRunner
+    include PathUtil
+
+    def initialize(path)
+      super(path)
+      @smp_dir = smp_dir(@dir)
+    end
+
+    def test_all
+      puts "***** test_all #{@prg} *****"
+      1.upto(999) do |i|
+        break unless test(i)
+      end
+    end
+
+    def test_one(n)
+      puts "***** test_one #{@prg} *****"
+      test(n)
+    end
+
+    def test(n)
+      id = format('%<q>s_%<n>03d', q: @q, n: n)
+      files = %w[in out exp].map { |ext| "#{@smp_dir}/#{id}.#{ext}" }
+      run_test(id, *files)
+    end
+  end
+end

data/lib/at_coder_friends/scraping_agent.rb

@@ -0,0 +1,135 @@
+# frozen_string_literal: true
+
+require 'uri'
+require 'mechanize'
+require 'logger'
+require 'English'
+
+module AtCoderFriends
+  # scrapes AtCoder contest site and
+  # - fetches problems
+  # - submits sources
+  class ScrapingAgent
+    include PathUtil
+
+    BASE_URL = 'https://atcoder.jp/'
+    XPATH_SECTION = '//h3[.="%<title>s"]/following-sibling::section'
+    LANG_TBL = {
+      'cxx'  => '3003',
+      'cs'   => '3006',
+      'java' => '3016',
+      'rb'   => '3024'
+    }.freeze
+
+    attr_reader :contest, :config, :agent
+
+    def initialize(contest, config)
+      @contest = contest
+      @config = config
+      @agent = Mechanize.new
+      # @agent.log = Logger.new(STDERR)
+    end
+
+    def common_url(path)
+      File.join(BASE_URL, path)
+    end
+
+    def contest_url(path)
+      File.join(BASE_URL, 'contests', contest, path)
+    end
+
+    def fetch_all
+      puts "***** fetch_all #{@contest} *****"
+      login
+      fetch_assignments.map do |q, url|
+        pbm = fetch_problem(q, url)
+        yield pbm if block_given?
+        pbm
+      end
+    end
+
+    def submit(path)
+      path, _dir, prg, _base, ext, q = split_prg_path(path)
+      puts "***** submit #{prg} *****"
+      src = File.read(path, encoding: Encoding::UTF_8)
+      login
+      post_src(q, ext, src)
+    end
+
+    def login
+      sleep 0.1
+      page = agent.get(common_url('login'))
+      form = page.forms[1]
+      form.field_with(name: 'username').value = config['user']
+      form.field_with(name: 'password').value = config['password']
+      sleep 0.1
+      form.submit
+    end
+
+    def fetch_assignments
+      url = contest_url('tasks')
+      puts "fetch list from #{url} ..."
+      sleep 0.1
+      page = agent.get(url)
+      ('A'..'Z').each_with_object({}) do |q, h|
+        link = page.link_with(text: q)
+        link && h[q] = link.href
+      end
+    end
+
+    def fetch_problem(q, url)
+      puts "fetch problem from #{url} ..."
+      sleep 0.1
+      page = agent.get(url)
+      Problem.new(q) do |pbm|
+        pbm.html = page.body
+        if contest == 'arc001'
+          page.search('//h3').each do |h3|
+            query = format(XPATH_SECTION, title: h3.content)
+            sections = page.search(query)
+            sections[0] && parse_section(pbm, h3, sections[0])
+          end
+        else
+          page.search('//*[./h3]').each do |section|
+            h3 = section.search('h3')[0]
+            parse_section(pbm, h3, section)
+          end
+        end
+      end
+    end
+
+    def parse_section(pbm, h3, section)
+      title = h3.content.strip
+      title.delete!("\u008f\u0090") # agc002
+      text = section.content
+      code = section.search('pre')[0]&.content || ''
+      case title
+      when /^制約$/
+        pbm.desc += text
+      when /^入出?力$/
+        pbm.desc += text
+        pbm.fmt = code
+      when /^入力例\s*(?<no>[\d０-９]+)$/
+        pbm.add_smp($LAST_MATCH_INFO[:no], :in, code)
+      when /^出力例\s*(?<no>[\d０-９]+)$/
+        pbm.add_smp($LAST_MATCH_INFO[:no], :exp, code)
+      end
+    end
+
+    def post_src(q, ext, src)
+      lang_id = LANG_TBL[ext.downcase]
+      raise AppError, ".#{ext} is not available." unless lang_id
+      sleep 0.1
+      page = agent.get(contest_url('submit'))
+      form = page.forms[1]
+      form.field_with(name: 'data.TaskScreenName') do |sel|
+        option = sel.options.find { |op| op.text.start_with?(q) }
+        option&.select || (raise AppError, "unknown problem:#{q}.")
+      end
+      form.add_field!('data.LanguageId', lang_id)
+      form.field_with(name: 'sourceCode').value = src
+      sleep 0.1
+      form.submit
+    end
+  end
+end

data/lib/at_coder_friends/test_runner.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+require 'rbconfig'
+
+module AtCoderFriends
+  # run tests for the specified program.
+  class TestRunner
+    include PathUtil
+
+    def initialize(path)
+      @path, @dir, @prg, @base, @ext, @q = split_prg_path(path)
+    end
+
+    # rubocop:disable Metrics/MethodLength
+    def run_test(id, infile, outfile, expfile)
+      return false unless File.exist?(infile) && File.exist?(expfile)
+
+      puts "==== #{id} ===="
+      ec = system("#{edit_cmd} < #{infile} > #{outfile}")
+
+      input, result, expected =
+        [infile, outfile, expfile].map { |file| File.read(file) }
+      puts '-- input --'
+      print input
+      puts '-- expected --'
+      print expected
+      puts '-- result --'
+      print result
+      if !ec
+        puts '!!!!! RE !!!!!'
+      elsif result != expected
+        puts '!!!!! WA !!!!!'
+      else
+        puts '<< OK >>'
+      end
+      true
+    end
+    # rubocop:enable Metrics/MethodLength
+
+    # rubocop:disable Metrics/MethodLength
+    def edit_cmd
+      case @ext
+      when 'java'
+        "java -cp #{@dir} Main"
+      when 'rb'
+        "ruby #{@dir}/#{@base}.rb"
+      when 'cs'
+        case which_os
+        when :windows
+          "#{@dir}/#{@base}.exe"
+        else
+          "mono #{@dir}/#{@base}.exe"
+        end
+      else # c, cxx
+        case which_os
+        when :windows
+          "#{@dir}/#{@base}.exe"
+        else
+          "#{@dir}/#{@base}"
+        end
+      end
+    end
+    # rubocop:enable Metrics/MethodLength
+
+    def which_os
+      @os ||= begin
+        case RbConfig::CONFIG['host_os']
+        when /mswin|msys|mingw|cygwin|bccwin|wince|emc/
+          :windows
+        when /darwin|mac os/
+          :macosx
+        when /linux/
+          :linux
+        when /solaris|bsd/
+          :unix
+        else
+          :unknown
+        end
+      end
+    end
+  end
+end