claude-on-rails 0.1.3 → 0.2.0

data/lib/claude_on_rails/swarm_builder.rb

@@ -1,192 +1,184 @@
+# frozen_string_literal: true
+
 module ClaudeOnRails
   class SwarmBuilder
     attr_reader :project_analysis
-    
+
     def initialize(project_analysis)
       @project_analysis = project_analysis
     end
-    
+
     def build
       {
         version: 1,
         swarm: {
-          name: "Rails Development Team",
-          main: "architect",
+          name: 'Rails Development Team',
+          main: 'architect',
           instances: build_instances
         }
       }
     end
-    
+
     private
-    
+
     def build_instances
       instances = {}
-      
+
       # Always include architect
       instances[:architect] = build_architect
-      
+
       # Core agents
       instances[:models] = build_models_agent
       instances[:controllers] = build_controllers_agent
-      
+
       # Conditional agents
-      unless project_analysis[:api_only]
-        instances[:views] = build_views_agent
-      end
-      
-      if project_analysis[:api_only]
-        instances[:api] = build_api_agent
-      end
-      
-      if project_analysis[:has_graphql]
-        instances[:graphql] = build_graphql_agent
-      end
-      
-      if project_analysis[:has_turbo] && !project_analysis[:api_only]
-        instances[:stimulus] = build_stimulus_agent
-      end
-      
+      instances[:views] = build_views_agent unless project_analysis[:api_only]
+
+      instances[:api] = build_api_agent if project_analysis[:api_only]
+
+      instances[:graphql] = build_graphql_agent if project_analysis[:has_graphql]
+
+      instances[:stimulus] = build_stimulus_agent if project_analysis[:has_turbo] && !project_analysis[:api_only]
+
       # Supporting agents
       instances[:services] = build_services_agent
       instances[:jobs] = build_jobs_agent
-      
-      if project_analysis[:test_framework]
-        instances[:tests] = build_tests_agent
-      end
-      
+
+      instances[:tests] = build_tests_agent if project_analysis[:test_framework]
+
       instances[:devops] = build_devops_agent
-      
+
       instances
     end
-    
+
     def build_architect
-      connections = ["models", "controllers"]
-      connections << "views" unless project_analysis[:api_only]
-      connections << "api" if project_analysis[:api_only]
-      connections << "graphql" if project_analysis[:has_graphql]
-      connections << "services"
-      connections << "jobs"
-      connections << "tests" if project_analysis[:test_framework]
-      connections << "devops"
-      
+      connections = %w[models controllers]
+      connections << 'views' unless project_analysis[:api_only]
+      connections << 'api' if project_analysis[:api_only]
+      connections << 'graphql' if project_analysis[:has_graphql]
+      connections << 'services'
+      connections << 'jobs'
+      connections << 'tests' if project_analysis[:test_framework]
+      connections << 'devops'
+
       {
         description: "Rails architect coordinating #{project_analysis[:api_only] ? 'API' : 'full-stack'} development",
-        directory: ".",
+        directory: '.',
         model: ClaudeOnRails.configuration.default_model,
         connections: connections,
-        prompt_file: ".claude-on-rails/prompts/architect.md",
+        prompt_file: '.claude-on-rails/prompts/architect.md',
         vibe: ClaudeOnRails.configuration.vibe_mode
       }
     end
-    
+
     def build_models_agent
       {
-        description: "ActiveRecord models, migrations, and database optimization specialist",
-        directory: "./app/models",
+        description: 'ActiveRecord models, migrations, and database optimization specialist',
+        directory: './app/models',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/models.md"
+        prompt_file: '.claude-on-rails/prompts/models.md'
       }
     end
-    
+
     def build_controllers_agent
-      connections = ["services"]
-      connections << "api" if project_analysis[:api_only]
-      
+      connections = ['services']
+      connections << 'api' if project_analysis[:api_only]
+
       {
-        description: "Rails controllers, routing, and request handling specialist",
-        directory: "./app/controllers",
+        description: 'Rails controllers, routing, and request handling specialist',
+        directory: './app/controllers',
         model: ClaudeOnRails.configuration.default_model,
         connections: connections.empty? ? nil : connections,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/controllers.md"
+        prompt_file: '.claude-on-rails/prompts/controllers.md'
       }.compact
     end
-    
+
     def build_views_agent
       connections = []
-      connections << "stimulus" if project_analysis[:has_turbo]
-      
+      connections << 'stimulus' if project_analysis[:has_turbo]
+
       {
-        description: "Rails views, layouts, partials, and asset pipeline specialist",
-        directory: "./app/views",
+        description: 'Rails views, layouts, partials, and asset pipeline specialist',
+        directory: './app/views',
         model: ClaudeOnRails.configuration.default_model,
         connections: connections.empty? ? nil : connections,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/views.md"
+        prompt_file: '.claude-on-rails/prompts/views.md'
       }.compact
     end
-    
+
     def build_api_agent
       {
-        description: "RESTful API design, serialization, and versioning specialist",
-        directory: "./app/controllers/api",
+        description: 'RESTful API design, serialization, and versioning specialist',
+        directory: './app/controllers/api',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/api.md"
+        prompt_file: '.claude-on-rails/prompts/api.md'
       }
     end
-    
+
     def build_graphql_agent
       {
-        description: "GraphQL schema, resolvers, and mutations specialist",
-        directory: "./app/graphql",
+        description: 'GraphQL schema, resolvers, and mutations specialist',
+        directory: './app/graphql',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/graphql.md"
+        prompt_file: '.claude-on-rails/prompts/graphql.md'
       }
     end
-    
+
     def build_stimulus_agent
       {
-        description: "Stimulus.js controllers and Turbo integration specialist",
-        directory: "./app/javascript",
+        description: 'Stimulus.js controllers and Turbo integration specialist',
+        directory: './app/javascript',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/stimulus.md"
+        prompt_file: '.claude-on-rails/prompts/stimulus.md'
       }
     end
-    
+
     def build_services_agent
       {
-        description: "Service objects, business logic, and design patterns specialist",
-        directory: "./app/services",
+        description: 'Service objects, business logic, and design patterns specialist',
+        directory: './app/services',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/services.md"
+        prompt_file: '.claude-on-rails/prompts/services.md'
       }
     end
-    
+
     def build_jobs_agent
       {
-        description: "Background jobs, ActiveJob, and async processing specialist",
-        directory: "./app/jobs",
+        description: 'Background jobs, ActiveJob, and async processing specialist',
+        directory: './app/jobs',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/jobs.md"
+        prompt_file: '.claude-on-rails/prompts/jobs.md'
       }
     end
-    
+
     def build_tests_agent
       test_dir = project_analysis[:test_framework] == 'RSpec' ? './spec' : './test'
-      
+
       {
         description: "#{project_analysis[:test_framework]} testing, factories, and test coverage specialist",
         directory: test_dir,
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/tests.md"
+        prompt_file: '.claude-on-rails/prompts/tests.md'
       }
     end
-    
+
     def build_devops_agent
       {
-        description: "Deployment, Docker, CI/CD, and production configuration specialist",
-        directory: "./config",
+        description: 'Deployment, Docker, CI/CD, and production configuration specialist',
+        directory: './config',
         model: ClaudeOnRails.configuration.default_model,
         allowed_tools: %w[Read Edit Write Bash Grep Glob LS],
-        prompt_file: ".claude-on-rails/prompts/devops.md"
+        prompt_file: '.claude-on-rails/prompts/devops.md'
       }
     end
   end
-end
+end

data/lib/claude_on_rails/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module ClaudeOnRails
-  VERSION = "0.1.3"
-end
+  VERSION = '0.2.0'
+end

data/lib/claude_on_rails.rb

@@ -1,29 +1,32 @@
-require "claude_on_rails/version"
-require "claude_on_rails/configuration"
-require "claude_on_rails/project_analyzer"
-require "claude_on_rails/swarm_builder"
+# frozen_string_literal: true
+
+require 'claude_on_rails/version'
+require 'claude_on_rails/configuration'
+require 'claude_on_rails/project_analyzer'
+require 'claude_on_rails/swarm_builder'
+require 'claude_on_rails/mcp_support'
+
+# Load railtie if Rails is defined
+require 'claude_on_rails/railtie' if defined?(Rails::Railtie)
 
 module ClaudeOnRails
   class Error < StandardError; end
-  
+
   class << self
-    attr_accessor :configuration
-    
     def configure
-      self.configuration ||= Configuration.new
       yield(configuration) if block_given?
     end
-    
+
     def configuration
       @configuration ||= Configuration.new
     end
-    
+
     def analyze_project(root_path = Rails.root)
       ProjectAnalyzer.new(root_path).analyze
     end
-    
+
     def build_swarm(project_analysis)
       SwarmBuilder.new(project_analysis).build
     end
   end
-end
+end

data/lib/generators/claude_on_rails/swarm/swarm_generator.rb

@@ -1,104 +1,164 @@
-require "rails/generators/base"
-require "claude_on_rails"
+# frozen_string_literal: true
+
+require 'rails/generators/base'
+require 'claude_on_rails'
 
 module ClaudeOnRails
   module Generators
     class SwarmGenerator < Rails::Generators::Base
-      source_root File.expand_path("templates", __dir__)
-      
+      source_root File.expand_path('templates', __dir__)
+
       class_option :api_only, type: :boolean, default: false,
-                   desc: "Generate swarm for API-only Rails application"
-      
+                              desc: 'Generate swarm for API-only Rails application'
+
       class_option :skip_tests, type: :boolean, default: false,
-                   desc: "Skip test agent in swarm configuration"
-                   
+                                desc: 'Skip test agent in swarm configuration'
+
       class_option :graphql, type: :boolean, default: false,
-                   desc: "Include GraphQL specialist agent"
-                   
+                             desc: 'Include GraphQL specialist agent'
+
       class_option :turbo, type: :boolean, default: true,
-                   desc: "Include Turbo/Stimulus specialist agents"
-      
+                           desc: 'Include Turbo/Stimulus specialist agents'
+
+      class_option :mcp_server, type: :boolean, default: true,
+                                desc: 'Include Rails MCP Server for enhanced documentation access'
+
       def analyze_project
-        say "Analyzing Rails project structure...", :green
+        say 'Analyzing Rails project structure...', :green
         @project_analysis = ClaudeOnRails.analyze_project(Rails.root)
-        
+
         # Auto-detect features
         @api_only = options[:api_only] || @project_analysis[:api_only]
         @has_graphql = options[:graphql] || @project_analysis[:has_graphql]
         @has_turbo = options[:turbo] && !@api_only
         @skip_tests = options[:skip_tests]
         @test_framework = @project_analysis[:test_framework]
-        
+
+        # Check for Rails MCP Server
+        @include_mcp_server = options[:mcp_server] && ClaudeOnRails::MCPSupport.available?
+
         say "Project type: #{@api_only ? 'API-only' : 'Full-stack Rails'}", :cyan
         say "Test framework: #{@test_framework}", :cyan if @test_framework
         say "GraphQL detected: #{@has_graphql ? 'Yes' : 'No'}", :cyan
-        
+        say "Rails MCP Server: #{@include_mcp_server ? 'Available' : 'Not available'}", :cyan
+
+        # Offer MCP setup if enabled but not available
+        offer_mcp_setup if options[:mcp_server] && !@include_mcp_server
+
         # Show which agents will be created
         say "\nAgents to be created:", :yellow
         agents.each do |agent|
           say "  - #{agent}", :cyan
         end
       end
-      
+
       def create_swarm_config
-        say "Generating swarm configuration...", :green
-        template "swarm.yml.erb", "claude-swarm.yml"
+        say 'Generating swarm configuration...', :green
+        template 'swarm.yml.erb', 'claude-swarm.yml'
       end
-      
+
       def create_claude_md
-        say "Creating CLAUDE.md configuration...", :green
-        template "CLAUDE.md.erb", "CLAUDE.md"
+        # Always create the ClaudeOnRails context file
+        template 'claude_on_rails_context.md', '.claude-on-rails/context.md'
+
+        if File.exist?('CLAUDE.md')
+          say 'CLAUDE.md already exists - adding ClaudeOnRails file reference...', :yellow
+
+          existing_content = File.read('CLAUDE.md')
+
+          # Check if file reference already exists
+          if existing_content.include?('.claude-on-rails/context.md')
+            say '✓ CLAUDE.md already references ClaudeOnRails context', :green
+          else
+            file_reference = "\n/file:.claude-on-rails/context.md\n"
+            append_to_file 'CLAUDE.md', file_reference
+            say '✓ Added ClaudeOnRails context reference to existing CLAUDE.md', :green
+          end
+        else
+          say 'Creating CLAUDE.md configuration...', :green
+          template 'CLAUDE.md.erb', 'CLAUDE.md'
+        end
       end
-      
+
       def create_agent_prompts
-        say "Setting up agent-specific prompts...", :green
-        directory "prompts", ".claude-on-rails/prompts"
+        say 'Setting up agent-specific prompts...', :green
+        directory 'prompts', '.claude-on-rails/prompts'
       end
-      
+
       def update_gitignore
-        say "Updating .gitignore...", :green
-        append_to_file ".gitignore", "\n# ClaudeOnRails\n.claude-on-rails/sessions/\nswarm.log\n"
+        say 'Updating .gitignore...', :green
+        gitignore_path = Rails.root.join('.gitignore')
+
+        # Create .gitignore if it doesn't exist
+        create_file '.gitignore', '' unless File.exist?(gitignore_path)
+
+        append_to_file '.gitignore', "\n# ClaudeOnRails\n.claude-on-rails/sessions/\n.claude-swarm/\nclaude-swarm.log\n"
       end
-      
+
       def display_next_steps
         say "\n✅ ClaudeOnRails swarm configuration created!", :green
         say "\nNext steps:", :yellow
-        say "1. Review and customize claude-swarm.yml for your project"
-        say "2. Start your Rails development swarm:"
-        say "   claude-swarm", :cyan
+
+        # Remove this section since we now handle it interactively in analyze_project
+
+        say '1. Review and customize claude-swarm.yml for your project'
+        say '2. Start your Rails development swarm:'
+        say '   claude-swarm', :cyan
         say "\nOnce the swarm is running, just describe what you want to build:"
         say '   > Add user authentication with social login', :cyan
         say "\nThe swarm will automatically coordinate the implementation across all layers!"
       end
-      
+
       private
-      
+
       def agents
         @agents ||= build_agent_list
       end
-      
+
       def build_agent_list
-        list = ["architect"]
-        list << "models" if File.directory?(Rails.root.join("app/models"))
-        list << "controllers" if File.directory?(Rails.root.join("app/controllers"))
-        list << "views" if !@api_only && File.directory?(Rails.root.join("app/views"))
-        list << "api" if @api_only && File.directory?(Rails.root.join("app/controllers/api"))
-        list << "graphql" if @has_graphql && File.directory?(Rails.root.join("app/graphql"))
-        list << "stimulus" if @has_turbo && File.directory?(Rails.root.join("app/javascript"))
-        list << "services" if File.directory?(Rails.root.join("app/services"))
-        list << "jobs" if File.directory?(Rails.root.join("app/jobs"))
-        
-        if !@skip_tests
-          if @test_framework == 'RSpec' && File.directory?(Rails.root.join("spec"))
-            list << "tests"
-          elsif @test_framework == 'Minitest' && File.directory?(Rails.root.join("test"))
-            list << "tests"
+        list = ['architect']
+        list << 'models' if File.directory?(Rails.root.join('app/models'))
+        list << 'controllers' if File.directory?(Rails.root.join('app/controllers'))
+        list << 'views' if !@api_only && File.directory?(Rails.root.join('app/views'))
+        list << 'api' if @api_only && File.directory?(Rails.root.join('app/controllers/api'))
+        list << 'graphql' if @has_graphql && File.directory?(Rails.root.join('app/graphql'))
+        list << 'stimulus' if @has_turbo && File.directory?(Rails.root.join('app/javascript'))
+        list << 'services' if File.directory?(Rails.root.join('app/services'))
+        list << 'jobs' if File.directory?(Rails.root.join('app/jobs'))
+
+        unless @skip_tests
+          case @test_framework
+          when 'RSpec'
+            list << 'tests' if File.directory?(Rails.root.join('spec'))
+          when 'Minitest'
+            list << 'tests' if File.directory?(Rails.root.join('test'))
           end
         end
-        
-        list << "devops" if File.directory?(Rails.root.join("config"))
+
+        list << 'devops' if File.directory?(Rails.root.join('config'))
         list
       end
+
+      def offer_mcp_setup
+        say "\n🎯 Rails MCP Server Enhancement Available!", :yellow
+        say "Rails MCP Server provides your AI agents with real-time Rails documentation.", :cyan
+
+        if yes?("Would you like to set it up now? (Y/n)", :green)
+          say "\nStarting Rails MCP Server setup...", :green
+          system('bundle exec rake claude_on_rails:setup_mcp')
+
+          # Re-check availability after setup
+          @include_mcp_server = ClaudeOnRails::MCPSupport.available?
+
+          if @include_mcp_server
+            say "\n✓ Rails MCP Server is now available!", :green
+          else
+            say "\nSetup was not completed. Continuing without MCP Server.", :yellow
+          end
+        else
+          say "\nYou can set it up later with: bundle exec rake claude_on_rails:setup_mcp", :cyan
+        end
+      end
     end
   end
-end
+end

data/lib/generators/claude_on_rails/swarm/templates/CLAUDE.md.erb

@@ -1,101 +1,3 @@
-# <%= Rails.application.class.module_parent_name %> Rails Development with ClaudeOnRails
+## ClaudeOnRails Configuration
 
-This project uses ClaudeOnRails to create an intelligent swarm of AI agents specialized in different aspects of Rails development.
-
-## Project Overview
-
-- **Application**: <%= Rails.application.class.module_parent_name %>
-- **Rails Version**: <%= Rails.version %>
-- **Ruby Version**: <%= RUBY_VERSION %>
-- **Type**: <%= @api_only ? 'API-only' : 'Full-stack Rails application' %>
-- **Test Framework**: <%= @test_framework || 'Not detected' %>
-<% if @has_graphql -%>
-- **GraphQL**: Enabled
-<% end -%>
-
-## How to Use
-
-Simply describe what you want to build or fix, and the swarm will automatically coordinate the implementation:
-
-```bash
-# Start the swarm
-claude-swarm orchestrate
-
-# Then just describe your task
-claude "Add user authentication with email confirmation"
-claude "Optimize the dashboard queries that are running slowly"
-claude "Create an API endpoint for mobile app integration"
-```
-
-## Swarm Architecture
-
-The following specialized agents work together to implement your requests:
-
-- **Architect**: Coordinates all development and makes high-level decisions
-- **Models**: Handles ActiveRecord models, migrations, and database design
-- **Controllers**: Manages request handling, routing, and controller logic
-<% unless @api_only -%>
-- **Views**: Creates and maintains views, layouts, and partials
-<% end -%>
-<% if @api_only -%>
-- **API**: Designs RESTful endpoints and handles serialization
-<% end -%>
-<% if @has_graphql -%>
-- **GraphQL**: Manages GraphQL schemas, types, and resolvers
-<% end -%>
-<% if @has_turbo -%>
-- **Stimulus**: Implements interactive features with Stimulus controllers
-<% end -%>
-- **Services**: Extracts business logic into service objects
-- **Jobs**: Handles background processing and async tasks
-<% unless @skip_tests -%>
-- **Tests**: Ensures comprehensive test coverage with <%= @test_framework %>
-<% end -%>
-- **DevOps**: Manages deployment and production configurations
-
-## Project Conventions
-
-### Code Style
-- Follow Rails conventions and best practices
-- Use RuboCop for Ruby style enforcement
-- Prefer clarity over cleverness
-- Write self-documenting code
-
-### Testing
-<% if @test_framework == 'RSpec' -%>
-- RSpec for all tests
-- Factory Bot for test data
-- Request specs for API endpoints
-- System specs for user interactions
-<% else -%>
-- Minitest for all tests
-- Fixtures or factories for test data
-- Integration tests for user flows
-- Unit tests for models and services
-<% end -%>
-
-### Git Workflow
-- Feature branches for new work
-- Descriptive commit messages
-- PR reviews before merging
-- Keep main branch deployable
-
-## Custom Patterns
-
-Add your project-specific patterns and conventions here:
-
-```yaml
-# Example: Custom service object pattern
-Services:
-  Pattern: Command pattern with Result objects
-  Location: app/services/
-  Naming: VerbNoun (e.g., CreateOrder, SendEmail)
-  Testing: Unit tests with mocked dependencies
-```
-
-## Notes
-
-- This configuration was generated by ClaudeOnRails
-- Customize agent prompts in `.claude-on-rails/prompts/`
-- Update this file with project-specific conventions
-- The swarm learns from your codebase patterns
+You are working on <%= Rails.application.class.module_parent_name %>, a <%= @api_only ? 'Rails API' : 'Rails' %> application. Review the ClaudeOnRails context file at @.claude-on-rails/context.md