#!/bin/bash
set -e

# ─── Usage ───────────────────────────────────────────────────────────────────
usage() {
  echo "Usage: $0 [options]"
  echo ""
  echo "Options:"
  echo "  -i, --id          Client ID (e.g. DFA-001)"
  echo "  -n, --name        Client name (e.g. 'DFA Tech Colo')"
  echo "  -s, --slug        Short slug for filenames (e.g. dfa_tech)"
  echo "  -w, --webhook     n8n webhook URL"
  echo "  -b, --billing     Billing model (hybrid|per-host|time-saved|tiered) default: hybrid"
  echo "  -e, --estimate    Human time estimate seconds (default: 2700)"
  echo "  -v, --vpn         VPN type (ipsec|openvpn) default: ipsec"
  echo "  -H, --hypervisor  Hypervisor type (proxmox|xcpng) default: proxmox"
  echo "  --proxmox-host    Proxmox host IP (optional)"
  echo "  --proxmox-token-id     Proxmox API token ID (e.g. ansible@pve!token-name)"
  echo "  --proxmox-token-secret Proxmox API token secret"
  echo "  --semaphore-url   Semaphore base URL (default: http://localhost:3000)"
  echo "  --semaphore-token Semaphore API token"
  echo "  --gitea-url       Gitea repo SSH URL"
  echo "  --project-name    Semaphore project name (default: 'Client - <n>')"
  echo ""
  echo "Example:"
  echo "  $0 -i ACME-001 -n 'Acme Corp' -s acme_corp -w https://n8n.voice1.me/webhook/xxx \\"
  echo "     --proxmox-host 10.1.2.3 --proxmox-token-id ansible@pve!ansible-token \\"
  echo "     --proxmox-token-secret abc123"
  exit 1
}

# ─── Defaults ────────────────────────────────────────────────────────────────
BILLING="hybrid"
ESTIMATE="2700"
VPN="ipsec"
HYPERVISOR="proxmox"
SEMAPHORE_URL="http://localhost:3000"
REPO_DIR="/opt/ansible-msp-automations"
GITEA_DEPLOY_KEY="/root/.ssh/gitea_ansible"
GITEA_REPO_URL="ssh://git@172.31.10.8:2222/VOICE1/ansible-msp-automations.git"
PROXMOX_HOST=""
PROXMOX_TOKEN_ID=""
PROXMOX_TOKEN_SECRET=""
PROJECT_NAME_OVERRIDE=""

# ─── Parse args ──────────────────────────────────────────────────────────────
while [[ $# -gt 0 ]]; do
  case $1 in
    -i|--id)                   CLIENT_ID="$2";               shift 2 ;;
    -n|--name)                 CLIENT_NAME="$2";             shift 2 ;;
    -s|--slug)                 CLIENT_SLUG="$2";             shift 2 ;;
    -w|--webhook)              WEBHOOK_URL="$2";             shift 2 ;;
    -b|--billing)              BILLING="$2";                 shift 2 ;;
    -e|--estimate)             ESTIMATE="$2";                shift 2 ;;
    -v|--vpn)                  VPN="$2";                     shift 2 ;;
    -H|--hypervisor)           HYPERVISOR="$2";              shift 2 ;;
    --proxmox-host)            PROXMOX_HOST="$2";            shift 2 ;;
    --proxmox-token-id)        PROXMOX_TOKEN_ID="$2";        shift 2 ;;
    --proxmox-token-secret)    PROXMOX_TOKEN_SECRET="$2";    shift 2 ;;
    --semaphore-url)           SEMAPHORE_URL="$2";           shift 2 ;;
    --semaphore-token)         SEMAPHORE_TOKEN="$2";         shift 2 ;;
    --gitea-url)               GITEA_REPO_URL="$2";          shift 2 ;;
    --project-name)            PROJECT_NAME_OVERRIDE="$2";   shift 2 ;;
    *) usage ;;
  esac
done

# ─── Validate required args ──────────────────────────────────────────────────
if [[ -z "${CLIENT_ID:-}" || -z "${CLIENT_NAME:-}" || -z "${CLIENT_SLUG:-}" || -z "${WEBHOOK_URL:-}" ]]; then
  echo "ERROR: --id, --name, --slug, and --webhook are required"
  usage
fi

# Load SEMAPHORE_TOKEN from env file if not passed as arg
if [[ -z "${SEMAPHORE_TOKEN:-}" && -f "/root/.semaphore_env" ]]; then
  source /root/.semaphore_env
fi

PROJECT_NAME="${PROJECT_NAME_OVERRIDE:-Client - ${CLIENT_NAME}}"
KEY_FILE="/root/.ssh/client_${CLIENT_SLUG}"
INVENTORY_DIR="$REPO_DIR/inventories/client_${CLIENT_SLUG}"
INVENTORY_REPO_PATH="inventories/client_${CLIENT_SLUG}/hosts.yml"

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  Onboarding client: $CLIENT_NAME ($CLIENT_ID)"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"

# ─── Step 1: Generate SSH key ─────────────────────────────────────────────────
echo ""
echo "[ 1/6 ] Generating SSH key..."
if [[ -f "$KEY_FILE" ]]; then
  echo "  ⚠  Key already exists at $KEY_FILE — skipping generation"
else
  ssh-keygen -t ed25519 -C "ansible-${CLIENT_SLUG}" -f "$KEY_FILE" -N ""
  echo "  ✓  Key generated: $KEY_FILE"
fi
echo ""
echo "  ┌─ Deploy this public key to all client hosts ───────────────────────"
echo "  │"
sed 's/^/  │  /' "$KEY_FILE.pub"
echo "  │"
echo "  └────────────────────────────────────────────────────────────────────"

# ─── Step 2: Create inventory from template ───────────────────────────────────
echo ""
echo "[ 2/6 ] Creating inventory from template..."

if [[ -d "$INVENTORY_DIR" ]]; then
  echo "  ⚠  Inventory directory already exists — skipping"
else
  cp -r "$REPO_DIR/inventories/client_template" "$INVENTORY_DIR"

  cat > "$INVENTORY_DIR/hosts.yml" << HOSTSEOF
---
all:
  vars:
    client_id: "${CLIENT_ID}"
    client_name: "${CLIENT_NAME}"
    billing_model: "${BILLING}"
    maintenance_window_start: "02:00"
    maintenance_window_end: "05:00"
    maintenance_window_tz: "UTC"
    change_freeze: false
    hypervisor_type: "${HYPERVISOR}"
    vpn_type: "${VPN}"
    auto_reboot: false
    human_estimate_seconds: ${ESTIMATE}

  children:
    linux_hosts:
      hosts: {}
      vars:
        ansible_user: root
        os_family: "debian"

    windows_hosts:
      hosts: {}
      vars:
        ansible_user: Administrator
        ansible_connection: winrm
        ansible_winrm_transport: ntlm
        ansible_winrm_server_cert_validation: validate
        ansible_port: 5986
HOSTSEOF

  cat > "$INVENTORY_DIR/group_vars/all.yml" << VARSEOF
---
# Client: ${CLIENT_NAME} (${CLIENT_ID})
# Onboarded: $(date +%Y-%m-%d)
# VPN: ${VPN}
# Hypervisor: ${HYPERVISOR}
# Billing: ${BILLING}

# Add client-specific overrides below
VARSEOF

  echo "  ✓  Inventory created at $INVENTORY_DIR"
fi

# ─── Step 3: Commit and push to Gitea ────────────────────────────────────────
echo ""
echo "[ 3/6 ] Committing to Gitea..."
cd "$REPO_DIR"
git add .
git commit -m "Onboard client: ${CLIENT_NAME} (${CLIENT_ID}) — inventory scaffold" \
  || echo "  ⚠  Nothing to commit"
git push origin main
echo "  ✓  Pushed to Gitea"

# ─── Step 4: Create Semaphore project via API ─────────────────────────────────
echo ""
echo "[ 4/6 ] Creating Semaphore project via API..."

if [[ -z "${SEMAPHORE_TOKEN:-}" ]]; then
  echo "  ✗  No semaphore token available."
  echo "     Set SEMAPHORE_TOKEN in /root/.semaphore_env or pass --semaphore-token"
  exit 1
fi

# 4a. Create project
PROJECT_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/projects" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{\"name\": \"$PROJECT_NAME\", \"alert\": false, \"max_parallel_tasks\": 0}")
PROJECT_ID=$(echo "$PROJECT_RESPONSE" | jq -r '.id')

if [[ "$PROJECT_ID" == "null" || -z "$PROJECT_ID" ]]; then
  echo "  ✗  Failed to create project"
  echo "     Response: $PROJECT_RESPONSE"
  exit 1
fi
echo "  ✓  Project created: $PROJECT_NAME (ID: $PROJECT_ID)"

# 4b. Add gitea-deploy key — dynamically read key file, jq for safe JSON encoding
GITEA_KEY_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/keys" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"gitea-deploy\",
    \"type\": \"ssh\",
    \"project_id\": $PROJECT_ID,
    \"ssh\": {
      \"login\": \"\",
      \"passphrase\": \"\",
      \"private_key\": $(jq -Rs . < "$GITEA_DEPLOY_KEY")
    }
  }")
GITEA_KEY_ID=$(echo "$GITEA_KEY_RESPONSE" | jq -r '.id')
echo "  ✓  gitea-deploy key added (ID: $GITEA_KEY_ID)"

# 4c. Add client SSH key
CLIENT_KEY_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/keys" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"client-${CLIENT_SLUG}-ssh\",
    \"type\": \"ssh\",
    \"project_id\": $PROJECT_ID,
    \"ssh\": {
      \"login\": \"\",
      \"passphrase\": \"\",
      \"private_key\": $(jq -Rs . < "$KEY_FILE")
    }
  }")
CLIENT_KEY_ID=$(echo "$CLIENT_KEY_RESPONSE" | jq -r '.id')
echo "  ✓  Client SSH key added (ID: $CLIENT_KEY_ID)"

# 4d. Add None key (required for inventory become_key_id)
NONE_KEY_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/keys" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"None\",
    \"type\": \"none\",
    \"project_id\": $PROJECT_ID
  }")
NONE_KEY_ID=$(echo "$NONE_KEY_RESPONSE" | jq -r '.id')
echo "  ✓  None key added (ID: $NONE_KEY_ID)"

# 4e. Add repository — uses dynamically obtained GITEA_KEY_ID (not hardcoded)
REPO_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/repositories" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"ansible-msp-automations\",
    \"project_id\": $PROJECT_ID,
    \"git_url\": \"$GITEA_REPO_URL\",
    \"git_branch\": \"main\",
    \"ssh_key_id\": $GITEA_KEY_ID
  }")
REPO_ID=$(echo "$REPO_RESPONSE" | jq -r '.id')
echo "  ✓  Repository linked (ID: $REPO_ID)"

# 4f. Add variable group with all vars including Proxmox
VARS_JSON=$(jq -n \
  --arg webhook  "$WEBHOOK_URL" \
  --arg cid      "$CLIENT_ID" \
  --arg cname    "$CLIENT_NAME" \
  --arg billing  "$BILLING" \
  --arg estimate "$ESTIMATE" \
  --arg phost    "$PROXMOX_HOST" \
  --arg ptid     "$PROXMOX_TOKEN_ID" \
  --arg ptsecret "$PROXMOX_TOKEN_SECRET" \
  '{
    N8N_WEBHOOK_URL:        $webhook,
    CLIENT_ID:              $cid,
    CLIENT_NAME:            $cname,
    BILLING_MODEL:          $billing,
    HUMAN_ESTIMATE_SECONDS: $estimate,
    PROXMOX_HOST:           $phost,
    PROXMOX_TOKEN_ID:       $ptid,
    PROXMOX_TOKEN_SECRET:   $ptsecret
  }')

ENV_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/environment" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"${CLIENT_SLUG}-vars\",
    \"project_id\": $PROJECT_ID,
    \"json\": $(echo "$VARS_JSON" | jq -Rs .),
    \"env\": \"{}\"
  }")
ENV_ID=$(echo "$ENV_RESPONSE" | jq -r '.id')
echo "  ✓  Variable group created (ID: $ENV_ID)"

# 4g. Add inventory (file type, pointing to Git repo path)
INVENTORY_RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/inventory" \
  -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
  -H "Content-Type: application/json" \
  -d "{
    \"name\": \"client-${CLIENT_SLUG}\",
    \"project_id\": $PROJECT_ID,
    \"inventory\": \"$INVENTORY_REPO_PATH\",
    \"ssh_key_id\": $CLIENT_KEY_ID,
    \"become_key_id\": $NONE_KEY_ID,
    \"type\": \"file\",
    \"repository_id\": $REPO_ID
  }")
INVENTORY_ID=$(echo "$INVENTORY_RESPONSE" | jq -r '.id')
echo "  ✓  Inventory created (ID: $INVENTORY_ID)"

# ─── Step 5: Create task templates ───────────────────────────────────────────
echo ""
echo "[ 5/6 ] Creating task templates..."

create_template() {
  local TNAME="$1"
  local PLAYBOOK="$2"
  local DESC="$3"
  RESPONSE=$(curl -s -X POST "$SEMAPHORE_URL/api/project/$PROJECT_ID/templates" \
    -H "Authorization: Bearer $SEMAPHORE_TOKEN" \
    -H "Content-Type: application/json" \
    -d "{
      \"project_id\":                  $PROJECT_ID,
      \"inventory_id\":                $INVENTORY_ID,
      \"repository_id\":               $REPO_ID,
      \"environment_id\":              $ENV_ID,
      \"name\":                        \"$TNAME\",
      \"playbook\":                    \"$PLAYBOOK\",
      \"arguments\":                   \"[]\",
      \"allow_override_args_in_task\": false,
      \"description\":                 \"$DESC\",
      \"app\":                         \"ansible\"
    }")
  echo "  ✓  $(echo "$RESPONSE" | jq -r '"Template: \(.name) (ID: \(.id))"')"
}

create_template "Preflight Check"  "playbooks/site_preflight.yml"   "Run safety checks on all hosts before maintenance"
create_template "Snapshot Test"    "playbooks/snapshot_pre.yml"     "Test pre-patch snapshot creation via Proxmox API"
create_template "Linux Patch"      "playbooks/linux_patch.yml"      "Full Linux patch run with version tracking"
create_template "Full Maintenance" "playbooks/site_maintenance.yml" "Full maintenance: snapshot, preflight, patch"

# ─── Step 6: Summary ──────────────────────────────────────────────────────────
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  ✓  Client onboarding complete: $CLIENT_NAME"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""
echo "  Semaphore project ID: $PROJECT_ID"
echo "  Inventory file:       $INVENTORY_DIR/hosts.yml"
echo ""
echo "  Next steps:"
echo "    1. Add hosts to: $INVENTORY_DIR/hosts.yml"
echo "    2. git add . && git commit -m 'Add hosts for ${CLIENT_NAME}' && git push origin main"
echo "    3. Semaphore → $PROJECT_NAME → Preflight Check → ▶ Run"
echo "    4. Semaphore → $PROJECT_NAME → Snapshot Test → ▶ Run"
echo "    5. Semaphore → $PROJECT_NAME → Linux Patch → ▶ Run"
echo ""
echo "  Public key to deploy to client hosts:"
sed 's/^/    /' "$KEY_FILE.pub"
echo ""