personality 0.1.4 → 0.1.5

data/lib/personality/mcp/server.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 require "mcp"
-require "mcp/transports/stdio"
+require "mcp/server/transports/stdio_transport"
 require "json"
 require_relative "../db"
 require_relative "../memory"
@@ -11,23 +11,36 @@ require_relative "../cart"
 module Personality
   module MCP
     class Server
-      def self.run
+      # Modes:
+      #   :all     - All tools (default)
+      #   :indexer - Only index_* tools (for local file indexing)
+      #   :core    - Only memory/cart tools (for remote/centralized)
+      VALID_MODES = %i[all indexer core].freeze
+
+      def self.run(mode: :all)
         DB.migrate!
-        new.start
+        new(mode: mode).start
       end
 
-      def initialize
+      def initialize(mode: :all)
+        @mode = VALID_MODES.include?(mode) ? mode : :all
+        server_name = case @mode
+        when :indexer then "indexer"
+        when :core then "core"
+        else "personality"
+        end
+
         @server = ::MCP::Server.new(
-          name: "core",
+          name: server_name,
           version: Personality::VERSION
         )
         @server.server_context = {}
         register_tools
-        register_resources
+        register_resources unless @mode == :indexer
       end
 
       def start
-        transport = ::MCP::Transports::StdioTransport.new(@server)
+        transport = ::MCP::Server::Transports::StdioTransport.new(@server)
         transport.open
       end
 
@@ -41,7 +54,7 @@ module Personality
 
       def register_memory_tools
         @server.define_tool(
-          name: "memory.store",
+          name: "memory_store",
           description: "Store a memory with subject and content. Automatically generates embedding.",
           input_schema: {
             type: "object",
@@ -58,7 +71,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "memory.recall",
+          name: "memory_recall",
           description: "Recall memories by semantic similarity to a query.",
           input_schema: {
             type: "object",
@@ -75,7 +88,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "memory.search",
+          name: "memory_search",
           description: "Search memories by subject or metadata.",
           input_schema: {
             type: "object",
@@ -90,7 +103,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "memory.forget",
+          name: "memory_forget",
           description: "Delete a memory by ID.",
           input_schema: {
             type: "object",
@@ -105,7 +118,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "memory.list",
+          name: "memory_list",
           description: "List all memory subjects and counts.",
           input_schema: {type: "object", properties: {}}
         ) do |server_context:, **|
@@ -118,7 +131,7 @@ module Personality
 
       def register_index_tools
         @server.define_tool(
-          name: "index.code",
+          name: "index_code",
           description: "Index code files in a directory for semantic search.",
           input_schema: {
             type: "object",
@@ -135,7 +148,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "index.docs",
+          name: "index_docs",
           description: "Index documentation files for semantic search.",
           input_schema: {
             type: "object",
@@ -151,7 +164,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "index.search",
+          name: "index_search",
           description: "Search indexed code and docs by semantic similarity.",
           input_schema: {
             type: "object",
@@ -174,7 +187,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "index.status",
+          name: "index_status",
           description: "Show indexing status and statistics.",
           input_schema: {
             type: "object",
@@ -188,7 +201,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "index.clear",
+          name: "index_clear",
           description: "Clear index for a project or all.",
           input_schema: {
             type: "object",
@@ -207,7 +220,7 @@ module Personality
 
       def register_cart_tools
         @server.define_tool(
-          name: "cart.list",
+          name: "cart_list",
           description: "List all personas.",
           input_schema: {type: "object", properties: {}}
         ) do |server_context:, **|
@@ -216,7 +229,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "cart.use",
+          name: "cart_use",
           description: "Switch active persona.",
           input_schema: {
             type: "object",
@@ -231,7 +244,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "cart.create",
+          name: "cart_create",
           description: "Create a new persona.",
           input_schema: {
             type: "object",
@@ -252,7 +265,7 @@ module Personality
 
       def register_persona_tools
         @server.define_tool(
-          name: "cart.teach",
+          name: "cart_teach",
           description: "Learn a persona from a training YAML file. Creates a .pcart cartridge and imports memories into the database.",
           input_schema: {
             type: "object",
@@ -278,7 +291,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "cart.show",
+          name: "cart_show",
           description: "Show persona details and LLM instructions for a cartridge.",
           input_schema: {
             type: "object",
@@ -320,7 +333,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "cart.instructions",
+          name: "cart_instructions",
           description: "Get the LLM persona instructions for the active or specified cart. Returns markdown formatted character instructions.",
           input_schema: {
             type: "object",
@@ -351,7 +364,7 @@ module Personality
         end
 
         @server.define_tool(
-          name: "cart.carts",
+          name: "cart_carts",
           description: "List available .pcart cartridge files with their metadata.",
           input_schema: {type: "object", properties: {}}
         ) do |server_context:, **|
@@ -396,7 +409,7 @@ module Personality
 
       def register_resource_tools
         @server.define_tool(
-          name: "resource.read",
+          name: "resource_read",
           description: "Read an MCP resource by URI. Available resources: memory://subjects (subjects with counts), memory://stats (total memories, date range), memory://recent (last 10 memories).",
           input_schema: {
             type: "object",
@@ -430,11 +443,151 @@ module Personality
       end
 
       def register_tools
-        register_memory_tools
-        register_index_tools
-        register_cart_tools
-        register_persona_tools
-        register_resource_tools
+        case @mode
+        when :indexer
+          # Local indexing only - no memory/cart/messaging
+          register_index_tools
+        when :core
+          # Centralized services - no local file indexing
+          register_memory_tools
+          register_cart_tools
+          register_persona_tools
+          register_resource_tools
+          register_messaging_tools
+          register_shell_tools
+        else # :all
+          register_memory_tools
+          register_index_tools
+          register_cart_tools
+          register_persona_tools
+          register_resource_tools
+          register_messaging_tools
+          register_shell_tools
+        end
+      end
+
+      # === Shell Tools ===
+
+      def register_shell_tools
+        @server.define_tool(
+          name: "shell_exec",
+          description: "Execute a shell command on junkpile (Ubuntu x86_64). Returns stdout, stderr, and exit code.",
+          input_schema: {
+            type: "object",
+            properties: {
+              command: {type: "string", description: "Shell command to execute"},
+              timeout: {type: "integer", description: "Timeout in seconds (default: 60, max: 300)"},
+              cwd: {type: "string", description: "Working directory (default: home)"}
+            },
+            required: %w[command]
+          }
+        ) do |command:, server_context:, **opts|
+          require "open3"
+          require "shellwords"
+
+          timeout = [opts[:timeout] || 60, 300].min
+          cwd = opts[:cwd] || ENV["HOME"]
+
+          # Wrap with timeout
+          full_cmd = "timeout #{timeout} bash -c #{Shellwords.escape(command)}"
+
+          stdout, stderr, status = Open3.capture3(full_cmd, chdir: cwd)
+
+          result = {
+            command: command,
+            cwd: cwd,
+            exit_code: status.exitstatus,
+            stdout: stdout,
+            stderr: stderr,
+            timed_out: status.exitstatus == 124
+          }
+          ::MCP::Tool::Response.new([{type: "text", text: JSON.generate(result)}])
+        end
+      end
+
+      # === Messaging Tools ===
+
+      SIGNAL_ACCOUNT = "+48600965497" # Moto G52 - BT's comm array
+      PILOT_NUMBER = "+48535329895"   # Adam's number
+      SIGNAL_CLI = "/home/linuxbrew/.linuxbrew/bin/signal-cli"
+
+      def register_messaging_tools
+        @server.define_tool(
+          name: "signal_send",
+          description: "Send a Signal message. Default sends from BT's comm array (+48600965497) to Pilot (+48535329895).",
+          input_schema: {
+            type: "object",
+            properties: {
+              message: {type: "string", description: "Message text to send"},
+              to: {type: "string", description: "Recipient phone number (default: Pilot's number)"},
+              from: {type: "string", description: "Sender account (default: BT's comm array)"}
+            },
+            required: %w[message]
+          }
+        ) do |message:, server_context:, **opts|
+          from = opts[:from] || SIGNAL_ACCOUNT
+          to = opts[:to] || PILOT_NUMBER
+
+          # Escape message for shell
+          escaped_message = message.gsub("'", "'\\''")
+          cmd = "#{SIGNAL_CLI} -a #{from} send -m '#{escaped_message}' #{to}"
+
+          output = `#{cmd} 2>&1`
+          success = $?.success?
+
+          result = {
+            success: success,
+            from: from,
+            to: to,
+            message: message,
+            output: output.strip
+          }
+          ::MCP::Tool::Response.new([{type: "text", text: JSON.generate(result)}])
+        end
+
+        @server.define_tool(
+          name: "signal_receive",
+          description: "Check for incoming Signal messages on BT's comm array.",
+          input_schema: {
+            type: "object",
+            properties: {
+              account: {type: "string", description: "Account to check (default: BT's comm array)"},
+              timeout: {type: "integer", description: "Timeout in seconds (default: 5)"}
+            }
+          }
+        ) do |server_context:, **opts|
+          account = opts[:account] || SIGNAL_ACCOUNT
+          timeout = opts[:timeout] || 5
+
+          cmd = "timeout #{timeout} #{SIGNAL_CLI} -a #{account} receive --json 2>&1"
+          output = `#{cmd}`
+
+          messages = []
+          output.each_line do |line|
+            next if line.strip.empty?
+            begin
+              msg = JSON.parse(line)
+              if msg["envelope"] && msg["envelope"]["dataMessage"]
+                data = msg["envelope"]["dataMessage"]
+                messages << {
+                  from: msg["envelope"]["source"],
+                  timestamp: msg["envelope"]["timestamp"],
+                  message: data["message"],
+                  group: data["groupInfo"]&.dig("groupId")
+                }
+              end
+            rescue JSON::ParserError
+              # Skip non-JSON lines
+            end
+          end
+
+          result = {
+            account: account,
+            message_count: messages.length,
+            messages: messages
+          }
+          ::MCP::Tool::Response.new([{type: "text", text: JSON.generate(result)}])
+        end
       end
 
       def read_memory_resource(uri)

data/lib/personality/mcp/tts_server.rb

@@ -40,17 +40,23 @@ module Personality
       def register_speak
         @server.define_tool(
           name: "speak",
-          description: "Speak text aloud using TTS. Synthesizes and plays audio in the background.",
+          description: "Speak text aloud using TTS. Synthesizes and plays audio in the background unless wait=true.",
           input_schema: {
             type: "object",
             properties: {
               text: {type: "string", description: "Text to speak aloud"},
-              voice: {type: "string", description: "Voice model name (optional, uses active voice if omitted)"}
+              voice: {type: "string", description: "Voice model name (optional, uses active voice if omitted)"},
+              language: {type: "string", description: "Language code (e.g. 'en', 'pl'). Auto-detected if omitted."},
+              wait: {type: "boolean", description: "Wait for playback to complete before returning (default: false)"}
             },
             required: %w[text]
           }
         ) do |text:, server_context:, **opts|
-          result = Personality::TTS.speak(text, voice: opts[:voice])
+          result = if opts[:wait]
+            Personality::TTS.speak_and_wait(text, voice: opts[:voice], language: opts[:language])
+          else
+            Personality::TTS.speak(text, voice: opts[:voice], language: opts[:language])
+          end
           ::MCP::Tool::Response.new([{type: "text", text: JSON.generate(result)}])
         end
       end
@@ -85,7 +91,8 @@ module Personality
         ) do |server_context:, **|
           voice = Personality::TTS.active_voice
           installed = !Personality::TTS.find_voice(voice).nil?
-          ::MCP::Tool::Response.new([{type: "text", text: JSON.generate({voice: voice, installed: installed})}])
+          backend = Personality::TTS.backend
+          ::MCP::Tool::Response.new([{type: "text", text: JSON.generate({voice: voice, installed: installed, backend: backend})}])
         end
       end