rails_code_auditor 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83cd23ea4027538587a1814cbf2a3472bd818fe0cb15bf492a3ab54b565c2e63
-  data.tar.gz: 40438e254bed8d07966559ac52df8a3063f427167fbc236a8eac0fba6a591c1d
+  metadata.gz: 02d182284c325102a56357851f1bf841c265b33d601c7af29726be9c1d3d80e0
+  data.tar.gz: 30afb60183c7b4fbc029b0b3341afac7c7a857364906caebe29d4a1e359f6773
 SHA512:
-  metadata.gz: e1cdd901ff82b3f2c2d5f0be65175a03b534a6d3a401c34a9c85861401ea2cf416162a1045fe34b2e8cf0ef79a3a220a1c1e9f87b81fff05d06b973735974227
-  data.tar.gz: 721d41c939a3e00321508a4c36033221e58b6f67a6fd1f0ee9df090e1ae289fafe0b65c94e0f07b7e20c6b3c4bad375463b763f00d440c7a96c29af529e6cde2
+  metadata.gz: a5c24bea5bf8a7b816eac7fc88b1d0b38bbb29fe69e0d5a02d9534b96a75b714f15df4ba65fba100c24d112267fee13b25ed13b3eaa096bf59d31ef8f028da72
+  data.tar.gz: 97c13bb474926f97522e39c43fb2f9bf8578d163cef77a697f0c43331fe311f314d207cec80d11ee606f88c073d45a0cc211f1b2cd3f1735fae9ba139b2d40a5

data/lib/rails_code_auditor/llm_client.rb

@@ -9,12 +9,13 @@ module RailsCodeAuditor
         tool.dup.tap do |entry|
           if entry.is_a?(Hash) && entry[:details].is_a?(String)
             entry[:details] = entry[:details].slice(0, 500) # Trim details to avoid LLM overload
-            entry[:details].gsub!(/\e\[[\d;]*m/, '')         # Remove ANSI color codes
+            entry[:details].gsub!(/\e\[[\d;]*m/, "") # Remove ANSI color codes
           end
         end
       end
     end
-    def self.score_with_llm(json_results)
+
+    def self.score_with_llm(json_results, endpoint: "http://localhost:11434/api/generate", model: "llama3")
       sanitized = sanitize_results(json_results)
       puts "[*] Scoring with LLM (LLaMA3)..."
 
@@ -34,9 +35,9 @@ module RailsCodeAuditor
         #{JSON.pretty_generate(sanitized)}
       PROMPT
 
-      uri = URI("http://localhost:11434/api/generate")
+      uri = URI(endpoint)
       body = {
-        model: "llama3",
+        model: model,
         prompt: prompt,
         stream: false
       }
@@ -48,24 +49,21 @@ module RailsCodeAuditor
       # Try to extract JSON from any surrounding text
       json_match = raw_output.match(/\{.*\}/m)
 
-      if json_match
-        parsed_scores = JSON.parse(json_match[0])
-        scored_results = parsed_scores.transform_values do |score|
-          remark = case score
-                  when 90..100 then "Excellent"
-                  when 75..89  then "Good"
-                  when 60..74  then "Average"
-                  else              "Needs Improvement"
-                  end
-          { score: score, remark: remark }
-        end
-        scored_results
-      else
-        raise "Response did not contain valid JSON"
+      raise "Response did not contain valid JSON" unless json_match
+
+      parsed_scores = JSON.parse(json_match[0])
+      parsed_scores.transform_values do |score|
+        remark = case score
+                 when 90..100 then "Excellent"
+                 when 75..89  then "Good"
+                 when 60..74  then "Average"
+                 else "Needs Improvement"
+                 end
+        { score: score, remark: remark }
       end
-    rescue => e
+    rescue StandardError => e
       puts "[!] LLM scoring failed: #{e.message}"
       nil
     end
   end
-end
+end

data/lib/rails_code_auditor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsCodeAuditor
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

data/lib/rails_code_auditor.rb

@@ -15,17 +15,41 @@ module RailsCodeAuditor
   def self.run(args)
     puts "[*] Running Rails Code Auditor..."
 
+    # Parse CLI options
+    use_llm     = args.include?("--use-llm")
+    model_arg   = args[args.index("--llm-model") + 1] if args.include?("--llm-model")
+    endpoint_arg = args[args.index("--llm-endpoint") + 1] if args.include?("--llm-endpoint")
+    format_arg = args[args.index("--format") + 1] if args.include?("--format")
+    output_arg = args[args.index("--output") + 1] if args.include?("--output")
+
     raw_results = Analyzer.run_all
     results = ReportGenerator.normalize(raw_results)
     results[:simplecov] = SimpleCovRunner.run
-    scores = if args.include?("--use-llm")
-               LlmClient.score_with_llm(results) || Scorer.score(results)
+    scores = if use_llm
+               LlmClient.score_with_llm(results, model: model_arg || "llama3",
+                                                 endpoint: endpoint_arg || "http://localhost:11434/api/generate") || Scorer.score(results)
              else
                Scorer.score(results)
              end
+
+    if format_arg == "json"
+      json_output = JSON.pretty_generate({ results: results, scores: scores })
+      if output_arg
+        File.write(output_arg, json_output)
+        puts "[✓] JSON report saved to #{output_arg}"
+      else
+        puts json_output
+      end
+      return
+    end
     graphs = Grapher.generate(scores)
     html_pdf_paths = USE_GROVER ? HtmlToPdfConverter.convert_all : []
-    PdfGenerator.generate(results, scores, graphs, html_pdf_paths)
-    puts "[✓] Audit complete. PDF report generated."
+    pdf_path = PdfGenerator.generate(results, scores, graphs, html_pdf_paths)
+    if output_arg
+      FileUtils.mv(pdf_path, output_arg)
+      puts "[✓] PDF report saved to #{output_arg}"
+    else
+      puts "[✓] Audit complete. PDF report generated at #{pdf_path}"
+    end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: rails_code_auditor
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - sivamanikandan
 bindir: exe
 cert_chain: []
-date: 2025-07-24 00:00:00.000000000 Z
+date: 2025-08-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: brakeman
@@ -247,7 +247,7 @@ licenses:
 metadata:
   allowed_push_host: https://rubygems.org
   homepage_uri: https://github.com/railsfactory-sivamanikandan/rails_code_auditor
-  source_code_uri: https://github.com/railsfactory-sivamanikandan/rails_code_auditor
+  documentation_uri: https://github.com/railsfactory-sivamanikandan/rails_code_auditor/blob/main/README.md
   changelog_uri: https://github.com/railsfactory-sivamanikandan/rails_code_auditor/CHANGELOG.md
 rdoc_options: []
 require_paths: