ai_rspec_writer 1.0.5 → 1.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5a56acb3b0f314948afc0cf7619369116cf7c6759ae3015af150e6476cc7ac46
-  data.tar.gz: 4981ef4684fe723b591ccdfed938e5684e62180e88157bbcfb5cd484706be8c6
+  metadata.gz: be389107c5f70774c3581dbf04170412e8a042adfca34873d9e7d3f73f6b1e33
+  data.tar.gz: 7164b14789a8de01a0e27487f0e66e6c14de572638b956f444bcb262e60a4464
 SHA512:
-  metadata.gz: a4de160d97ab90474b3062d8b9865bdb459e722420fc52b2fd5085349512000310bc01eef750cb9c4a6d65c256e0d4fbf63900aca9898d41f71702813b94bc3f
-  data.tar.gz: 8cc60641cd834cdd101de19863bd1281a21d001d3d3d85880dbefbfbc3feb84edc141a0deac4ddd6955ca627c7ecac257cea5a640ef95b2538d662320e3b5223
+  metadata.gz: b38dd7f0a8d2a04e3176a6add0bce16ca0b64daae60305f930f8851129e89ed316c1fbe03164cc980f38be2b38ca32d2ebe2ce583d2df5fb21fc56bfd03bce8e
+  data.tar.gz: d197257e867ae0cdeb5b06f2ec262415140575c1c9d77a16b54dd1e93a25d2182644bb71314afb89d595068564d3e8c49132e51d5ecfbb1af561ae723da2535a

data/bin/ai_rspec_writer

@@ -6,7 +6,7 @@ require "optparse"
 require_relative "../lib/ai_rspec_writer"
 require_relative "../lib/ai_rspec_writer/schema_extractor"
 
-if AiRspecWriter.configuration.chatgpt_api_key.nil? || AiRspecWriter.configuration.gemini_api_key.nil?
+if AiRspecWriter.configuration.chatgpt_api_key.nil? && AiRspecWriter.configuration.gemini_api_key.nil?
   puts "❌ API key is missing.".red
   puts "ℹ️  Set it in `.env`, or use `export CHATGPT_API_KEY=your-key` `export GEMINI_API_KEY=your-key`.".yellow
   exit 1

data/lib/ai_rspec_writer/ai_providers/gemini_generator.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
-require 'gemini-ai'
+require 'net/http'
+require 'json'
+require 'uri'
 require 'colorize'
 require 'tty-spinner'
 
@@ -8,20 +10,19 @@ module AIRspecWriter
   module AIProviders
     # Generates RSpec tests using Google's Gemini AI API
     class GeminiGenerator
+      BASE_URL = 'https://generativelanguage.googleapis.com/v1beta/models'
+      
       def initialize(ec: nil, schema: nil)
         @extra_comment = ec
         @schema = schema
-        @client = Gemini.new(
-          credentials: {
-            service: 'generative-language-api',
-            api_key: AiRspecWriter.configuration.gemini_api_key
-          },
-          options: { model: AiRspecWriter.configuration.gemini_model, server_sent_events: true }
-        )
+        @api_key = AiRspecWriter.configuration.gemini_api_key
+        @model = AiRspecWriter.configuration.gemini_model
+        
+        validate_configuration
       end
 
       def generate_spec(file_content)
-        fixed_promt = "Write RSpec tests for this Ruby code with all factory files and rubocop standards:"
+        fixed_prompt = "Write RSpec tests for this Ruby code with all factory files and rubocop standards:"
         my_schema = "Check my table definition details: #{@schema}" if @schema
 
         # Define loading spinner
@@ -31,19 +32,13 @@ module AIRspecWriter
         spinner.auto_spin
 
         begin
-          result = @client.generate_content({
-                                              contents: { 
-                                                role: "user", 
-                                                parts: { 
-                                                  text: "#{fixed_promt}\n\n#{file_content}\n\n#{@extra_comment}\n\n#{my_schema}" 
-                                                } 
-                                              }
-                                            })
-                                                  
-          if result["candidates"].first["content"]["parts"].first["text"].present?
+          content_text = build_content_text(fixed_prompt, file_content, my_schema)
+          response_text = call_gemini_api(content_text)
+          
+          if response_text&.strip&.length&.positive?
             # Stop the spinner when AI response is ready
             spinner.success("✅ Done!")
-            return result["candidates"].first["content"]["parts"].first["text"]
+            return response_text
           else
             spinner.error('(Gemini AI returned an empty response.)')
             return nil
@@ -53,6 +48,94 @@ module AIRspecWriter
           exit 1
         end
       end
+
+      private
+
+      def validate_configuration
+        raise ConfigurationError, "Gemini API key is required" if @api_key.nil? || @api_key.empty?
+        raise ConfigurationError, "Gemini model is required" if @model.nil? || @model.empty?
+      end
+
+      def build_content_text(fixed_prompt, file_content, my_schema)
+        content_parts = [fixed_prompt, file_content, @extra_comment, my_schema].compact
+        content_parts.join("\n\n")
+      end
+
+      def call_gemini_api(content_text)
+        url = build_api_url
+        payload = build_payload(content_text)
+        
+        response = make_http_request(url, payload)
+        parse_response(response)
+      end
+
+      def build_api_url
+        "#{BASE_URL}/#{@model}:generateContent?key=#{@api_key}"
+      end
+
+      def build_payload(content_text)
+        {
+          contents: [
+            {
+              role: "user",
+              parts: [
+                { text: content_text }
+              ]
+            }
+          ],
+          generationConfig: {
+            temperature: 0.7,
+            topK: 40,
+            topP: 0.95,
+            maxOutputTokens: 8192
+          }
+        }
+      end
+
+      def make_http_request(url, payload)
+        uri = URI(url)
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.read_timeout = 60 # Set timeout for long responses
+        
+        request = Net::HTTP::Post.new(uri)
+        request['Content-Type'] = 'application/json'
+        request.body = payload.to_json
+        
+        response = http.request(request)
+        
+        unless response.is_a?(Net::HTTPSuccess)
+          raise ApiError, "HTTP #{response.code}: #{response.body}"
+        end
+        
+        response
+      end
+
+      def parse_response(response)
+        data = JSON.parse(response.body)
+        
+        if data['error']
+          raise ApiError, "API error: #{data['error']['message']}"
+        end
+        
+        # Extract the generated text following the same pattern as your original code
+        candidates = data['candidates']
+        return nil if candidates.nil? || candidates.empty?
+        
+        first_candidate = candidates.first
+        content = first_candidate['content']
+        return nil if content.nil?
+        
+        parts = content['parts']
+        return nil if parts.nil? || parts.empty?
+        
+        parts.first['text']
+      rescue JSON::ParserError => e
+        raise ApiError, "Failed to parse response: #{e.message}"
+      end
+
+      class ApiError < StandardError; end
+      class ConfigurationError < StandardError; end
     end
   end
-end
+end

data/lib/ai_rspec_writer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiRspecWriter
-  VERSION = "1.0.5"
+  VERSION = "1.0.6"
 end

data/lib/ai_rspec_writer.rb

@@ -4,6 +4,11 @@ require_relative "ai_rspec_writer/generator"
 require_relative "ai_rspec_writer/spec_file_handler"
 require "colorize"
 
+# Load Rails configuration only if inside a Rails application
+if defined?(Rails::Generators)
+  require_relative "generators/ai_rspec_writer/install_generator"
+end
+
 module AiRspecWriter
   class Error < StandardError; end
   class << self

data/lib/generators/ai_rspec_writer/install_generator.rb

@@ -0,0 +1,104 @@
+require "rails/generators"
+require "tty-spinner"
+
+module AiRspecWriter
+  class InstallGenerator < Rails::Generators::Base
+    source_root File.expand_path("templates", __dir__)
+
+    def install_ai_rspec_writer
+      puts "\n🚀 Starting `ai_rspec_writer` setup...\n".yellow
+      check_rspec_installed
+      configure_spec_helper
+      configure_rails_helper
+      puts "\n✅ `ai_rspec_writer` setup completed successfully!\n".green
+    end
+
+    private
+
+    # Check if RSpec is installed, otherwise install it
+    def check_rspec_installed
+      spinner = TTY::Spinner.new("🔍 Checking if RSpec is installed... :spinner")
+      spinner.auto_spin
+
+      if File.exist?("spec/rails_helper.rb") && File.exist?("spec/spec_helper.rb")
+        spinner.success("✅ RSpec is already installed.")
+      else
+        spinner.stop("⚠️ RSpec not found. Installing...".red)
+        generate "rspec:install"
+        puts "✅ RSpec installed successfully.".green
+      end
+    end
+
+    # Append content to spec_helper.rb
+    def configure_spec_helper
+      spec_helper_path = "spec/spec_helper.rb"
+      spinner = TTY::Spinner.new("🔧 Configuring `spec_helper.rb`... :spinner")
+      spinner.auto_spin
+
+      if File.exist?(spec_helper_path)
+        original_content = File.read(spec_helper_path)
+        new_content = "#{spec_file_content}\n\n" + original_content
+        File.write(spec_helper_path, new_content)
+        spinner.success("✅ Updated `spec_helper.rb` with AI RSpec settings.")
+      else
+        spinner.error("⚠️ `spec_helper.rb` not found. Skipping...".red)
+      end
+    end
+
+    # Append content to rails_helper.rb after `RSpec.configure do |config|`
+    def configure_rails_helper
+      rails_helper_path = "spec/rails_helper.rb"
+      target_line = "RSpec.configure do |config|"
+      spinner = TTY::Spinner.new("🔧 Configuring `rails_helper.rb`... :spinner")
+      spinner.auto_spin
+
+      if File.exist?(rails_helper_path)
+        content = File.readlines(rails_helper_path)
+        index = content.find_index { |line| line.strip == target_line.strip }
+
+        if index
+          content.insert(index + 1, "#{rails_helper_content}\n")
+          File.write(rails_helper_path, content.join)
+          spinner.success("✅ Successfully inserted configurations into `rails_helper.rb`.")
+        else
+          spinner.error("⚠️ `RSpec.configure` block not found in `rails_helper.rb`. Skipping...".red)
+        end
+      else
+        spinner.error("⚠️ `rails_helper.rb` not found. Skipping...".red)
+      end
+    end
+
+    # Content for spec_helper.rb
+    def spec_file_content
+      <<~RUBY
+        # Auto-generated by ai_rspec_writer
+        require 'simplecov'
+        SimpleCov.start 'rails' do
+          add_filter '/spec/'
+          add_filter '/config/'
+          add_filter '/vendor/'
+          add_filter '/db/'
+          add_filter '/bin/'
+        end
+        # Auto-generated by ai_rspec_writer
+      RUBY
+    end
+
+    # Content for rails_helper.rb with proper indentation
+    def rails_helper_content
+      <<~RUBY.indent(2)
+        # Auto-generated by ai_rspec_writer
+        config.include FactoryBot::Syntax::Methods
+
+        # Include Shoulda Matchers
+        Shoulda::Matchers.configure do |shoulda_config|
+          shoulda_config.integrate do |with|
+            with.test_framework :rspec
+            with.library :rails
+          end
+        end
+        # Auto-generated by ai_rspec_writer
+      RUBY
+    end
+  end
+end

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: ai_rspec_writer
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.6
 platform: ruby
 authors:
 - Aniruddha Mirajkar
-autorequire: 
 bindir: bin
 cert_chain: []
-date: 2025-03-04 00:00:00.000000000 Z
+date: 2025-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec-rails
@@ -157,6 +156,7 @@ files:
 - lib/ai_rspec_writer/schema_extractor.rb
 - lib/ai_rspec_writer/spec_file_handler.rb
 - lib/ai_rspec_writer/version.rb
+- lib/generators/ai_rspec_writer/install_generator.rb
 homepage: https://github.com/aniruddhami/ai_rspec_writer
 licenses:
 - MIT
@@ -165,7 +165,6 @@ metadata:
   source_code_uri: https://github.com/aniruddhami/ai_rspec_writer
   changelog_uri: https://github.com/aniruddhami/ai_rspec_writer/blob/main/CHANGELOG.md
   github_repo: https://github.com/aniruddhami/ai_rspec_writer
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -180,8 +179,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.1
-signing_key: 
+rubygems_version: 3.6.6
 specification_version: 4
 summary: A gem to generate RSpec tests using AI (ChatGPT, GeminiAI).
 test_files: []