rails-active-mcp 0.1.6 → 0.1.7

data/lib/rails_active_mcp/stdio_server.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'json'
+require 'io/console'
 require 'logger'
 
 module RailsActiveMcp
@@ -10,51 +11,112 @@ module RailsActiveMcp
 
     def initialize
       @tools = {}
-      @logger = Logger.new(STDERR) # Log to stderr to avoid interfering with stdout
-      @logger.level = ENV['RAILS_MCP_DEBUG'] ? Logger::DEBUG : Logger::ERROR
-      @logger.formatter = proc do |severity, datetime, progname, msg|
-        "[#{datetime}] [RAILS-MCP] #{severity}: #{msg}\n"
-      end
+      @mcp_server = McpServer.new
+      @running = false
+
+      # Ensure all logging goes to stderr, never stdout
+      setup_logging
       register_default_tools
-      @logger.info "Rails Active MCP Server initialized with #{@tools.size} tools"
+      log_to_stderr "Rails Active MCP Server initialized with #{@tools.size} tools", level: :info
+    end
+
+    def start
+      @running = true
+      log_to_stderr 'Rails Active MCP Server started successfully', level: :info
+
+      # Send initial notification to stderr first, then stdout for MCP
+      send_notification('info', 'Rails Active MCP Server started successfully')
+
+      process_requests
+    end
+
+    def stop
+      @running = false
+      log_to_stderr 'Rails Active MCP Server stopped', level: :info
+    end
+
+    private
+
+    def setup_logging
+      # Configure Rails Active MCP logger to use stderr
+      return unless RailsActiveMcp.respond_to?(:logger=)
+
+      stderr_logger = Logger.new($stderr)
+      stderr_logger.level = ENV['RAILS_MCP_DEBUG'] == '1' ? Logger::DEBUG : Logger::INFO
+      stderr_logger.formatter = proc do |severity, datetime, progname, msg|
+        "[#{datetime.strftime('%Y-%m-%d %H:%M:%S %z')}] [RAILS-MCP] #{severity}: #{msg}\n"
+      end
+      RailsActiveMcp.logger = stderr_logger
     end
 
-    def run
-      @logger.info 'Starting Rails Active MCP Stdio Server'
-      send_log_notification('info', 'Rails Active MCP Server started successfully')
+    def process_requests
+      while @running
+        begin
+          line = $stdin.gets
+          break if line.nil?
+
+          line = line.strip
+          next if line.empty?
 
-      STDIN.each_line do |line|
-        line = line.strip
-        next if line.empty?
+          log_to_stderr "Received request: #{line}", level: :debug
 
-        @logger.debug "Received request: #{line}" if ENV['RAILS_MCP_DEBUG']
-        data = JSON.parse(line)
+          request = JSON.parse(line)
+          log_to_stderr "Processing method: #{request['method']}", level: :debug
 
-        @logger.debug "Processing method: #{data['method']}" if ENV['RAILS_MCP_DEBUG']
-        response = handle_jsonrpc_request(data)
+          response = @mcp_server.handle_jsonrpc_request(request)
 
-        if response
-          @logger.debug "Sending response: #{response.to_json}" if ENV['RAILS_MCP_DEBUG']
-          puts response.to_json
-          STDOUT.flush
+          log_to_stderr "Sending response: #{response.to_json}", level: :debug
+
+          # Only JSON responses go to stdout
+          $stdout.puts response.to_json
+          $stdout.flush
+        rescue JSON::ParserError => e
+          log_to_stderr "JSON parse error: #{e.message}", level: :error
+          send_error_response(nil, -32_700, 'Parse error')
+        rescue StandardError => e
+          log_to_stderr "Server error: #{e.message}", level: :error
+          log_to_stderr e.backtrace.join("\n"), level: :debug
+          send_error_response(nil, -32_603, 'Internal error')
         end
-      rescue JSON::ParserError => e
-        @logger.error "JSON Parse Error: #{e.message}"
-        send_log_notification('error', "JSON Parse Error: #{e.message}")
-        error_response = jsonrpc_error(nil, -32_700, 'Parse error')
-        puts error_response.to_json
-        STDOUT.flush
-      rescue StandardError => e
-        @logger.error "Unexpected error: #{e.message}"
-        @logger.error e.backtrace.join("\n")
-        send_log_notification('error', "Server error: #{e.message}")
-        error_response = jsonrpc_error(nil, -32_603, 'Internal error')
-        puts error_response.to_json
-        STDOUT.flush
       end
     end
 
-    private
+    def send_notification(level, message)
+      notification = {
+        jsonrpc: JSONRPC_VERSION,
+        method: 'notifications/message',
+        params: {
+          level: level,
+          data: message
+        }
+      }
+
+      # Notifications go to stdout for MCP protocol
+      $stdout.puts notification.to_json
+      $stdout.flush
+    end
+
+    def send_error_response(id, code, message)
+      error_response = {
+        jsonrpc: JSONRPC_VERSION,
+        id: id,
+        error: {
+          code: code,
+          message: message
+        }
+      }
+
+      $stdout.puts error_response.to_json
+      $stdout.flush
+    end
+
+    def log_to_stderr(message, level: :info)
+      return unless ENV['RAILS_MCP_DEBUG'] == '1' || level == :error || level == :info
+
+      timestamp = Time.now.strftime('%Y-%m-%d %H:%M:%S %z')
+      warn "[#{timestamp}] [RAILS-MCP] #{level.to_s.upcase}: #{message}"
+      $stderr.flush
+    end
 
     def handle_jsonrpc_request(data)
       case data['method']
@@ -121,16 +183,16 @@ module RailsActiveMcp
       tool = @tools[tool_name]
       return jsonrpc_error(data['id'], -32_602, "Tool '#{tool_name}' not found") unless tool
 
-      @logger.info "Executing tool: #{tool_name}"
-      send_log_notification('info', "Executing tool: #{tool_name}")
+      log_to_stderr "Executing tool: #{tool_name}", level: :info
+      send_notification('info', "Executing tool: #{tool_name}")
 
       begin
         start_time = Time.now
         result = tool[:handler].call(arguments)
         execution_time = Time.now - start_time
 
-        @logger.info "Tool #{tool_name} completed in #{execution_time}s"
-        send_log_notification('info', "Tool #{tool_name} completed successfully")
+        log_to_stderr "Tool #{tool_name} completed in #{execution_time}s", level: :info
+        send_notification('info', "Tool #{tool_name} completed successfully")
 
         {
           jsonrpc: JSONRPC_VERSION,
@@ -141,9 +203,9 @@ module RailsActiveMcp
           }
         }
       rescue StandardError => e
-        @logger.error "Tool execution error: #{e.message}"
-        @logger.error e.backtrace.first(5).join("\n") if ENV['RAILS_MCP_DEBUG']
-        send_log_notification('error', "Tool #{tool_name} failed: #{e.message}")
+        log_to_stderr "Tool execution error: #{e.message}", level: :error
+        log_to_stderr e.backtrace.first(5).join("\n"), level: :debug if ENV['RAILS_MCP_DEBUG']
+        send_notification('error', "Tool #{tool_name} failed: #{e.message}")
 
         {
           jsonrpc: JSONRPC_VERSION,
@@ -316,8 +378,10 @@ module RailsActiveMcp
       end
 
       begin
-        # Try to load Rails environment if not already loaded
-        require_relative '../../../config/environment' if !defined?(Rails) && File.exist?('config/environment.rb')
+        # Check if Rails environment is available
+        unless defined?(Rails)
+          return 'Rails environment not loaded. Please ensure the MCP server is started from your Rails app directory.'
+        end
 
         model_class = model_name.constantize
         unless defined?(ActiveRecord) && model_class < ActiveRecord::Base
@@ -352,8 +416,10 @@ module RailsActiveMcp
       end
 
       begin
-        # Try to load Rails environment if not already loaded
-        require_relative '../../../config/environment' if !defined?(Rails) && File.exist?('config/environment.rb')
+        # Check if Rails environment is available
+        unless defined?(Rails)
+          return 'Rails environment not loaded. Please ensure the MCP server is started from your Rails app directory.'
+        end
 
         model_class = args['model'].constantize
         unless defined?(ActiveRecord) && model_class < ActiveRecord::Base
@@ -440,22 +506,6 @@ module RailsActiveMcp
       output.join("\n")
     end
 
-    def send_log_notification(level, message)
-      notification = {
-        jsonrpc: JSONRPC_VERSION,
-        method: 'notifications/message',
-        params: {
-          level: level,
-          data: message
-        }
-      }
-
-      puts notification.to_json
-      STDOUT.flush
-    rescue StandardError => e
-      @logger.error "Failed to send log notification: #{e.message}"
-    end
-
     def jsonrpc_error(id, code, message)
       {
         jsonrpc: JSONRPC_VERSION,

data/lib/rails_active_mcp/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsActiveMcp
-  VERSION = '0.1.6'
+  VERSION = '0.1.7'
 end

data/lib/rails_active_mcp.rb

@@ -47,11 +47,16 @@ module RailsActiveMcp
       @server ||= McpServer.new
     end
 
-    # Add logger accessor
+    # Logger accessor - configured by railtie or defaults to stderr
     attr_accessor :logger
 
     def logger
-      @logger ||= defined?(Rails) && Rails.respond_to?(:logger) && Rails.logger ? Rails.logger : Logger.new(STDOUT)
+      @logger ||= Logger.new(STDERR).tap do |logger|
+        logger.level = Logger::INFO
+        logger.formatter = proc do |severity, datetime, progname, msg|
+          "[#{datetime}] #{severity} -- RailsActiveMcp: #{msg}\n"
+        end
+      end
     end
   end
 end

data/rails_active_mcp.gemspec

@@ -30,12 +30,13 @@ Gem::Specification.new do |spec|
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
 
-  # Dependencies
-  spec.add_runtime_dependency 'concurrent-ruby', '~> 1.3.5'
-  spec.add_runtime_dependency 'rails', '~> 7.0'
+  # Runtime dependencies - more flexible Rails version support
+  spec.add_runtime_dependency 'concurrent-ruby', '~> 1.3'
+  spec.add_runtime_dependency 'rails', '>= 6.1', '< 8.0'
 
+  # Core dependencies
   spec.add_dependency 'json', '~> 2.0'
-  spec.add_dependency 'rack', '~> 3.0'
+  spec.add_dependency 'rack', '>= 2.0', '< 4.0'
   spec.add_dependency 'timeout', '~> 0.4'
   spec.add_dependency 'webrick', '~> 1.8'
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: rails-active-mcp
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.7
 platform: ruby
 authors:
 - Brandyn Britton
 bindir: exe
 cert_chain: []
-date: 2025-06-26 00:00:00.000000000 Z
+date: 2025-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -15,28 +15,34 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.3.5
+        version: '1.3'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.3.5
+        version: '1.3'
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '6.1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.1'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: '8.0'
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
@@ -55,16 +61,22 @@ dependencies:
   name: rack
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.0'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '4.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.0'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '4.0'
 - !ruby/object:Gem::Dependency
   name: timeout
   requirement: !ruby/object:Gem::Requirement
@@ -206,8 +218,11 @@ files:
 - ".idea/rails-active-mcp-gem.iml"
 - ".idea/vcs.xml"
 - README.md
+- app/controllers/rails_active_mcp/mcp_controller.rb
 - changelog.md
+- claude_desktop_config.json
 - docs/DEBUGGING.md
+- docs/GENERATOR_TESTING.md
 - docs/README.md
 - exe/rails-active-mcp-server
 - lib/generators/rails_active_mcp/install/install_generator.rb