legion-tty 0.4.41 → 0.5.0

data/lib/legion/tty/components/digital_rain.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
 require_relative '../theme'
 
 module Legion
@@ -7,6 +8,11 @@ module Legion
     module Components
       # rubocop:disable Metrics/ClassLength
       class DigitalRain
+        include Legion::Logging::Helper
+
+        class << self
+          include Legion::Logging::Helper
+        end
         # rubocop:disable Naming/VariableNumber
         FADE_SHADES = %i[
           purple_12 purple_11 purple_10 purple_9 purple_8
@@ -40,7 +46,7 @@ module Legion
                                    .map { |s| s.name.sub(/^lex-/, '') }
           gems.empty? ? FALLBACK_NAMES : gems
         rescue StandardError => e
-          Legion::Logging.debug("extension_names failed: #{e.message}") if defined?(Legion::Logging)
+          log.debug { "extension_names failed: #{e.message}" }
           FALLBACK_NAMES
         end
 

data/lib/legion/tty/components/markdown_view.rb

@@ -1,15 +1,20 @@
 # frozen_string_literal: true
 
 require 'tty-markdown'
+require 'legion/logging'
 
 module Legion
   module TTY
     module Components
       module MarkdownView
+        class << self
+          include Legion::Logging::Helper
+        end
+
         def self.render(text, width: 80)
           ::TTY::Markdown.parse(text, width: width)
         rescue StandardError => e
-          Legion::Logging.warn("markdown render failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "markdown render failed: #{e.message}" }
           "#{text}\n(markdown render error: #{e.message})"
         end
       end

data/lib/legion/tty/components/message_stream.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'English'
+require 'legion/logging'
 require_relative '../theme'
 
 module Legion
@@ -8,6 +9,8 @@ module Legion
     module Components
       # rubocop:disable Metrics/ClassLength
       class MessageStream
+        include Legion::Logging::Helper
+
         attr_reader :messages, :scroll_offset
         attr_accessor :mute_system, :silent_mode, :highlights, :filter, :truncate_limit, :wrap_width, :show_numbers,
                       :colorize, :show_timestamps
@@ -183,7 +186,7 @@ module Legion
           require_relative 'markdown_view'
           MarkdownView.render(text, width: width)
         rescue StandardError => e
-          Legion::Logging.warn("render_markdown failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "render_markdown failed: #{e.message}" }
           text
         end
 
@@ -208,7 +211,7 @@ module Legion
             result.gsub(pattern) { "#{HIGHLIGHT_COLOR}#{$LAST_MATCH_INFO}#{HIGHLIGHT_RESET}" }
           end
         rescue StandardError => e
-          Legion::Logging.warn("apply_highlights failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "apply_highlights failed: #{e.message}" }
           text
         end
 

data/lib/legion/tty/components/model_picker.rb

@@ -1,9 +1,13 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Components
       class ModelPicker
+        include Legion::Logging::Helper
+
         def initialize(current_provider: nil, current_model: nil)
           @current_provider = current_provider
           @current_model = current_model
@@ -38,7 +42,7 @@ module Legion
           end
           prompt.select('Select model:', choices, per_page: 10)
         rescue ::TTY::Reader::InputInterrupt, Interrupt => e
-          Legion::Logging.debug("model picker cancelled: #{e.message}") if defined?(Legion::Logging)
+          log.debug { "model picker cancelled: #{e.message}" }
           nil
         end
       end

data/lib/legion/tty/components/progress_panel.rb

@@ -1,11 +1,14 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
 require_relative '../theme'
 
 module Legion
   module TTY
     module Components
       class ProgressPanel
+        include Legion::Logging::Helper
+
         attr_reader :title, :total, :current
 
         def initialize(title:, total:, output: $stdout)
@@ -62,7 +65,7 @@ module Legion
             width: 40
           )
         rescue LoadError => e
-          Legion::Logging.debug("tty-progressbar not available: #{e.message}") if defined?(Legion::Logging)
+          log.debug { "tty-progressbar not available: #{e.message}" }
           nil
         end
       end

data/lib/legion/tty/components/session_picker.rb

@@ -1,9 +1,13 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Components
       class SessionPicker
+        include Legion::Logging::Helper
+
         def initialize(session_store:)
           @session_store = session_store
         end
@@ -20,7 +24,7 @@ module Legion
           choices << { name: '+ New session', value: :new }
           prompt.select('Select session:', choices, per_page: 10)
         rescue ::TTY::Reader::InputInterrupt, Interrupt => e
-          Legion::Logging.debug("session picker cancelled: #{e.message}") if defined?(Legion::Logging)
+          log.debug { "session picker cancelled: #{e.message}" }
           nil
         end
       end

data/lib/legion/tty/components/status_bar.rb

@@ -15,6 +15,9 @@ module Legion
         end
 
         def notify(message:, level: :info, ttl: 5)
+          priority = level_to_priority(level)
+          return if priority != :urgent && !Legion::TTY::NotificationGate.should_deliver?(priority: priority)
+
           @notifications << Notification.new(message: message, level: level, ttl: ttl)
         end
 
@@ -122,6 +125,10 @@ module Legion
           Theme.c(:muted, "#{scroll[:current]}/#{scroll[:total]}")
         end
 
+        def level_to_priority(level)
+          { info: :ambient, success: :low, warning: :normal, error: :urgent }[level] || :normal
+        end
+
         def format_number(num)
           num.to_s.chars.reverse.each_slice(3).map(&:join).join(',').reverse
         end

data/lib/legion/tty/components/table_view.rb

@@ -1,15 +1,21 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Components
       module TableView
+        class << self
+          include Legion::Logging::Helper
+        end
+
         def self.render(headers:, rows:, width: 80)
           require 'tty-table'
           table = ::TTY::Table.new(header: headers, rows: rows)
           table.render(:unicode, width: width, padding: [0, 1]) || ''
         rescue StandardError => e
-          Legion::Logging.warn("table render failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "table render failed: #{e.message}" }
           "Table render error: #{e.message}"
         end
       end

data/lib/legion/tty/components/tool_call_parser.rb

@@ -1,11 +1,14 @@
 # frozen_string_literal: true
 
 require 'legion/json'
+require 'legion/logging'
 
 module Legion
   module TTY
     module Components
       class ToolCallParser
+        include Legion::Logging::Helper
+
         OPEN_TAG = '<tool_call>'
         CLOSE_TAG = '</tool_call>'
         MAX_BUFFER = 4096
@@ -97,7 +100,7 @@ module Legion
 
           @on_tool_call.call(name: name, args: args)
         rescue StandardError => e
-          Legion::Logging.warn("emit_tool_call failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "emit_tool_call failed: #{e.message}" }
           @on_text.call("#{OPEN_TAG}#{json_str}#{CLOSE_TAG}")
         end
 

data/lib/legion/tty/daemon_client.rb

@@ -4,6 +4,7 @@ require 'net/http'
 require 'uri'
 require 'fileutils'
 require 'legion/json'
+require 'legion/logging'
 
 module Legion
   module TTY
@@ -12,11 +13,14 @@ module Legion
 
       # rubocop:disable Metrics/ClassLength
       class << self
+        include Legion::Logging::Helper
+
         def configure(daemon_url: 'http://127.0.0.1:4567', cache_file: nil, timeout: 5)
           @daemon_url = daemon_url
           @cache_file = cache_file || File.expand_path('~/.legionio/catalog.json')
           @timeout = timeout
           @manifest = nil
+          log.info { "TTY daemon client configured daemon_url=#{@daemon_url} timeout=#{@timeout}" }
         end
 
         def available?
@@ -26,7 +30,7 @@ module Legion
           end
           response.code.to_i == 200
         rescue StandardError => e
-          Legion::Logging.debug("daemon available? check failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :debug, operation: 'tty.daemon_client.available?', daemon_url: daemon_url)
           false
         end
 
@@ -37,12 +41,9 @@ module Legion
           end
           return nil unless response.code.to_i == 200
 
-          body = Legion::JSON.load(response.body)
-          @manifest = body[:data]
-          write_cache(@manifest)
-          @manifest
+          store_manifest(Legion::JSON.load(response.body)[:data])
         rescue StandardError => e
-          Legion::Logging.warn("fetch_manifest failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.fetch_manifest', daemon_url: daemon_url)
           nil
         end
 
@@ -53,7 +54,7 @@ module Legion
 
           @manifest = Legion::JSON.load(File.read(@cache_file))
         rescue StandardError => e
-          Legion::Logging.warn("cached_manifest failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.cached_manifest', cache_file: @cache_file)
           nil
         end
 
@@ -80,29 +81,47 @@ module Legion
 
           uri = URI("#{daemon_url}/api/llm/chat")
           payload = Legion::JSON.dump({ message: message, model: model, provider: provider })
+          log.debug { "TTY chat request model=#{model} provider=#{provider} message_length=#{message.to_s.length}" }
           response = post_json(uri, payload)
 
           return nil unless response && SUCCESS_CODES.include?(response.code.to_i)
 
           Legion::JSON.load(response.body)
         rescue StandardError => e
-          Legion::Logging.warn("chat failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.chat',
+                              daemon_url: daemon_url, model: model, provider: provider)
           nil
         end
 
         def inference(messages:, tools: [], model: nil, provider: nil, timeout: 120)
+          log.debug { "TTY inference model=#{model} provider=#{provider} msgs=#{Array(messages).size}" }
           response = post_inference(messages: messages, tools: tools, model: model,
                                     provider: provider, timeout: timeout)
           return inference_error_result(response) unless SUCCESS_CODES.include?(response.code.to_i)
 
-          body = Legion::JSON.load(response.body)
-          data = body[:data] || body
-          { status: :ok, data: data }
+          parse_inference_response(response)
         rescue StandardError => e
-          Legion::Logging.warn("inference failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.inference',
+                              daemon_url: daemon_url, model: model, provider: provider, timeout: timeout)
           { status: :unavailable, error: { message: e.message } }
         end
 
+        def run_tool(name:, args: {})
+          uri = URI("#{daemon_url}/api/tools/run")
+          payload = Legion::JSON.dump({ name: name, args: args })
+          response = post_json(uri, payload)
+          if SUCCESS_CODES.include?(response.code.to_i)
+            body = Legion::JSON.load(response.body)
+            { status: :ok, data: body[:data] || body }
+          else
+            { status: :error, error: "HTTP #{response.code}" }
+          end
+        rescue StandardError => e
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.run_tool',
+                              daemon_url: daemon_url, name: name)
+          { status: :unavailable }
+        end
+
         def reset!
           @daemon_url = nil
           @cache_file = nil
@@ -112,6 +131,19 @@ module Legion
 
         private
 
+        def parse_inference_response(response)
+          body = Legion::JSON.load(response.body)
+          data = body[:data] || body
+          { status: :ok, data: data }
+        end
+
+        def store_manifest(data)
+          @manifest = data
+          write_cache(@manifest)
+          log.info { "TTY fetched daemon manifest entries=#{Array(@manifest).size}" }
+          @manifest
+        end
+
         def post_inference(messages:, tools:, model:, provider:, timeout:)
           uri = URI("#{daemon_url}/api/llm/inference")
           payload = Legion::JSON.dump({ messages: messages, tools: tools,
@@ -151,8 +183,9 @@ module Legion
 
           FileUtils.mkdir_p(File.dirname(@cache_file))
           File.write(@cache_file, Legion::JSON.dump(data))
+          log.debug { "TTY daemon manifest cache updated file=#{@cache_file}" }
         rescue StandardError => e
-          Legion::Logging.warn("write_cache failed: #{e.message}") if defined?(Legion::Logging)
+          handle_exception(e, level: :warn, operation: 'tty.daemon_client.write_cache', cache_file: @cache_file)
           nil
         end
       end

data/lib/legion/tty/keybinding_manager.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'legion/json'
+require 'legion/logging'
 require 'fileutils'
 
 module Legion
@@ -15,6 +16,8 @@ module Legion
     #
     # User overrides loaded from ~/.legionio/keybindings.json at boot.
     class KeybindingManager
+      include Legion::Logging::Helper
+
       CONTEXTS = %i[global chat dashboard extensions config command_palette session_picker history].freeze
 
       OVERRIDES_PATH = File.expand_path('~/.legionio/keybindings.json')
@@ -108,7 +111,7 @@ module Legion
         raw = Legion::JSON.parse(File.read(@overrides_path), symbolize_names: true)
         raw.each { |key, cfg| apply_override(key, cfg) }
       rescue Legion::JSON::ParseError => e
-        Legion::Logging.warn("keybindings load failed: #{e.message}") if defined?(Legion::Logging)
+        log.warn { "keybindings load failed: #{e.message}" }
       end
 
       private

data/lib/legion/tty/notification_gate.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require 'legion/logging'
+
+module Legion
+  module TTY
+    module NotificationGate
+      class << self
+        include Legion::Logging::Helper
+
+        def should_deliver?(priority: :normal)
+          return true unless gaia_gate_available?
+
+          Legion::Gaia::NotificationGate.instance.should_notify?(priority: priority)
+        rescue StandardError => e
+          log.debug { "NotificationGate.should_deliver? failed: #{e.message}" }
+          true
+        end
+
+        def update_presence(availability:, activity: nil)
+          return unless gaia_gate_available?
+
+          Legion::Gaia::NotificationGate.instance.update_presence(
+            availability: availability, activity: activity
+          )
+        rescue StandardError => e
+          log.debug { "NotificationGate.update_presence failed: #{e.message}" }
+          nil
+        end
+
+        private
+
+        def gaia_gate_available?
+          defined?(Legion::Gaia::NotificationGate) &&
+            Legion::Gaia::NotificationGate.respond_to?(:instance)
+        end
+      end
+    end
+  end
+end

data/lib/legion/tty/notify.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     # Notify — OS-level terminal notification dispatcher with auto-detection.
@@ -20,6 +22,8 @@ module Legion
       BACKENDS = %w[iterm2 kitty ghostty notify_send osascript bell].freeze
 
       class << self
+        include Legion::Logging::Helper
+
         # Send a notification.
         # @param message [String] notification body
         # @param title [String] notification title
@@ -82,7 +86,7 @@ module Legion
           else notify_bell
           end
         rescue StandardError => e
-          Legion::Logging.warn("Notify dispatch failed: #{e.message}") if defined?(Legion::Logging)
+          log.warn { "Notify dispatch failed: #{e.message}" }
           notify_bell
         end
 

data/lib/legion/tty/screens/chat/export_commands.rb

@@ -1,10 +1,14 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Screens
       class Chat < Base
         module ExportCommands
+          include Legion::Logging::Helper
+
           private
 
           def handle_export(input)
@@ -15,7 +19,7 @@ module Legion
             @message_stream.add_message(role: :system, content: "Exported to: #{path}")
             :handled
           rescue StandardError => e
-            Legion::Logging.warn("handle_export failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "handle_export failed: #{e.message}" }
             @message_stream.add_message(role: :system, content: "Export failed: #{e.message}")
             :handled
           end
@@ -129,7 +133,7 @@ module Legion
             @message_stream.add_message(role: :system, content: "Bookmarks exported to: #{path}")
             :handled
           rescue StandardError => e
-            Legion::Logging.warn("handle_bookmark failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "handle_bookmark failed: #{e.message}" }
             @message_stream.add_message(role: :system, content: "Bookmark export failed: #{e.message}")
             :handled
           end
@@ -152,7 +156,7 @@ module Legion
             end
             :handled
           rescue StandardError => e
-            Legion::Logging.warn("handle_tee failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "handle_tee failed: #{e.message}" }
             @message_stream.add_message(role: :system, content: "Tee error: #{e.message}")
             :handled
           end
@@ -162,7 +166,7 @@ module Legion
 
             File.open(@tee_path, 'a') { |f| f.puts(line) }
           rescue StandardError => e
-            Legion::Logging.warn("tee_message failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "tee_message failed: #{e.message}" }
             nil
           end
         end

data/lib/legion/tty/screens/chat/message_commands.rb

@@ -1,10 +1,14 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Screens
       class Chat < Base
         module MessageCommands
+          include Legion::Logging::Helper
+
           INJECT_VALID_ROLES = %w[user assistant system].freeze
           TRANSFORM_OPS = %w[upcase downcase reverse strip squeeze].freeze
 
@@ -98,7 +102,7 @@ module Legion
             display_grep_results(results, pattern_str)
             :handled
           rescue RegexpError => e
-            Legion::Logging.warn("handle_grep regex error: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "handle_grep regex error: #{e.message}" }
             @message_stream.add_message(role: :system, content: "Invalid regex: #{e.message}")
             :handled
           end
@@ -302,11 +306,11 @@ module Legion
           def copy_to_clipboard(text)
             IO.popen('pbcopy', 'w') { |io| io.write(text) }
           rescue Errno::ENOENT => e
-            Legion::Logging.debug("pbcopy not available: #{e.message}") if defined?(Legion::Logging)
+            log.debug { "pbcopy not available: #{e.message}" }
             begin
               IO.popen('xclip -selection clipboard', 'w') { |io| io.write(text) }
             rescue Errno::ENOENT => e
-              Legion::Logging.debug("xclip not available: #{e.message}") if defined?(Legion::Logging)
+              log.debug { "xclip not available: #{e.message}" }
               nil
             end
           end
@@ -398,7 +402,7 @@ module Legion
             parsed = ::JSON.parse(raw, symbolize_names: true)
             parsed.is_a?(Array) ? parsed : []
           rescue StandardError => e
-            Legion::Logging.warn("load_favorites failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "load_favorites failed: #{e.message}" }
             []
           end
 

data/lib/legion/tty/screens/chat/model_commands.rb

@@ -1,10 +1,14 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Screens
       class Chat < Base
         module ModelCommands
+          include Legion::Logging::Helper
+
           private
 
           def handle_model(input)
@@ -24,7 +28,7 @@ module Legion
             @message_stream.add_message(role: :system,
                                         content: "Model preference set to: #{name} (applied on next daemon request)")
           rescue StandardError => e
-            Legion::Logging.warn("switch_model failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "switch_model failed: #{e.message}" }
             @message_stream.add_message(role: :system, content: "Failed to switch model: #{e.message}")
           end
 
@@ -40,7 +44,7 @@ module Legion
 
             Legion::LLM.chat(provider: name)
           rescue StandardError => e
-            Legion::Logging.warn("try_provider_switch failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "try_provider_switch failed: #{e.message}" }
             nil
           end
 

data/lib/legion/tty/screens/chat/session_commands.rb

@@ -1,10 +1,14 @@
 # frozen_string_literal: true
 
+require 'legion/logging'
+
 module Legion
   module TTY
     module Screens
       class Chat < Base
         module SessionCommands
+          include Legion::Logging::Helper
+
           private
 
           def handle_save(input)
@@ -142,7 +146,7 @@ module Legion
             @last_autosave = Time.now
             @status_bar.notify(message: 'Autosaved', level: :info, ttl: 2)
           rescue StandardError => e
-            Legion::Logging.warn("check_autosave failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "check_autosave failed: #{e.message}" }
             nil
           end
 
@@ -154,7 +158,7 @@ module Legion
             end
             @session_store.save(@session_name, messages: @message_stream.messages)
           rescue StandardError => e
-            Legion::Logging.warn("auto_save_session failed: #{e.message}") if defined?(Legion::Logging)
+            log.warn { "auto_save_session failed: #{e.message}" }
             nil
           end