#!/bin/bash
set -euo pipefail

ARTIFACT_DIR="/opt/api-artifacts"
FRONTEND_ARTIFACT_DIR="/opt/frontend-artifacts"
FRONTEND_WEB_DIR="/var/www/app"
CURRENT_LINK="$ARTIFACT_DIR/current"
BLUE_LINK="$ARTIFACT_DIR/blue"
GREEN_LINK="$ARTIFACT_DIR/green"
SERVICE_NAME="api-server.service"

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'

log_info() { echo -e "${GREEN}[INFO]${NC} $1"; }
log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }

# Get commit SHA if provided
COMMIT_SHA="${1:-}"

if [ -L "$CURRENT_LINK" ]; then
    CURRENT_TARGET=$(readlink "$CURRENT_LINK")
    if [ "$CURRENT_TARGET" = "blue" ]; then
        NEW_TARGET="green"
    else
        NEW_TARGET="blue"
    fi
else
    NEW_TARGET="blue"
fi

NEW_TARGET_LINK="$ARTIFACT_DIR/$NEW_TARGET"

if [ ! -L "$NEW_TARGET_LINK" ]; then
    log_error "Target deployment not found: $NEW_TARGET_LINK"
    log_error "Run deploy.sh first"
    exit 1
fi

NEW_BUILD=$(readlink "$NEW_TARGET_LINK")
log_info "Promoting $NEW_TARGET to production"
log_info "New backend build: $NEW_BUILD"

# Check for frontend staging if commit SHA provided
if [ -n "$COMMIT_SHA" ] && [ -d "$FRONTEND_ARTIFACT_DIR/staging/$COMMIT_SHA" ]; then
    log_info "Frontend build found: $COMMIT_SHA"
    DEPLOY_FRONTEND=true
else
    log_warn "No frontend build found for this commit"
    DEPLOY_FRONTEND=false
fi

read -p "Continue with promotion? (yes/no): " -r
if [[ ! $REPLY =~ ^[Yy][Ee][Ss]$ ]]; then
    log_warn "Promotion cancelled"
    exit 0
fi

log_info "Switching current symlink to $NEW_TARGET"
ln -sfn "$NEW_TARGET" "$CURRENT_LINK"

# Deploy frontend if available
if [ "$DEPLOY_FRONTEND" = true ]; then
    log_info "Deploying frontend to $FRONTEND_WEB_DIR..."
    
    # Backup current frontend
    if [ -d "$FRONTEND_WEB_DIR" ] && [ "$(ls -A $FRONTEND_WEB_DIR)" ]; then
        BACKUP_DIR="$FRONTEND_ARTIFACT_DIR/backups/$(date +%Y%m%d_%H%M%S)"
        mkdir -p "$BACKUP_DIR"
        cp -r "$FRONTbackend deployment: $NEW_TARGET ($NEW_BUILD)"
    if [ "$DEPLOY_FRONTEND" = true ]; then
        log_info "Active frontend deployment: $COMMIT_SHA"
        log_info "Frontend URL: https://app.nitrokite.com"
    fi
        log_info "Backed up current frontend to $BACKUP_DIR"
    fi
    
    # Deploy new frontend
    rm -rf "$FRONTEND_WEB_DIR"/*
    cp -r "$FRONTEND_ARTIFACT_DIR/staging/$COMMIT_SHA"/* "$FRONTEND_WEB_DIR/"
    chown -R www-data:www-data "$FRONTEND_WEB_DIR"
    log_info "Frontend deployed successfully"
fi

log_info "Reloading systemd service..."
systemctl reload-or-restart "$SERVICE_NAME"

sleep 5

log_info "Running health check..."
if "$ARTIFACT_DIR/scripts/health-check.sh"; then
    log_info "Promotion successful!"
    log_info "Active deployment: $NEW_TARGET ($NEW_BUILD)"
else
    log_error "Health check failed after promotion!"
    log_warn "Run rollback.sh to revert"
    exit 1
fi