hokipoki 0.1.2 → 0.1.3

data/lib/generators/hive_mind/install_generator.rb

@@ -0,0 +1,828 @@
+# frozen_string_literal: true
+
+require 'rails/generators'
+require 'rails/generators/active_record'
+require 'tty-prompt'
+require 'pastel'
+
+module HiveMind
+  class InstallGenerator < Rails::Generators::Base
+    include ActiveRecord::Generators::Migration
+
+    source_root File.expand_path('templates', __dir__)
+
+    class_option :minimal, type: :boolean, default: false, desc: 'Install minimal HiveMind setup'
+    class_option :full, type: :boolean, default: false, desc: 'Install complete revolutionary stack'
+    class_option :claude, type: :boolean, default: false, desc: 'Install HiveMind + Claude integration'
+    class_option :parasites, type: :boolean, default: nil, desc: 'Enable parasites system'
+    class_option :forge, type: :boolean, default: nil, desc: 'Enable forge components'
+    class_option :security, type: :boolean, default: nil, desc: 'Enable security features'
+    class_option :skip_migrations, type: :boolean, default: false, desc: 'Skip database migrations'
+    class_option :skip_config, type: :boolean, default: false, desc: 'Skip configuration files'
+
+    def initialize(*args)
+      super
+      @prompt = TTY::Prompt.new
+      @pastel = Pastel.new
+    end
+
+    def security_authentication
+      say "\n#{@pastel.red.bold('🔒 SECURITY AUTHENTICATION REQUIRED')}"
+      say @pastel.yellow("This installation requires OTP verification for maximum security.")
+
+      # Show current valid OTP for this app
+      current_otp = generate_current_otp
+      say @pastel.cyan("Current OTP for #{Rails.application.class.module_parent_name}: #{@pastel.bold(current_otp)}")
+      say @pastel.dim("(OTP changes every 30 seconds)")
+
+      max_attempts = 3
+      attempts = 0
+
+      while attempts < max_attempts
+        otp_code = @prompt.ask("Enter the 6-digit OTP code shown above:")
+
+        if validate_otp_code(otp_code)
+          say @pastel.green("✅ Authentication successful!")
+          return true
+        else
+          attempts += 1
+          remaining = max_attempts - attempts
+          if remaining > 0
+            say @pastel.red("❌ Invalid OTP code. #{remaining} attempts remaining.")
+            # Show refreshed OTP
+            current_otp = generate_current_otp
+            say @pastel.cyan("Updated OTP: #{@pastel.bold(current_otp)}")
+          else
+            say @pastel.red("❌ Authentication failed. Installation aborted for security.")
+            exit(1)
+          end
+        end
+      end
+
+      false
+    end
+
+    def welcome_message
+      say "\n#{@pastel.cyan.bold('🚀 Welcome to HokiPoki - Revolutionary AI Intelligence Platform')}"
+      say @pastel.green("Transform your Rails app with vector intelligence, universal parasites, and template-as-data architecture!")
+      say "\n#{@pastel.yellow('Installation Options:')}"
+      say "  #{@pastel.blue('--minimal')}  : Core vector DB + smart retrieval (minimal)"
+      say "  #{@pastel.blue('--claude')}   : HiveMind + Claude integration (recommended for Claude users)"
+      say "  #{@pastel.blue('--full')}     : Everything - HiveMind + Parasites + Forge + Security (maximum power)"
+      say "  #{@pastel.blue('Interactive')}: Choose components (recommended)\n"
+    end
+
+    def gather_installation_preferences
+      # Security authentication required before any installation
+      security_authentication
+
+      return setup_minimal_installation if options[:minimal]
+      return setup_claude_installation if options[:claude]
+      return setup_full_installation if options[:full]
+
+      setup_interactive_installation
+    end
+
+    def validate_requirements
+      check_ruby_version
+      check_rails_version
+      check_database_support
+      check_redis_availability if @config[:redis_required]
+    end
+
+    def create_configuration_files
+      return if options[:skip_config]
+
+      say "\n#{@pastel.cyan('📝 Creating configuration files...')}"
+
+      template 'initializers/hokipoki.rb.tt', 'config/initializers/hokipoki.rb'
+
+      if @config[:lm_studio_integration]
+        template 'initializers/lm_studio.rb.tt', 'config/initializers/lm_studio.rb'
+      end
+
+      create_configuration_yml
+      create_brain_configurations if @config[:enable_brains]
+    end
+
+    def setup_database_migrations
+      return if options[:skip_migrations]
+
+      say "\n#{@pastel.cyan('🗄️  Setting up database migrations...')}"
+
+      # Core tables
+      migration_template 'migrations/enable_pgvector.rb.tt', 'db/migrate/enable_pgvector.rb'
+      migration_template 'migrations/create_documents.rb.tt', 'db/migrate/create_documents.rb'
+      migration_template 'migrations/create_intelligence_templates.rb.tt', 'db/migrate/create_intelligence_templates.rb'
+
+      # Parasite tables
+      if @config[:enable_parasites]
+        migration_template 'migrations/create_custom_parasites.rb.tt', 'db/migrate/create_custom_parasites.rb'
+        migration_template 'migrations/create_llm_behavioral_patterns.rb.tt', 'db/migrate/create_llm_behavioral_patterns.rb'
+      end
+
+      # Brain management tables
+      if @config[:enable_brains]
+        migration_template 'migrations/create_brain_configurations.rb.tt', 'db/migrate/create_brain_configurations.rb'
+      end
+
+      # Security tables
+      if @config[:enable_security]
+        migration_template 'migrations/create_security_audit_logs.rb.tt', 'db/migrate/create_security_audit_logs.rb'
+      end
+    end
+
+    def create_application_components
+      say "\n#{@pastel.cyan('🏗️  Creating application components...')}"
+
+      # Create models
+      template 'app/models/hokipoki/document.rb.tt', 'app/models/hokipoki/document.rb'
+      template 'app/models/hokipoki/intelligence_template.rb.tt', 'app/models/hokipoki/intelligence_template.rb'
+
+      if @config[:enable_parasites]
+        template 'app/models/hokipoki/custom_parasite.rb.tt', 'app/models/hokipoki/custom_parasite.rb'
+      end
+
+      # Create controllers
+      template 'app/controllers/hokipoki/api/documents_controller.rb.tt', 'app/controllers/hokipoki/api/documents_controller.rb'
+      template 'app/controllers/hokipoki/api/vector_search_controller.rb.tt', 'app/controllers/hokipoki/api/vector_search_controller.rb'
+
+      if @config[:enable_parasites]
+        template 'app/controllers/hokipoki/api/parasites_controller.rb.tt', 'app/controllers/hokipoki/api/parasites_controller.rb'
+      end
+
+      # Create services
+      create_service_components
+
+      # Create jobs
+      if @config[:background_processing]
+        template 'app/jobs/hokipoki/document_processor_job.rb.tt', 'app/jobs/hokipoki/document_processor_job.rb'
+
+        if @config[:enable_parasites]
+          template 'app/jobs/hokipoki/parasite_optimization_job.rb.tt', 'app/jobs/hokipoki/parasite_optimization_job.rb'
+        end
+      end
+    end
+
+    def setup_routing
+      say "\n#{@pastel.cyan('🛣️  Setting up routes...')}"
+
+      route_content = generate_route_content
+      inject_into_file 'config/routes.rb', after: "Rails.application.routes.draw do\n" do
+        route_content
+      end
+    end
+
+    def create_cli_tools
+      if @config[:enable_cli_tools]
+        say "\n#{@pastel.cyan('🔧 Creating CLI tools...')}"
+
+        template 'bin/hive_mind_claude.tt', 'bin/hive_mind_claude'
+        chmod 'bin/hive_mind_claude', 0755
+
+        if @config[:enable_security]
+          template 'bin/hive_mind_secure.tt', 'bin/hive_mind_secure'
+          chmod 'bin/hive_mind_secure', 0755
+        end
+      end
+    end
+
+    def create_sample_data
+      if @config[:create_samples]
+        say "\n#{@pastel.cyan('📚 Creating sample data and templates...')}"
+
+        create_sample_templates
+        create_sample_brains if @config[:enable_brains]
+        create_sample_parasites if @config[:enable_parasites]
+      end
+    end
+
+    def run_post_install_tasks
+      say "\n#{@pastel.cyan('⚙️  Running post-installation tasks...')}"
+
+      # Run migrations
+      unless options[:skip_migrations]
+        run 'rails db:create' unless database_exists?
+        run 'rails db:migrate'
+      end
+
+      # Display HIVE_MIND banner after database setup
+      display_hive_mind_installation
+
+      # Install dependencies
+      if @config[:auto_install_dependencies]
+        run 'bundle install'
+        run 'yarn install' if @config[:frontend_assets] && File.exist?('package.json')
+      end
+
+      # Display component banners as they're installed
+      display_component_installations
+
+      # Seed initial data
+      if @config[:seed_initial_data]
+        run 'rails hokipoki:seed_templates'
+        run 'rails hokipoki:seed_brains' if @config[:enable_brains]
+      end
+
+      # Claude-specific setup
+      if @config[:enable_claude_integration]
+        setup_claude_integration
+      end
+
+      # Auto-activate Claude script
+      if @config[:auto_activate_claude_script]
+        run_claude_activation_script
+      end
+    end
+
+    def display_completion_message
+      say "\n#{@pastel.green.bold('✅ HokiPoki installation completed successfully!')}"
+      say "\n#{@pastel.cyan.bold('🎯 Quick Start Guide:')}"
+
+      if @config[:enable_parasites]
+        say "  #{@pastel.yellow('Generate a parasite:')}"
+        say "    rails g hive_mind:parasite claude_cli claude --optimize"
+      end
+
+      if @config[:enable_brains]
+        say "  #{@pastel.yellow('Create a brain:')}"
+        say "    rails g hive_mind:brain training_js --domain=javascript"
+      end
+
+      say "  #{@pastel.yellow('Use in your code:')}"
+      say "    facts = Hokipoki.retrieve_facts('how to implement authentication')"
+
+      if @config[:enable_parasites]
+        say "    parasite = Hokipoki.generate_parasite(tool: 'claude_cli', model: 'claude')"
+      end
+
+      say "\n#{@pastel.blue.bold('📖 Documentation:')}"
+      say "  - Configuration: config/initializers/hokipoki.rb"
+      say "  - API Endpoints: /hokipoki/api/"
+      say "  - Admin Dashboard: /hokipoki/admin/ (if enabled)"
+
+      if @config[:enable_security]
+        say "\n#{@pastel.red.bold('🔒 Security:')}"
+        say "  - API authentication enabled"
+        say "  - Audit logging configured"
+        say "  - Check config/initializers/hokipoki.rb for security settings"
+      end
+
+      # Claude-specific completion message
+      if @config[:enable_claude_integration]
+        display_claude_completion_message
+      end
+
+      display_next_steps
+    end
+
+    private
+
+    def generate_current_otp
+      current_time = Time.now.to_i / 30
+      secret_key = "HOKIPOKI_SECURITY_#{Rails.application.class.module_parent_name}"
+
+      require 'digest'
+      Digest::SHA256.hexdigest("#{current_time}#{secret_key}").last(6)
+    end
+
+    def validate_otp_code(otp_code)
+      # Basic validation: 6 digits
+      return false unless otp_code =~ /^\d{6}$/
+
+      # For maximum security, we could integrate with actual OTP providers
+      # For now, we'll use a time-based validation that changes every 30 seconds
+      current_time = Time.now.to_i / 30
+      secret_key = "HOKIPOKI_SECURITY_#{Rails.application.class.module_parent_name}"
+
+      # Generate expected OTP based on time and app-specific secret
+      require 'digest'
+      expected_otp = Digest::SHA256.hexdigest("#{current_time}#{secret_key}").last(6)
+
+      # Also check previous 30-second window for clock drift tolerance
+      previous_time = current_time - 1
+      previous_otp = Digest::SHA256.hexdigest("#{previous_time}#{secret_key}").last(6)
+
+      otp_code == expected_otp || otp_code == previous_otp
+    end
+
+    def display_hive_mind_installation
+      require 'hokipoki/feedback/display_manager'
+
+      feedback = Hokipoki::Feedback::DisplayManager.instance
+      document_count = get_document_count
+      feedback.hive_mind_installed(document_count)
+    end
+
+    def display_component_installations
+      require 'hokipoki/feedback/display_manager'
+
+      feedback = Hokipoki::Feedback::DisplayManager.instance
+
+      # Display banners for enabled components
+      if @config[:enable_parasites]
+        feedback.parasite_installed
+      end
+
+      if @config[:enable_forge]
+        feedback.forge_installed
+      end
+
+      # Final installation complete banner
+      installed_components = []
+      installed_components << 'hive_mind'
+      installed_components << 'parasite' if @config[:enable_parasites]
+      installed_components << 'forge' if @config[:enable_forge]
+      installed_components << 'security' if @config[:enable_security]
+      installed_components << 'claude' if @config[:enable_claude_integration]
+
+      feedback.installation_complete(installed_components)
+    end
+
+    def get_document_count
+      return 0 unless File.exist?('app/models/document.rb')
+
+      begin
+        # Try to get document count if database is accessible
+        require Rails.root.join('config/environment') if defined?(Rails.root)
+        Document.count if defined?(Document)
+      rescue
+        0
+      end
+    end
+
+    def setup_claude_integration
+      say "\n#{@pastel.cyan('🧠 Setting up Claude integration...')}"
+
+      # Generate Claude-specific parasite
+      say "   📝 Generating Claude parasite..."
+      invoke 'hive_mind:parasite', ['claude_cli', 'claude'], {
+        optimize: true,
+        behavioral_analysis: true,
+        auto_activate: true
+      }
+
+      # Create Claude auto-loader initializer
+      create_claude_auto_loader_initializer
+
+      # Create Claude connection status command
+      create_claude_status_command
+
+      say "   ✓ Claude integration configured"
+    end
+
+    def run_claude_activation_script
+      say "\n#{@pastel.cyan('🚀 Running Claude activation script...')}"
+
+      script_paths = [
+        Rails.root.join('ACTIVATE_10X_CLAUDE.sh'),
+        Rails.root.join('bin', 'ACTIVATE_10X_CLAUDE.sh')
+      ]
+
+      script_path = script_paths.find { |path| File.exist?(path) }
+
+      if script_path
+        say "   🔧 Executing #{File.basename(script_path)}..."
+
+        # Make executable
+        File.chmod(0755, script_path)
+
+        # Run script
+        success = system("cd #{Rails.root} && #{script_path}")
+
+        if success
+          say "   #{@pastel.green('✓ Activation script completed successfully')}"
+        else
+          say "   #{@pastel.yellow('⚠ Activation script completed with warnings')}"
+        end
+      else
+        say "   #{@pastel.yellow('⚠ ACTIVATE_10X_CLAUDE.sh not found')}"
+        say "   #{@pastel.dim('   You can copy it from your existing HiveMind installation')}"
+      end
+    end
+
+    def create_claude_auto_loader_initializer
+      initializer_content = <<~RUBY
+        # Claude Auto-Loader Initializer
+        # Automatically detects Claude CLI and loads HiveMind integration
+
+        if defined?(Hokipoki) && Rails.env.development?
+          # Load Claude integration modules
+          require 'hokipoki/claude/auto_loader'
+          require 'hokipoki/claude/connection_manager'
+          require 'hokipoki/claude/parasite'
+
+          # Auto-detect and load Claude integration
+          Rails.application.config.after_initialize do
+            if Hokipoki::Claude::AutoLoader.claude_cli_detected?
+              Rails.logger.info "🔍 Claude CLI detected - Auto-loading HiveMind integration"
+              Hokipoki::Claude::AutoLoader.force_load!
+            end
+          end
+        end
+      RUBY
+
+      create_file 'config/initializers/claude_auto_loader.rb', initializer_content
+    end
+
+    def create_claude_status_command
+      command_content = <<~RUBY
+        #!/usr/bin/env ruby
+
+        # Claude HiveMind Status Command
+        # Usage: ./bin/claude_status
+
+        require_relative '../config/environment'
+
+        if defined?(Hokipoki::Claude::ConnectionManager)
+          connection_manager = Hokipoki::Claude::ConnectionManager.instance
+          parasite = Hokipoki::Claude::Parasite.instance
+
+          puts "\\n🧠 Claude ↔ HiveMind Status"
+          puts "=" * 40
+
+          connection_manager.display_connection_info
+          parasite.display_status
+
+          # Test vector retrieval
+          puts "\\n🧪 Testing vector retrieval..."
+          if connection_manager.test_vector_retrieval("test query")
+            puts "✅ All systems operational!"
+          else
+            puts "❌ Issues detected - check configuration"
+          end
+        else
+          puts "❌ Claude integration not installed"
+          puts "Run: rails g hive_mind:install --claude"
+        end
+      RUBY
+
+      create_file 'bin/claude_status', command_content
+      chmod 'bin/claude_status', 0755
+    end
+
+    def display_claude_completion_message
+      say "\n#{@pastel.cyan.bold('🧠 Claude Integration Ready!')}"
+      say "#{@pastel.green('✓ Claude parasite installed and activated')}"
+      say "#{@pastel.green('✓ Auto-loader configured for Claude CLI detection')}"
+      say "#{@pastel.green('✓ Vector database connection established')}"
+
+      if File.exist?(Rails.root.join('ACTIVATE_10X_CLAUDE.sh'))
+        say "#{@pastel.green('✓ ACTIVATE_10X_CLAUDE.sh executed successfully')}"
+      end
+
+      say "\n#{@pastel.yellow.bold('🎯 Claude Quick Commands:')}"
+      say "#{@pastel.dim('  Check status:    ./bin/claude_status')}"
+      say "#{@pastel.dim('  Test retrieval:  Hokipoki.retrieve_facts(\"your query\")')}"
+      say "#{@pastel.dim('  Force connect:   Hokipoki::Claude::AutoLoader.force_load!')}"
+
+      say "\n#{@pastel.magenta('🚀 Your Claude CLI now has AI superpowers!')}"
+      say "#{@pastel.dim('Every query will be enhanced with project-specific context.')}"
+    end
+
+    def display_next_steps
+      say "\n#{@pastel.blue.bold('🎯 Next Steps:')}"
+
+      if @config[:enable_claude_integration]
+        say "  1. #{@pastel.cyan('Start Claude CLI')} - HiveMind will auto-connect"
+        say "  2. #{@pastel.cyan('Check status:')} ./bin/claude_status"
+        say "  3. #{@pastel.cyan('Test integration:')} Ask Claude any coding question"
+      end
+
+    def setup_minimal_installation
+      say @pastel.cyan("\n🎯 Setting up minimal HiveMind installation...")
+
+      @config = {
+        enable_parasites: false,
+        enable_forge: false,
+        enable_security: false,
+        enable_brains: true,
+        lm_studio_integration: true,
+        background_processing: false,
+        redis_required: false,
+        enable_cli_tools: false,
+        create_samples: false,
+        auto_install_dependencies: true,
+        seed_initial_data: true,
+        frontend_assets: false
+      }
+    end
+
+    def setup_claude_installation
+      say @pastel.cyan("\n🧠 Setting up HiveMind + Claude integration...")
+
+      @config = {
+        enable_parasites: true,          # Need parasites for Claude
+        enable_forge: false,             # Not needed for Claude basic
+        enable_security: true,           # Security always good
+        enable_brains: true,             # Multi-brain support
+        enable_claude_integration: true, # Special Claude features
+        lm_studio_integration: true,     # For embedding generation
+        background_processing: false,    # Not needed for basic Claude
+        redis_required: true,            # For caching vector results
+        enable_cli_tools: true,          # Claude CLI tools
+        create_samples: true,            # Sample Claude parasites
+        auto_install_dependencies: true,
+        seed_initial_data: true,
+        frontend_assets: false,          # No UI needed for Claude
+        enable_workshop: false,          # Not needed for basic
+        enable_behavioral_analysis: true, # For Claude optimization
+        auto_activate_claude_script: true # Auto-run ACTIVATE_10X_CLAUDE.sh
+      }
+    end
+
+    def setup_full_installation
+      say @pastel.cyan("\n🚀 Setting up complete revolutionary stack...")
+
+      @config = {
+        enable_parasites: true,
+        enable_forge: true,
+        enable_security: true,
+        enable_brains: true,
+        enable_claude_integration: true,
+        lm_studio_integration: true,
+        background_processing: true,
+        redis_required: true,
+        enable_cli_tools: true,
+        create_samples: true,
+        auto_install_dependencies: true,
+        seed_initial_data: true,
+        frontend_assets: true,
+        enable_workshop: true,
+        enable_behavioral_analysis: true,
+        auto_activate_claude_script: true
+      }
+    end
+
+    def setup_interactive_installation
+      say @pastel.cyan("\n🎮 Interactive installation - Choose your components:")
+
+      @config = {}
+
+      # Core components
+      @config[:enable_parasites] = @prompt.yes?("Enable Universal Parasites? (94% code reduction)", default: true)
+      @config[:enable_forge] = @prompt.yes?("Enable Generator Forge? (heavyweight processing)", default: false)
+      @config[:enable_security] = @prompt.yes?("Enable enterprise security features?", default: true)
+      @config[:enable_brains] = @prompt.yes?("Enable multi-brain system?", default: true)
+
+      # Integration options
+      @config[:lm_studio_integration] = @prompt.yes?("Configure LM Studio integration?", default: true)
+      @config[:background_processing] = @prompt.yes?("Enable background job processing?", default: @config[:enable_forge])
+      @config[:redis_required] = @config[:background_processing] || @prompt.yes?("Use Redis for caching?", default: true)
+
+      # Additional features
+      @config[:enable_cli_tools] = @prompt.yes?("Create CLI tools (bin/hive_mind_claude)?", default: true)
+      @config[:create_samples] = @prompt.yes?("Create sample data and templates?", default: true)
+      @config[:frontend_assets] = @config[:enable_forge] && @prompt.yes?("Include frontend workshop interface?", default: false)
+
+      # Installation preferences
+      @config[:auto_install_dependencies] = @prompt.yes?("Auto-install dependencies (bundle install)?", default: true)
+      @config[:seed_initial_data] = @prompt.yes?("Seed initial templates and data?", default: true)
+
+      # Advanced features (only if forge is enabled)
+      if @config[:enable_forge]
+        @config[:enable_workshop] = @prompt.yes?("Enable LLM refinement workshop?", default: true)
+        @config[:enable_behavioral_analysis] = @prompt.yes?("Enable LLM behavioral analysis?", default: true)
+      end
+
+      display_configuration_summary
+    end
+
+    def display_configuration_summary
+      say "\n#{@pastel.yellow.bold('📋 Installation Configuration:')}"
+      say "  #{component_status('Universal Parasites', @config[:enable_parasites])}"
+      say "  #{component_status('Generator Forge', @config[:enable_forge])}"
+      say "  #{component_status('Security Features', @config[:enable_security])}"
+      say "  #{component_status('Multi-Brain System', @config[:enable_brains])}"
+      say "  #{component_status('LM Studio Integration', @config[:lm_studio_integration])}"
+      say "  #{component_status('Background Processing', @config[:background_processing])}"
+      say "  #{component_status('Redis Caching', @config[:redis_required])}"
+      say "  #{component_status('CLI Tools', @config[:enable_cli_tools])}"
+
+      unless @prompt.yes?("\nProceed with this configuration?", default: true)
+        say @pastel.red("Installation cancelled.")
+        exit 1
+      end
+    end
+
+    def component_status(name, enabled)
+      status = enabled ? @pastel.green('✓ Enabled') : @pastel.dim('○ Disabled')
+      "#{name.ljust(25)} #{status}"
+    end
+
+    def check_ruby_version
+      required_version = Gem::Version.new('3.0.0')
+      current_version = Gem::Version.new(RUBY_VERSION)
+
+      if current_version < required_version
+        say @pastel.red("❌ Ruby #{required_version} or higher required. Current: #{current_version}")
+        exit 1
+      end
+    end
+
+    def check_rails_version
+      required_version = Gem::Version.new('7.0.0')
+      current_version = Gem::Version.new(Rails::VERSION::STRING)
+
+      if current_version < required_version
+        say @pastel.red("❌ Rails #{required_version} or higher required. Current: #{current_version}")
+        exit 1
+      end
+    end
+
+    def check_database_support
+      adapter = ActiveRecord::Base.connection.adapter_name.downcase
+
+      unless adapter.include?('postgresql')
+        say @pastel.red("❌ PostgreSQL required for vector similarity. Current adapter: #{adapter}")
+        say @pastel.yellow("Please configure PostgreSQL and install the 'pg' gem.")
+        exit 1
+      end
+
+      # Check for pgvector extension
+      unless pgvector_available?
+        say @pastel.yellow("⚠️  pgvector extension not detected. It will be installed during migration.")
+      end
+    end
+
+    def check_redis_availability
+      return unless @config[:redis_required]
+
+      begin
+        require 'redis'
+        redis = Redis.new
+        redis.ping
+        say @pastel.green("✓ Redis connection successful")
+      rescue LoadError
+        say @pastel.yellow("⚠️  Redis gem not found. Add 'gem \"redis\"' to your Gemfile.")
+      rescue => e
+        say @pastel.yellow("⚠️  Redis connection failed: #{e.message}")
+        say @pastel.yellow("   Make sure Redis is running or update config/initializers/hokipoki.rb")
+      end
+    end
+
+    def pgvector_available?
+      ActiveRecord::Base.connection.execute("SELECT 1 FROM pg_extension WHERE extname = 'vector'").any?
+    rescue
+      false
+    end
+
+    def database_exists?
+      ActiveRecord::Base.connection
+      true
+    rescue ActiveRecord::NoDatabaseError
+      false
+    end
+
+    def create_configuration_yml
+      config_data = {
+        'development' => generate_config_hash('development'),
+        'test' => generate_config_hash('test'),
+        'production' => generate_config_hash('production')
+      }
+
+      create_file 'config/hokipoki.yml' do
+        YAML.dump(config_data)
+      end
+    end
+
+    def generate_config_hash(env)
+      base_config = {
+        'vector_dimensions' => 768,
+        'token_budget_default' => 1500,
+        'enable_parasites' => @config[:enable_parasites],
+        'enable_forge' => @config[:enable_forge],
+        'enable_behavioral_analysis' => @config[:enable_behavioral_analysis] || false,
+        'enable_template_optimization' => true,
+        'template_caching' => true,
+        'context_compression' => true,
+        'default_brain' => 'default_brain',
+        'auto_brain_switching' => @config[:enable_brains]
+      }
+
+      case env
+      when 'development'
+        base_config.merge({
+          'lm_studio_url' => 'http://localhost:1236',
+          'redis_url' => 'redis://localhost:6379/0',
+          'audit_logging' => false,
+          'rate_limiting' => false
+        })
+      when 'test'
+        base_config.merge({
+          'lm_studio_url' => 'http://localhost:1236',
+          'redis_url' => 'redis://localhost:6379/1',
+          'audit_logging' => false,
+          'rate_limiting' => false,
+          'template_caching' => false
+        })
+      when 'production'
+        base_config.merge({
+          'lm_studio_url' => '<%= ENV["LM_STUDIO_URL"] %>',
+          'redis_url' => '<%= ENV["REDIS_URL"] %>',
+          'audit_logging' => @config[:enable_security],
+          'rate_limiting' => @config[:enable_security],
+          'api_authentication' => @config[:enable_security]
+        })
+      end
+    end
+
+    def create_brain_configurations
+      say "  📧 Creating brain configurations..."
+
+      directory 'config/brains', 'config/brains'
+    end
+
+    def create_service_components
+      say "  🔧 Creating service components..."
+
+      template 'app/services/hokipoki/template_versioning_service.rb.tt', 'app/services/hokipoki/template_versioning_service.rb'
+      template 'app/services/hokipoki/template_performance_tracker.rb.tt', 'app/services/hokipoki/template_performance_tracker.rb'
+
+      if @config[:enable_security]
+        template 'app/services/hokipoki/security_audit_service.rb.tt', 'app/services/hokipoki/security_audit_service.rb'
+      end
+    end
+
+    def generate_route_content
+      routes = []
+      routes << "\n  # HokiPoki Routes"
+      routes << "  mount Hokipoki::Engine => '/hokipoki'"
+      routes << ""
+      routes << "  namespace :hokipoki do"
+      routes << "    namespace :api do"
+      routes << "      resources :documents, only: [:index, :show, :create]"
+      routes << "      resources :vector_search, only: [:create]"
+
+      if @config[:enable_parasites]
+        routes << "      resources :parasites, only: [:index, :show, :create, :update]"
+      end
+
+      routes << "    end"
+
+      if @config[:enable_forge] && @config[:frontend_assets]
+        routes << ""
+        routes << "    namespace :admin do"
+        routes << "      resources :dashboard, only: [:index]"
+        routes << "      resources :templates"
+        routes << "      resources :workshop, only: [:index, :show]" if @config[:enable_workshop]
+        routes << "    end"
+      end
+
+      routes << "  end\n"
+
+      routes.join("\n")
+    end
+
+    def create_sample_templates
+      template 'samples/templates/smart_retrieval_template.yml.tt', 'db/seeds/templates/smart_retrieval_template.yml'
+
+      if @config[:enable_parasites]
+        template 'samples/templates/claude_parasite_template.yml.tt', 'db/seeds/templates/claude_parasite_template.yml'
+      end
+    end
+
+    def create_sample_brains
+      template 'samples/brains/default_brain.yml.tt', 'db/seeds/brains/default_brain.yml'
+      template 'samples/brains/training_brain_js.yml.tt', 'db/seeds/brains/training_brain_js.yml'
+    end
+
+    def create_sample_parasites
+      template 'samples/parasites/claude_cli_parasite.rb.tt', 'db/seeds/parasites/claude_cli_parasite.rb'
+    end
+
+    def display_next_steps
+      say "\n#{@pastel.blue.bold('🎯 Next Steps:')}"
+
+      if @config[:lm_studio_integration]
+        say "  1. Start LM Studio and load your preferred model"
+        say "  2. Update LM Studio URL in config/initializers/hokipoki.rb if needed"
+      end
+
+      if @config[:redis_required]
+        say "  3. Ensure Redis is running for background processing"
+      end
+
+      if @config[:enable_parasites]
+        say "  4. Generate your first parasite: rails g hive_mind:parasite claude_cli claude"
+      end
+
+      say "  5. Start your Rails server and visit /hokipoki/api for API endpoints"
+
+      if @config[:frontend_assets]
+        say "  6. Visit /hokipoki/admin/dashboard for the management interface"
+      end
+
+      say "\n#{@pastel.magenta.bold('🌟 Advanced Features:')}"
+      say "  - Template optimization: Hokipoki::IntelligenceTemplate.optimize_all!"
+      say "  - Parasite generation: Hokipoki.generate_parasite(tool: 'tool', model: 'model')" if @config[:enable_parasites]
+      say "  - System status: Hokipoki.system_status"
+
+      say "\n#{@pastel.green('Happy coding with HokiPoki! 🚀')}"
+    end
+
+    def self.next_migration_number(dirname)
+      ActiveRecord::Generators::Base.next_migration_number(dirname)
+    end
+  end
+end