#!/bin/bash
set -e

INPUT=$(cat)
SESSION_ID=$(echo "$INPUT" | jq -r '.session_id')
PLUGIN_ROOT="${CLAUDE_PLUGIN_ROOT:-$(dirname "$0")/..}"
HUD_DIR="$HOME/.claude/hud"
EVENT_FIFO="$HUD_DIR/events/$SESSION_ID.fifo"
PID_FILE="$HUD_DIR/pids/$SESSION_ID.pid"
LOG_FILE="$HUD_DIR/logs/$SESSION_ID.log"

mkdir -p "$HUD_DIR/events" "$HUD_DIR/logs" "$HUD_DIR/pids"

rm -f "$EVENT_FIFO"
mkfifo "$EVENT_FIFO"

if [ ! -d "$PLUGIN_ROOT/tui/node_modules" ]; then
  cd "$PLUGIN_ROOT/tui"
  if command -v bun &> /dev/null; then
    bun install --silent 2>/dev/null || true
  elif command -v npm &> /dev/null; then
    npm install --silent 2>/dev/null || true
  fi
fi

if [ ! -f "$PLUGIN_ROOT/tui/dist/index.js" ]; then
  cd "$PLUGIN_ROOT/tui"
  if command -v bun &> /dev/null; then
    bun run build 2>/dev/null || true
  elif command -v npm &> /dev/null; then
    npm run build 2>/dev/null || true
  fi
fi

if command -v bun &> /dev/null; then
  HUD_CMD="bun $PLUGIN_ROOT/tui/dist/index.js --session $SESSION_ID --fifo $EVENT_FIFO"
else
  HUD_CMD="node $PLUGIN_ROOT/tui/dist/index.js --session $SESSION_ID --fifo $EVENT_FIFO"
fi

launch_split_pane() {
  # tmux - most reliable split pane support
  if [ -n "$TMUX" ]; then
    tmux split-window -h -l 50 "$HUD_CMD" 2>/dev/null && return 0
  fi

  # iTerm2 on macOS
  if [ "$TERM_PROGRAM" = "iTerm.app" ]; then
    osascript -e "
      tell application \"iTerm2\"
        tell current session of current window
          split vertically with default profile command \"$HUD_CMD\"
        end tell
      end tell
    " 2>/dev/null && return 0
  fi

  # Kitty terminal
  if [ -n "$KITTY_PID" ]; then
    kitty @ launch --location=vsplit --cwd=current $HUD_CMD 2>/dev/null && return 0
  fi

  # WezTerm
  if [ "$TERM_PROGRAM" = "WezTerm" ]; then
    wezterm cli split-pane --right --percent 25 -- $HUD_CMD 2>/dev/null && return 0
  fi

  # Zellij
  if [ -n "$ZELLIJ" ]; then
    zellij run -f -- $HUD_CMD 2>/dev/null && return 0
  fi

  # Windows Terminal (WSL)
  if [ -n "$WT_SESSION" ]; then
    wt.exe -w 0 sp -H -s 0.25 wsl.exe $HUD_CMD 2>/dev/null && return 0
  fi

  # macOS Terminal.app - open in new window
  if [ "$(uname)" = "Darwin" ] && [ "$TERM_PROGRAM" = "Apple_Terminal" ]; then
    osascript -e "
      tell application \"Terminal\"
        do script \"$HUD_CMD\"
        set bounds of front window to {100, 100, 500, 600}
      end tell
    " 2>/dev/null && return 0
  fi

  # Linux with xterm available - open in new window
  if [ "$(uname)" = "Linux" ] && command -v xterm &> /dev/null; then
    xterm -geometry 50x40 -e "$HUD_CMD" &
    echo $! > "$PID_FILE"
    return 0
  fi

  # Fallback: run in background with logging
  nohup $HUD_CMD > "$LOG_FILE" 2>&1 &
  echo $! > "$PID_FILE"
  return 0
}

launch_split_pane

exit 0