actionmcp 0.2.3 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 51df9fa26245d233eebbbf193729618ee8c0f8cc7aa6c680d1a2daac97b9a19e
-  data.tar.gz: b3b898d71781a538ee1826c64294018494e4824a0971768adadcb516be2dd2df
+  metadata.gz: 0a1cd6d70ca62ff8ecf4d25c00e29d9592bc9d80ca46db50330a2f818a66ce1f
+  data.tar.gz: 259e35b08f9627aa4165157b71436f4fdd4c0c9eaedc2ae41ddb6f0bb8c3fdde
 SHA512:
-  metadata.gz: a5304f933ec4c0b4e97ca263c2c6833a380dd99ae3beca60e00709c71dbe921024e9572bf1a2ba38e69729d8c8d7728c6d1ab2d3e854cdbfc720c8c494fb59fc
-  data.tar.gz: c79eac91747d34080ae55326d41005e0a71e2ab940abbb858f35e6601797e6ea923f0e4a21bf718456a0a3a6688a05c833ec41e5717d03bd6fcf4eaff2aa554b
+  metadata.gz: c213f50edbec5dc96af78b3090f562dae9fe479a405a44d573927ef65c1c042d6a43ef174297bca3bc92280a372d5a3a74c61aec5a8dfc3e1381550c56b27397
+  data.tar.gz: 343bee907d44e4e7869372b5ffa250c6d628c8789ef51a4690b311c494d6b5b20199837dbbc5f39959e36b8a71d3ce2942b82685b4076897ddc393e59e73d8e3

data/app/controllers/action_mcp/application_controller.rb

@@ -0,0 +1,13 @@
+module ActionMCP
+  class ApplicationController < ActionController::Metal
+    abstract!
+    ActionController::API.without_modules(:StrongParameters, :ParamsWrapper).each do |left|
+      include left
+    end
+    include Engine.routes.url_helpers
+
+    def session_key
+      @session_key = "action_mcp-sessions-#{session_id}"
+    end
+  end
+end

data/app/controllers/action_mcp/messages_controller.rb

@@ -0,0 +1,51 @@
+module ActionMCP
+  class MessagesController < ApplicationController
+    # @route POST / (sse_in)
+    def create
+      begin
+        handle_post_message(params, response)
+      rescue => e
+        head :internal_server_error
+      end
+      head response.status
+    end
+
+    private
+
+    def transport
+      @transport ||= Transport.new(session_key)
+    end
+
+    def transport_handler
+      TransportHandler.new(transport)
+    end
+
+    def json_rpc_handler
+      @json_rpc_handler ||= ActionMCP::JsonRpcHandler.new(transport_handler)
+    end
+
+    def handle_post_message(params, response)
+      json_rpc_handler.call(params)
+
+      response.status = :accepted
+    rescue StandardError => e
+      response.status = :bad_request
+    end
+
+    def session_id
+      params[:session_id]
+    end
+
+    class Transport
+      attr_reader :session_key, :adapter
+      def initialize(session_key)
+        @session_key = session_key
+        @adapter = ActionMCP::Server.server.pubsub
+      end
+
+      def write(data)
+        adapter.broadcast(session_key, data.to_json)
+      end
+    end
+  end
+end

data/app/controllers/action_mcp/sse_controller.rb

@@ -0,0 +1,151 @@
+module ActionMCP
+  class SSEController < ApplicationController
+    HEARTBEAT_INTERVAL = 10
+    INITIALIZATION_TIMEOUT = 2
+    include ActionController::Live
+
+    # @route GET /sse (sse_out)
+    def events
+      # Set headers first
+      response.headers["X-Accel-Buffering"] = "no"
+      response.headers["Content-Type"] = "text/event-stream"
+      response.headers["Cache-Control"] = "no-cache"
+      response.headers["Connection"] = "keep-alive"
+
+      listener = nil
+      begin
+        # Now start streaming - send endpoint
+        send_endpoint_event(sse_in_url)
+
+        # Start listener and process messages via the transport
+        listener = SseListener.new(session_key)
+        if listener.start do |message|
+          begin
+            Rails.logger.debug "Processing message in controller: #{message.inspect} (#{message.class})"
+
+            # Send with proper SSE formatting
+            sse = SSE.new(response.stream)
+            sse.write(message)
+          rescue => e
+            Rails.logger.error "Error sending SSE message: #{e.class} - #{e.message}\n#{e.backtrace.join("\n")}"
+          end
+        end
+
+          # Heartbeat loop
+          until response.stream.closed?
+            sleep HEARTBEAT_INTERVAL
+            send_ping!
+          end
+        else
+          Rails.logger.error "Listener failed to activate for session: #{session_id}"
+          raise "Failed to establish subscription"
+        end
+      rescue ActionController::Live::ClientDisconnected, IOError => e
+        Rails.logger.debug "SSE: Expected disconnection: #{e.message}"
+      rescue => e
+        Rails.logger.error "SSE: Unexpected error: #{e.class} - #{e.message}\n#{e.backtrace.join("\n")}"
+      ensure
+        listener&.stop
+        response.stream.close
+        Rails.logger.debug "SSE: Connection closed for session: #{session_id}"
+      end
+    end
+
+    private
+
+    def send_endpoint_event(messages_url)
+      endpoint = "#{messages_url}?session_id=#{session_id}"
+      SSE.new(response.stream,
+              event: "endpoint")
+         .write(endpoint)
+    end
+
+    def send_ping!
+      SSE.new(response.stream,
+              event: "ping")
+         .write(Time.now.to_i)
+    end
+
+    def default_url_options
+      { host: request.host, port: request.port }
+    end
+
+    def session_id
+      @session_id ||= SecureRandom.hex(6)
+    end
+  end
+
+  class SseListener
+    attr_reader :session_key, :adapter
+
+    def initialize(session_key)
+      @session_key = session_key
+      @adapter = ActionMCP::Server.server.pubsub
+      @stopped = false
+      @subscription_active = false
+    end
+
+    # Start listening using ActionCable's PostgreSQL adapter
+    def start(&callback)
+      Rails.logger.debug "Starting listener for channel: #{session_key}"
+
+      # Set up success callback
+      success_callback = -> {
+        Rails.logger.debug "Successfully subscribed to channel: #{session_key}"
+        @subscription_active = true
+      }
+
+      # Set up message callback with detailed debugging
+      message_callback = ->(raw_message) {
+        Rails.logger.debug "Received raw message via adapter: #{raw_message.inspect} (#{raw_message.class})"
+
+        begin
+          # Try to parse the message if it's JSON
+          message = raw_message.is_a?(String) ? JSON.parse(raw_message) : raw_message
+          Rails.logger.debug "Processed message: #{message.inspect}"
+
+          # Send the message to the callback
+          callback.call(message) if callback && !@stopped
+        rescue => e
+          Rails.logger.error "Error processing message: #{e.class} - #{e.message}"
+          # Still try to send the raw message as a fallback
+          callback.call(raw_message) if callback && !@stopped
+        end
+      }
+
+      # Subscribe using the ActionCable adapter
+      adapter.subscribe(session_key, message_callback, success_callback)
+
+      # Give some time for the subscription to be established
+      sleep 1.5
+
+      # Check if subscription was successful
+      if @subscription_active
+        Rails.logger.debug "Subscription confirmed active for: #{session_key}"
+        true
+      else
+        Rails.logger.error "Failed to activate subscription for: #{session_key}"
+        false
+      end
+    end
+
+    def stop
+      Rails.logger.debug "Stopping listener for: #{session_key}"
+      @stopped = true
+
+      # Unsubscribe using the correct method signature
+      begin
+        # Create a dummy callback that matches the one we provided in start
+        dummy_callback = ->(_) { }
+        adapter.unsubscribe(session_key, dummy_callback)
+        Rails.logger.debug "Unsubscribed from: #{session_key}"
+      rescue => e
+        Rails.logger.error "Error unsubscribing from #{session_key}: #{e.message}"
+      end
+    end
+
+    def active?
+      @subscription_active
+    end
+  end
+end

data/config/routes.rb

@@ -0,0 +1,4 @@
+ActionMCP::Engine.routes.draw do
+  get "/", to: "sse#events", as: :sse_out
+  post "/", to: "messages#create", as: :sse_in
+end

data/lib/action_mcp/version.rb

@@ -2,7 +2,7 @@
 
 require_relative "gem_version"
 module ActionMCP
-  VERSION = "0.2.3"
+  VERSION = "0.2.4"
 
   class << self
     alias version gem_version

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: actionmcp
 version: !ruby/object:Gem::Version
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors:
 - Abdelkader Boudih
@@ -105,6 +105,10 @@ files:
 - MIT-LICENSE
 - README.md
 - Rakefile
+- app/controllers/action_mcp/application_controller.rb
+- app/controllers/action_mcp/messages_controller.rb
+- app/controllers/action_mcp/sse_controller.rb
+- config/routes.rb
 - exe/actionmcp_cli
 - lib/action_mcp.rb
 - lib/action_mcp/capability.rb