at_coder_friends 0.5.2 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c37cfc3181ae4abcc3d837e5a312987af2e9758e
-  data.tar.gz: b6573770d47b1d73cf94b4a048cb9b7cbcd49b26
+  metadata.gz: cfcd8c547d0916bc1db5c76b5c698e410ddb3541
+  data.tar.gz: 03e06ac13b32c2d55166cf1cefa83ad9cfbe3fa2
 SHA512:
-  metadata.gz: 5a21187eab2ff6217744414daa64251e8b1602bbc5e2d8c5e05d6f9b1dff6f9cfd209a5d8136900e9ab65f03289d3338743e01f7eb4043b94908956b09b4863b
-  data.tar.gz: a3557e02966c8189d06c1133cbc519cb1a53dd31e385d433eba628232d3557ae1c1f9e96d2edcbf406c92954e8c358adb3e4ee9c2e4d1c4a041a87b52da95e58
+  metadata.gz: d73018f888e13b7a37a19b8945b879513caf4952ff5bdcbe9412c413a84d05c4c055fb00166e97fe3b5420637b0c87c767796f9c2835e82c879451f854b2fc9d
+  data.tar.gz: 602aa37bf5f315a9bccfbd3e676f19b0651c4687fbe351e65d670359c956b74e230ebcedade158928cd1486092947af1bfb34c1504632b7389d1029da2afb70a

data/.rubocop.yml

@@ -2,6 +2,7 @@ inherit_from: .rubocop_todo.yml
 
 AllCops:
   Exclude:
+    - 'templates/**/*'
     - 'spec/fixtures/**/*'
     - 'regression/**/*'
   TargetRubyVersion: 2.3

data/.rubocop_todo.yml

@@ -2,10 +2,7 @@ Metrics/AbcSize:
   Max: 20
 
 Metrics/ClassLength:
-  Max: 135
-
-Metrics/ModuleLength:
-  Max: 120
+  Max: 125
 
 Metrics/MethodLength:
   Max: 15

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    at_coder_friends (0.5.2)
+    at_coder_friends (0.6.0)
       colorize (~> 0.8.1)
       launchy (~> 2.4.3)
       mechanize (~> 2.0)

data/README.md

@@ -87,7 +87,7 @@ See [CONFIGURATION.md](docs/CONFIGURATION.md) for details.
 ## Notes
 
 - Compilation is not supported.
-- Source generator supports only ruby and C++.
+- Source generator supports only Ruby and C++ in default.
 
 ## For Sublime Text user
 

data/config/default.yml

@@ -1,5 +1,8 @@
 user: ~
 password: ~
+generators:
+  - RubyBuiltin
+  - CxxBuiltin
 ext_settings:
   'awk':
     submit_lang: '3506' # Awk (mawk)

data/docs/CONFIGURATION.md

@@ -20,9 +20,14 @@ Its contents are merged with the default settings.
 ```.at_coder_friends.yml```で新しい設定項目の追加や、既存の設定項目の変更をすると、  
 その内容がデフォルト設定にマージされます。
 
+[/config/default.yml](/config/default.yml)
+
 ```YAML
 user: ~
 password: ~
+generators:
+  - RubyBuiltin
+  - CxxBuiltin
 ext_settings:
   'awk':
     submit_lang: '3506' # Awk (mawk)
@@ -115,15 +120,43 @@ ext_settings:
 
 - user  
   AtCoder username  
-  If omitted, you are prompted to enter it on the first run.  
+  If omitted, you are prompted to enter it on the first login.  
   AtCoderのユーザ名  
-  省略した場合、初回の実行時に入力を求められます
+  省略した場合、初回のログイン時に入力を求められます
 
 - password  
   AtCoder password  
-  If omitted, you are prompted to enter it on the first run.  
+  If omitted, you are prompted to enter it on the first login.  
   AtCoderのパスワード  
-  省略した場合、初回の実行時に入力を求められます
+  省略した場合、初回のログイン時に入力を求められます
+
+- generators  
+  List of source generator class names  
+  In default, ```RubyBuiltin``` and ```CxxBuiltin``` are available.  
+  For those other than above, corresponding plugin will be used  
+  if it has been installed (?).  
+  ソーススケルトンを生成するジェネレータのクラス名（リスト形式）  
+  既定の状態では「RubyBuiltin」と「CxxBuiltin」が利用でき、  
+  上記以外については、対応するプラグインがインストールされていれば  
+  それが使われます（？）  
+
+  For example, if ```RubyAlt``` is specified as generator name,  
+  following plugin will be used:
+  
+    |                 |                                                   |
+    |-----------------|---------------------------------------------------|
+    |Gem Name         |```at_coder_friends-generator-ruby_alt```          |
+    |Require Statement|```require 'at_coder_friends/generator/ruby_alt'```|
+    |Main Class Name  |```AtCoderFriends::Generator::RubyAlt```           |
+
+- generator_settings
+  - _(generator name)_  
+    Settings for each generator  
+    For details, see the manual of each generator.  
+    [Settings for built-in generators](#builtin-generator-settings)  
+    ジェネレータ毎の設定  
+    詳細は各ジェネレータのマニュアルを参照してください  
+    [組込ジェネレータの設定](#builtin-generator-settings)
 
 - ext_settings  
   Extension specific settings  
@@ -159,10 +192,23 @@ ext_settings:
         Execution command for Linux(optional)  
         Linux専用実行コマンド(あれば設定)
 
+<a id="builtin-generator-settings"></a>
+### Settings for RubyBuiltin generator
+| Setting | Description | Default |
+|---------|-------------|---------|
+|default_template|Source template file path|[/templates/ruby_builtin_default.rb](/templates/ruby_builtin_default.rb)|
+|interactive_template|Source template file path for interactive problems|[/templates/ruby_builtin_interactive.rb](/templates/ruby_builtin_interactive.rb)|
+
+### Settings for CxxBuiltin generator
+| Setting | Description | Default |
+|---------|-------------|---------|
+|default_template|Source template file path|[/templates/cxx_builtin_default.cxx](/templates/cxx_builtin_default.cxx)|
+|interactive_template|Source template file path for interactive problems|[/templates/cxx_builtin_interactive.cxx](/templates/cxx_builtin_interactive.cxx)|
+
 ### Language ID list (2019/09/16)
 
 | Language ID | Description |
-|---|---|
+|-------------|-------------|
 |3003|C++14 (GCC 5.4.1)|
 |3001|Bash (GNU bash v4.3.11)|
 |3002|C (GCC 5.4.1)|
@@ -229,13 +275,32 @@ during execution.
 test_cmd 文字列中の以下の変数には、  
 実行時に対象ファイルのパス情報が展開されます。
 
-| Variable | Descripton |
-|---|---|
-|{dir}|the directory name of the file<br>ファイルの置かれているディレクトリ名|
-|{base}|the file name excluding the extension<br>拡張子を除いたファイル名|
+| Variable | Descripton                                                     |
+|----------|----------------------------------------------------------------|
+|{dir}     |the directory name of the file<br>ファイルの置かれているディレクトリ名|
+|{base}    |the file name excluding the extension<br>拡張子を除いたファイル名   |
 
 ## Examples of overwriting settings
 
+- Use only Ruby source generator  
+  Rubyのソースジェネレータのみ使用
+  ```YAML
+  generators:
+    - RubyBuiltin
+  ```
+- Do not use source generator  
+  ソースジェネレータを使用しない
+  ```YAML
+  generators: ~
+  ```
+- change C++ source template  
+  C++のソース雛型を変更
+  ```YAML
+  generator_settings:
+    CxxBuiltin:
+      default_template: /path/to/template
+  ```
+
 - Test .py with local Python  
   .pyをローカル環境のPythonでテスト
   ```YAML

data/lib/at_coder_friends.rb

@@ -15,12 +15,17 @@ require 'at_coder_friends/scraping/custom_test'
 require 'at_coder_friends/scraping/submission'
 require 'at_coder_friends/scraping/tasks'
 require 'at_coder_friends/scraping/agent'
-require 'at_coder_friends/parser/page_parser'
-require 'at_coder_friends/parser/format_parser'
-require 'at_coder_friends/parser/constraints_parser'
+require 'at_coder_friends/parser/section_wrapper'
+require 'at_coder_friends/parser/sections'
+require 'at_coder_friends/parser/sample_data'
+require 'at_coder_friends/parser/input_format'
+require 'at_coder_friends/parser/constraints'
+require 'at_coder_friends/parser/interactive'
+require 'at_coder_friends/parser/binary'
 require 'at_coder_friends/parser/main'
-require 'at_coder_friends/ruby_generator'
-require 'at_coder_friends/cxx_generator'
+require 'at_coder_friends/generator/cxx_builtin'
+require 'at_coder_friends/generator/ruby_builtin'
+require 'at_coder_friends/generator/main'
 require 'at_coder_friends/emitter'
 require 'at_coder_friends/context'
 require 'at_coder_friends/cli'

data/lib/at_coder_friends/cli.rb

@@ -92,17 +92,14 @@ module AtCoderFriends
       raise AppError, "#{path} is not empty." \
         if Dir.exist?(path) && !Dir["#{path}/*"].empty?
 
-      rb_gen = RubyGenerator.new
-      cxx_gen = CxxGenerator.new
       ctx.scraping_agent.fetch_all do |pbm|
         Parser::Main.process(pbm)
-        rb_gen.process(pbm)
-        cxx_gen.process(pbm)
+        ctx.generator.process(pbm)
         ctx.emitter.emit(pbm)
       end
     end
 
-    def test_one(id = 1)
+    def test_one(id = '001')
       ctx.sample_test_runner.test_one(id)
     end
 

data/lib/at_coder_friends/context.rb

@@ -22,6 +22,10 @@ module AtCoderFriends
       @scraping_agent ||= Scraping::Agent.new(self)
     end
 
+    def generator
+      @generator ||= Generator::Main.new(self)
+    end
+
     def sample_test_runner
       @sample_test_runner ||= TestRunner::Sample.new(self)
     end

data/lib/at_coder_friends/emitter.rb

@@ -14,8 +14,8 @@ module AtCoderFriends
     end
 
     def emit(pbm)
-      pbm.smps.each { |smp| emit_sample(pbm, smp) }
-      pbm.srcs.each { |src| emit_source(pbm, src) }
+      pbm.samples.each { |smp| emit_sample(pbm, smp) }
+      pbm.sources.each { |src| emit_source(pbm, src) }
     end
 
     def emit_sample(pbm, smp)

data/lib/at_coder_friends/generator/cxx_builtin.rb

@@ -0,0 +1,191 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  module Generator
+    module CxxBuiltinConstants
+      ACF_HOME = File.realpath(File.join(__dir__, '..', '..', '..'))
+      TMPL_DIR = File.join(ACF_HOME, 'templates')
+      DEFAULT_TMPL = File.join(TMPL_DIR, 'cxx_builtin_default.cxx')
+      INTERACTIVE_TMPL = File.join(TMPL_DIR, 'cxx_builtin_interactive.cxx')
+      SCANF_FMTS = [
+        'scanf("%<fmt>s", %<addr>s);',
+        'REP(i, %<sz1>s) scanf("%<fmt>s", %<addr>s);',
+        'REP(i, %<sz1>s) REP(j, %<sz2>s) scanf("%<fmt>s", %<addr>s);'
+      ].freeze
+      FMT_FMTS = { number: '%d', string: '%s', char: '%s' }.freeze
+      ADDR_FMTS = {
+        single: {
+          number: '&%<v>s',
+          string: '%<v>s'
+        },
+        harray: {
+          number: '%<v>s + i',
+          string: '%<v>s[i]',
+          char: '%<v>s'
+        },
+        varray: {
+          number: '%<v>s + i',
+          string: '%<v>s[i]'
+        },
+        matrix: {
+          number: '&%<v>s[i][j]',
+          string: '%<v>s[i][j]',
+          char: '%<v>s[i]'
+        }
+      }.freeze
+      DEFAULT_OUTPUT = <<~TEXT
+        int ans = 0;
+        printf("%d\\n", ans);
+      TEXT
+      BINARY_OUTPUT_FMT = <<~TEXT
+        bool cond = false;
+        puts(cond ? "%s" : "%s");
+      TEXT
+    end
+
+    # generates C++ source code from problem description
+    class CxxBuiltin
+      include CxxBuiltinConstants
+
+      attr_reader :cfg, :pbm
+
+      def initialize(cfg = nil)
+        @cfg = cfg || {}
+      end
+
+      def process(pbm)
+        src = generate(pbm)
+        pbm.add_src(:cxx, src)
+      end
+
+      def generate(pbm)
+        @pbm = pbm
+        src = File.read(select_template)
+        src = embed_lines(src, '/*** URL ***/', [pbm.url])
+        src = embed_lines(src, '/*** CONSTS ***/', gen_consts)
+        src = embed_lines(src, '/*** DCLS ***/', gen_decls)
+        src = embed_lines(src, '/*** INPUTS ***/', gen_inputs)
+        embed_lines(src, '/*** OUTPUT ***/', gen_output.split("\n"))
+      end
+
+      def embed_lines(src, pat, lines)
+        re = Regexp.escape(pat)
+        src.gsub(
+          /^(.*)#{re}(.*)$/,
+          lines.map { |s| '\1' + s + '\2' }.join("\n")
+        )
+      end
+
+      def select_template(interactive = pbm.options.interactive)
+        interactive ? interactive_template : default_template
+      end
+
+      def default_template
+        cfg['default_template'] || DEFAULT_TMPL
+      end
+
+      def interactive_template
+        cfg['interactive_template'] || INTERACTIVE_TMPL
+      end
+
+      def gen_consts(constraints = pbm.constraints)
+        constraints
+          .select { |c| c.type == :max }
+          .map { |c| "const int #{c.name.upcase}_MAX = #{c.value};" }
+      end
+
+      def gen_decls(inpdefs = pbm.formats)
+        inpdefs.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
+          gen_varray_decl(inpdef)
+        when :matrix
+          gen_matrix_decl(inpdef)
+        end
+      end
+
+      def gen_single_decl(inpdef)
+        names = inpdef.names
+        case inpdef.item
+        when :number
+          dcl = names.join(', ')
+          "int #{dcl};"
+        when :string
+          names.map { |v| "char #{v}[#{v.upcase}_MAX + 1];" }
+        end
+      end
+
+      def gen_harray_decl(inpdef)
+        v = inpdef.names[0]
+        sz = gen_arr_size(inpdef.size)[0]
+        case inpdef.item
+        when :number
+          "int #{v}[#{sz}];"
+        when :string
+          "char #{v}[#{sz}][#{v.upcase}_MAX + 1];"
+        when :char
+          "char #{v}[#{sz} + 1];"
+        end
+      end
+
+      def gen_varray_decl(inpdef)
+        names = inpdef.names
+        sz = gen_arr_size(inpdef.size)[0]
+        case inpdef.item
+        when :number
+          names.map { |v| "int #{v}[#{sz}];" }
+        when :string
+          names.map { |v| "char #{v}[#{sz}][#{v.upcase}_MAX + 1];" }
+        end
+      end
+
+      def gen_matrix_decl(inpdef)
+        v = inpdef.names[0]
+        sz1, sz2 = gen_arr_size(inpdef.size)
+        case inpdef.item
+        when :number
+          "int #{v}[#{sz1}][#{sz2}];"
+        when :string
+          "char #{v}[#{sz1}][#{sz2}][#{v.upcase}_MAX + 1];"
+        when :char
+          "char #{v}[#{sz1}][#{sz2} + 1];"
+        end
+      end
+
+      def gen_arr_size(szs)
+        szs.map { |sz| sz =~ /\D/ ? "#{sz.upcase}_MAX" : sz }
+      end
+
+      def gen_inputs(inpdefs = pbm.formats)
+        inpdefs.map { |inpdef| gen_input(inpdef) }.flatten
+      end
+
+      # rubocop:disable Metrics/AbcSize
+      def gen_input(inpdef)
+        dim = inpdef.size.size - (inpdef.item == :char ? 1 : 0)
+        scanf = SCANF_FMTS[dim]
+        sz1, sz2 = inpdef.size
+        fmt = FMT_FMTS[inpdef.item] * inpdef.names.size
+        addr_fmt = ADDR_FMTS[inpdef.container][inpdef.item]
+        addr = inpdef.names.map { |v| format(addr_fmt, v: v) }.join(', ')
+        format(scanf, sz1: sz1, sz2: sz2, fmt: fmt, addr: addr)
+      end
+      # rubocop:enable Metrics/AbcSize
+
+      def gen_output(vs = pbm.options.binary_values)
+        if vs
+          format(BINARY_OUTPUT_FMT, *vs)
+        else
+          DEFAULT_OUTPUT
+        end
+      end
+    end
+  end
+end

data/lib/at_coder_friends/generator/main.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module AtCoderFriends
+  module Generator
+    # entry point of source generation
+    class Main
+      attr_reader :ctx
+
+      def initialize(ctx)
+        @ctx = ctx
+        @cache = {}
+      end
+
+      def process(pbm)
+        generators = ctx.config.dig('generators') || []
+        generators.each do |gen_name|
+          gen_obj = load_obj(gen_name)
+          gen_obj.process(pbm)
+        end
+      end
+
+      def load_obj(gen_name)
+        @cache[gen_name] ||= begin
+          gen_class = load_class(gen_name)
+          gen_cnf = config_for(gen_name)
+          gen_class.new(gen_cnf)
+        end
+      end
+
+      def load_class(gen_name)
+        unless AtCoderFriends::Generator.const_defined?(gen_name)
+          require "at_coder_friends/generator/#{to_snake(gen_name)}"
+        end
+        AtCoderFriends::Generator.const_get(gen_name)
+      rescue LoadError
+        raise AppError, "plugin load error : generator #{gen_name} not found."
+      end
+
+      def config_for(gen_name)
+        ctx.config.dig('generator_settings', gen_name) || {}
+      end
+
+      def to_snake(str)
+        str
+          .gsub(/::/, '/')
+          .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
+          .gsub(/([a-z\d])([A-Z])/, '\1_\2')
+          .tr('-', '_')
+          .downcase
+      end
+    end
+  end
+end