brainiac 0.0.1

data/monitor/menubar.rb

@@ -0,0 +1,295 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Brainiac macOS Menu Bar Plugin (xbar/SwiftBar)
+# Reads from monitor daemon socket and outputs xbar-format text
+# Mirrors monitor/waybar.rb patterns for macOS-native display
+
+require "json"
+require "net/http"
+require "shellwords"
+require "socket"
+require "time"
+require "uri"
+
+SERVER_URL = "http://localhost:4567"
+SOCKET_PATH = "/tmp/brainiac-monitor.sock"
+CONFIG_PATH = File.expand_path("~/.brainiac/waybar.json")
+DEFAULT_EMOJI = "❓"
+SELF_PATH = File.realpath(__FILE__)
+
+def load_config
+  JSON.parse(File.read(CONFIG_PATH))
+rescue StandardError => e
+  warn "Failed to load waybar.json: #{e.message}"
+  {}
+end
+
+CONFIG = load_config.freeze
+
+def load_agent_config
+  agents = {}
+  (CONFIG["agents"] || []).each do |agent|
+    agents[agent["name"].downcase] = { emoji: agent["emoji"], color: agent["color"] }
+  end
+  agents
+end
+
+AGENTS = load_agent_config.freeze
+FIZZY_ACCOUNT_ID = CONFIG["fizzy_account_id"]
+DISCORD_GUILD_ID = CONFIG["discord_guild_id"]
+
+def fetch_state
+  socket = UNIXSocket.new(SOCKET_PATH)
+  data = socket.read
+  socket.close
+  JSON.parse(data)
+rescue Errno::ENOENT
+  { "sessions" => [], "count" => 0, "recent" => [], "error" => "daemon not running" }
+rescue StandardError => e
+  { "sessions" => [], "count" => 0, "recent" => [], "error" => e.message }
+end
+
+def fetch_deployments
+  uri = URI("#{SERVER_URL}/api/deployments")
+  response = Net::HTTP.get_response(uri)
+  JSON.parse(response.body)["deployments"] || []
+rescue StandardError
+  nil
+end
+
+DEPLOY_RECENT_WINDOW = 30 * 60
+
+def deploy_dot(dep)
+  status = dep["last_deploy_status"]
+  if status == "deploying"
+    "🟠"
+  elsif status == "failed"
+    "💥"
+  elsif dep["status"] == "occupied"
+    deploy_time = dep["last_deploy_at"] || dep["deployed_at"]
+    recent = deploy_time && (Time.now - Time.parse(deploy_time)) < DEPLOY_RECENT_WINDOW
+    recent ? "🚀" : "🔴"
+  else
+    "🟢"
+  end
+end
+
+LOG_VIEWER_PATH = File.join(File.dirname(SELF_PATH), "view-logs-macos.rb")
+DEPLOY_SCRIPT_PATH = File.join(File.dirname(SELF_PATH), "deploy-env-macos.rb")
+
+ANSI_REGEX = /\e\[[0-9;]*[a-zA-Z]|\e\[\?[0-9;]*[a-zA-Z]/
+LOG_PREVIEW_LINES = 15
+LOG_LINE_MAX = 80
+LOG_FONT = "SFMono-Regular"
+LOG_SIZE = 12
+
+def tail_log(log_file, lines: LOG_PREVIEW_LINES)
+  return [] unless log_file && File.exist?(log_file)
+
+  raw = `tail -n 50 #{log_file.shellescape} 2>/dev/null`
+  raw.encode("UTF-8", invalid: :replace, undef: :replace, replace: "")
+     .lines
+     .map { |l| l.gsub(ANSI_REGEX, "").gsub(/[^[:print:]\t]/, "").strip }
+     .reject(&:empty?)
+     .last(lines)
+rescue StandardError
+  []
+end
+
+def format_log_line(text)
+  text.length > LOG_LINE_MAX ? "#{text[0, LOG_LINE_MAX]}…" : text
+end
+
+def format_elapsed(seconds)
+  return "#{seconds}s" if seconds < 60
+
+  minutes = seconds / 60
+  return "#{minutes}m" if minutes < 60
+
+  "#{minutes / 60}h"
+end
+
+def format_context(card_key)
+  return "" unless card_key
+
+  if card_key.start_with?("discord-")
+    "Discord"
+  elsif card_key.start_with?("card-")
+    "##{card_key.split("-")[1]}"
+  else
+    card_key
+  end
+end
+
+def time_ago(iso_string)
+  return nil unless iso_string
+
+  seconds = (Time.now - Time.parse(iso_string)).to_i
+  "#{format_elapsed(seconds)} ago"
+rescue StandardError
+  nil
+end
+
+def log_action(log_file)
+  return "" unless log_file
+
+  " | shell=#{LOG_VIEWER_PATH} param1=#{log_file} terminal=false refresh=false"
+end
+
+OPEN_SCRIPT = File.join(File.dirname(SELF_PATH), "open-action.sh")
+
+def full_log_action(log_file)
+  return "" unless log_file
+
+  " | shell=#{OPEN_SCRIPT} param1=#{log_file.shellescape} terminal=false refresh=false"
+end
+
+def prompt_url(card_key)
+  return nil unless card_key
+
+  if card_key.start_with?("card-")
+    card_num = card_key.split("-")[1]
+    "https://app.fizzy.do/#{FIZZY_ACCOUNT_ID}/cards/#{card_num}" if FIZZY_ACCOUNT_ID && card_num
+  elsif card_key.start_with?("discord-") && DISCORD_GUILD_ID
+    parts = card_key.split("-")
+    # discord-AGENT-CHANNEL_ID-MESSAGE_ID (agent name may contain hyphens, IDs are last two numeric parts)
+    channel_id = parts[-2]
+    message_id = parts[-1]
+    "https://discord.com/channels/#{DISCORD_GUILD_ID}/#{channel_id}/#{message_id}" if channel_id && message_id
+  end
+end
+
+def prompt_action(card_key)
+  url = prompt_url(card_key)
+  return "" unless url
+
+  " | shell=#{OPEN_SCRIPT} param1=#{url} terminal=false refresh=false"
+end
+
+def worktree_path(log_file, card_key)
+  return nil unless log_file && card_key&.start_with?("card-")
+
+  dir = File.dirname(log_file, 2)
+  dir if File.directory?(dir) && dir != "/"
+end
+
+def worktree_action(log_file, card_key)
+  path = worktree_path(log_file, card_key)
+  return "" unless path
+
+  " | shell=#{OPEN_SCRIPT} param1=#{path.shellescape} terminal=false refresh=false"
+end
+
+COLOR_MAP = {
+  "red" => "#ff5555", "green" => "#50fa7b", "blue" => "#8be9fd",
+  "yellow" => "#f1fa8c", "cyan" => "#8be9fd", "magenta" => "#ff79c6",
+  "purple" => "#bd93f9", "pink" => "#ff79c6", "white" => "#f8f8f2"
+}.freeze
+
+def hex_color(name)
+  COLOR_MAP[name] || name
+end
+
+def generate_output
+  state = fetch_state
+  deployments = fetch_deployments
+
+  return ["⚠️", "---", state["error"], "---", "Refresh | refresh=true"].join("\n") if state["error"] && !deployments
+
+  sessions = state["sessions"] || []
+  recent = state["recent"] || []
+  lines = []
+
+  # Title line — agent emojis + deploy dots
+  parts = []
+  parts << sessions.map { |s| AGENTS.dig(s["agent"]&.downcase, :emoji) || DEFAULT_EMOJI }.join(" ") if sessions.any?
+  parts << deployments.map { |d| deploy_dot(d) }.join if deployments&.any?
+  title = parts.any? ? parts.join(" ") : "💤"
+  lines << title
+  lines << "---"
+
+  # Active sessions
+  if sessions.any?
+    lines << "Active | size=12"
+    sessions.each do |s|
+      agent_key = (s["agent"] || "").downcase
+      emoji = AGENTS.dig(agent_key, :emoji) || DEFAULT_EMOJI
+      color = AGENTS.dig(agent_key, :color)
+      color_str = color ? " color=#{hex_color(color)}" : ""
+      context = format_context(s["card_key"])
+      elapsed = format_elapsed(s["elapsed_seconds"] || 0)
+      lines << "#{emoji} #{s["agent"]}: #{context} (#{elapsed}) |#{color_str}"
+
+      tail_log(s["log_file"]).each do |line|
+        lines << "-- #{format_log_line(line)} | font=#{LOG_FONT} size=#{LOG_SIZE}"
+      end
+      lines << "-- ---" if s["log_file"]
+      lines << "-- Tail Log#{log_action(s["log_file"])}" if s["log_file"]
+      lines << "-- View Full Log#{full_log_action(s["log_file"])}" if s["log_file"]
+      lines << "-- Open Prompt#{prompt_action(s["card_key"])}" unless prompt_url(s["card_key"]).nil?
+      wt = worktree_path(s["log_file"], s["card_key"])
+      lines << "-- Open Worktree#{worktree_action(s["log_file"], s["card_key"])}" if wt
+    end
+  else
+    lines << "No active sessions | size=12"
+  end
+
+  # Recent completed sessions
+  if recent.any?
+    lines << "---"
+    lines << "Recent | size=12"
+    recent.each do |s|
+      agent_key = (s["agent"] || "").downcase
+      emoji = AGENTS.dig(agent_key, :emoji) || DEFAULT_EMOJI
+      context = format_context(s["card_key"])
+      ago = time_ago(s["finished_at"]) || "?"
+      lines << "#{emoji} #{s["agent"]}: #{context} — #{ago}"
+
+      tail_log(s["log_file"]).each do |line|
+        lines << "-- #{format_log_line(line)} | font=#{LOG_FONT} size=#{LOG_SIZE}"
+      end
+      lines << "-- ---" if s["log_file"]
+      lines << "-- Tail Log#{log_action(s["log_file"])}" if s["log_file"]
+      lines << "-- View Full Log#{full_log_action(s["log_file"])}" if s["log_file"]
+      lines << "-- Open Prompt#{prompt_action(s["card_key"])}" unless prompt_url(s["card_key"]).nil?
+      wt = worktree_path(s["log_file"], s["card_key"])
+      lines << "-- Open Worktree#{worktree_action(s["log_file"], s["card_key"])}" if wt
+    end
+  end
+
+  # Deployments
+  if deployments&.any?
+    lines << "---"
+    lines << "Deployments | size=12"
+    deployments.each do |d|
+      label = d["label"] || d["env"]
+      env = d["env"]
+      dot = deploy_dot(d)
+      if d["status"] == "occupied"
+        card = d["card_number"] ? "##{d["card_number"]}" : d["branch"] || "unknown"
+        ago = time_ago(d["deployed_at"])
+        status_label = case d["last_deploy_status"]
+                       when "deploying" then " — deploying…"
+                       when "failed" then " — FAILED"
+                       else ""
+                       end
+        line = "#{dot} #{label}: #{card}#{status_label}#{" (#{ago})" if ago}"
+        url = d["url"]
+        lines << (url ? "#{line} | href=#{url}" : line)
+      else
+        ago = time_ago(d["cleared_at"])
+        last = d["last_card"] ? " (was ##{d["last_card"]})" : ""
+        lines << "#{dot} #{label}: Available#{" #{ago}" if ago}#{last}"
+      end
+      lines << "-- Deploy to #{label} | shell=#{DEPLOY_SCRIPT_PATH} param1=#{env} terminal=false refresh=true"
+      lines << "-- Open #{label} | shell=#{OPEN_SCRIPT} param1=#{d["url"]} terminal=false refresh=false" if d["status"] == "occupied" && d["url"]
+    end
+  end
+
+  lines << "---"
+  lines << "Refresh | refresh=true"
+  lines.join("\n")
+end
+
+puts generate_output

data/monitor/open-action.sh

@@ -0,0 +1,15 @@
+#!/bin/bash
+# Opens a URL in browser, a directory in Finder, or a file in the default viewer.
+# Used by the SwiftBar menubar plugin for prompt links, worktrees, and full log views.
+
+arg="$1"
+
+if [[ "$arg" == https://discord.com/* ]]; then
+  open "discord://${arg#https://}"
+elif [[ "$arg" == http* ]]; then
+  open "$arg"
+elif [[ -d "$arg" ]]; then
+  open -a Kiro "$arg"
+elif [[ -f "$arg" ]]; then
+  open -a "Console" "$arg" 2>/dev/null || open "$arg"
+fi

data/monitor/setup-menubar.rb

@@ -0,0 +1,78 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# One-time setup script to install Brainiac menubar plugin into xbar or SwiftBar
+# Run this once — the plugin will then auto-refresh on its configured interval
+
+PLUGIN_APPS = [
+  {
+    name: "SwiftBar",
+    plugin_dir: File.expand_path("~/Library/Application Support/SwiftBar/Plugins"),
+    app_path: "/Applications/SwiftBar.app"
+  },
+  {
+    name: "xbar",
+    plugin_dir: File.expand_path("~/Library/Application Support/xbar/plugins"),
+    app_path: "/Applications/xbar.app"
+  }
+].freeze
+
+SYMLINK_NAME = "brainiac.2s.rb"
+SOURCE_PATH = File.join(File.dirname(File.expand_path(__FILE__)), "menubar.rb")
+
+def detect_plugin_app
+  PLUGIN_APPS.each do |app|
+    return { name: app[:name], plugin_dir: app[:plugin_dir] } if Dir.exist?(app[:plugin_dir]) || File.exist?(app[:app_path])
+  end
+  nil
+end
+
+def install_plugin(plugin_dir, source_path)
+  FileUtils.mkdir_p(plugin_dir)
+  link_path = File.join(plugin_dir, SYMLINK_NAME)
+
+  # Remove existing symlink/file if present
+  File.delete(link_path) if File.exist?(link_path) || File.symlink?(link_path)
+
+  File.symlink(source_path, link_path)
+rescue StandardError => e
+  warn "✗ Failed to create symlink: #{e.message}"
+  warn "  Source: #{source_path}"
+  warn "  Target: #{link_path}"
+  exit 1
+end
+
+def verify_executable!(path) # rubocop:disable Naming/PredicateMethod
+  unless File.executable?(path)
+    File.chmod(0o755, path)
+    warn "  Fixed executable permission on #{path}"
+  end
+  File.executable?(path)
+end
+
+# --- Main ---
+
+require "fileutils"
+
+app = detect_plugin_app
+
+unless app
+  puts "No xbar or SwiftBar installation detected."
+  puts ""
+  puts "Install one of the following to use the Brainiac menu bar plugin:"
+  puts "  • xbar:     https://xbarapp.com"
+  puts "  • SwiftBar: https://github.com/swiftbar/SwiftBar"
+  puts ""
+  puts "After installing, re-run this script:"
+  puts "  ruby #{__FILE__}"
+  exit 0
+end
+
+puts "Detected #{app[:name]}"
+install_plugin(app[:plugin_dir], SOURCE_PATH)
+verify_executable!(SOURCE_PATH)
+
+link_path = File.join(app[:plugin_dir], SYMLINK_NAME)
+puts "✓ Installed Brainiac plugin into #{app[:name]}"
+puts "  Symlink: #{link_path} → #{SOURCE_PATH}"
+puts "  Refresh interval: 2s"

data/monitor/setup-waybar-deploy-envs.rb

@@ -0,0 +1,121 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# One-time setup: replaces the single brainiac-deployments module
+# with per-environment modules so each dot gets its own border/click.
+
+require "json"
+require "fileutils"
+
+WAYBAR_CONFIG = File.expand_path("~/.config/waybar/config.jsonc")
+DEPLOY_SCRIPT = File.expand_path("~/.brainiac/bin/waybar-deploy-env")
+DEPLOYMENTS_CONFIG = File.expand_path("~/.brainiac/deployments.json")
+WAYBAR_STYLE = File.expand_path("~/.config/waybar/style.css")
+
+# Create wrapper script that resolves from server.root
+wrapper_dir = File.expand_path("~/.brainiac/bin")
+FileUtils.mkdir_p(wrapper_dir)
+File.write(DEPLOY_SCRIPT, <<~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", "waybar-deploy-env.rb")
+    if File.exist?(script)
+      ARGV.unshift if ARGV.empty?
+      load script
+      exit
+    end
+  end
+  require "json"
+  puts({ text: "", tooltip: "Brainiac server root not found", class: "error" }.to_json)
+SCRIPT
+File.chmod(0o755, DEPLOY_SCRIPT)
+
+def load_config
+  content = File.read(WAYBAR_CONFIG)
+  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
+
+deployments = JSON.parse(File.read(DEPLOYMENTS_CONFIG))
+envs = deployments["environments"].keys
+
+config = load_config
+
+# Remove old single deployments module from all bar positions
+%w[modules-left modules-center modules-right].each do |pos|
+  next unless config[pos]
+
+  config[pos].reject! { |m| m.to_s.include?("brainiac-deploy") }
+end
+config.delete("custom/brainiac-deployments")
+
+# Remove any existing per-env modules
+config.each_key do |key|
+  config.delete(key) if key.start_with?("custom/brainiac-deploy-")
+end
+
+# Insert per-env modules into modules-center, before custom/brainiac
+center = config["modules-center"] || []
+zc_idx = center.index("custom/brainiac") || center.length
+envs.each_with_index do |env, i|
+  mod_name = "custom/brainiac-deploy-#{env}"
+  center.insert(zc_idx + i, mod_name) unless center.include?(mod_name)
+end
+config["modules-center"] = center
+
+# Add module configs for each env
+envs.each do |env|
+  mod_name = "custom/brainiac-deploy-#{env}"
+  config[mod_name] = {
+    "exec" => "#{DEPLOY_SCRIPT} #{env}",
+    "return-type" => "json",
+    "interval" => 30,
+    "format" => "{}",
+    "tooltip" => true,
+    "escape" => false,
+    "on-click" => "#{DEPLOY_SCRIPT} #{env} --click",
+    "on-click-right" => "#{DEPLOY_SCRIPT} #{env} --deploy"
+  }
+end
+
+save_config(config)
+puts "✓ Added per-environment deploy modules: #{envs.map { |e| "custom/brainiac-deploy-#{e}" }.join(", ")}"
+
+# Update CSS — remove old single-module styles, add per-env styles
+style = File.read(WAYBAR_STYLE)
+
+# Remove old block
+style.gsub!(%r{/\* Brainiac deployment environment dots \*/.*?(?=\n\n|\n/\*|\z)}m, "")
+style.gsub!(/\n*#custom-brainiac-deployments[^{]*\{[^}]*\}\n*/m, "")
+
+# Add new per-env styles
+unless style.include?("#custom-brainiac-deploy-")
+  css = <<~CSS
+
+    /* Brainiac per-environment deploy dots */
+    [id^="custom-brainiac-deploy-"] {
+      font-size: 28px;
+      padding: 0 6px;
+      border-radius: 8px;
+      border: 2px solid transparent;
+    }
+
+    [id^="custom-brainiac-deploy-"].deploy-recent {
+      border: 2px solid #4488ff;
+    }
+
+    [id^="custom-brainiac-deploy-"].deploy-failed {
+      border: 2px solid #ff4444;
+    }
+  CSS
+  File.write(WAYBAR_STYLE, "#{style.strip}\n#{css}")
+  puts "✓ Updated waybar CSS with per-environment border styles"
+end
+
+puts "✓ Restart waybar to apply: omarchy restart waybar"

data/monitor/setup-waybar-deployments.rb

@@ -0,0 +1,96 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# One-time setup: adds Brainiac deployments module to waybar config
+# and repositions the agent session module with more breathing room.
+
+require "json"
+
+WAYBAR_CONFIG = File.expand_path("~/.config/waybar/config.jsonc")
+DEPLOY_SCRIPT = File.expand_path("~/Code/brainiac/monitor/waybar-deployments.rb")
+WAYBAR_STYLE  = File.expand_path("~/.config/waybar/style.css")
+
+def load_config
+  content = File.read(WAYBAR_CONFIG)
+  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
+
+config = load_config
+
+# Remove any existing deployment module
+config["modules-center"]&.reject! { |m| m.to_s.include?("brainiac-deploy") }
+config["modules-right"]&.reject! { |m| m.to_s.include?("brainiac-deploy") }
+config.delete("custom/brainiac-deployments")
+
+# Move agent session module from modules-right to modules-center (after indicators)
+if config["modules-right"]&.delete("custom/brainiac")
+  config["modules-center"] ||= []
+  config["modules-center"] << "custom/brainiac" unless config["modules-center"].include?("custom/brainiac")
+end
+
+# Add deployments module right before agent sessions in modules-center
+center = config["modules-center"] || []
+zc_idx = center.index("custom/brainiac")
+if zc_idx
+  center.insert(zc_idx, "custom/brainiac-deployments") unless center.include?("custom/brainiac-deployments")
+else
+  center << "custom/brainiac-deployments" unless center.include?("custom/brainiac-deployments")
+end
+
+# Add module config
+config["custom/brainiac-deployments"] = {
+  "exec" => DEPLOY_SCRIPT,
+  "return-type" => "json",
+  "interval" => 30,
+  "format" => "{}",
+  "tooltip" => true,
+  "format-alt" => "{}",
+  "escape" => false,
+  "on-click" => "#{DEPLOY_SCRIPT} --click",
+  "on-click-right" => "#{DEPLOY_SCRIPT} --deploy"
+}
+
+save_config(config)
+
+# Add CSS for the deployments module
+style = File.read(WAYBAR_STYLE)
+unless style.include?("#custom-brainiac-deployments")
+  css = <<~CSS
+
+    /* Brainiac deployment environment dots */
+    #custom-brainiac-deployments {
+      margin-left: 100px;
+      margin-right: 40px;
+      font-size: 14px;
+    }
+  CSS
+  File.write(WAYBAR_STYLE, style + css)
+  puts "✓ Added deployment styles to waybar CSS"
+end
+
+# Add padding-right to agent sessions module
+unless style.include?("padding-right") && style.include?("#custom-brainiac")
+  updated_style = File.read(WAYBAR_STYLE)
+  if updated_style.include?("#custom-brainiac {")
+    updated_style.sub!(/(#custom-brainiac\s*\{[^}]*)(\})/) do
+      block = Regexp.last_match(1)
+      close = Regexp.last_match(2)
+      if block.include?("padding-right")
+        "#{block}#{close}"
+      else
+        "#{block}\n  padding-right: 100px;\n#{close}"
+      end
+    end
+    File.write(WAYBAR_STYLE, updated_style)
+    puts "✓ Added padding-right to agent session module"
+  end
+end
+
+puts "✓ Deployments module added to waybar config"
+puts "  Positioned: [deploy-dots] [agent-sessions] in center bar"
+puts "  Restart waybar to apply: omarchy restart waybar"