#!/bin/bash
# scripts/sync-guacamole-oauth.sh
# Sync Guacamole OAuth credentials from Authentik to Vault

set -euo pipefail

AUTHENTIK_NAMESPACE="authentik"
VAULT_NAMESPACE="vault"
VAULT_SECRET_PATH="secret/guacamole-oauth"

echo "=== Syncing Guacamole OAuth Credentials from Authentik to Vault ==="

# Wait for Authentik to be fully ready
echo "Waiting for Authentik to be ready..."
kubectl -n "${AUTHENTIK_NAMESPACE}" wait --for=condition=Ready pod -l app.kubernetes.io/name=authentik,app.kubernetes.io/component=server --timeout=60s || {
  echo "WARNING: Authentik may not be fully ready yet. Continuing anyway..."
}

# Get Authentik pod name
AUTHENTIK_POD=$(kubectl get pods -n "${AUTHENTIK_NAMESPACE}" -l app.kubernetes.io/name=authentik,app.kubernetes.io/component=server -o jsonpath='{.items[0].metadata.name}')

if [ -z "$AUTHENTIK_POD" ]; then
  echo "ERROR: Could not find Authentik server pod"
  exit 1
fi

echo "Found Authentik pod: ${AUTHENTIK_POD}"

# Retrieve OAuth2 provider credentials from Authentik
echo "Retrieving Guacamole OAuth credentials from Authentik..."

# Try to get credentials via ak command
CLIENT_ID=$(kubectl exec -n "${AUTHENTIK_NAMESPACE}" "${AUTHENTIK_POD}" -- ak list_providers --type oauth2 2>/dev/null | grep -A 20 '"name": "guacamole"' | grep '"client_id"' | cut -d'"' -f4 || echo "")
CLIENT_SECRET=$(kubectl exec -n "${AUTHENTIK_NAMESPACE}" "${AUTHENTIK_POD}" -- ak list_providers --type oauth2 2>/dev/null | grep -A 20 '"name": "guacamole"' | grep '"client_secret"' | cut -d'"' -f4 || echo "")

# If command fails, try using PostgreSQL directly
if [ -z "$CLIENT_ID" ] || [ -z "$CLIENT_SECRET" ]; then
  echo "Attempting to retrieve credentials from Authentik PostgreSQL database..."

  CNPG_POD=$(kubectl get pods -n "${AUTHENTIK_NAMESPACE}" -l cnpg.io/cluster=pg-authentik,role=primary -o jsonpath='{.items[0].metadata.name}')

  if [ -z "$CNPG_POD" ]; then
    echo "ERROR: Could not find Authentik PostgreSQL pod"
    exit 1
  fi

  # Get PostgreSQL password
  PG_PASSWORD=$(kubectl get secret -n "${AUTHENTIK_NAMESPACE}" pg-authentik-app -o jsonpath='{.data.password}' | base64 -d)

  # Query the database for guacamole provider credentials
  PROVIDER_DATA=$(kubectl exec -n "${AUTHENTIK_NAMESPACE}" "${CNPG_POD}" -c postgres -- env PGPASSWORD="${PG_PASSWORD}" psql -U authentik -d authentik -t -c \
    "SELECT client_id, client_secret FROM authentik_providers_oauth2_oauth2provider WHERE name='guacamole';" 2>/dev/null || echo "")

  if [ -n "$PROVIDER_DATA" ]; then
    CLIENT_ID=$(echo "$PROVIDER_DATA" | awk '{print $1}' | tr -d ' ')
    CLIENT_SECRET=$(echo "$PROVIDER_DATA" | awk '{print $3}' | tr -d ' ')
  fi
fi

# Validate credentials were retrieved
if [ -z "$CLIENT_ID" ] || [ -z "$CLIENT_SECRET" ]; then
  echo "ERROR: Failed to retrieve Guacamole OAuth credentials from Authentik"
  echo ""
  echo "Manual steps required:"
  echo "1. Access Authentik admin panel at https://auth.kube.huskypup.net"
  echo "2. Navigate to: Applications > Providers > guacamole"
  echo "3. Copy the Client ID and Client Secret"
  echo "4. Store them in Vault manually with:"
  echo "   kubectl exec -n vault <vault-pod> -- vault kv put ${VAULT_SECRET_PATH} client-id=\"<CLIENT_ID>\" client-secret=\"<CLIENT_SECRET>\""
  exit 1
fi

echo "Successfully retrieved credentials:"
echo "  Client ID: ${CLIENT_ID}"
echo "  Client Secret: ${CLIENT_SECRET:0:10}..." # Only show first 10 chars

# Check if Vault is unsealed and ready
echo "Checking Vault status..."
VAULT_POD=$(kubectl get pods -n "${VAULT_NAMESPACE}" -l app.kubernetes.io/name=vault -o jsonpath='{.items[0].metadata.name}')

if [ -z "$VAULT_POD" ]; then
  echo "ERROR: Vault pod not found"
  exit 1
fi

VAULT_STATUS=$(kubectl exec -n "${VAULT_NAMESPACE}" "${VAULT_POD}" -- vault status -format=json 2>/dev/null || echo "{}")
SEALED=$(echo "$VAULT_STATUS" | grep -o '"sealed":[^,}]*' | cut -d':' -f2 | tr -d ' ')

if [ "$SEALED" = "true" ]; then
  echo "ERROR: Vault is sealed. Please unseal Vault first."
  exit 1
fi

# Store credentials in Vault
echo "Storing Guacamole OAuth credentials in Vault at ${VAULT_SECRET_PATH}..."
kubectl exec -n "${VAULT_NAMESPACE}" "${VAULT_POD}" -- vault kv put "${VAULT_SECRET_PATH}" \
  client-id="${CLIENT_ID}" \
  client-secret="${CLIENT_SECRET}"

if [ $? -eq 0 ]; then
  echo "✓ Successfully stored Guacamole OAuth credentials in Vault"
  echo ""
  echo "The External Secret Operator will now sync these credentials to the guacamole namespace."
  echo "You can verify with:"
  echo "  kubectl get externalsecret -n guacamole guacamole-oauth"
  echo "  kubectl get secret -n guacamole guacamole-oauth-secret"
else
  echo "ERROR: Failed to store credentials in Vault"
  exit 1
fi

echo ""
echo "=== Guacamole OAuth Sync Complete ==="