atcoder_tools 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 21578ce616e39fd778e17427eb0a1bee8a13fe4b2161331b83601b26b389cfdb
-  data.tar.gz: 0cb19e3f8c40624a6215ebe903ed64579ee4446655a2ef3d2eb6be384c95ac6b
+  metadata.gz: e5d0c7a03c8491024c16d6ae87c288f04954180fe6bd3e189db5e3d8118faebf
+  data.tar.gz: 4967f7572872f9e90d1a30753a6c9c2c810392a5a78ee4936580de2b2016fe02
 SHA512:
-  metadata.gz: b0c8ce53618f75d4eb7202d8439599fe1e0eaeb6b0d0fef6c9fe0c7a8824293bcc38a7954a4c139f15f3fa81cf590e3a45e32b369ab311843a190f1a0012e7f0
-  data.tar.gz: 06cc6d61fd01079fbc5795ff0d7f4f1f4b352b178033eb39354f7ebd8e53825c36bbef892b5015b9c3b068f413aaea5e0a41311597b04ac3f12e2fc9c8440053
+  metadata.gz: b36e09e331d51158b46712cfa49e585b18e1be0e72385463b49fcc4bb0cc86060ffed4968469e9be0fd9dc48d8e1f4bae559ab98e526a0938e9b343c47510282
+  data.tar.gz: d758c5a868d0df29e73e9be659f6499eb383301dfca75835ccdcd210e34634d7307df61e7a5ac525a5f560a119dff2ebefa18858871aab27f5ca999c8ba8ccec

data/.envrc

@@ -0,0 +1 @@
+ATCODER_TOOLS_ENV=development

data/.github/workflows/deploy.yml

@@ -0,0 +1,54 @@
+name: Publish Gem
+
+on:
+  push:
+    tags:
+      - v*
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v2
+    - uses: actions/setup-ruby@v1
+      with:
+        ruby-version: '2.7'
+
+    - name: Bundle install
+      run: bundle
+
+    - name: Build Gem
+      run: rake build
+    
+    - name: Publish gem
+      uses: dawidd6/action-publish-gem@v1
+      with:
+        api_key: ${{secrets.RUBYGEMS_API_KEY}}
+
+    # - name: mkdir gem
+    #   run: mkdir ~/.gem
+
+    # - name: Add api key
+    #   run: "echo :rubygems_api_key: $RUBYGEMS_API_KEY > ~/.gem/credentials"
+    #   env:
+    #     RUBYGEMS_API_KEY: ${{ secrets.RUBYGEMS_API_KEY }}
+
+    # - name: test
+    #   run: cat ~/.gem/credentials
+
+    # - name: test2
+    #   run: git status
+
+    # - name: test3
+    #   run: "echo $RUBYGEMS_API_KEY"
+    #   env:
+    #     RUBYGEMS_API_KEY: ${{ secrets.RUBYGEMS_API_KEY }}
+
+    # - name: set git username
+    #   run: git config user.name "Aitaro Chaya"
+      
+    # - name: set git email
+    #   run: git config user.email "aitaro.chaya@gmail.com"
+
+    # - name: Release Gem
+    #   run: rake release

data/.gitignore

@@ -9,3 +9,6 @@
 
 # rspec failure tracking
 .rspec_status
+test.rb
+
+Gemfile.lock

data/README.md

@@ -1,47 +1,51 @@
 # AtcoderTools
+Atcoderの競技環境を簡単にsetupするためのツールです。
 
-TODOTODOTODO
-
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/atcoder_tools`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
-
-## Installation
-
-Add this line to your application's Gemfile:
-
-```ruby
-gem 'atcoder_tools'
+## インストール
+```bash
+$ gem install atcoder_tools
 ```
 
-And then execute:
-
-    $ bundle install
+## 使い方
+### atcoderにログイン
+```
+$ atcoder_tools login
+```
 
-Or install it yourself as:
+### コンテスト用ファイルの作成
+```
+$ atcoder_tools create abc170
+```
 
-    $ gem install atcoder_tools
+### コンテスト用自動デバッグ
+```
+$ atcoder_tools start
+```
 
-## Usage
+### コンテストの提出
 ```
-$ atcoder_tools abc170
+$ atcoder_tools submit abc170
 ```
 
-## Development
+### コンテストの削除
+```
+$ atcoder_tools delete abc170
+```
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+### MODEについて
+- NONE: なにもおこらない
+- DEBUG: 標準入力で、testcaseが渡されたら状態で実行される。
+- RUN: 単純にファイルを実行。標準入力は自分で渡す。
+- TEST: rspecを用いたテスト自動実行。どれが通っていてどれが通っていないかわかる。
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+## Development
+本リポジトリをクローンし、path指定して、gemをインストールするとデバッグできる。
 
 ## Contributing
-
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/atcoder_tools. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/atcoder_tools/blob/master/CODE_OF_CONDUCT.md).
-
+issueやプルリク適当に投げてくれてかまいません。
 
 ## License
-
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
 
 ## Code of Conduct
-
 Everyone interacting in the AtcoderTools project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/atcoder_tools/blob/master/CODE_OF_CONDUCT.md).

data/atcoder_tools.gemspec

@@ -31,4 +31,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency "listen"
   spec.add_dependency "rspec"
   spec.add_dependency 'nokogiri'
+  spec.add_dependency 'mechanize'
+  spec.add_dependency 'tty-prompt'
 end

data/lib/atcoder_tools/atcoder.rb

@@ -0,0 +1,72 @@
+require_relative 'settings'
+require_relative 'log'
+
+class Atcoder
+  include Log
+
+  def initialize
+    @settings = Settings.new
+    @agent = Mechanize.new
+    @agent.user_agent_alias = 'Mac Mozilla'
+  end
+
+  def self.test_login(username, password)
+    atcoder = Atcoder.new
+    return atcoder.login(username=username, password=password)
+  end
+
+  def self.submit(task)
+    atcoder = Atcoder.new
+    atcoder.login
+    atcoder.submit(task)
+  end
+
+  # 引数はtaskクラス
+  def task_url(task)
+    "https://atcoder.jp/contests/#{task.contest.name}/tasks/#{task.contest.name}_#{task.name}"
+  end
+
+  # 引数はcontest, taskクラス
+  def submit(task)
+    response = nil
+    @agent.get(task_url(task)) do |page|
+      # 1つ目のフォームはログアウト用なので2つ目を使用   
+      form = page.forms.last
+      form.field_with(name: 'data.LanguageId').value = "4049"
+      form.field_with(name: 'sourceCode').value   = task.code
+      response = @agent.submit(form)
+    end
+
+    if response.code.to_i == 200
+      return true
+    end
+    return false
+  end
+
+  def login(username=nil, password=nil)
+
+    username ||= @settings.credentials.username
+    password ||= @settings.credentials.password
+
+    response = nil
+    log_debug "try login to atcoder"
+    @agent.get("https://atcoder.jp/login") do |page|
+      # 1つ目のフォームはログアウト用なので2つ目を使用
+      form = page.forms.last
+      form.field_with(name: 'username').value = username
+      form.field_with(name: 'password').value = password
+      response = @agent.submit(form)
+    end
+
+    # print(response.response)
+    res = response.body.include?("Welcome, #{username}.")
+    if res
+      log_debug "login secceed!"
+    else
+      log_warn "login falied!"
+    end
+
+    return res
+  end
+
+end

data/lib/atcoder_tools/cli.rb

@@ -1,50 +1,156 @@
-
 require "thor"
 require 'fileutils'
 require "listen"
+require 'mechanize'
 require_relative 'contest'
 require_relative 'task'
+require_relative 'settings'
+require_relative 'atcoder'
+require "tty-prompt"
+require_relative 'log'
+
 
 module AtcoderTools
   class CLI < Thor
-    desc 'create [contest name]', 'create contest workspace'
+    include Log
+    Prompt = TTY::Prompt.new
+    desc 'login', 'login to atcoder'
+    def login
+      # 認証情報を書き込むので、gitignoreを設定する
+      unless Dir.exist?('.atcoder')
+        Dir.mkdir('.atcoder')
+      end
+      File.open(".atcoder/.gitignore", mode = "w"){|f|
+        f.write("*")  # ファイルに書き込む
+      }
+      loop do
+        prompt = TTY::Prompt.new
+        username = prompt.ask("What is your atcoder username?", required: true)
+        password = prompt.mask("What is your atcoder password?", required: true)
+        puts ""
+        log_info 'starting test login.'
+
+        if Atcoder.test_login(username, password)
+          settings = Settings.new
+          settings.username = username
+          settings.password = password
+          settings.save!
+          log_info 'OK!'
+          break
+        else
+          log_error "Your username or password is invalid! try again!"
+        end
+      end
+    end
+
+    desc 'logout', 'logout from atcoder.'
+    def logout
+      settings = Settings.new
+      settings.destroy_credentials!
+      settings.save!
+    end
+
+    desc 'language [lang?]', '言語を選択できます。現在対応している言語は ruby, c++(gcc) です。'
+    def language(lang='')
+      unless ['ruby', 'c++(gcc)'].include?(lang)
+        lang = Prompt.select("choose your default language.", ['ruby', 'c++(gcc)'])
+      end
+
+      # コマンドが通っているかテスト
+      log_info "Checking path ..."
+      res = case lang
+            when 'ruby'
+              !!system('ruby -v')
+            when 'c++(gcc)'
+              !!system('c++ -v')
+            end
+
+      unless res
+        raise "Cannot find path of #{lang}. Is `$ #{lang} -v` working?"
+      end
+
+      settings = Settings.new
+      settings.language = lang
+      settings.save!
+      puts
+      log_info 'Default language successfully changed!'
+    end
+
+    desc 'create [contest name]', 'create contest workspace. contest name は actoderのurlより取得できます。(ex. abc173)'
     def create(contest_name)
       contest = Contest.new(contest_name)
+      contest.check_validity!
       contest.create!
   
-      puts 'successfully created'
+      log_info 'successfully created'
     end
 
     desc 'start', 'start run'
     def start
-      # system('bundle exec guard -d --guardfile .atcoder/Guardfile')
       listener = Listen.to('.', ignore: /.atcoder\/.*/) do |modified, added, removed|
-        # puts "modified absolute path: #{modified}"
-        # puts "added absolute path: #{added}"
-        # puts "removed absolute path: #{removed}"
         if modified[0]
           contest_name ,task_name = modified[0].split('/')[-2..-1]
-          task_name = task_name[..-4] # .rb抜き出し
-
-          puts("#{contest_name}/#{task_name}.rb was changed")
+          task_name = task_name.scan(/(.*)\..*/)[0][0]
 
           contest = Contest.new(contest_name)
           # thor と名前空間がかぶっているため
           task = ::Task.new(contest, task_name)
+          # 最後に動かしたcontest, taskを記憶
+          settings = Settings.new
+          settings.current_contest = contest
+          settings.current_task = task
+          settings.save!
+
           task.run
         end
       end
-      listener.start # not blocking
+      log_info 'started actoder_tools session.'
+      listener.start
       sleep
     end
 
-    desc 'submit [contest name]', 'submit'
-    def submit(contest_name, task_name)
-      puts("まだ実装してないよ！")
+    desc 'submit', 'ソースコードを提出します。提出する問題は対話形式で指定できます。'
+    def submit
+      prompt = TTY::Prompt.new
+      # contest_name = prompt.select("Choose your submit contest?", Contest.list.map(&:name))
+      contest_name = Prompt.select("Choose your submit contest?") do |submit|
+        Contest.list.each_with_index do |contest, i|
+          submit.choice contest.name
+          if Settings.new.current_contest&.name == contest.name
+            submit.default i + 1
+          end
+        end
+      end
+
+      # contestによっては6種類ない可能性がある。
+      task_list = ['a', 'b', 'c', 'd', 'e', 'f']
+      task_name = prompt.select("Choose your tesk?") do |submit|
+        task_list.each_with_index do |task, i|
+          submit.choice task
+          if Settings.new.current_task&.name == task
+            submit.default i + 1
+          end
+        end
+      end
+
+      contest = Contest.new(contest_name)
+      # thor と名前空間がかぶっているため
+      task = ::Task.new(contest, task_name)
+      Atcoder.submit(task)
     end
 
-    desc 'delete [contest name]', 'delete'
-    def delete(contest_name)
+    desc 'delete [contest_name?]', 'delete'
+    def delete(contest_name=nil)
+      unless contest_name
+        contest_name = Prompt.select("Choose your submit contest?") do |submit|
+          Contest.list.each_with_index do |contest, i|
+            submit.choice contest.name
+            if Settings.new.current_contest&.name == contest.name
+              submit.default i + 1
+            end
+          end
+        end
+      end
       FileUtils.rm_rf(contest_name)
       FileUtils.rm_rf(".atcoder/#{contest_name}")      
       puts 'successfully deleted'

data/lib/atcoder_tools/contest.rb

@@ -3,6 +3,7 @@ require 'fileutils'
 require 'nokogiri'
 require 'erb'
 require_relative 'task'
+require_relative 'settings'
 
 class Contest
   attr_reader :name
@@ -11,14 +12,39 @@ class Contest
     @name = name
   end
 
-  def url(task_name)
-    "https://atcoder.jp/contests/#{@name}/tasks/#{@name}_#{task_name}"
+  def check_validity!
+    # 実在するかcheck
+    begin
+      URI.open(url)
+    rescue OpenURI::HTTPError
+      raise 'contest name が存在しません。'
+    end
+  end
+
+  def url
+    "https://atcoder.jp/contests/#{@name}"
+  end
+
+  def submission_url
+    "https://atcoder.jp/contests/#{@name}/submissions/me"
   end
 
   def create!
     FileUtils.mkdir_p @name
     tasks = ['a','b','c','d','e','f'].map{|task_name| Task.new(self, task_name)}
     tasks.map(&:create!)
+    settings = Settings.new
+    settings.current_contest = @name
+    settings.contest_language = self, settings.language
+    settings.save!
+  end
+
+  def language
+    Settings.new.contest_language(self)
+  end
+
+  def self.list
+    Dir.glob("./*").select{|dir| FileTest.directory?(dir)}.map{|dir| Contest.new dir[2..-1] }
   end
 
 end

data/lib/atcoder_tools/log.rb

@@ -0,0 +1,25 @@
+require 'logger'
+module Log
+  LoggerIns = Logger.new(STDOUT)
+  if ENV['ATCODER_TOOLS_ENV'] == 'development'
+    LoggerIns.level = Logger::DEBUG
+  else
+    LoggerIns.level = Logger::INFO
+  end
+  
+  def log_debug(msg)
+    LoggerIns.debug(msg)
+  end
+ 
+  def log_info(msg)
+    LoggerIns.info(msg)
+  end
+
+  def log_warn(msg)
+    LoggerIns.warn(msg)
+  end
+
+  def log_error(msg)
+    LoggerIns.error(msg)
+  end
+end

data/lib/atcoder_tools/settings.rb

@@ -0,0 +1,100 @@
+require 'yaml'
+require 'base64'
+require 'fileutils'
+
+class Settings
+  Credentials = Struct.new(:username, :password)
+
+  def initialize
+    if File.exist?(".atcoder/settings.yml")
+      @settings = YAML.load_file(".atcoder/settings.yml")
+    else
+      @settings = {}
+    end
+  end
+
+  def current_contest
+    @settings['current_contest'] && Contest.new(@settings['current_contest'])
+  end
+
+  def ambiguous_contest(contest)
+    case contest
+    when String
+      Contest.new(contest)
+    when Contest
+      contest
+    else
+      raise ArgumentError, "引数は StringまたはContestです。"
+    end
+  end
+
+  def current_contest=(contest)
+    @settings['current_contest'] = ambiguous_contest(contest).name
+
+  end
+
+  def current_task
+    current_contest && @settings['current_task'] && Task.new(current_contest, @settings['current_task'])
+  end
+
+  def current_task=(task)
+    case task
+    when String
+      task_name = task
+    when Task
+      task_name = task.name
+    else
+      raise ArgumentError, "引数は StringまたはTaskです。"
+    end
+    @settings['current_task'] = task_name
+  end
+
+  def language
+    @settings['language'] || 'ruby'
+  end
+
+  def language=(lang)
+    @settings['language'] = lang
+  end
+
+  def contest_language(contest)
+    if @settings[ambiguous_contest(contest).name]
+      @settings[ambiguous_contest(contest).name]['language']
+    else
+      language
+    end
+  end
+
+  # contest, languageの順に代入
+  def contest_language=(data)
+    name = ambiguous_contest(data[0]).name
+    @settings[name] ||= {}
+    @settings[name]['language'] = data[1]
+  end
+
+  def password=(password)
+    @settings['password'] = Base64.strict_encode64(password)
+  end
+
+  def username=(username)
+    @settings['username'] = username
+  end
+
+  def destroy_credentials!
+    @settings.delete('username')
+    @settings.delete('password')
+  end
+
+  def credentials
+    unless @settings.key?('username') && @settings.key?('password')
+      raise "先にログインしてください！"
+    end
+    return Credentials.new(@settings['username'], Base64.decode64(@settings['password']))
+  end
+
+  def save!
+    @settings['updated_at'] = Time.now
+    FileUtils.mkdir_p '.atcoder'
+    open('.atcoder/settings.yml' , 'w') {|f| YAML.dump(@settings , f ) }
+  end
+end

data/lib/atcoder_tools/sources/spec.rb.erb

@@ -3,8 +3,10 @@ require 'rspec'
 RSpec.describe 'test' do
   <% @tests.each do |test| %>
   it "test-<%= test[:no] %>" do
-    io = IO.popen("ruby <%= task_file_path %>", "w+")
-    io.puts("<%= test[:input] %>")
+    io = IO.popen("<%= command %>", "w+")
+    <% test[:inputs].each do |input| %>
+    io.puts("<%= input %>")
+    <% end %>
     io.close_write
     expect(io.readlines.join).to eq("<%= test[:output] %>")
   end

data/lib/atcoder_tools/sources/task.cpp.erb

@@ -0,0 +1,20 @@
+// ------------------------------------------------------------
+// THIS IS METADATA FOR ATCODER_TOOLS
+// BE CAREFULL WHEN YOU CHANGE THE VALUE
+// 
+// METADATA::START<% inputs.each_with_index do |input, i| %>
+// test-<%=i+1%>
+// input: "<%= input.gsub(/\R/, '\\n') %>"
+// output: "<%= outputs[i].gsub(/\R/, '\\n') %>"<% end %>
+// MODE: TEST
+// METADATA::FINISH
+// 
+// you can add your original test case 
+// you can select MODE from NONE, DEBUG, RUN, TEST
+// problem_url: <%= url %>
+// submission_url: <%= contest.submission_url %>
+// -------------------------------------------------------------
+
+// snippets
+// n = gets.to_i
+// arr = gets.split(' ').map(&:to_i)

data/lib/atcoder_tools/sources/task.rb.erb

@@ -11,7 +11,8 @@
 # 
 # you can add your original test case 
 # you can select MODE from NONE, DEBUG, RUN, TEST
-# problem_url: <%=url%>
+# problem_url: <%= url %>
+# submission_url: <%= contest.submission_url %>
 # -------------------------------------------------------------
 
 # snippets

data/lib/atcoder_tools/task.rb

@@ -3,102 +3,154 @@ require 'fileutils'
 require 'nokogiri'
 require 'erb'
 
-# module Atcoder
-  class Task
-    def initialize(contest, name)
-      @contest = contest
-      @name = name
-      @tests = []
-      @test_outputs = []
-    end
+class Task
+  attr_reader :contest, :name
+
+  # contest引数はcontestクラス
+  def initialize(contest, name)
+    @contest = contest
+    @name = name
+    @tests = []
+    @test_outputs = []
+  end
+
+  def url
+    "https://atcoder.jp/contests/#{@contest.name}/tasks/#{@contest.name}_#{@name}"
+  end
 
-    def url
-      "https://atcoder.jp/contests/#{@contest.name}/tasks/#{@contest.name}_#{@name}"
+  def task_file_path
+    extension = nil
+    case @contest.language
+    when 'ruby'
+      extension = 'rb'
+    when 'c++(gcc)'
+      extension = 'cpp'
     end
+    "#{@contest.name}/#{@name}.#{extension}"
+  end
+
+  def compile_file_path
+    "#{@contest.name}/#{@name}"
+  end
 
-    def task_file_path
-      "#{@contest.name}/#{@name}.rb"
+  def test_file_path
+    ".atcoder/#{@contest.name}/#{@name}/spec.rb"
+  end
+
+  def template_task_file_path
+    extension = nil
+    case @contest.language
+    when 'ruby'
+      extension = 'rb'
+    when 'c++(gcc)'
+      extension = 'cpp'
     end
+    File.dirname(__dir__)+"/atcoder_tools/sources/task.#{extension}.erb"
+  end
 
-    def test_file_path
-      ".atcoder/#{@contest.name}/#{@name}/spec.rb"
+  def create!
+    html = URI.open(url) do |f|
+      charset = f.charset
+      f.read
     end
 
-    def create!
-      html = URI.open(url) do |f|
-        charset = f.charset
-        f.read
+    doc = Nokogiri::HTML.parse(html, nil, 'utf8')
+    samples = doc.css('.lang-ja > .part > section > pre').map { |e| e.children.text }
+    inputs, outputs = samples.partition.with_index { |_sample, i| i.even? }
+
+    erb = ERB.new(File.read(template_task_file_path))
+    File.write(task_file_path, erb.result(binding))
+  end
+
+  def run
+    read_meta_data
+    puts("#{@contest.name}/#{@name} is running in #{@mode} MODE")
+    update_spec
+    if @mode == "DEBUG"
+      compile
+      io = IO.popen(command, "w+")
+      @tests[0][:inputs].each do |input|
+        io.puts(input)
       end
+      io.close_write
+      puts io.readlines
+    end
+    if @mode == "RUN"
+      compile
+      system("echo put your inputs")
+      system(command)
+    end
+    if @mode == "TEST"
+      compile
+      system("bundle exec rspec #{test_file_path}")
+    end
+  end
 
-      doc = Nokogiri::HTML.parse(html, nil, 'utf8')
-      samples = doc.css('.lang-ja > .part > section > pre').map { |e| e.children.text }
-      inputs, outputs = samples.partition.with_index { |_sample, i| i.even? }
+  def compile
+    case contest.language
+    when 'c++(gcc)'
+      p 'hoge'
+      system("c++ #{task_file_path} -o #{@contest.name}/#{@name}")
+    end
+  end
 
-      erb = ERB.new(File.read(File.dirname(__dir__)+"/atcoder_tools/sources/task.rb.erb"))
-      File.write(task_file_path, erb.result(binding))
+  def command
+    case contest.language
+    when 'ruby'
+      "ruby #{task_file_path}"
+    when 'c++(gcc)'
+      "#{compile_file_path}"
     end
+  end
 
-    def run
-      read_meta_data
-      update_spec
-      if @mode == "DEBUG"
-        io = IO.popen("ruby #{task_file_path}", "w+")
-        io.puts(@tests[0][:input])
-        io.close_write
-        puts io.readlines
+  def read_meta_data
+    meta_flag = false
+    test_no = nil
+    tests = []
+    mode = nil
+    File.foreach(task_file_path) do |line|
+      line.chomp!
+      if line == "# METADATA::START" || line == "// METADATA::START"
+        meta_flag = true
+        next
       end
-      if @mode == "RUN"
-        system("echo put your inputs")
-        system("ruby #{task_file_path}")
+      if line == "# METADATA::FINISH" || line == "// METADATA::FINISH"
+        meta_flag = false
+        next
       end
-      if @mode == "TEST"
-        system("bundle exec rspec #{test_file_path}")
+      unless meta_flag
+        next
       end
-    end
-
-    def read_meta_data
-      meta_flag = false
-      test_no = nil
-      tests = []
-      mode = nil
-      File.foreach(task_file_path) do |line|
-        line.chomp!
-        if line == "# METADATA::START"
-          meta_flag = true
-          next
-        end
-        if line == "# METADATA::FINISH"
-          meta_flag = false
-          next
-        end
-        unless meta_flag
-          next
-        end
-        if line =~ /\#\stest-/
-          test_no = line.scan(/\#\stest-(\d+)/)[0][0].to_i
-          tests.push({ no: test_no })
-        end
-        if line =~ /\#\sinput:/
-          test = tests.find{|test| test[:no] == test_no}
-          test.merge!({ input: line.scan(/\#\sinput\:\s\"(.*)\"/)[0][0] })
-        end
-        if line =~ /\#\soutput:/
-          test = tests.find{|test| test[:no] == test_no}
-          test.merge!({ output: line.scan(/\#\soutput\:\s\"(.*)\"/)[0][0] })
-        end
-        if line =~ /\#\sMODE:/
-          mode = line.scan(/\#\sMODE:\s(.+)/)[0][0]
-        end
+      if line =~ /(\#|\/\/)\stest-/
+        test_no = line.scan(/(\#|\/\/)\stest-(\d+)/)[0][1].to_i
+        tests.push({ no: test_no })
+      end
+      if line =~ /(\#|\/\/)\sinput:/
+        test = tests.find{|test| test[:no] == test_no}
+        input_str = line.scan(/(\#|\/\/)\sinput\:\s\"(.*)\"/)[0][1]
+        inputs = input_str.chomp.split('\\n')
+        test.merge!({ inputs: inputs })
+      end
+      if line =~ /(\#|\/\/)\soutput:/
+        test = tests.find{|test| test[:no] == test_no}
+        test.merge!({ output: line.scan(/(\#|\/\/)\soutput\:\s\"(.*)\"/)[0][1] })
+      end
+      if line =~ /(\#|\/\/)\sMODE:/
+        mode = line.scan(/(\#|\/\/)\sMODE:\s(.+)/)[0][1]
       end
-      @tests = tests
-      @mode = mode
     end
+    @tests = tests
+    @mode = mode
+  end
 
-    def update_spec
-      FileUtils.mkdir_p ".atcoder/#{@contest.name}/#{@name}"
-      erb = ERB.new(File.read(File.dirname(__dir__)+"/atcoder_tools/sources/spec.rb.erb"))
-      File.write(test_file_path, erb.result(binding))
-    end
+  def code
+    File.read(task_file_path)
+  end
 
+  def update_spec
+    FileUtils.mkdir_p ".atcoder/#{@contest.name}/#{@name}"
+    erb = ERB.new(File.read(File.dirname(__dir__)+"/atcoder_tools/sources/spec.rb.erb"))
+    File.write(test_file_path, erb.result(binding))
   end
-# end
+
+end

data/lib/atcoder_tools/version.rb

@@ -1,3 +1,3 @@
 module AtcoderTools
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: atcoder_tools
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Aitaro Chaya
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-06-20 00:00:00.000000000 Z
+date: 2020-07-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -66,6 +66,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: mechanize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: tty-prompt
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Write a longer description or delete this line.
 email:
 - aitaro.chaya@gmail.com
@@ -74,12 +102,12 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".envrc"
+- ".github/workflows/deploy.yml"
 - ".gitignore"
 - ".rspec"
-- ".travis.yml"
 - CODE_OF_CONDUCT.md
 - Gemfile
-- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
@@ -88,9 +116,13 @@ files:
 - bin/setup
 - exe/atcoder_tools
 - lib/atcoder_tools.rb
+- lib/atcoder_tools/atcoder.rb
 - lib/atcoder_tools/cli.rb
 - lib/atcoder_tools/contest.rb
+- lib/atcoder_tools/log.rb
+- lib/atcoder_tools/settings.rb
 - lib/atcoder_tools/sources/spec.rb.erb
+- lib/atcoder_tools/sources/task.cpp.erb
 - lib/atcoder_tools/sources/task.rb.erb
 - lib/atcoder_tools/task.rb
 - lib/atcoder_tools/version.rb

data/.travis.yml

@@ -1,6 +0,0 @@
----
-language: ruby
-cache: bundler
-rvm:
-  - 2.7.1
-before_install: gem install bundler -v 2.1.4

data/Gemfile.lock

@@ -1,50 +0,0 @@
-PATH
-  remote: .
-  specs:
-    atcoder_tools (0.1.0)
-      listen
-      nokogiri
-      rspec
-      thor
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    diff-lcs (1.3)
-    ffi (1.13.1)
-    listen (3.2.1)
-      rb-fsevent (~> 0.10, >= 0.10.3)
-      rb-inotify (~> 0.9, >= 0.9.10)
-    mini_portile2 (2.4.0)
-    nokogiri (1.10.9)
-      mini_portile2 (~> 2.4.0)
-    rake (12.3.3)
-    rb-fsevent (0.10.4)
-    rb-inotify (0.10.1)
-      ffi (~> 1.0)
-    rspec (3.9.0)
-      rspec-core (~> 3.9.0)
-      rspec-expectations (~> 3.9.0)
-      rspec-mocks (~> 3.9.0)
-    rspec-core (3.9.2)
-      rspec-support (~> 3.9.3)
-    rspec-expectations (3.9.2)
-      diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.9.0)
-    rspec-mocks (3.9.1)
-      diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.9.0)
-    rspec-support (3.9.3)
-    thor (1.0.1)
-
-PLATFORMS
-  ruby
-
-DEPENDENCIES
-  atcoder_tools!
-  listen (~> 3.2)
-  rake (~> 12.0)
-  rspec (~> 3.0)
-
-BUNDLED WITH
-   2.1.4