brainiac 0.0.1

data/monitor/setup-waybar-module.rb

@@ -0,0 +1,113 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# One-time setup script to add Brainiac module to waybar config
+# Run this once, then the module will update dynamically without config rewrites
+
+require "json"
+require "fileutils"
+
+WAYBAR_CONFIG = File.expand_path("~/.config/waybar/config.jsonc")
+WAYBAR_SCRIPT = File.expand_path("~/.brainiac/bin/waybar-status")
+
+# Create a wrapper script that resolves the running server's waybar.rb dynamically
+wrapper_dir = File.expand_path("~/.brainiac/bin")
+FileUtils.mkdir_p(wrapper_dir)
+wrapper_path = File.join(wrapper_dir, "waybar-status")
+File.write(wrapper_path, <<~SCRIPT)
+  #!/usr/bin/env ruby
+  # Resolves the running Brainiac server's waybar module dynamically.
+  # This allows worktrees / branches to work without reconfiguring waybar.
+
+  root_file = File.expand_path("~/.brainiac/server.root")
+  if File.exist?(root_file)
+    server_root = File.read(root_file).strip
+    waybar_script = File.join(server_root, "monitor", "waybar.rb")
+    if File.exist?(waybar_script)
+      load waybar_script
+      exit
+    end
+  end
+
+  # Fallback: no server root known, try the API directly
+  require "json"
+  require "net/http"
+
+  begin
+    uri = URI("http://localhost:4567/api/status")
+    response = Net::HTTP.get_response(uri)
+    if response.is_a?(Net::HTTPSuccess)
+      data = JSON.parse(response.body)
+      sessions = data["sessions"] || []
+      if sessions.empty?
+        puts({ text: "💤", tooltip: "No active agent sessions", class: "idle" }.to_json)
+      else
+        puts({ text: "🟢 \#{sessions.size}", tooltip: sessions.map { |s| s["agent"] }.join(", "), class: "working" }.to_json)
+      end
+    else
+      puts({ text: "⚠️", tooltip: "Brainiac Error: HTTP \#{response.code}", class: "error" }.to_json)
+    end
+  rescue StandardError => e
+    puts({ text: "⚠️", tooltip: "Brainiac Error: \#{e.message}", class: "error" }.to_json)
+  end
+SCRIPT
+File.chmod(0o755, wrapper_path)
+
+def load_config
+  content = File.read(WAYBAR_CONFIG)
+  # Strip comments for JSON parsing
+  json_content = content.lines.reject { |line| line.strip.start_with?("//") }.join
+  JSON.parse(json_content)
+end
+
+def save_config(config)
+  File.write(WAYBAR_CONFIG, JSON.pretty_generate(config))
+end
+
+# Load current config
+config = load_config
+
+# Remove old brainiac modules if they exist (from all module arrays)
+%w[modules-left modules-center modules-right].each do |section|
+  next unless config[section].is_a?(Array)
+
+  config[section].reject! { |m| ["custom/brainiac", "group/brainiac-agents"].include?(m.to_s) }
+end
+config.each_key do |key|
+  config.delete(key) if ["custom/brainiac", "group/brainiac-agents"].include?(key.to_s)
+end
+
+# Add single dynamic module at the end of modules-center (after deploy envs)
+config["modules-center"] ||= []
+config["modules-center"].push("custom/brainiac")
+
+# Add module config
+config["custom/brainiac"] = {
+  "exec" => WAYBAR_SCRIPT,
+  "return-type" => "json",
+  "interval" => 3,
+  "format" => "{}",
+  "tooltip" => true,
+  "on-click" => File.expand_path("~/.brainiac/bin/waybar-logs").to_s
+}
+
+# Create on-click wrapper too
+logs_wrapper = File.expand_path("~/.brainiac/bin/waybar-logs")
+File.write(logs_wrapper, <<~SCRIPT)
+  #!/usr/bin/env ruby
+  root_file = File.expand_path("~/.brainiac/server.root")
+  if File.exist?(root_file)
+    server_root = File.read(root_file).strip
+    script = File.join(server_root, "monitor", "view-logs-rofi.rb")
+    exec("ruby", script) if File.exist?(script)
+  end
+  warn "Brainiac server root not found"
+SCRIPT
+File.chmod(0o755, logs_wrapper)
+
+# Save updated config
+save_config(config)
+
+puts "✓ Brainiac module added to waybar config"
+puts "  Module will update every 3 seconds without config rewrites"
+puts "  Restart waybar to apply: omarchy restart waybar"

data/monitor/setup-xbar-plugin.rb

@@ -0,0 +1,35 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# One-time setup: symlinks the Brainiac xbar plugin into xbar's plugin directory
+# Run this once on macOS after installing xbar
+
+require "fileutils"
+
+XBAR_PLUGIN_DIR = File.expand_path("~/Library/Application Support/xbar/plugins")
+PLUGIN_SOURCE = File.expand_path("xbar.3s.rb", __dir__)
+PLUGIN_DEST = File.join(XBAR_PLUGIN_DIR, "brainiac.3s.rb")
+
+unless RUBY_PLATFORM.match?(/darwin/i)
+  puts "⚠ This script is for macOS only (xbar doesn't run on Linux)"
+  exit 1
+end
+
+unless File.directory?(XBAR_PLUGIN_DIR)
+  puts "⚠ xbar plugin directory not found: #{XBAR_PLUGIN_DIR}"
+  puts "  Install xbar first: https://xbarapp.com"
+  exit 1
+end
+
+if File.exist?(PLUGIN_DEST)
+  puts "Removing existing plugin at #{PLUGIN_DEST}"
+  File.delete(PLUGIN_DEST)
+end
+
+File.symlink(PLUGIN_SOURCE, PLUGIN_DEST)
+File.chmod(0o755, PLUGIN_SOURCE)
+
+puts "✓ Brainiac xbar plugin installed"
+puts "  #{PLUGIN_SOURCE} → #{PLUGIN_DEST}"
+puts "  Refresh interval: 3 seconds"
+puts "  Restart xbar to activate"

data/monitor/view-logs-macos.rb

@@ -0,0 +1,210 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Brainiac macOS Log Viewer
+# Opens agent log files in a terminal (wezterm by default, configurable via waybar.json)
+# Set "log_viewer_command" in ~/.brainiac/waybar.json to override (supports wezterm/iTerm/Terminal.app)
+# Uses macOS notifications via osascript for status messages
+
+require "json"
+require "socket"
+
+SOCKET_PATH = "/tmp/brainiac-monitor.sock"
+CONFIG_PATH = File.expand_path("~/.brainiac/waybar.json")
+
+# Load agent configuration from JSON
+# Returns [agents_hash, default_emoji] tuple
+def load_agent_config
+  config = JSON.parse(File.read(CONFIG_PATH))
+  agents = {}
+  config["agents"].each do |agent|
+    agents[agent["name"].downcase] = agent["emoji"]
+  end
+  default_emoji = config["default_emoji"] || "❓"
+  [agents, default_emoji]
+rescue StandardError => e
+  warn "Failed to load waybar.json: #{e.message}"
+  [{}, "❓"]
+end
+
+AGENTS, DEFAULT_EMOJI = load_agent_config
+
+# Read agent state from daemon socket
+# Returns hash with sessions/count/last_update, or error hash if daemon unavailable
+def fetch_state
+  socket = UNIXSocket.new(SOCKET_PATH)
+  data = socket.read
+  socket.close
+  JSON.parse(data)
+rescue Errno::ENOENT
+  { "sessions" => [], "count" => 0, "error" => "daemon not running" }
+rescue StandardError => e
+  { "sessions" => [], "count" => 0, "error" => e.message }
+end
+
+def format_elapsed(seconds)
+  return "#{seconds}s" if seconds < 60
+
+  minutes = seconds / 60
+  return "#{minutes}m" if minutes < 60
+
+  hours = minutes / 60
+  "#{hours}h"
+end
+
+# Send a macOS notification via osascript
+# Silently fails if osascript is unavailable
+def notify(title, message)
+  escaped_title = title.gsub('"', '\\"')
+  escaped_message = message.gsub('"', '\\"')
+  system("osascript", "-e", "display notification \"#{escaped_message}\" with title \"#{escaped_title}\"")
+rescue StandardError => e
+  warn "Notification failed: #{e.message}"
+end
+
+def format_context(card_key)
+  if card_key.start_with?("discord-")
+    "Discord chat"
+  elsif card_key.start_with?("card-")
+    card_key.split("-")[1]
+  else
+    card_key
+  end
+end
+
+def load_log_viewer_command
+  config = JSON.parse(File.read(CONFIG_PATH))
+  config["log_viewer_command"]
+rescue StandardError
+  nil
+end
+
+DEFAULT_LOG_VIEWER = "/opt/homebrew/bin/wezterm"
+LOG_VIEWER_COMMAND = load_log_viewer_command || DEFAULT_LOG_VIEWER
+
+# Find an existing wezterm pane that's already tailing the given log file
+def find_wezterm_pane_for(log_file)
+  json = `#{LOG_VIEWER_COMMAND} cli list --format json 2>/dev/null`
+  panes = JSON.parse(json)
+  panes.find { |p| p["title"]&.include?(log_file) || p["cwd"]&.include?(log_file) }
+rescue StandardError
+  nil
+end
+
+# Get the window ID of the first wezterm window
+def find_wezterm_window_id
+  json = `#{LOG_VIEWER_COMMAND} cli list --format json 2>/dev/null`
+  panes = JSON.parse(json)
+  panes.first&.dig("window_id")
+rescue StandardError
+  nil
+end
+
+def open_log(log_file)
+  escaped_path = log_file.gsub("'", "'\\\\''")
+
+  if LOG_VIEWER_COMMAND.include?("wezterm")
+    wezterm_running = system("pgrep -qf WezTerm")
+    if wezterm_running
+      # Check if this log is already being tailed in an existing tab
+      existing_pane = find_wezterm_pane_for(log_file)
+      if existing_pane
+        system(LOG_VIEWER_COMMAND, "cli", "activate-pane", "--pane-id", existing_pane["pane_id"].to_s)
+      else
+        # Spawn as a new tab in the first available window
+        window_id = find_wezterm_window_id
+        args = [LOG_VIEWER_COMMAND, "cli", "spawn"]
+        args += ["--window-id", window_id.to_s] if window_id
+        args += ["--", "tail", "-f", log_file]
+        system(*args)
+      end
+    else
+      system(LOG_VIEWER_COMMAND, "start", "--", "tail", "-f", log_file)
+      sleep 0.5 # Give wezterm time to launch before trying to activate
+    end
+    system("open", "-a", "WezTerm")
+    system("osascript", "-e", 'tell application "System Events" to set frontmost of process "WezTerm" to true')
+  elsif LOG_VIEWER_COMMAND.include?("iTerm")
+    script = <<~APPLESCRIPT
+      tell application "iTerm"
+        activate
+        create window with default profile command "tail -f '#{escaped_path}'"
+      end tell
+    APPLESCRIPT
+    system("osascript", "-e", script)
+  elsif LOG_VIEWER_COMMAND.include?("Terminal")
+    system("osascript", "-e", "tell application \"Terminal\" to do script \"tail -f '#{escaped_path}'\"")
+    system("osascript", "-e", 'tell application "Terminal" to activate')
+  else
+    system(LOG_VIEWER_COMMAND, "tail", "-f", log_file)
+  end
+end
+
+# --- Main invocation logic ---
+
+# Mode 1: xbar submenu click — log file path passed as param1 (ARGV[0])
+if ARGV[0]
+  log_file = ARGV[0]
+  unless File.exist?(log_file)
+    notify("Brainiac", "Log file not found: #{log_file}")
+    exit 1
+  end
+  open_log(log_file)
+  exit 0
+end
+
+# Mode 2+: standalone invocation — fetch state from daemon
+state = fetch_state
+sessions = state["sessions"] || []
+
+if state["error"]
+  notify("Brainiac", state["error"])
+  exit 1
+end
+
+if sessions.empty?
+  notify("Brainiac", "No active agent sessions")
+  exit 0
+end
+
+# Single session — open directly
+if sessions.size == 1
+  log_file = sessions[0]["log_file"]
+  if log_file && File.exist?(log_file)
+    open_log(log_file)
+  else
+    notify("Brainiac", "Log file not found: #{log_file}")
+  end
+  exit 0
+end
+
+# Multiple sessions — use fzf if available
+if system("which fzf > /dev/null 2>&1")
+  options = sessions.map do |s|
+    agent = s["agent"]
+    elapsed = format_elapsed(s["elapsed_seconds"])
+    context = format_context(s["card_key"])
+    emoji = AGENTS[agent.downcase] || DEFAULT_EMOJI
+    "#{emoji} #{agent}: #{context} (#{elapsed})|#{s["log_file"]}"
+  end
+
+  menu_text = options.join("\n")
+  selected = `echo "#{menu_text}" | fzf --prompt="Agent Logs: "`.strip
+
+  unless selected.empty?
+    log_file = selected.split("|").last
+    if File.exist?(log_file)
+      open_log(log_file)
+    else
+      notify("Brainiac", "Log file not found: #{log_file}")
+    end
+  end
+else
+  # No fzf — open first session's log
+  log_file = sessions[0]["log_file"]
+  if log_file && File.exist?(log_file)
+    open_log(log_file)
+  else
+    notify("Brainiac", "Log file not found: #{log_file}")
+  end
+end

data/monitor/view-logs-rofi.rb

@@ -0,0 +1,194 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Brainiac Log Viewer (Rofi version)
+# Shows a rofi menu to select which agent log to tail
+# Non-blocking, safe for waybar on-click
+
+require "json"
+require "net/http"
+require "socket"
+
+SOCKET_PATH = "/tmp/brainiac-monitor.sock"
+API_URL = "http://localhost:4567/api/status"
+CONFIG_PATH = File.expand_path("~/.brainiac/waybar.json")
+
+# Load agent configuration from JSON
+def load_agent_config
+  config = JSON.parse(File.read(CONFIG_PATH))
+  agents = {}
+  config["agents"].each do |agent|
+    agents[agent["name"].downcase] = agent["emoji"]
+  end
+  default_emoji = config["default_emoji"] || "❓"
+  [agents, default_emoji]
+rescue StandardError => e
+  warn "Failed to load waybar.json: #{e.message}"
+  [{}, "❓"]
+end
+
+AGENTS, DEFAULT_EMOJI = load_agent_config
+
+def fetch_state_from_socket
+  socket = UNIXSocket.new(SOCKET_PATH)
+  data = socket.read
+  socket.close
+  JSON.parse(data)
+end
+
+def fetch_state_from_api
+  uri = URI(API_URL)
+  response = Net::HTTP.get_response(uri)
+  return nil unless response.is_a?(Net::HTTPSuccess)
+
+  JSON.parse(response.body)
+end
+
+def fetch_state
+  # Prefer socket (daemon mode) — faster, no HTTP overhead
+  fetch_state_from_socket
+rescue Errno::ENOENT, Errno::ECONNREFUSED
+  # Daemon not running — fall back to direct API call
+  data = fetch_state_from_api
+  unless data
+    system("notify-send", "Brainiac", "Server not reachable")
+    exit 1
+  end
+  data
+rescue StandardError => e
+  system("notify-send", "Brainiac Error", e.message)
+  exit 1
+end
+
+def format_elapsed(seconds)
+  return "#{seconds}s" if seconds < 60
+
+  minutes = seconds / 60
+  return "#{minutes}m" if minutes < 60
+
+  hours = minutes / 60
+  "#{hours}h"
+end
+
+state = fetch_state
+sessions = state["sessions"] || []
+
+if sessions.empty?
+  system("notify-send", "Brainiac", "No active agent sessions")
+  exit 0
+end
+
+INFRA_CMDS = %w[kiro-cli-chat ruby-lsp clangd gopls].freeze
+
+# Build menu entries: sessions + their child processes
+entries = []
+sessions.each do |s|
+  agent = s["agent"]
+  elapsed = format_elapsed(s["elapsed_seconds"])
+  card_key = s["card_key"]
+  context = if card_key.start_with?("discord-")
+              "Discord chat"
+            elsif card_key.start_with?("card-")
+              card_key.split("-")[1]
+            else
+              card_key
+            end
+  emoji = AGENTS[agent.downcase] || DEFAULT_EMOJI
+  entries << { display: "#{emoji} #{agent}: #{context} (#{elapsed})", type: :log, log: s["log_file"] }
+  entries << { display: "   ⛔ Kill session: #{agent} (#{context})", type: :kill_session, card_key: card_key, agent: agent }
+
+  (s["children"] || []).each do |c|
+    cmd_short = c["cmd"].to_s.split("/").last.to_s.split.first.to_s
+    cmd_short = c["cmd"].to_s[0..40] if cmd_short.empty?
+    next if INFRA_CMDS.any? { |ic| cmd_short.start_with?(ic) }
+
+    entries << {
+      display: "   └ 🔪 Kill: #{cmd_short} (#{format_elapsed(c["elapsed_seconds"])}) [PID #{c["pid"]}]",
+      type: :kill_child, pid: c["pid"], cmd: cmd_short
+    }
+  end
+end
+
+# If only one session with no children, open log directly
+if entries.size == 1 && entries[0][:type] == :log
+  spawn("alacritty", "-e", "tail", "-f", entries[0][:log]) if entries[0][:log]
+  exit 0
+end
+
+def find_launcher
+  %w[rofi fuzzel wofi zenity].find { |cmd| system("which #{cmd} > /dev/null 2>&1") }
+end
+
+def run_menu(launcher, entries)
+  menu_text = entries.map { |e| e[:display] }.join("\n")
+  case launcher
+  when "rofi"
+    IO.popen(%w[rofi -dmenu -i -p] + ["Agent Sessions"], "r+") do |io|
+      io.puts menu_text
+      io.close_write
+      io.read.strip
+    end
+  when "fuzzel"
+    IO.popen(%w[fuzzel --dmenu --prompt] + ["Agent Sessions: "], "r+") do |io|
+      io.puts menu_text
+      io.close_write
+      io.read.strip
+    end
+  when "wofi"
+    IO.popen(%w[wofi --dmenu --prompt] + ["Agent Sessions"], "r+") do |io|
+      io.puts menu_text
+      io.close_write
+      io.read.strip
+    end
+  when "zenity"
+    IO.popen(["zenity", "--list", "--title", "Agent Sessions", "--column", "Session", "--width", "600", "--height", "400"], "r+",
+             err: "/dev/null") do |io|
+      entries.each { |e| io.puts e[:display] }
+      io.close_write
+      io.read.strip
+    end
+  end
+end
+
+launcher = find_launcher
+unless launcher
+  system("notify-send", "Brainiac", "No menu launcher found (install rofi, fuzzel, wofi, or zenity)")
+  exit 1
+end
+
+selected_line = run_menu(launcher, entries)
+
+unless selected_line.to_s.empty?
+  selected = entries.find { |e| e[:display].strip == selected_line.strip }
+  if selected
+    case selected[:type]
+    when :log
+      spawn("alacritty", "-e", "tail", "-f", selected[:log]) if selected[:log]
+    when :kill_session
+      card_key = selected[:card_key]
+      uri = URI("http://localhost:4567/api/sessions/kill/#{card_key}")
+      response = Net::HTTP.post(uri, "", { "Content-Type" => "application/json" })
+      if response.is_a?(Net::HTTPSuccess)
+        system("notify-send", "Brainiac", "Killed session: #{selected[:agent]}")
+      else
+        system("notify-send", "Brainiac", "Failed to kill session: #{selected[:agent]}")
+      end
+    when :kill_child
+      pid = selected[:pid]
+      begin
+        Process.kill("TERM", pid)
+      rescue StandardError
+        nil
+      end
+      Thread.new do
+        sleep 3
+        begin
+          Process.kill("KILL", pid)
+        rescue StandardError
+          nil
+        end
+      end
+      system("notify-send", "Brainiac", "Killed #{selected[:cmd]} (PID #{pid})")
+    end
+  end
+end

data/monitor/view-logs.rb

@@ -0,0 +1,119 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Brainiac Log Viewer
+# Shows a rofi menu to select which agent log to tail
+
+require "json"
+require "net/http"
+require "socket"
+
+SOCKET_PATH = "/tmp/brainiac-monitor.sock"
+API_URL = "http://localhost:4567/api/status"
+CONFIG_PATH = File.expand_path("~/.brainiac/waybar.json")
+
+# Load agent configuration from JSON
+def load_agent_config
+  config = JSON.parse(File.read(CONFIG_PATH))
+  agents = {}
+  config["agents"].each do |agent|
+    agents[agent["name"].downcase] = agent["emoji"]
+  end
+  default_emoji = config["default_emoji"] || "❓"
+  [agents, default_emoji]
+rescue StandardError => e
+  warn "Failed to load waybar.json: #{e.message}"
+  [{}, "❓"]
+end
+
+AGENTS, DEFAULT_EMOJI = load_agent_config
+
+def fetch_state_from_socket
+  socket = UNIXSocket.new(SOCKET_PATH)
+  data = socket.read
+  socket.close
+  JSON.parse(data)
+end
+
+def fetch_state_from_api
+  uri = URI(API_URL)
+  response = Net::HTTP.get_response(uri)
+  return nil unless response.is_a?(Net::HTTPSuccess)
+
+  JSON.parse(response.body)
+end
+
+def fetch_state
+  # Prefer socket (daemon mode) — faster, no HTTP overhead
+  fetch_state_from_socket
+rescue Errno::ENOENT, Errno::ECONNREFUSED
+  # Daemon not running — fall back to direct API call
+  data = fetch_state_from_api
+  unless data
+    puts "Error: Server not reachable"
+    exit 1
+  end
+  data
+rescue StandardError => e
+  puts "Error: #{e.message}"
+  exit 1
+end
+
+def format_elapsed(seconds)
+  return "#{seconds}s" if seconds < 60
+
+  minutes = seconds / 60
+  return "#{minutes}m" if minutes < 60
+
+  hours = minutes / 60
+  "#{hours}h"
+end
+
+state = fetch_state
+sessions = state["sessions"] || []
+
+if sessions.empty?
+  system("notify-send", "Brainiac", "No active agent sessions")
+  exit 0
+end
+
+# If only one session, open it directly
+if sessions.size == 1
+  log_file = sessions[0]["log_file"]
+  exec("alacritty", "-e", "tail", "-f", log_file) if log_file
+  exit 0
+end
+
+# Multiple sessions: use fzf if available, otherwise just open the first one
+if system("which fzf > /dev/null 2>&1")
+  # Build fzf menu
+  options = sessions.map do |s|
+    agent = s["agent"]
+    elapsed = format_elapsed(s["elapsed_seconds"])
+
+    card_key = s["card_key"]
+    context = if card_key.start_with?("discord-")
+                "Discord chat"
+              elsif card_key.start_with?("card-")
+                card_key.split("-")[1]
+              else
+                card_key
+              end
+
+    emoji = AGENTS[agent.downcase] || DEFAULT_EMOJI
+
+    "#{emoji} #{agent}: #{context} (#{elapsed})|#{s["log_file"]}"
+  end
+
+  menu_text = options.join("\n")
+  selected = `echo "#{menu_text}" | fzf --prompt="Agent Logs: "`.strip
+
+  unless selected.empty?
+    log_file = selected.split("|").last
+    exec("alacritty", "-e", "tail", "-f", log_file)
+  end
+else
+  # No menu system, just open the first log
+  log_file = sessions[0]["log_file"]
+  exec("alacritty", "-e", "tail", "-f", log_file) if log_file
+end