vercon 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8472e1a2fafc85a47fa7fd66553ecb16dd051d867165ff3441b8a883cb62b007
-  data.tar.gz: 9bbcc1a5af63dc57d0d74a0b7d2a089cc78ec4b814868b858f3ce6a059bb881e
+  metadata.gz: 77ce1057ad9f1213ed089114b632bf87c6a600ec04aa718fc990951312083f94
+  data.tar.gz: b5902118f9fd5b59841181da6ab0131ed8d57697565ee39805e6e7174a763943
 SHA512:
-  metadata.gz: bf335632885d7cee074bb756448c501a8b06cf31d8090d29827b98b836cf3b7ca644163302ef048161fbc48c226bd8643397f68ad3a5e87f82907015a5285c69
-  data.tar.gz: abafa240143904a41c04b121b55b4fb16b2fb4d81d27ca3977733c7151f625b28febf50f4ff6604c2b263e66963a4e09207f8c842ba382378a439d3ca7d9307e
+  metadata.gz: 230f23e688c23bc89b9fddb01acb821b77290a93acafaa80d0be80059886e05204ee9862133fbad8e67ca18166d5bd2f1bc35d809e5db42e355dfee447db90d6
+  data.tar.gz: cc1ecd4620fc12c490f5612fd6c4e74e5e12518db460ce8d96adc0bf05d96082f10300818124c248fb7082983b56497e59ddde196224be45a66979a9718a0e51

data/.simplecov

@@ -0,0 +1,3 @@
+SimpleCov.start do
+  add_filter "spec"
+end

data/README.md

@@ -1,34 +1,44 @@
 # Vercon
 
-TODO: Delete this and the text below, and describe your gem
+Vercon - a handy little gem that takes the pain out of writing tests for your Ruby projects. 🔥
 
-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/vercon`. To experiment with that code, run `bin/console` for an interactive prompt.
+The name Vercon comes from the Latin "verum conantur", meaning "they strive for truth". Vercon automatically generate test files and even though they will not always will be perfect, but at least it will save some time writting them manually.
 
-## Installation
+It's build on top of Claude 3. It sends the source code of the Ruby file alongside with available factory names and current test file (in case one exists).
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+Claude analyzes your code to understand how it works, then uses that knowledge to put together relevant tests. 
 
-Install the gem and add to the application's Gemfile by executing:
+Just let the AI handle the tedious test writing for you 🚀 
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+Give Vercon a try and save yourself some time and headaches when it comes to testing your Ruby apps. 
 
-If bundler is not being used to manage dependencies, install the gem by executing:
+Easy, efficient, no fuss 👌
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+## How to use
 
-## Usage
+Install the gem by executing:
 
-TODO: Write usage instructions here
+    $ gem install vercon
+
+It will require from you Claude Api Token so greb it from [here](https://console.anthropic.com/settings/keys)
+
+Initialize the gem by executing:
+
+    $ vercon init
+
+Generate test:
+
+    $ vercon generate <relative ruby file path>
 
 ## Development
 
 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.
 
-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 the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+To install this gem onto your local machine, run `bundle exec rake install`. 
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/vercon.
+Bug reports and pull requests are welcome on GitHub at [vercon repo](https://github.com/AlexBeznoss/vercon).
 
 ## License
 

data/Rakefile

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require 'bundler/gem_tasks'
-require 'rspec/core/rake_task'
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
 
 RSpec::Core::RakeTask.new(:spec)
 
-require 'standard/rake'
+require "standard/rake"
 
 task default: %i[spec standard]

data/exe/vercon

@@ -3,7 +3,7 @@
 
 # require "vercon"
 # TODO: change back
-require_relative '../lib/vercon'
+require_relative "../lib/vercon"
 
 cli = Dry::CLI.new(Vercon::CLI)
 

data/lib/vercon/claude.rb

@@ -1,53 +1,75 @@
 # frozen_string_literal: true
 
-require 'httpx'
+require "httpx"
 
 module Vercon
   class Claude
-    BASE_URL = 'https://api.anthropic.com'
+    BASE_URL = "https://api.anthropic.com"
 
     def initialize
       config = Vercon::Config.new
 
-      @api_token = config.token
+      @api_token = config.claude_token
       @claude_model = config.claude_model
     end
 
     def submit(model: nil, system: nil, max_tokens: 4096, temperature: 0.2, stop_sequences: nil, user: nil, # rubocop:disable Metrics/ParameterLists
-               messages: nil)
+      messages: nil, tools: nil)
       body = {
         model: model || @claude_model,
         system: system,
         max_tokens: max_tokens,
         temperature: temperature,
         stop_sequences: stop_sequences,
-        messages: messages || [{ role: 'user', content: user }]
-      }.reject { |_, v| v.nil? || v == '' }
+        messages: messages || [{role: "user", content: user}],
+        tools: tools
+      }.reject { |_, v| v.nil? || ["", [], {}].include?(v) }
 
-      client.post('/v1/messages', body: body.to_json).then { |res| prepare_response(res.json) }
+      client.post("/v1/messages", body: body.to_json).then { |res| prepare_response(res.json) }
     end
 
     private
 
     def extra_headers
-      { 'x-api-key' => @api_token, 'anthropic-version' => '2023-06-01' }
+      {"x-api-key" => @api_token, "anthropic-version" => "2023-06-01"}
     end
 
     def client
       @client ||=
         HTTPX
-        .plugin(:retries)
-        .with(headers: { 'Content-Type' => 'application/json', 'Cache-Control' => 'no-cache' })
-        .with(headers: extra_headers)
-        .with(origin: BASE_URL)
-        .with(ssl: { alpn_protocols: %w[http/1.1] })
-        .with(timeout: { keep_alive_timeout: 180 })
+          .plugin(:retries)
+          .with(headers: {"Content-Type" => "application/json", "Cache-Control" => "no-cache",
+                          "anthropic-version" => "2023-06-01", "anthropic-beta" => "tools-2024-04-04"})
+          .with(headers: extra_headers)
+          .with(origin: BASE_URL)
+          .with(ssl: {alpn_protocols: %w[http/1.1]})
+          .with(
+            timeout: {
+              connect_timeout: 10,
+              read_timeout: 400,
+              keep_alive_timeout: 480,
+              request_timeout: 400,
+              operation_timeout: 400
+            }
+          )
     end
 
     def prepare_response(response)
-      return { error: response.dig('error', 'message') } if response['type'] == 'error'
+      return {error: response.dig("error", "message")} if response["type"] == "error"
 
-      { text: response.dig('content', 0, 'text') }
+      texts = []
+      tools = []
+
+      response["content"].each do |msg|
+        case msg["type"]
+        when "text"
+          texts << msg["text"].gsub(%r{\n?</?thinking>\n?}, "")
+        when "tool_use"
+          tools << msg.slice("name", "id", "input").transform_keys(&:to_sym)
+        end
+      end
+
+      {text: texts.join("\n"), tools: tools}
     end
   end
 end

data/lib/vercon/cli.rb

@@ -1,15 +1,17 @@
 # frozen_string_literal: true
 
-require 'dry/cli'
+require "dry/cli"
 
-require_relative 'commands/init'
-require_relative 'commands/generate'
+require_relative "commands/init"
+require_relative "commands/generate"
+require_relative "commands/version"
 
 module Vercon
   class CLI
     extend Dry::CLI::Registry
 
-    register 'init', Commands::Init, aliases: ['i']
-    register 'generate', Commands::Generate, aliases: ['g']
+    register "init", Commands::Init, aliases: ["i"]
+    register "generate", Commands::Generate, aliases: ["g"]
+    register "version", Commands::Version, aliases: ["v", "-v", "--version"]
   end
 end

data/lib/vercon/commands/generate.rb

@@ -1,28 +1,33 @@
 # frozen_string_literal: true
 
-require 'prism'
-require 'dry/files'
-require 'tty-spinner'
-require 'tty-pager'
-require 'tty-editor'
+require "prism"
+require "dry/files"
+require "tty-spinner"
+require "rouge"
+require "tty-editor"
 
 module Vercon
   module Commands
     class Generate < Dry::CLI::Command
-      desc 'Generate test file'
-
-      argument :path, desc: 'Path to the ruby file'
-
-      option :edit_prompt, type: :boolean, default: false, aliases: ['e'],
-                           desc: 'Edit prompt before submitting to claude'
-      option :output_path, type: :string, default: nil, aliases: ['o'],
-                           desc: 'Path to save test file'
-      option :stdout, type: :boolean, default: false, aliases: ['s'],
-                      desc: 'Output test file to stdout instead of writing to test file'
-      option :force, type: :boolean, default: false, aliases: ['f'],
-                     desc: 'Force overwrite of existing test file'
-      option :open, type: :boolean, default: false, aliases: ['p'],
-                    desc: 'Open test file in editor after generation'
+      AUTOFIXERS = {
+        standard: "bundle exec standardrb --fix %{file} > /dev/null 2>&1",
+        rubocop: "bundle exec rubocop -a %{file} > /dev/null 2>&1"
+      }
+
+      desc "Generate test file"
+
+      argument :path, desc: "Path to the ruby file"
+
+      option :edit_prompt, type: :boolean, default: false, aliases: ["e"],
+        desc: "Edit prompt before submitting to claude"
+      option :output_path, type: :string, default: nil, aliases: ["o"],
+        desc: "Path to save test file"
+      option :stdout, type: :boolean, default: false, aliases: ["s"],
+        desc: "Output test file to stdout instead of writing to test file"
+      option :force, type: :boolean, default: false, aliases: ["f"],
+        desc: "Force overwrite of existing test file"
+      option :open, type: :boolean, default: nil, aliases: ["p"],
+        desc: "Open test file in editor after generation"
 
       def initialize
         @config = Vercon::Config.new
@@ -41,10 +46,15 @@ module Vercon
         current_test = files.exist?(output_path) ? files.read(output_path) : nil
 
         result = generate_test_file(path, opts, current_test)
+        return if result.nil?
+
+        result = run_autofixes(result)
 
         if opts[:stdout]
-          pager = TTY::Pager.new
-          pager.page(result)
+          formatter = Rouge::Formatters::Terminal256.new(Rouge::Themes::Base16::Monokai.new)
+          lexer = Rouge::Lexers::Ruby.new
+
+          stdout.puts(formatter.format(lexer.lex(result)))
           return
         end
 
@@ -54,13 +64,11 @@ module Vercon
 
         files.write(output_path, result)
 
-        run_rubocop(output_path) if include_gem?('rubocop') || include_gem?('standard')
-
         stdout.ok("Test file saved at \"#{output_path}\" 🥳")
 
-        return unless opts[:open]
-
-        TTY::Editor.new(raise_on_failure: true).open(output_path)
+        if opts[:open] == true || (opts[:open].nil? && config.open_by_default?)
+          TTY::Editor.new(raise_on_failure: true).open(output_path)
+        end
       end
 
       private
@@ -69,29 +77,29 @@ module Vercon
 
       def can_generate?(path, _opts)
         unless config.exists?
-          stdout.error('Config file does not exist. Run `vercon init` to create a config file.')
+          stdout.error("Config file does not exist. Run `vercon init` to create a config file.")
           return false
         end
 
         if path.nil? || path.empty?
-          stdout.error('Path to ruby file is blank.')
+          stdout.error("Path to ruby file is blank.")
           return false
         end
 
         unless files.exist?(path)
-          stdout.error('Ruby file does not exist.')
+          stdout.error("Ruby file does not exist.")
           return false
         end
 
         expanded_path = files.expand_path(path)
 
         if Prism.parse_file_failure?(expanded_path)
-          stdout.error('Looks like the ruby file has syntax errors. Fix them before generating tests.')
+          stdout.error("Looks like the ruby file has syntax errors. Fix them before generating tests.")
           return false
         end
 
-        unless include_gem?('rspec')
-          stdout.error('RSpec is not installed. Vercon requires RSpec to generate test files.')
+        unless include_gem?("rspec")
+          stdout.error("RSpec is not installed. Vercon requires RSpec to generate test files.")
           return false
         end
 
@@ -99,15 +107,11 @@ module Vercon
       end
 
       def generate_test_file_path(path, _opts)
-        system, user, stop_sequence = Vercon::Prompt.for_test_path(path: path)
-        spinner = TTY::Spinner.new('[:spinner] Preparing spec file path...', format: :flip)
+        prompt = Vercon::Prompt.for_test_path(path: path)
+        spinner = TTY::Spinner.new("[:spinner] Preparing spec file path...", format: :flip)
         spinner.auto_spin
 
-        result = Vercon::Claude.new.submit(
-          model: config.class::LOWEST_CLAUDE_MODEL,
-          system: system, user: user,
-          stop_sequences: [stop_sequence]
-        )
+        result = Vercon::Claude.new.submit(**prompt.merge(model: config.class::LOWEST_CLAUDE_MODEL))
         spinner.stop
         stdout.erase(lines: 1)
 
@@ -116,28 +120,28 @@ module Vercon
           return
         end
 
-        path = result[:text].match(/RSPEC FILE PATH: "(.+)"/)[1]
+        path, = result[:text].match(/RSPEC FILE PATH: "(.+)"/).captures
 
         if stdout.no?("Corresponding test file path should be \"#{path}\". Correct?")
-          path = stdout.ask('Enter a relative path of corresponding test:')
+          path = stdout.ask("Enter a relative path of corresponding test:")
         end
 
         path
       end
 
       def generate_test_file(path, opts, current_test)
-        factories = Vercon::Factories.new.load if include_gem?('factory_bot')
-        system, user, stop_sequence = Vercon::Prompt.for_test_generation(
+        factories = Vercon::Factories.new.load if include_gem?("factory_bot")
+        prompt = Vercon::Prompt.for_test_generation(
           path: path, source: files.read(path),
           factories: factories, current_test: current_test
         )
-        system, user = ask_for_edits(system, user) if opts[:edit_prompt]
-        return if system.nil? || user.nil?
+        prompt = ask_for_edits(**prompt) if opts[:edit_prompt]
+        return if prompt[:system].nil? || prompt[:user].nil? || prompt[:tools].nil?
 
-        spinner = TTY::Spinner.new('[:spinner] Generating spec file...', format: :flip)
+        spinner = TTY::Spinner.new("[:spinner] Generating spec file...", format: :flip)
         spinner.auto_spin
 
-        result = Vercon::Claude.new.submit(system: system, user: user, stop_sequences: [stop_sequence])
+        result = Vercon::Claude.new.submit(**prompt)
         spinner.stop
         stdout.erase(lines: 1)
 
@@ -146,49 +150,85 @@ module Vercon
           return
         end
 
-        result[:text].match(/TEST SOURCE CODE:\n```ruby\n(.+)\n```/m)[1]
+        tool = result[:tools].find { |tool| tool[:name] == "write_test_file" }
+
+        if tool.nil?
+          stdout.error('Claude did not return the "write_test_file" tool. Aborting generation.')
+          return nil
+        end
+
+        source = tool.dig(:input, "source_code")
+        source = source.match(/```ruby\n(.+)\n```/m).captures.first if source.include?("```ruby")
+
+        source
       end
 
-      def run_rubocop(path)
-        spinner = TTY::Spinner.new('[:spinner] Running RuboCop...', format: :flip)
-        spinner.auto_spin
+      def run_autofixes(source)
+        file = Tempfile.new("source_spec.rb")
+        file.write(source)
+        file.open
 
-        system("bundle exec rubocop -A #{files.expand_path(path)} > /dev/null 2>&1")
+        AUTOFIXERS.each do |name, command|
+          next unless include_gem?(name.to_s)
 
-        spinner.stop
-        stdout.erase(lines: 1)
+          spinner = TTY::Spinner.new("[:spinner] Running #{name}...", format: :flip)
+          spinner.auto_spin
+
+          system(format(command, file: file.path))
+
+          spinner.stop
+          stdout.erase(lines: 1)
+        end
+
+        file.read
+      ensure
+        file.unlink
       end
 
-      def ask_for_edits(system, user)
-        path = '~/.vercon_prompt.txt'
-        text = <<~EOF.strip
-          Please, do not remove magick comments :)
+      def ask_for_edits(system:, user:, tools: nil)
+        path = "~/.vercon_prompt.txt"
+        text = []
+        text << <<~EOF.strip
+          Please, do not remove magick comments like <System prompt> and others :)
+          #{tools.nil? ? "" : "When chaning tools, make sure to keep the general schema section intact, change descriptions only."}
+
           <System prompt>
           #{system}
+
           <User prompt>
           #{user}
         EOF
+        text << <<~EOF.strip if tools
+          <Tools>
+          #{JSON.pretty_generate(tools)}
+        EOF
 
-        files.write(path, text)
+        files.write(path, text.join("\n\n"))
 
         TTY::Editor.new(raise_on_failure: true).open(path)
-        TTY::Spinner.new('[:spinner] Waiting for changes...', format: :flip).run { sleep(rand(1..3)) }
+        TTY::Spinner.new("[:spinner] Waiting for changes...", format: :flip).run { sleep(rand(1..3)) }
         stdout.erase(lines: 1)
 
-        if stdout.no?('Can we proceed?')
-          stdout.error('Generation aborted!')
-          return []
+        if stdout.no?("Can we proceed?")
+          stdout.error("Generation aborted!")
+          return {}
         end
 
-        system, user = files.read(path).match(/<System prompt>\n(.+)\n<User prompt>\n(.+)/m).captures
+        result = files.read(path)
+        system, user = result.match(/<System prompt>(.+)\n<User prompt>(.+)/m).captures
+        if tools
+          user = user.match(/^(.+)\n\n<Tools>/m).captures.first
+          tools = result.match(/<Tools>(.+)/m)&.captures&.first
+          tools = JSON.parse(tools)
+        end
 
-        [system, user]
+        {system: system, user: user, tools: tools}.reject { |_, v| v.nil? }
       ensure
         files.delete(path)
       end
 
       def include_gem?(name)
-        files.read('Gemfile').include?(name)
+        files.read("Gemfile").include?(name)
       end
     end
   end

data/lib/vercon/commands/init.rb

@@ -3,10 +3,11 @@
 module Vercon
   module Commands
     class Init < Dry::CLI::Command
-      desc 'Initialize vercon config'
+      desc "Initialize vercon config"
 
-      option :token, desc: 'Claude API token'
-      option :claude_model, desc: 'Claude model to use by default'
+      option :claude_token, desc: "Claude API token"
+      option :claude_model, desc: "Claude model to use by default"
+      option :open, type: :boolean, default: nil, desc: "Open generated test file by default"
 
       def initialize
         @stdout = Vercon::Stdout.new
@@ -18,26 +19,23 @@ module Vercon
       end
 
       def call(**opts)
-        token_changed = setup_token(opts)
-        claude_changed = setup_claude_model(opts)
+        setup_token(opts)
+        setup_claude_model(opts)
+        setup_default_open(opts)
 
-        if token_changed || claude_changed
-          @stdout.ok("Config file #{@config_existed ? 'updated' : 'created'}!")
-        else
-          @stdout.warn('Config file is not touched.')
-        end
+        @stdout.ok("Config file #{@config_existed ? "updated" : "created"}!")
       end
 
       private
 
       def setup_token(opts)
-        if @config.token && @stdout.no?("Claude API token already set to `#{@config.token}`. Do you want to replace it?")
+        if @config.claude_token && @stdout.no?("Claude API token already set to `#{@config.claude_token}`. Do you want to replace it?")
           return
         end
 
-        token = opts[:token]
-        token ||= @stdout.ask('Provide your Claude API token:')
-        @config.token = token
+        token = opts[:claude_token]
+        token ||= @stdout.ask("Provide your Claude API token:")
+        @config.claude_token = token
       end
 
       def setup_claude_model(opts)
@@ -46,10 +44,24 @@ module Vercon
         end
 
         model = opts[:claude_model]
-        model ||= @stdout.select('Select Claude model that will be used by default:', Vercon::Config::CLAUDE_MODELS,
-                                 default: Vercon::Config::DEFAULT_CLAUDE_MODEL, cycle: true)
+        model ||= @stdout.select("Select Claude model that will be used by default:", Vercon::Config::CLAUDE_MODELS,
+          default: Vercon::Config::DEFAULT_CLAUDE_MODEL, cycle: true)
         @config.claude_model = model
       end
+
+      def setup_default_open(opts)
+        open = opts[:open]
+        if open.nil?
+          open = @stdout.select(
+            "Open generated test file by default?",
+            {Yes: true, No: false},
+            default: "No",
+            cycle: true
+          )
+        end
+
+        @config.open_by_default = open
+      end
     end
   end
 end

data/lib/vercon/commands/version.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative "../version"
+
+module Vercon
+  module Commands
+    class Version < Dry::CLI::Command
+      desc "Print version"
+
+      def initialize
+        @stdout = Vercon::Stdout.new
+
+        super
+      end
+
+      def call(*)
+        @stdout.puts ::Vercon::VERSION
+      end
+    end
+  end
+end

data/lib/vercon/config.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require 'yaml'
-require 'dry/files'
+require "yaml"
+require "dry/files"
 
 module Vercon
   class Config
@@ -10,9 +10,9 @@ module Vercon
       claude-3-sonnet-20240229
       claude-3-opus-20240229
     ].freeze
-    DEFAULT_CLAUDE_MODEL = 'claude-3-sonnet-20240229'
-    LOWEST_CLAUDE_MODEL = 'claude-3-haiku-20240307'
-    PATH = '~/.vercon.yml'
+    DEFAULT_CLAUDE_MODEL = "claude-3-sonnet-20240229"
+    LOWEST_CLAUDE_MODEL = "claude-3-haiku-20240307"
+    PATH = "~/.vercon.yml"
 
     def initialize
       @files = Dry::Files.new
@@ -25,21 +25,28 @@ module Vercon
       !@config.empty?
     end
 
-    def token
-      @config['claude_token']
+    def claude_token
+      @config["claude_token"]
     end
 
-    def token=(value)
-      @config['claude_token'] = value
-      @files.write(@files.expand_path(PATH), YAML.safe_dump(@config))
+    def claude_model
+      @config["claude_model"]
     end
 
-    def claude_model
-      @config['claude_model']
+    def open_by_default?
+      @config["open_by_default"].nil? ? false : @config["open_by_default"]
     end
 
-    def claude_model=(value)
-      @config['claude_model'] = value
+    %i[claude_token claude_model open_by_default].each do |method|
+      define_method(:"#{method}=") do |value|
+        @config[method.to_s] = value
+        write_config
+      end
+    end
+
+    private
+
+    def write_config
       @files.write(@files.expand_path(PATH), YAML.safe_dump(@config))
     end
   end

data/lib/vercon/factories.rb

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
-require 'dry/files'
-require 'prism'
+require "dry/files"
+require "prism"
 
 module Vercon
   class Factories
-    PATH = './spec/factories'
+    PATH = "./spec/factories"
 
     def initialize
       @files = Dry::Files.new
@@ -14,7 +14,7 @@ module Vercon
     def load
       return unless @files.directory?(PATH)
 
-      Dir[@files.expand_path(@files.join(PATH, '**', '*.rb'))].map do |file_path|
+      Dir[@files.expand_path(@files.join(PATH, "**", "*.rb"))].map do |file_path|
         load_factory(file_path)
       end.flatten.compact
     end

data/lib/vercon/prompt.rb

@@ -2,11 +2,11 @@
 
 module Vercon
   class Prompt
-    class << self
-      END_SEQUENCE = '<END RESULT>'
+    END_SEQUENCE = "<END RESULT>"
 
+    class << self
       def for_test_path(path:)
-        system_prompt = <<~PROMPT.strip
+        system = <<~PROMPT.strip
           You are tasked as a professional Ruby and Ruby on Rails developer specialising in writing comprehensive RSpec unit tests for a Ruby class. You will receive a path to ruby file. Your objective is to generate a path corresponding RSpec unit test file. Make sure to use common practices used in Ruby on Rails community for structuring test file paths.
 
           Provide a in the following format:
@@ -17,15 +17,15 @@ module Vercon
           Make sure to include "#{END_SEQUENCE}" at the end of your test source code. It's required.
         PROMPT
 
-        user_prompt = <<~PROMPT.strip
+        user = <<~PROMPT.strip
           PATH: #{path.inspect}
         PROMPT
 
-        [system_prompt, user_prompt, END_SEQUENCE]
+        {system: system, user: user, stop_sequences: [END_SEQUENCE]}
       end
 
       def for_test_generation(path:, source:, factories: nil, current_test: nil)
-        system_prompt = <<~PROMPT.strip
+        system = <<~PROMPT.strip
           You are a professional Ruby developer specializing in writing comprehensive RSpec unit tests for a Ruby class. Your objective is to ensure each public method within the class is thoroughly tested. Below are the specifics you must adhere to:
 
           - Structure:
@@ -59,25 +59,18 @@ module Vercon
             - This includes testing with empty or nil values, large or small input values, and any other relevant scenarios specific to the class.
             - Think critically about potential edge cases and ensure they are adequately covered in the tests.
 
-          Provide the result in the following format:
-
-          TEST SOURCE CODE:
-          ```ruby
-          <the RSpec tests, without any additional comments or markdown instructions>
-          ```
-          #{END_SEQUENCE}
-
-          Make sure to include "#{END_SEQUENCE}" at the end of your test source code.
-
           If the user provides "CURRENT RSPEC FILE", use it as a base for your tests, improve it, and extend it according to the Ruby file being tested. Ensure that the existing tests are updated to meet the specified requirements and that new tests are added to achieve comprehensive coverage of the class's public methods.
 
           Remember to focus on testing the behavior of the class rather than its implementation details. Aim for concise, readable, and maintainable tests that provide confidence in the correctness of the class. Use descriptive test names, keep tests isolated from each other, and follow RSpec best practices and conventions throughout your test suite.
+
+          After you have created the RSpec tests for the Ruby class, make sure to call "write_test_file" tool with the test source code.
+          You shouldn't comment your thinking process.
         PROMPT
 
-        user_prompt = ["PATH: #{path.inspect}"]
+        user = ["PATH: #{path.inspect}"]
 
         if factories
-          user_prompt << <<~PROMPT.strip
+          user << <<~PROMPT.strip
             AVAILABLE FACTORIES:
             ```json
             #{JSON.dump(factories)}
@@ -85,7 +78,7 @@ module Vercon
           PROMPT
         end
 
-        user_prompt << <<~PROMPT.strip
+        user << <<~PROMPT.strip
           CODE:
           ```ruby
           #{source}
@@ -93,7 +86,7 @@ module Vercon
         PROMPT
 
         if current_test
-          user_prompt << <<~PROMPT.strip
+          user << <<~PROMPT.strip
             CURRENT RSPEC FILE:
             ```ruby
             #{current_test}
@@ -101,7 +94,24 @@ module Vercon
           PROMPT
         end
 
-        [system_prompt, user_prompt.join("\n"), END_SEQUENCE]
+        tools = [
+          {
+            name: "write_test_file",
+            description: "Tool to write the test file to disk",
+            input_schema: {
+              type: "object",
+              properties: {
+                source_code: {
+                  type: "string",
+                  description: "The source code of the test file to be written to disk. Make sure to include the complete test file content. Do not include any additional information, comments or markup."
+                }
+              },
+              required: ["source_code"]
+            }
+          }
+        ]
+
+        {system: system, user: user.join("\n"), tools: tools}
       end
     end
   end

data/lib/vercon/stdout.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'tty-prompt'
+require "tty-prompt"
 
 module Vercon
   class Stdout
@@ -12,7 +12,7 @@ module Vercon
 
     def write(message)
       @stdout.puts(message)
-      @lines += 1
+      @lines += message.count("\n") + 1
     end
 
     def erase(lines: nil)
@@ -23,16 +23,16 @@ module Vercon
     end
 
     %i[ask yes? no? say ok warn error mask select].each do |method|
-      define_method(method) do |*args, &block|
-        @lines += 1
-        @prompt.send(method, *args, &block)
+      define_method(method) do |*args, **params, &block|
+        @lines += args.first.count("\n") + 1
+        @prompt.send(method, *args, **params, &block)
       end
     end
 
     %i[puts print].each do |method|
-      define_method(method) do |*args, &block|
-        @lines += 1
-        @stdout.send(method, *args, &block)
+      define_method(method) do |*args, **params, &block|
+        @lines += args.first.count("\n") + 1
+        @stdout.send(method, *args, **params, &block)
       end
     end
   end

data/lib/vercon/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Vercon
-  VERSION = '0.0.1'
+  VERSION = "0.0.3"
 end

data/lib/vercon.rb

@@ -1,12 +1,12 @@
 # frozen_string_literal: true
 
-require_relative 'vercon/version'
-require_relative 'vercon/config'
-require_relative 'vercon/stdout'
-require_relative 'vercon/cli'
-require_relative 'vercon/claude'
-require_relative 'vercon/prompt'
-require_relative 'vercon/factories'
+require_relative "vercon/version"
+require_relative "vercon/config"
+require_relative "vercon/stdout"
+require_relative "vercon/cli"
+require_relative "vercon/claude"
+require_relative "vercon/prompt"
+require_relative "vercon/factories"
 
 module Vercon
   class Error < StandardError; end

data/vercon.gemspec

@@ -1,22 +1,22 @@
 # frozen_string_literal: true
 
-require_relative 'lib/vercon/version'
+require_relative "lib/vercon/version"
 
 Gem::Specification.new do |spec|
-  spec.name = 'vercon'
+  spec.name = "vercon"
   spec.version = Vercon::VERSION
-  spec.authors = ['Alex Beznos']
-  spec.email = ['beznosa@yahoo.com']
+  spec.authors = ["Alex Beznos"]
+  spec.email = ["beznosa@yahoo.com"]
 
-  spec.summary = 'CLI tool to generate test files with Cloude 3'
-  spec.description = 'CLI tool to generate test files with Cloude 3.'
-  spec.homepage = 'https://github.com/AlexBeznoss/vercon'
-  spec.license = 'MIT'
-  spec.required_ruby_version = '>= 3.0'
-  spec.required_rubygems_version = '>= 3.3.11'
+  spec.summary = "CLI tool to generate test files with Cloude 3"
+  spec.description = "CLI tool to generate test files with Cloude 3."
+  spec.homepage = "https://github.com/AlexBeznoss/vercon"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 3.0"
+  spec.required_rubygems_version = ">= 3.3.11"
 
-  spec.metadata['homepage_uri'] = spec.homepage
-  spec.metadata['source_code_uri'] = spec.homepage
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
@@ -26,18 +26,18 @@ Gem::Specification.new do |spec|
         f.start_with?(*%w[bin/ test/ spec/ features/ .git .circleci appveyor Gemfile])
     end
   end
-  spec.bindir = 'exe'
+  spec.bindir = "exe"
   spec.executables = [spec.name]
-  spec.require_paths = ['lib']
+  spec.require_paths = ["lib"]
 
-  spec.add_dependency 'dry-cli', '~> 1.0', '< 2'
-  spec.add_dependency 'dry-files', '~> 1.0', '< 2'
-  spec.add_dependency 'httpx', '~> 1.2.3'
-  spec.add_dependency 'prism', '~> 0.24'
-  spec.add_dependency 'tty-editor', '~> 0.7.0'
-  spec.add_dependency 'tty-pager', '~> 0.14'
-  spec.add_dependency 'tty-prompt', '~> 0.23.1'
-  spec.add_dependency 'tty-spinner', '~> 0.9.3'
+  spec.add_dependency "dry-cli", "~> 1.0", "< 2"
+  spec.add_dependency "dry-files", "~> 1.0", "< 2"
+  spec.add_dependency "httpx", "~> 1.2.3"
+  spec.add_dependency "prism", "~> 0.24"
+  spec.add_dependency "rouge", "~> 4.2.1"
+  spec.add_dependency "tty-editor", "~> 0.7.0"
+  spec.add_dependency "tty-prompt", "~> 0.23.1"
+  spec.add_dependency "tty-spinner", "~> 0.9.3"
 
   # For more information and examples about making a new gem, check out our
   # guide at: https://bundler.io/guides/creating_gem.html

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vercon
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Alex Beznos
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-04-15 00:00:00.000000000 Z
+date: 2024-04-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: dry-cli
@@ -79,33 +79,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0.24'
 - !ruby/object:Gem::Dependency
-  name: tty-editor
+  name: rouge
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.7.0
+        version: 4.2.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.7.0
+        version: 4.2.1
 - !ruby/object:Gem::Dependency
-  name: tty-pager
+  name: tty-editor
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: 0.7.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: 0.7.0
 - !ruby/object:Gem::Dependency
   name: tty-prompt
   requirement: !ruby/object:Gem::Requirement
@@ -143,6 +143,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".rspec"
+- ".simplecov"
 - ".standard.yml"
 - ".tool-versions"
 - LICENSE.txt
@@ -154,6 +155,7 @@ files:
 - lib/vercon/cli.rb
 - lib/vercon/commands/generate.rb
 - lib/vercon/commands/init.rb
+- lib/vercon/commands/version.rb
 - lib/vercon/config.rb
 - lib/vercon/factories.rb
 - lib/vercon/prompt.rb