ai_refactor 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 758af953e626b18190ef1e2252e730ea636d773a53b63052aaa7aba6213b49a2
-  data.tar.gz: 2bf63e217e7c3647e9cc26a9112f72eb19b81700f5784730a15ec0b229889963
+  metadata.gz: fa9421b9791dda3c02b930e63d0fa7bd3dc587d121a0b3748a0f9d8c649f98a3
+  data.tar.gz: a5cb385968939847ebb1ceb1d663b1926e410fce0571fa2ab7463440347c0f50
 SHA512:
-  metadata.gz: 76e034568b78234e7b66c756601be73976f3367c542112b6cfc88c47ab3edb31016a8507201eee382a33ed2c7d041707295fcf140cfecc859a1d6e5b99c52d35
-  data.tar.gz: bb00b41760079e5203c5bcadbff10c09b7fc701ec12a1c25623dbff85862e6c3de0416d341ad41c22d03dad54fdadc97ccb33e2a02756186a65ad9e4dfb17d4d
+  metadata.gz: 6ff997f0533fe95da7f0361789d88b4a73ecd8fbd82a1f54179fbce3b6a3b466c40330ae615a00f850e549212fe2463ba60ca4467cc0d9f65bbd6679f326dff1
+  data.tar.gz: 5898b17ebaa96c24a89cae1ff741a53d3afbb7a6c4844f52a271ddfecc1ebc1923026324601cb98e56b5aed0d673bba5d583d388a7a936e4cc86b272e1858bd8

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ai_refactor (0.2.0)
+    ai_refactor (0.3.0)
       colorize (< 2.0)
       open3 (< 2.0)
       ruby-openai (>= 3.4.0, < 5.0)

data/exe/ai_refactor

@@ -14,7 +14,6 @@ supported_names = AIRefactor::Refactors.names
 option_parser = OptionParser.new do |parser|
   parser.banner = "Usage: ai_refactor REFACTOR_TYPE INPUT_FILE_OR_DIR [options]\n\nWhere REFACTOR_TYPE is one of: #{supported_names}\n\n"
 
-  # todo: support for a sort of generic process which uses a custom prompt file
   parser.on("-p", "--prompt PROMPT_FILE", String, "Specify path to a text file that contains the ChatGPT 'system' prompt.") do |f|
     options[:prompt_file_path] = f
   end
@@ -27,11 +26,11 @@ option_parser = OptionParser.new do |parser|
     options[:ai_model] = m
   end
 
-  parser.on(nil, "--temperature TEMP", Float, "Specify the temperature parameter for ChatGPT (default 0.7).") do |p|
+  parser.on("--temperature TEMP", Float, "Specify the temperature parameter for ChatGPT (default 0.7).") do |p|
     options[:ai_temperature] = p
   end
 
-  parser.on(nil, "--max-tokens MAX_TOKENS", Integer, "Specify the max number of tokens of output ChatGPT can generate. Max will depend on the size of the prompt (default 1500)") do |m|
+  parser.on("--max-tokens MAX_TOKENS", Integer, "Specify the max number of tokens of output ChatGPT can generate. Max will depend on the size of the prompt (default 1500)") do |m|
     options[:ai_max_tokens] = m
   end
 
@@ -47,18 +46,23 @@ option_parser = OptionParser.new do |parser|
     options[:debug] = true
   end
 
-  supported_refactors.each do |_name, refactorer|
-    refactorer.command_line_options.each do |option|
-      parser.on(option[:short], option[:long], option[:type], option[:help]) do |o|
-        options[option[:key]] = o
-      end
-    end
-  end
-
   parser.on("-h", "--help", "Prints this help") do
     puts parser
     exit
   end
+
+  parser.separator ""
+
+  supported_refactors.each do |name, refactorer|
+    parser.separator "For refactor type '#{name}':" if refactorer.command_line_options.size.positive?
+    refactorer.command_line_options.each do |option|
+      args = [option[:long], option[:type], option[:help]]
+      args.unshift(option[:short]) if option[:short]
+      parser.on(*args) do |o|
+        options[option[:key]] = o.nil? ? true : o
+      end
+    end
+  end
 end
 
 option_parser.parse!

data/lib/ai_refactor/base_refactor.rb

@@ -21,6 +21,17 @@ module AIRefactor
 
     private
 
+    def can_overwrite_output_file?(output_path)
+      logger.info "Do you wish to overwrite #{output_path}? (y/n)"
+      answer = $stdin.gets.chomp
+      unless answer == "y" || answer == "Y"
+        logger.warn "Skipping #{input_file}..."
+        self.failed_message = "Skipped as output file already exists"
+        return false
+      end
+      true
+    end
+
     def prompt_file_path
       self.class.prompt_file_path
     end

data/lib/ai_refactor/refactors/generic.rb

@@ -5,14 +5,20 @@ module AIRefactor
     class Generic < BaseRefactor
       def run
         logger.verbose "Generic refactor to #{input_file}... (using user supplied prompt #{prompt_file_path})"
+        logger.verbose "Write output to #{output_file_path}..." if output_file_path
 
         processor = AIRefactor::FileProcessor.new(
           input_path: input_file,
           prompt_file_path: prompt_file_path,
           ai_client: ai_client,
-          logger: logger
+          logger: logger,
+          output_path: output_file_path
         )
 
+        if processor.output_exists?
+          return false unless can_overwrite_output_file?(output_file_path)
+        end
+
         logger.verbose "Converting #{input_file}..."
 
         begin
@@ -38,11 +44,38 @@ module AIRefactor
           return false
         end
 
-        output_content
+        output_file_path ? true : output_content
       end
 
       private
 
+      def output_file_path
+        return output_file_path_from_template if output_template_path
+
+        path = options[:output_file_path]
+        return unless path
+
+        if path == true
+          input_file
+        else
+          path
+        end
+      end
+
+      def output_template_path
+        options[:output_template_path]
+      end
+
+      def output_file_path_from_template
+        path = output_template_path.gsub("[FILE]", File.basename(input_file))
+          .gsub("[NAME]", File.basename(input_file, ".*"))
+          .gsub("[DIR]", File.dirname(input_file))
+          .gsub("[REFACTOR]", self.class.refactor_name)
+          .gsub("[EXT]", File.extname(input_file))
+        raise "Output template could not be used" unless path.length.positive?
+        path
+      end
+
       def prompt_file_path
         specified_prompt_path = options[:prompt_file_path]
         if specified_prompt_path&.length&.positive?
@@ -61,6 +94,23 @@ module AIRefactor
         def prompt_file_path
           raise "Generic refactor requires prompt file to be user specified."
         end
+
+        def command_line_options
+          [
+            {
+              key: :output_file_path,
+              long: "--output [FILE]",
+              type: String,
+              help: "Write output to file instead of stdout. If no path provided will overwrite input file (will prompt to overwrite existing files)"
+            },
+            {
+              key: :output_template_path,
+              long: "--output-template TEMPLATE",
+              type: String,
+              help: "Write outputs to files instead of stdout. The template is used to create the output name, where the it can have substitutions, '[FILE]', '[NAME]', '[DIR]', '[REFACTOR]' & '[EXT]'. Eg `[DIR]/[NAME]_[REFACTOR][EXT]` (will prompt to overwrite existing files)"
+            }
+          ]
+        end
       end
     end
   end

data/lib/ai_refactor/refactors/rspec_to_minitest_rails.rb

@@ -35,13 +35,7 @@ module AIRefactor
         )
 
         if processor.output_exists?
-          logger.info "Do you wish to overwrite #{output_path}? (y/n)"
-          answer = $stdin.gets.chomp
-          unless answer == "y" || answer == "Y"
-            logger.warn "Skipping #{input_file}..."
-            self.failed_message = "Skipped as output test file already exists"
-            return false
-          end
+          return false unless can_overwrite_output_file?(output_path)
         end
 
         logger.verbose "Converting #{input_file}..."

data/lib/ai_refactor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AIRefactor
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_refactor
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Stephen Ierodiaconou
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-24 00:00:00.000000000 Z
+date: 2023-05-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize