rubyn-code 0.1.0 → 0.2.0

data/lib/rubyn_code/mcp/client.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
 module RubynCode
   module MCP
@@ -9,7 +9,8 @@ module RubynCode
     class Client
       INITIALIZE_TIMEOUT = 10
 
-      ClientError = Class.new(RubynCode::Error)
+      class ClientError < RubynCode::Error
+      end
 
       attr_reader :name, :transport
 
@@ -41,7 +42,7 @@ module RubynCode
       #
       # @return [Array<Hash>] tool definitions in JSON Schema format
       def tools
-        @tools_cache ||= discover_tools
+        @tools ||= discover_tools
       end
 
       # Invokes a tool on the MCP server.
@@ -53,10 +54,10 @@ module RubynCode
       def call_tool(tool_name, arguments = {})
         ensure_connected!
 
-        @transport.send_request("tools/call", {
-          name: tool_name,
-          arguments: arguments
-        })
+        @transport.send_request('tools/call', {
+                                  name: tool_name,
+                                  arguments: arguments
+                                })
       end
 
       # Gracefully disconnects from the MCP server.
@@ -107,28 +108,28 @@ module RubynCode
       private
 
       def perform_initialize
-        result = @transport.send_request("initialize", {
-          protocolVersion: "2024-11-05",
-          capabilities: {
-            tools: {}
-          },
-          clientInfo: {
-            name: "rubyn-code",
-            version: RubynCode::VERSION
-          }
-        })
-
-        @server_info = result&.dig("serverInfo")
-        @server_capabilities = result&.dig("capabilities")
-
-        @transport.send_notification("notifications/initialized") if @transport.respond_to?(:send_notification)
+        result = @transport.send_request('initialize', {
+                                           protocolVersion: '2024-11-05',
+                                           capabilities: {
+                                             tools: {}
+                                           },
+                                           clientInfo: {
+                                             name: 'rubyn-code',
+                                             version: RubynCode::VERSION
+                                           }
+                                         })
+
+        @server_info = result&.dig('serverInfo')
+        @server_capabilities = result&.dig('capabilities')
+
+        @transport.send_notification('notifications/initialized') if @transport.respond_to?(:send_notification)
       end
 
       def discover_tools
         ensure_connected!
 
-        result = @transport.send_request("tools/list")
-        result&.fetch("tools", []) || []
+        result = @transport.send_request('tools/list')
+        result&.fetch('tools', []) || []
       end
 
       def ensure_connected!

data/lib/rubyn_code/mcp/config.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
 module RubynCode
   module MCP
@@ -17,7 +17,7 @@ module RubynCode
     #     }
     #   }
     module Config
-      CONFIG_FILENAME = ".rubyn-code/mcp.json"
+      CONFIG_FILENAME = '.rubyn-code/mcp.json'
 
       ENV_VAR_PATTERN = /\$\{([^}]+)\}/
 
@@ -32,14 +32,14 @@ module RubynCode
 
           raw = File.read(config_path)
           data = JSON.parse(raw)
-          servers = data["mcpServers"] || {}
+          servers = data['mcpServers'] || {}
 
           servers.map do |name, server_def|
             {
               name: name,
-              command: server_def["command"],
-              args: Array(server_def["args"]),
-              env: expand_env(server_def["env"] || {})
+              command: server_def['command'],
+              args: Array(server_def['args']),
+              env: expand_env(server_def['env'] || {})
             }
           end
         rescue JSON::ParserError => e
@@ -73,7 +73,7 @@ module RubynCode
             env_name = ::Regexp.last_match(1)
             ENV.fetch(env_name) do
               warn "[MCP::Config] Environment variable #{env_name} is not set"
-              ""
+              ''
             end
           end
         end

data/lib/rubyn_code/mcp/sse_transport.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "faraday"
-require "json"
-require "uri"
+require 'faraday'
+require 'json'
+require 'uri'
 
 module RubynCode
   module MCP
@@ -15,8 +15,11 @@ module RubynCode
     class SSETransport
       DEFAULT_TIMEOUT = 30 # seconds
 
-      TransportError = Class.new(RubynCode::Error)
-      TimeoutError = Class.new(TransportError)
+      class TransportError < RubynCode::Error
+      end
+
+      class TimeoutError < TransportError
+      end
 
       # @param url [String] the SSE endpoint URL of the MCP server
       # @param timeout [Integer] default timeout in seconds per request
@@ -36,7 +39,7 @@ module RubynCode
       # @return [void]
       # @raise [TransportError] if the connection cannot be established
       def start!
-        raise TransportError, "Transport already started" if @connected
+        raise TransportError, 'Transport already started' if @connected
 
         @pending_responses = {}
         @sse_thread = Thread.new { run_sse_listener }
@@ -61,14 +64,14 @@ module RubynCode
       # @raise [TransportError] on protocol or server errors
       # @raise [TimeoutError] if the response is not received in time
       def send_request(method, params = {})
-        raise TransportError, "Transport is not connected" unless @connected
+        raise TransportError, 'Transport is not connected' unless @connected
 
         id = next_request_id
         queue = Queue.new
         @mutex.synchronize { @pending_responses[id] = queue }
 
         request = {
-          jsonrpc: "2.0",
+          jsonrpc: '2.0',
           id: id,
           method: method,
           params: params
@@ -111,7 +114,7 @@ module RubynCode
         @connection ||= Faraday.new(url: base_url) do |f|
           f.options.timeout = @timeout
           f.options.open_timeout = @timeout
-          f.headers["Content-Type"] = "application/json"
+          f.headers['Content-Type'] = 'application/json'
           f.adapter Faraday.default_adapter
         end
       end
@@ -121,9 +124,7 @@ module RubynCode
           req.body = JSON.generate(request)
         end
 
-        unless response.success?
-          raise TransportError, "MCP server returned HTTP #{response.status}: #{response.body}"
-        end
+        raise TransportError, "MCP server returned HTTP #{response.status}: #{response.body}" unless response.success?
       rescue Faraday::Error => e
         raise TransportError, "Failed to send request to MCP server: #{e.message}"
       end
@@ -138,8 +139,8 @@ module RubynCode
           @mutex.synchronize { @pending_responses.delete(id) }
         end
 
-        if result.is_a?(Hash) && result.key?("error")
-          err = result["error"]
+        if result.is_a?(Hash) && result.key?('error')
+          err = result['error']
           raise TransportError, "MCP error (#{err['code']}): #{err['message']}"
         end
 
@@ -150,11 +151,11 @@ module RubynCode
         sse_connection = Faraday.new(url: base_url) do |f|
           f.options.timeout = nil # Keep-alive
           f.options.open_timeout = @timeout
-          f.headers["Accept"] = "text/event-stream"
+          f.headers['Accept'] = 'text/event-stream'
           f.adapter Faraday.default_adapter
         end
 
-        buffer = +""
+        buffer = +''
 
         sse_connection.get(@url) do |req|
           req.options.on_data = proc do |chunk, _bytes, _env|
@@ -181,10 +182,10 @@ module RubynCode
 
         raw.each_line do |line|
           line = line.chomp
-          if line.start_with?("event:")
-            event_type = line.sub("event:", "").strip
-          elsif line.start_with?("data:")
-            data_lines << line.sub("data:", "").strip
+          if line.start_with?('event:')
+            event_type = line.sub('event:', '').strip
+          elsif line.start_with?('data:')
+            data_lines << line.sub('data:', '').strip
           end
         end
 
@@ -195,9 +196,9 @@ module RubynCode
 
       def handle_sse_event(event)
         case event[:type]
-        when "endpoint"
+        when 'endpoint'
           @post_endpoint = event[:data]
-        when "message"
+        when 'message'
           dispatch_message(event[:data])
         else
           dispatch_message(event[:data])
@@ -206,16 +207,16 @@ module RubynCode
 
       def dispatch_message(data)
         message = JSON.parse(data)
-        return unless message.is_a?(Hash) && message.key?("id")
+        return unless message.is_a?(Hash) && message.key?('id')
 
-        id = message["id"]
+        id = message['id']
         queue = @mutex.synchronize { @pending_responses[id] }
         return unless queue
 
-        if message.key?("error")
+        if message.key?('error')
           queue.push(message)
         else
-          queue.push(message["result"])
+          queue.push(message['result'])
         end
       rescue JSON::ParserError
         # Ignore malformed messages

data/lib/rubyn_code/mcp/stdio_transport.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require "json"
-require "open3"
-require "timeout"
+require 'json'
+require 'open3'
+require 'timeout'
 
 module RubynCode
   module MCP
@@ -14,8 +14,11 @@ module RubynCode
     class StdioTransport
       DEFAULT_TIMEOUT = 30 # seconds
 
-      TransportError = Class.new(RubynCode::Error)
-      TimeoutError = Class.new(TransportError)
+      class TransportError < RubynCode::Error
+      end
+
+      class TimeoutError < TransportError
+      end
 
       # @param command [String] executable to spawn
       # @param args [Array<String>] arguments for the command
@@ -39,7 +42,7 @@ module RubynCode
       # @return [void]
       # @raise [TransportError] if the process fails to start
       def start!
-        raise TransportError, "Transport already started" if alive?
+        raise TransportError, 'Transport already started' if alive?
 
         @stdin, @stdout, @stderr, @wait_thread = Open3.popen3(@env, @command, *@args)
       rescue Errno::ENOENT => e
@@ -54,11 +57,11 @@ module RubynCode
       # @raise [TransportError] on protocol or server errors
       # @raise [TimeoutError] if the response is not received within the timeout
       def send_request(method, params = {})
-        raise TransportError, "Transport is not running" unless alive?
+        raise TransportError, 'Transport is not running' unless alive?
 
         id = next_request_id
         request = {
-          jsonrpc: "2.0",
+          jsonrpc: '2.0',
           id: id,
           method: method,
           params: params
@@ -74,10 +77,10 @@ module RubynCode
       # @param params [Hash] parameters for the notification
       # @return [void]
       def send_notification(method, params = {})
-        raise TransportError, "Transport is not running" unless alive?
+        raise TransportError, 'Transport is not running' unless alive?
 
         notification = {
-          jsonrpc: "2.0",
+          jsonrpc: '2.0',
           method: method,
           params: params
         }
@@ -92,7 +95,7 @@ module RubynCode
         return unless alive?
 
         begin
-          send_notification("notifications/cancelled")
+          send_notification('notifications/cancelled')
           @stdin&.close
         rescue IOError, Errno::EPIPE
           # Process may already be gone
@@ -138,7 +141,7 @@ module RubynCode
         Timeout.timeout(@timeout, TimeoutError, "MCP server did not respond within #{@timeout}s") do
           loop do
             line = @stdout.gets
-            raise TransportError, "MCP server closed stdout unexpectedly" if line.nil?
+            raise TransportError, 'MCP server closed stdout unexpectedly' if line.nil?
 
             line = line.strip
             next if line.empty?
@@ -147,17 +150,17 @@ module RubynCode
             next unless message
 
             # Skip notifications (no id field)
-            next unless message.key?("id")
+            next unless message.key?('id')
 
             # Skip responses for other requests
-            next unless message["id"] == expected_id
+            next unless message['id'] == expected_id
 
-            if message.key?("error")
-              err = message["error"]
+            if message.key?('error')
+              err = message['error']
               raise TransportError, "MCP error (#{err['code']}): #{err['message']}"
             end
 
-            return message["result"]
+            return message['result']
           end
         end
       end
@@ -172,9 +175,9 @@ module RubynCode
         return unless @wait_thread
 
         pid = @wait_thread.pid
-        Process.kill("TERM", pid)
+        Process.kill('TERM', pid)
         sleep(0.5)
-        Process.kill("KILL", pid) if @wait_thread.alive?
+        Process.kill('KILL', pid) if @wait_thread.alive?
       rescue Errno::ESRCH, Errno::EPERM
         # Process already gone or we lack permissions
       end

data/lib/rubyn_code/mcp/tool_bridge.rb

@@ -34,10 +34,10 @@ module RubynCode
         # @param tool_def [Hash] tool definition with "name", "description", "inputSchema"
         # @return [Class] the newly created and registered tool class
         def build_tool_class(mcp_client, tool_def)
-          remote_name = tool_def["name"]
+          remote_name = tool_def['name']
           tool_name = "mcp_#{sanitize_name(remote_name)}"
-          description = tool_def["description"] || "MCP tool: #{remote_name}"
-          input_schema = tool_def["inputSchema"] || {}
+          description = tool_def['description'] || "MCP tool: #{remote_name}"
+          input_schema = tool_def['inputSchema'] || {}
           parameters = build_parameters_from_schema(input_schema)
 
           klass = Class.new(Tools::Base) do
@@ -60,8 +60,8 @@ module RubynCode
             define_method(:format_result) do |result|
               case result
               when Hash
-                if result.key?("content")
-                  extract_content(result["content"])
+                if result.key?('content')
+                  extract_content(result['content'])
                 else
                   JSON.generate(result)
                 end
@@ -74,13 +74,13 @@ module RubynCode
 
             define_method(:extract_content) do |content|
               Array(content).map do |block|
-                case block["type"]
-                when "text"
-                  block["text"]
-                when "image"
+                case block['type']
+                when 'text'
+                  block['text']
+                when 'image'
                   "[image: #{block['mimeType']}]"
-                when "resource"
-                  block.dig("resource", "text") || "[resource: #{block.dig('resource', 'uri')}]"
+                when 'resource'
+                  block.dig('resource', 'text') || "[resource: #{block.dig('resource', 'uri')}]"
                 else
                   block.to_s
                 end
@@ -90,13 +90,13 @@ module RubynCode
 
           # Build parameter definitions from JSON Schema
           klass.define_singleton_method(:build_parameters) do |schema|
-            properties = schema["properties"] || {}
-            required = schema["required"] || []
+            properties = schema['properties'] || {}
+            required = schema['required'] || []
 
             properties.each_with_object({}) do |(name, prop), params|
               params[name.to_sym] = {
-                type: map_json_type(prop["type"]),
-                description: prop["description"] || "",
+                type: map_json_type(prop['type']),
+                description: prop['description'] || '',
                 required: required.include?(name)
               }
             end
@@ -104,12 +104,12 @@ module RubynCode
 
           klass.define_singleton_method(:map_json_type) do |json_type|
             case json_type
-            when "string"  then :string
-            when "integer" then :integer
-            when "number"  then :number
-            when "boolean" then :boolean
-            when "array"   then :array
-            when "object"  then :object
+            when 'string'  then :string
+            when 'integer' then :integer
+            when 'number'  then :number
+            when 'boolean' then :boolean
+            when 'array'   then :array
+            when 'object'  then :object
             else :string
             end
           end
@@ -123,13 +123,13 @@ module RubynCode
         # @param schema [Hash] JSON Schema with "properties" and "required"
         # @return [Hash]
         def build_parameters_from_schema(schema)
-          properties = schema["properties"] || {}
-          required = schema["required"] || []
+          properties = schema['properties'] || {}
+          required = schema['required'] || []
 
           properties.each_with_object({}) do |(name, prop), params|
             params[name.to_sym] = {
-              type: map_json_type(prop["type"]),
-              description: prop["description"] || "",
+              type: map_json_type(prop['type']),
+              description: prop['description'] || '',
               required: required.include?(name)
             }
           end
@@ -141,12 +141,12 @@ module RubynCode
         # @return [Symbol]
         def map_json_type(json_type)
           case json_type
-          when "string"  then :string
-          when "integer" then :integer
-          when "number"  then :number
-          when "boolean" then :boolean
-          when "array"   then :array
-          when "object"  then :object
+          when 'string'  then :string
+          when 'integer' then :integer
+          when 'number'  then :number
+          when 'boolean' then :boolean
+          when 'array'   then :array
+          when 'object'  then :object
           else :string
           end
         end
@@ -156,7 +156,7 @@ module RubynCode
         # @param name [String] the original tool name
         # @return [String] sanitized name
         def sanitize_name(name)
-          name.to_s.gsub(/[^a-zA-Z0-9_]/, "_").gsub(/_+/, "_").downcase
+          name.to_s.gsub(/[^a-zA-Z0-9_]/, '_').gsub(/_+/, '_').downcase
         end
       end
     end

data/lib/rubyn_code/memory/RUBYN.md

@@ -0,0 +1,17 @@
+# Layer 12: Memory
+
+Persistent cross-session memory backed by SQLite.
+
+## Classes
+
+- **`Store`** — Writes memories to the `memories` table. Each memory has content, category
+  (`code_pattern`, `user_preference`, `project_convention`, `error_resolution`, `decision`),
+  and a retention tier (`short`, `medium`, `long`).
+
+- **`Search`** — Full-text search across memories. Filters by category and tier.
+  Used by the agent to recall context from previous sessions.
+
+- **`SessionPersistence`** — Saves and restores session state (conversation, tasks, costs)
+  across REPL sessions. Keyed by session ID in the `sessions` table.
+
+- **`Models`** — Data objects mapping to/from SQLite memory rows.

data/lib/rubyn_code/memory/models.rb

@@ -33,13 +33,13 @@ module RubynCode
       end
 
       # @return [Boolean]
-      def short? = tier == "short"
+      def short? = tier == 'short'
 
       # @return [Boolean]
-      def medium? = tier == "medium"
+      def medium? = tier == 'medium'
 
       # @return [Boolean]
-      def long? = tier == "long"
+      def long? = tier == 'long'
 
       # @return [Hash]
       def to_h

data/lib/rubyn_code/memory/search.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require "json"
+require 'json'
 
 module RubynCode
   module Memory
@@ -24,16 +24,16 @@ module RubynCode
       # @param limit [Integer] maximum results (default 10)
       # @return [Array<MemoryRecord>]
       def search(query, tier: nil, category: nil, limit: 10)
-        conditions = ["m.project_path = ?"]
+        conditions = ['m.project_path = ?']
         params = [@project_path]
 
         if tier
-          conditions << "m.tier = ?"
+          conditions << 'm.tier = ?'
           params << tier
         end
 
         if category
-          conditions << "m.category = ?"
+          conditions << 'm.category = ?'
           params << category
         end
 
@@ -127,20 +127,20 @@ module RubynCode
       # @param row [Hash]
       # @return [MemoryRecord]
       def build_record(row)
-        metadata = parse_json(row["metadata"])
+        metadata = parse_json(row['metadata'])
 
         MemoryRecord.new(
-          id: row["id"],
-          project_path: row["project_path"],
-          tier: row["tier"],
-          category: row["category"],
-          content: row["content"],
-          relevance_score: row["relevance_score"].to_f,
-          access_count: row["access_count"].to_i,
-          last_accessed_at: row["last_accessed_at"],
-          expires_at: row["expires_at"],
+          id: row['id'],
+          project_path: row['project_path'],
+          tier: row['tier'],
+          category: row['category'],
+          content: row['content'],
+          relevance_score: row['relevance_score'].to_f,
+          access_count: row['access_count'].to_i,
+          last_accessed_at: row['last_accessed_at'],
+          expires_at: row['expires_at'],
           metadata: metadata,
-          created_at: row["created_at"]
+          created_at: row['created_at']
         )
       end
 
@@ -152,9 +152,9 @@ module RubynCode
       def touch_accessed(records)
         return if records.empty?
 
-        now = Time.now.utc.strftime("%Y-%m-%d %H:%M:%S")
+        now = Time.now.utc.strftime('%Y-%m-%d %H:%M:%S')
         ids = records.map(&:id)
-        placeholders = (["?"] * ids.size).join(", ")
+        placeholders = (['?'] * ids.size).join(', ')
 
         @db.execute(
           "UPDATE memories SET access_count = access_count + 1, last_accessed_at = ? WHERE id IN (#{placeholders})",