#!/bin/bash
set -euo pipefail

# =============================================================================
# wechat-claude-code cross-platform daemon manager
# Supports: macOS (launchd) / Linux (systemd + nohup fallback)
# =============================================================================

DATA_DIR="${HOME}/.wechat-claude-code"
PROJECT_DIR="$(cd "$(dirname "$0")/.." && pwd)"
SERVICE_NAME="wechat-claude-code"

# Platform detection
OS_TYPE="$(uname -s)"

# =============================================================================
# macOS (launchd) functions
# =============================================================================

macos_plist_label() {
  echo "com.wechat-claude-code.bridge"
}

macos_plist_path() {
  echo "${HOME}/Library/LaunchAgents/$(macos_plist_label).plist"
}

macos_is_loaded() {
  launchctl print "gui/$(id -u)/$(macos_plist_label)" &>/dev/null
}

macos_start() {
  local plist_label="$(macos_plist_label)"
  local plist_path="$(macos_plist_path)"
  local node_bin="$(command -v node || echo '/usr/local/bin/node')"

  if macos_is_loaded; then
    echo "Already running (or plist loaded)"
    exit 0
  fi

  mkdir -p "$DATA_DIR/logs"

  # Collect Anthropic/Claude env vars for plist
  local plist_extra_env=""
  for var in ANTHROPIC_AUTH_TOKEN ANTHROPIC_API_KEY ANTHROPIC_BASE_URL CLAUDE_API_KEY; do
    if [ -n "${!var:-}" ]; then
      plist_extra_env="${plist_extra_env}    <key>${var}</key>
    <string>${!var}</string>
"
    fi
  done

  cat > "$plist_path" <<PLIST
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
  <key>Label</key>
  <string>${plist_label}</string>
  <key>ProgramArguments</key>
  <array>
    <string>${node_bin}</string>
    <string>${PROJECT_DIR}/dist/main.js</string>
    <string>start</string>
  </array>
  <key>WorkingDirectory</key>
  <string>${PROJECT_DIR}</string>
  <key>RunAtLoad</key>
  <true/>
  <key>KeepAlive</key>
  <true/>
  <key>StandardOutPath</key>
  <string>${DATA_DIR}/logs/stdout.log</string>
  <key>StandardErrorPath</key>
  <string>${DATA_DIR}/logs/stderr.log</string>
  <key>EnvironmentVariables</key>
  <dict>
    <key>PATH</key>
    <string>${HOME}/.local/bin:${node_bin%/*}:/usr/local/bin:/opt/homebrew/bin:/usr/bin:/bin</string>
${plist_extra_env}  </dict>
</dict>
</plist>
PLIST

  launchctl load "$plist_path"
  echo "Started wechat-claude-code daemon (macOS launchd)"
}

macos_stop() {
  local plist_label="$(macos_plist_label)"
  local plist_path="$(macos_plist_path)"

  launchctl bootout "gui/$(id -u)/${plist_label}" 2>/dev/null || true
  rm -f "$plist_path"
  echo "Stopped wechat-claude-code daemon (macOS launchd)"
}

macos_status() {
  if macos_is_loaded; then
    local pid=$(pgrep -f "dist/main.js start" 2>/dev/null | head -1)
    if [ -n "$pid" ]; then
      echo "Running (PID: $pid)"
    else
      echo "Loaded but not running"
    fi
  else
    echo "Not running"
  fi
}

macos_logs() {
  local log_dir="${DATA_DIR}/logs"
  if [ -d "$log_dir" ]; then
    local latest=$(ls -t "${log_dir}"/bridge-*.log 2>/dev/null | head -1)
    if [ -n "$latest" ]; then
      tail -100 "$latest"
    else
      echo "No bridge logs found. Checking stdout/stderr:"
      for f in "${log_dir}"/stdout.log "${log_dir}"/stderr.log; do
        if [ -f "$f" ]; then
          echo "=== $(basename "$f") ==="
          tail -30 "$f"
        fi
      done
    fi
  else
    echo "No logs found"
  fi
}

# =============================================================================
# Linux (systemd) functions
# =============================================================================

linux_ensure_user_session() {
  if [ -z "${XDG_RUNTIME_DIR:-}" ]; then
    export XDG_RUNTIME_DIR="/run/user/$(id -u)"
    mkdir -p "$XDG_RUNTIME_DIR" 2>/dev/null || true
  fi
  if [ -z "${DBUS_SESSION_BUS_ADDRESS:-}" ]; then
    export DBUS_SESSION_BUS_ADDRESS="unix:path=${XDG_RUNTIME_DIR}/bus"
  fi
}

linux_service_file() {
  echo "${HOME}/.config/systemd/user/${SERVICE_NAME}.service"
}

linux_pid_file() {
  echo "${DATA_DIR}/${SERVICE_NAME}.pid"
}

linux_node_bin() {
  local node_bin="$(command -v node 2>/dev/null || echo '')"
  if [ -z "$node_bin" ]; then
    local nvm_default="${NVM_DIR:-${HOME}/.nvm}/versions/node"
    if [ -d "$nvm_default" ]; then
      node_bin="$(find "$nvm_default" -name "node" -type f 2>/dev/null | head -1)"
    fi
  fi
  echo "${node_bin:-/usr/bin/node}"
}

linux_systemd_available() {
  linux_ensure_user_session
  systemctl --user list-units &>/dev/null
}

linux_create_service_file() {
  local service_file="$(linux_service_file)"
  local node_bin="$(linux_node_bin)"

  mkdir -p "$(dirname "$service_file")"

  # Collect Anthropic/Claude env vars to pass through to the service
  local extra_env=""
  for var in ANTHROPIC_AUTH_TOKEN ANTHROPIC_API_KEY ANTHROPIC_BASE_URL CLAUDE_API_KEY; do
    if [ -n "${!var:-}" ]; then
      extra_env="${extra_env}Environment=${var}=${!var}
"
    fi
  done

  cat > "$service_file" <<SERVICE
[Unit]
Description=WeChat Claude Code Bridge
Documentation=https://github.com/Wechat-ggGitHub/wechat-claude-code
After=network.target

[Service]
Type=simple
ExecStart=${node_bin} ${PROJECT_DIR}/dist/main.js start
WorkingDirectory=${PROJECT_DIR}
Restart=always
RestartSec=10
Environment=PATH=${HOME}/.local/bin:${node_bin%/*}:/usr/local/bin:/usr/bin:/bin
${extra_env}StandardOutput=append:${DATA_DIR}/logs/stdout.log
StandardError=append:${DATA_DIR}/logs/stderr.log
NoNewPrivileges=true
PrivateTmp=true

[Install]
WantedBy=default.target
SERVICE

  chmod 644 "$service_file"
}

linux_reload_daemon() {
  linux_ensure_user_session
  systemctl --user daemon-reload 2>/dev/null || true
}

linux_direct_start() {
  local pid_file="$(linux_pid_file)"
  local node_bin="$(linux_node_bin)"

  if [ -f "$pid_file" ]; then
    local old_pid=$(cat "$pid_file" 2>/dev/null)
    if [ -n "$old_pid" ] && kill -0 "$old_pid" 2>/dev/null; then
      echo "Already running (PID: $old_pid)"
      exit 0
    fi
    rm -f "$pid_file"
  fi

  mkdir -p "$DATA_DIR/logs"

  echo "Starting wechat-claude-code daemon (direct mode)..."
  nohup "$node_bin" "${PROJECT_DIR}/dist/main.js" start \
    >> "$DATA_DIR/logs/stdout.log" \
    2>> "$DATA_DIR/logs/stderr.log" &
  local pid=$!
  echo "$pid" > "$pid_file"
  echo "Started (PID: $pid)"
  echo "Logs: $DATA_DIR/logs/stdout.log"
}

linux_direct_stop() {
  local pid_file="$(linux_pid_file)"

  if [ ! -f "$pid_file" ]; then
    echo "Not running (no PID file)"
    exit 0
  fi

  local pid=$(cat "$pid_file" 2>/dev/null)
  if [ -z "$pid" ]; then
    rm -f "$pid_file"
    echo "Stopped"
    exit 0
  fi

  if kill -0 "$pid" 2>/dev/null; then
    kill "$pid" 2>/dev/null || true
    local count=0
    while kill -0 "$pid" 2>/dev/null && [ $count -lt 10 ]; do
      sleep 1
      count=$((count + 1))
    done
    kill -9 "$pid" 2>/dev/null || true
    echo "Stopped (PID: $pid)"
  else
    echo "Process not running (cleaning up PID file)"
  fi

  rm -f "$pid_file"
}

linux_direct_status() {
  local pid_file="$(linux_pid_file)"

  if [ ! -f "$pid_file" ]; then
    echo "Not running"
    exit 0
  fi

  local pid=$(cat "$pid_file" 2>/dev/null)
  if [ -z "$pid" ]; then
    echo "Not running (invalid PID file)"
    exit 0
  fi

  if kill -0 "$pid" 2>/dev/null; then
    echo "Running (PID: $pid)"
  else
    echo "Not running (stale PID file)"
  fi
}

linux_start() {
  if linux_systemd_available; then
    local service_file="$(linux_service_file)"

    if systemctl --user is-active --quiet "${SERVICE_NAME}" 2>/dev/null; then
      echo "Already running"
      exit 0
    fi

    mkdir -p "$DATA_DIR/logs"
    linux_create_service_file
    linux_reload_daemon

    systemctl --user start "${SERVICE_NAME}"
    systemctl --user enable "${SERVICE_NAME}" 2>/dev/null || true
    echo "Started wechat-claude-code daemon (Linux systemd)"
  else
    echo "Note: systemd user session not available, using direct mode"
    echo "To enable systemd mode, run: 'loginctl enable-linger $(whoami)'"
    echo ""
    linux_direct_start
  fi
}

linux_stop() {
  if linux_systemd_available && systemctl --user cat "${SERVICE_NAME}" &>/dev/null; then
    systemctl --user stop "${SERVICE_NAME}" 2>/dev/null || true
    systemctl --user disable "${SERVICE_NAME}" 2>/dev/null || true
    echo "Stopped wechat-claude-code daemon (Linux systemd)"
  else
    linux_direct_stop
  fi
}

linux_restart() {
  linux_stop
  sleep 1
  linux_start
}

linux_status() {
  if linux_systemd_available && systemctl --user cat "${SERVICE_NAME}" &>/dev/null; then
    if systemctl --user is-active --quiet "${SERVICE_NAME}" 2>/dev/null; then
      local pid=$(systemctl --user show-property --value=MainPID "${SERVICE_NAME}" 2>/dev/null)
      if [ -n "$pid" ] && [ "$pid" != "0" ]; then
        echo "Running (PID: $pid)"
      else
        echo "Active"
      fi
    else
      echo "Not running"
    fi

    if systemctl --user cat "${SERVICE_NAME}" &>/dev/null; then
      echo ""
      systemctl --user status "${SERVICE_NAME}" --no-pager 2>/dev/null || true
    fi
  else
    linux_direct_status
  fi
}

linux_logs() {
  if command -v journalctl >/dev/null 2>&1; then
    if journalctl --user --unit="${SERVICE_NAME}" --quiet &>/dev/null; then
      echo "=== systemd journal logs (last 100 lines) ==="
      journalctl --user --unit="${SERVICE_NAME}" --no-pager -n 100 2>/dev/null || true
      echo ""
      echo "=== File logs ==="
    fi
  fi

  local log_dir="${DATA_DIR}/logs"
  if [ -d "$log_dir" ]; then
    for f in "${log_dir}"/stdout.log "${log_dir}"/stderr.log; do
      if [ -f "$f" ]; then
        echo "=== $(basename "$f") ==="
        tail -50 "$f"
        echo ""
      fi
    done
  else
    echo "No logs found"
  fi
}

# =============================================================================
# Main dispatcher
# =============================================================================

main() {
  local command="${1:-}"

  case "$OS_TYPE" in
    Darwin)
      case "$command" in
        start)   macos_start ;;
        stop)    macos_stop ;;
        restart) macos_stop; sleep 1; macos_start ;;
        status)  macos_status ;;
        logs)    macos_logs ;;
        *)
          echo "Usage: daemon.sh {start|stop|restart|status|logs}"
          echo "Platform: macOS (launchd)"
          exit 1
          ;;
      esac
      ;;
    Linux)
      case "$command" in
        start)   linux_start ;;
        stop)    linux_stop ;;
        restart) linux_restart ;;
        status)  linux_status ;;
        logs)    linux_logs ;;
        *)
          echo "Usage: daemon.sh {start|stop|restart|status|logs}"
          echo "Platform: Linux (systemd)"
          exit 1
          ;;
      esac
      ;;
    *)
      echo "Error: Unsupported platform '$OS_TYPE'"
      echo "Supported platforms: macOS (Darwin), Linux"
      exit 1
      ;;
  esac
}

main "$@"