#!/bin/bash

# <xbar.title>hstack</xbar.title>
# <xbar.version>1.0.0</xbar.version>
# <xbar.author>hstack</xbar.author>
# <xbar.author.github>happier-dev</xbar.author.github>
# <xbar.desc>Monitor and control your Happier stacks from the menu bar</xbar.desc>
# <xbar.dependencies>node</xbar.dependencies>
# <swiftbar.hideAbout>true</swiftbar.hideAbout>
# <swiftbar.hideRunInTerminal>true</swiftbar.hideRunInTerminal>
# <swiftbar.hideLastUpdated>false</swiftbar.hideLastUpdated>
# <swiftbar.hideDisablePlugin>true</swiftbar.hideDisablePlugin>
# <swiftbar.hideSwiftBar>true</swiftbar.hideSwiftBar>
# <swiftbar.refreshOnOpen>false</swiftbar.refreshOnOpen>

# ============================================================================
# Configuration
# ============================================================================

# SwiftBar runs with a minimal environment, so users often won't have
# HAPPIER_STACK_HOME_DIR / HAPPIER_STACK_WORKSPACE_DIR exported.
# Treat <canonicalHomeDir>/.env as the canonical pointer file (written by `hstack init`).
# Default: ~/.happier-stack/.env
CANONICAL_HOME_DIR="${HAPPIER_STACK_CANONICAL_HOME_DIR:-$HOME/.happier-stack}"
CANONICAL_ENV_FILE="$CANONICAL_HOME_DIR/.env"

_dotenv_get_quick() {
  # Usage: _dotenv_get_quick /path/to/env KEY
  local file="$1"
  local key="$2"
  [[ -n "$file" && -n "$key" && -f "$file" ]] || return 0
  local line
  line="$(grep -E "^${key}=" "$file" 2>/dev/null | head -n 1 || true)"
  [[ -n "$line" ]] || return 0
  local v="${line#*=}"
  v="${v%$'\r'}"
  # Strip simple surrounding quotes.
  if [[ "$v" == \"*\" && "$v" == *\" ]]; then v="${v#\"}"; v="${v%\"}"; fi
  if [[ "$v" == \'*\' && "$v" == *\' ]]; then v="${v#\'}"; v="${v%\'}"; fi
  echo "$v"
}

_expand_home_quick() {
  local p="$1"
  if [[ "$p" == "~/"* ]]; then
    echo "$HOME/${p#~/}"
  else
    echo "$p"
  fi
}

_home_from_canonical=""
if [[ -f "$CANONICAL_ENV_FILE" ]]; then
  _home_from_canonical="$(_dotenv_get_quick "$CANONICAL_ENV_FILE" "HAPPIER_STACK_HOME_DIR")"
fi
_home_from_canonical="$(_expand_home_quick "${_home_from_canonical:-}")"

export HAPPIER_STACK_HOME_DIR="${HAPPIER_STACK_HOME_DIR:-${_home_from_canonical:-$CANONICAL_HOME_DIR}}"
export HAPPIER_STACK_CANONICAL_HOME_DIR="$CANONICAL_HOME_DIR"

# Colors
GREEN="#34C759"
RED="#FF3B30"
YELLOW="#FFCC00"
GRAY="#8E8E93"
BLUE="#007AFF"

# ============================================================================
# Load libs
# ============================================================================

hstack_ROOT_DIR="${HAPPIER_STACK_CLI_ROOT_DIR:-$HAPPIER_STACK_HOME_DIR}"
LIB_DIR="$hstack_ROOT_DIR/extras/swiftbar/lib"
if [[ ! -f "$LIB_DIR/utils.sh" ]]; then
  echo "hstack"
  echo "---"
  echo "SwiftBar libs missing at: $LIB_DIR"
  echo "↪ run: hstack menubar install"
  exit 0
fi

# shellcheck source=/dev/null
source "$LIB_DIR/utils.sh"
hstack_ROOT_DIR="$(resolve_hstack_root_dir)"
LIB_DIR="$hstack_ROOT_DIR/extras/swiftbar/lib"
# shellcheck source=/dev/null
source "$LIB_DIR/icons.sh"
# shellcheck source=/dev/null
source "$LIB_DIR/system.sh"
# shellcheck source=/dev/null
source "$LIB_DIR/git.sh"
# shellcheck source=/dev/null
source "$LIB_DIR/render.sh"

# ============================================================================
# Menu
# ============================================================================

PRIMARY_STACK_RAW="${HAPPIER_STACK_SWIFTBAR_PRIMARY_STACK:-${HAPPIER_STACK_STACK:-main}}"
PRIMARY_STACK="${PRIMARY_STACK_RAW:-main}"

hstack_BIN="$(resolve_hstack_bin)"
hstack_TERM="$hstack_ROOT_DIR/extras/swiftbar/hstack-term.sh"
TAILSCALE_URL="$(get_tailscale_url)"
if swiftbar_is_sandboxed; then
  # Never probe Tailscale (global machine state) when sandboxing.
  TAILSCALE_URL=""
fi
MAIN_ENV_FILE="$(resolve_main_env_file)"
MENUBAR_MODE="$(resolve_menubar_mode)"

ensure_launchctl_cache

if [[ -z "$MAIN_ENV_FILE" ]]; then
  MAIN_ENV_FILE="$(resolve_stack_env_file "$PRIMARY_STACK")"
fi
if [[ -z "${HAPPIER_STACK_ENV_FILE:-}" ]] && [[ -n "$MAIN_ENV_FILE" ]]; then
  export HAPPIER_STACK_ENV_FILE="$MAIN_ENV_FILE"
fi
MAIN_SERVER_COMPONENT="$(resolve_main_server_component)"
if [[ "$PRIMARY_STACK" == "main" ]]; then
  MAIN_PORT="$(resolve_main_port)"
else
  MAIN_PORT="$(resolve_stack_server_port "$PRIMARY_STACK" "$MAIN_ENV_FILE")"
fi

HAPPIER_HOME_DIR="$(resolve_stack_base_dir "$PRIMARY_STACK" "$MAIN_ENV_FILE")"
CLI_HOME_DIR="$(resolve_stack_cli_home_dir "$PRIMARY_STACK" "$MAIN_ENV_FILE")"
LOGS_DIR="$HAPPIER_HOME_DIR/logs"
MAIN_LABEL="$(resolve_stack_label "$PRIMARY_STACK")"

MAIN_COLLECT="$(collect_stack_status "$MAIN_PORT" "$CLI_HOME_DIR" "$MAIN_LABEL" "$HAPPIER_HOME_DIR")"
IFS=$'\t' read -r MAIN_LEVEL MAIN_SERVER_STATUS MAIN_SERVER_PID MAIN_SERVER_METRICS MAIN_DAEMON_STATUS MAIN_DAEMON_PID MAIN_DAEMON_METRICS MAIN_DAEMON_UPTIME MAIN_LAST_HEARTBEAT MAIN_LAUNCHAGENT_STATUS MAIN_AUTOSTART_PID MAIN_AUTOSTART_METRICS <<<"$MAIN_COLLECT"
MAIN_DAEMON_STATE_FILE="$(resolve_preferred_daemon_state_pair "$CLI_HOME_DIR" | cut -d'|' -f1)"
for v in MAIN_SERVER_PID MAIN_SERVER_METRICS MAIN_DAEMON_PID MAIN_DAEMON_METRICS MAIN_DAEMON_UPTIME MAIN_LAST_HEARTBEAT MAIN_AUTOSTART_PID MAIN_AUTOSTART_METRICS; do
  if [[ "${!v}" == "-" ]]; then
    printf -v "$v" '%s' ""
  fi
done

# Menu bar icon
MENU_STATUS_ICON_B64="$(status_icon_b64 "$MAIN_LEVEL" 18)"
if [[ -n "$MENU_STATUS_ICON_B64" ]]; then
  echo " | image=$MENU_STATUS_ICON_B64"
else
  STATUS_COLOR="$(color_for_level "$MAIN_LEVEL")"
  ICON_B64="$(get_menu_icon_b64)"
  if [[ -n "$ICON_B64" ]]; then
    echo "● | templateImage=$ICON_B64 color=$STATUS_COLOR"
  else
    echo "hstack"
  fi
fi

echo "---"
echo "hstack | size=14 font=SF Pro Display"
echo "---"

# Mode (selfhost vs dev)
if [[ "$MENUBAR_MODE" == "selfhost" ]]; then
  echo "Mode: Selfhost | sfimage=house"
else
  echo "Mode: Dev | sfimage=hammer"
fi
if [[ -n "$hstack_BIN" ]]; then
  if [[ "$MENUBAR_MODE" == "selfhost" ]]; then
    echo "--Switch to Dev mode | bash=$hstack_BIN param1=menubar param2=mode param3=dev dir=$hstack_ROOT_DIR terminal=false refresh=true"
  else
    echo "--Switch to Selfhost mode | bash=$hstack_BIN param1=menubar param2=mode param3=selfhost dir=$hstack_ROOT_DIR terminal=false refresh=true"
  fi
fi
echo "---"

# Main stack (inline)
if [[ "$PRIMARY_STACK" == "main" ]]; then
  # If the main env file points at a named stack (common in wrapper installs),
  # render the concrete stack name to avoid ambiguity.
  DECLARED_MAIN_STACK_NAME="$(dotenv_get "$MAIN_ENV_FILE" "HAPPIER_STACK_STACK")"
  if [[ -n "$DECLARED_MAIN_STACK_NAME" && "$DECLARED_MAIN_STACK_NAME" != "main" ]]; then
    echo "Stack: $DECLARED_MAIN_STACK_NAME"
  else
    echo "Main stack"
  fi
else
  echo "Stack: $PRIMARY_STACK"
fi
echo "---"
export MAIN_LEVEL="$MAIN_LEVEL"
render_stack_info "" "$PRIMARY_STACK" "$MAIN_PORT" "$MAIN_SERVER_COMPONENT" "$HAPPIER_HOME_DIR" "$CLI_HOME_DIR" "$MAIN_LABEL" "$MAIN_ENV_FILE" "$TAILSCALE_URL" "$MAIN_SERVER_METRICS" "$MAIN_DAEMON_METRICS" "$MAIN_AUTOSTART_METRICS"
render_component_server "" "$PRIMARY_STACK" "$MAIN_PORT" "$MAIN_SERVER_COMPONENT" "$MAIN_SERVER_STATUS" "$MAIN_SERVER_PID" "$MAIN_SERVER_METRICS" "$TAILSCALE_URL" "$MAIN_LABEL"
render_component_daemon "" "$MAIN_DAEMON_STATUS" "$MAIN_DAEMON_PID" "$MAIN_DAEMON_METRICS" "$MAIN_DAEMON_UPTIME" "$MAIN_LAST_HEARTBEAT" "$MAIN_DAEMON_STATE_FILE" "$PRIMARY_STACK"
render_component_autostart "" "$PRIMARY_STACK" "$MAIN_LABEL" "$MAIN_LAUNCHAGENT_STATUS" "$MAIN_AUTOSTART_PID" "$MAIN_AUTOSTART_METRICS" "$LOGS_DIR"
render_component_tailscale "" "$PRIMARY_STACK" "$TAILSCALE_URL"

echo "---"
if [[ "$MENUBAR_MODE" == "selfhost" ]]; then
  echo "Maintenance | sfimage=wrench.and.screwdriver"
  if [[ -n "$hstack_BIN" ]]; then
    UPDATE_JSON="${hstack_ROOT_DIR}/cache/update.json"
    update_available=""
    latest=""
    current=""
    if [[ -f "$UPDATE_JSON" ]]; then
      update_available="$(grep -oE '\"updateAvailable\"[[:space:]]*:[[:space:]]*(true|false)' "$UPDATE_JSON" 2>/dev/null | head -1 | grep -oE '(true|false)' || true)"
      latest="$(grep -oE '\"latest\"[[:space:]]*:[[:space:]]*\"[^\"]+\"' "$UPDATE_JSON" 2>/dev/null | head -1 | sed -E 's/.*\"latest\"[[:space:]]*:[[:space:]]*\"([^\"]+)\".*/\\1/' || true)"
      current="$(grep -oE '\"current\"[[:space:]]*:[[:space:]]*\"[^\"]+\"' "$UPDATE_JSON" 2>/dev/null | head -1 | sed -E 's/.*\"current\"[[:space:]]*:[[:space:]]*\"([^\"]+)\".*/\\1/' || true)"
    fi
    if [[ "$update_available" == "true" && -n "$latest" ]]; then
      echo "--Update available: ${current:-current} → ${latest} | color=$YELLOW"
    else
      echo "--Updates: up to date | color=$GRAY"
    fi
    echo "--Check for updates | bash=$hstack_BIN param1=self param2=check dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Update hstack runtime | bash=$hstack_BIN param1=self param2=update dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Doctor | bash=$hstack_BIN param1=doctor dir=$hstack_ROOT_DIR terminal=false refresh=true"
  else
    echo "--⚠️ hstack not found (run: npx @happier-dev/stack@latest init)" 
  fi
else
  echo "Stacks | sfimage=server.rack"
  STACKS_PREFIX="--"

  if [[ -n "$hstack_BIN" ]]; then
    echo "${STACKS_PREFIX}New stack (interactive) | bash=$hstack_TERM param1=stack param2=new param3=--interactive dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "${STACKS_PREFIX}List stacks | bash=$hstack_TERM param1=stack param2=list dir=$hstack_ROOT_DIR terminal=false"
    print_sep "$STACKS_PREFIX"
  fi

  STACKS_DIR="$(resolve_stacks_storage_root)"
  if [[ -d "$STACKS_DIR" ]]; then
    STACK_NAMES="$(
      {
        ls -1 "$STACKS_DIR" 2>/dev/null || true
      } | sort -u
    )"
    if [[ -z "$STACK_NAMES" ]]; then
      echo "${STACKS_PREFIX}No stacks found | color=$GRAY"
    fi
    for s in $STACK_NAMES; do
      env_file="$(resolve_stack_env_file "$s")"
      [[ -f "$env_file" ]] || continue

      # Ports may be ephemeral (runtime-only). Do not skip stacks if the env file does not pin a port.
      port="$(resolve_stack_server_port "$s" "$env_file")"

      server_component="$(dotenv_get "$env_file" "HAPPIER_STACK_SERVER_COMPONENT")"
      [[ -n "$server_component" ]] || server_component="happier-server-light"

      base_dir="$(resolve_stack_base_dir "$s" "$env_file")"
      cli_home_dir="$(resolve_stack_cli_home_dir "$s" "$env_file")"
      label="$(resolve_stack_label "$s")"

      COLLECT="$(collect_stack_status "$port" "$cli_home_dir" "$label" "$base_dir")"
      IFS=$'\t' read -r LEVEL SERVER_STATUS SERVER_PID SERVER_METRICS DAEMON_STATUS DAEMON_PID DAEMON_METRICS DAEMON_UPTIME LAST_HEARTBEAT LAUNCHAGENT_STATUS AUTOSTART_PID AUTOSTART_METRICS <<<"$COLLECT"
      DAEMON_STATE_FILE="$(resolve_preferred_daemon_state_pair "$cli_home_dir" | cut -d'|' -f1)"
      for v in SERVER_PID SERVER_METRICS DAEMON_PID DAEMON_METRICS DAEMON_UPTIME LAST_HEARTBEAT AUTOSTART_PID AUTOSTART_METRICS; do
        if [[ "${!v}" == "-" ]]; then
          printf -v "$v" '%s' ""
        fi
      done

      render_stack_overview_item "Stack: $s" "$LEVEL" "$STACKS_PREFIX"
      export STACK_LEVEL="$LEVEL"
      render_stack_info "${STACKS_PREFIX}--" "$s" "$port" "$server_component" "$base_dir" "$cli_home_dir" "$label" "$env_file" "" "$SERVER_METRICS" "$DAEMON_METRICS" "$AUTOSTART_METRICS"
      render_component_server "${STACKS_PREFIX}--" "$s" "$port" "$server_component" "$SERVER_STATUS" "$SERVER_PID" "$SERVER_METRICS" "" "$label"
      render_component_daemon "${STACKS_PREFIX}--" "$DAEMON_STATUS" "$DAEMON_PID" "$DAEMON_METRICS" "$DAEMON_UPTIME" "$LAST_HEARTBEAT" "$DAEMON_STATE_FILE" "$s"
      render_component_autostart "${STACKS_PREFIX}--" "$s" "$label" "$LAUNCHAGENT_STATUS" "$AUTOSTART_PID" "$AUTOSTART_METRICS" "$base_dir/logs"
      render_component_tailscale "${STACKS_PREFIX}--" "$s" ""
      render_components_menu "${STACKS_PREFIX}--" "stack" "$s" "$env_file"
    done
  else
    echo "${STACKS_PREFIX}No stacks dir found at: $(shorten_path "$STACKS_DIR" 52) | color=$GRAY"
  fi

  echo "---"
  render_components_menu "" "main" "$PRIMARY_STACK" "$MAIN_ENV_FILE"

  echo "Worktrees | sfimage=arrow.triangle.branch"
  if [[ -z "$hstack_BIN" ]]; then
    echo "--⚠️ hstack not found (run: npx @happier-dev/stack@latest init)"
  else
    echo "--Use (interactive) | bash=$hstack_TERM param1=wt param2=use param3=--interactive dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--New (interactive) | bash=$hstack_TERM param1=wt param2=new param3=--interactive dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--PR worktree (prompt) | bash=$hstack_ROOT_DIR/extras/swiftbar/wt-pr.sh dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Sync mirrors (all) | bash=$hstack_BIN param1=wt param2=sync-all dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Update all (dry-run) | bash=$hstack_TERM param1=wt param2=update-all param3=--dry-run dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Update all (apply) | bash=$hstack_BIN param1=wt param2=update-all dir=$hstack_ROOT_DIR terminal=false refresh=true"
  fi

  echo "---"
  echo "Setup / Tools"
  if [[ -z "$hstack_BIN" ]]; then
    echo "--⚠️ hstack not found (run: npx @happier-dev/stack@latest init)"
  else
    echo "--Setup (guided) | bash=$hstack_TERM param1=setup dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Bootstrap (clone/install) | bash=$hstack_TERM param1=bootstrap dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--CLI link (install happier wrapper) | bash=$hstack_TERM param1=cli:link dir=$hstack_ROOT_DIR terminal=false refresh=true"
    echo "--Mobile dev helper | bash=$hstack_TERM param1=mobile dir=$hstack_ROOT_DIR terminal=false"
  fi
fi

echo "---"
echo "Refresh | sfimage=arrow.clockwise refresh=true"
echo "---"
echo "Refresh interval | sfimage=timer"
SET_INTERVAL="$hstack_ROOT_DIR/extras/swiftbar/set-interval.sh"
echo "--10s | bash=$SET_INTERVAL param1=10s dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--30s | bash=$SET_INTERVAL param1=30s dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--1m | bash=$SET_INTERVAL param1=1m dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--5m (recommended) | bash=$SET_INTERVAL param1=5m dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--10m | bash=$SET_INTERVAL param1=10m dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--15m | bash=$SET_INTERVAL param1=15m dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--30m | bash=$SET_INTERVAL param1=30m dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--1h | bash=$SET_INTERVAL param1=1h dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--2h | bash=$SET_INTERVAL param1=2h dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--6h | bash=$SET_INTERVAL param1=6h dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--12h | bash=$SET_INTERVAL param1=12h dir=$hstack_ROOT_DIR terminal=false refresh=true"
echo "--1d | bash=$SET_INTERVAL param1=1d dir=$hstack_ROOT_DIR terminal=false refresh=true"

exit 0