deploy.ps1

#!/usr/bin/env pwsh

param(
    [Parameter(Mandatory=$true)]
    [string]$Account,

    [Parameter(Mandatory=$true)]
    [string]$Region,

    [string]$Environment = "dev",

    [string]$Profile = "default",

    [string]$ResourcePrefix = "as2",

    [Alias("y")]
    [switch]$AutoApprove,

    [Alias("q")]
    [switch]$Quiet
)

# Set error action preference
$ErrorActionPreference = "Stop"

# Function to check if command exists
function Test-Command {
    param($Command)
    $null = Get-Command $Command -ErrorAction SilentlyContinue
    return $?
}

# Function to write output (respects quiet mode)
function Write-Step {
    param($Message, $Color = "White")
    if (-not $Quiet) {
        Write-Host $Message -ForegroundColor $Color
    }
}

Write-Host "=== AS2 Application Deployment Script ===" -ForegroundColor Green
Write-Host "Account: $Account" -ForegroundColor Yellow
Write-Host "Region: $Region" -ForegroundColor Yellow
Write-Host "Environment: $Environment" -ForegroundColor Yellow
Write-Host "Profile: $Profile" -ForegroundColor Yellow
Write-Host "Resource Prefix: $ResourcePrefix" -ForegroundColor Yellow
Write-Host ""

# Confirmation prompt (unless auto-approve)
if (-not $AutoApprove) {
    $response = Read-Host "Proceed with deployment? [Y/n]"
    if ($response -eq "n" -or $response -eq "N" -or $response -eq "no" -or $response -eq "No") {
        Write-Host "Deployment cancelled" -ForegroundColor Yellow
        exit 0
    }
}

# Track deployment progress
$TotalSteps = 4
$CurrentStep = 0

function Step-Start {
    param($Message)
    $script:CurrentStep++
    Write-Host "`n[$script:CurrentStep/$TotalSteps] $Message" -ForegroundColor Blue
}

function Step-Done {
    param($Message)
    Write-Host "    ✓ $Message" -ForegroundColor Green
}

function Step-Fail {
    param($Message)
    Write-Host "    ✗ $Message" -ForegroundColor Red
}

# Check prerequisites
Step-Start "Checking prerequisites..."

if (-not (Test-Command "aws")) {
    Write-Error "AWS CLI is not installed or not in PATH"
    exit 1
}

if (-not (Test-Command "cdk")) {
    Write-Error "AWS CDK is not installed or not in PATH"
    exit 1
}

if (-not (Test-Command "mvn")) {
    Write-Error "Maven is not installed or not in PATH"
    exit 1
}

if (-not (Test-Command "npm")) {
    Write-Error "Node.js/npm is not installed or not in PATH"
    exit 1
}

Step-Done "All prerequisites installed"

# Set AWS profile
$env:AWS_PROFILE = $Profile

# Verify AWS credentials
Write-Step "Verifying AWS credentials..." "Blue"
try {
    $identity = aws sts get-caller-identity --output json | ConvertFrom-Json
    Step-Done "AWS credentials verified for account: $($identity.Account)"
} catch {
    Write-Error "Failed to verify AWS credentials. Please configure AWS CLI with 'aws configure'"
    exit 1
}

# Navigate to project root
$projectRoot = $PSScriptRoot
Set-Location $projectRoot

Write-Step "Project root: $projectRoot" "Yellow"

# Generate date suffix for unique bucket names
$DateSuffix = Get-Date -Format "yyyyMMdd"

# Generate base resource names
$FrontendBucket = "$ResourcePrefix-frontend-bucket-$DateSuffix"
$SendBucket = "$ResourcePrefix-send-bucket-$DateSuffix"
$ReceiveBucket = "$ResourcePrefix-receive-bucket-$DateSuffix"
$SqsQueue = "$ResourcePrefix-sqs-queue-$DateSuffix"

# Display actual bucket names (with suffix for non-prod)
if ($Environment -eq "prod") {
    $DisplayFrontendBucket = $FrontendBucket
    $DisplaySendBucket = $SendBucket
    $DisplayReceiveBucket = $ReceiveBucket
    $DisplaySqsQueue = $SqsQueue
} else {
    $DisplayFrontendBucket = "$FrontendBucket-$Environment-$Region"
    $DisplaySendBucket = "$SendBucket-$Environment-$Region"
    $DisplayReceiveBucket = "$ReceiveBucket-$Environment-$Region"
    $DisplaySqsQueue = "$SqsQueue-$Environment-$Region"
}

# Stack names with env + region suffix for multi-instance support
$StackSuffix = "-$Environment-$Region"
$S3EventStack = "S3EventLambdaStack$StackSuffix"
$SqsStack = "SQSLambdaStack$StackSuffix"
$AS2ReceiverStack = "AS2ReceiverLambdaStack$StackSuffix"
$ManageStack = "ManageLambdaStack$StackSuffix"
$FrontendStack = "FrontendDeploymentStack$StackSuffix"

Write-Host "Using resource names:" -ForegroundColor Blue
Write-Host "  Frontend Bucket: $DisplayFrontendBucket" -ForegroundColor White
Write-Host "  Send Bucket: $DisplaySendBucket" -ForegroundColor White
Write-Host "  Receive Bucket: $DisplayReceiveBucket" -ForegroundColor White
Write-Host "  SQS Queue: $DisplaySqsQueue" -ForegroundColor White
Write-Host ""

# Step 2: Build backend
Step-Start "Building backend..."
Set-Location "backend"

try {
    if ($Quiet) {
        mvn clean package -DskipTests=true -q
    } else {
        mvn clean package -DskipTests=true
    }
    if ($LASTEXITCODE -ne 0) {
        throw "Maven build failed"
    }
    Step-Done "Backend JAR built"
} catch {
    Step-Fail "Backend build failed"
    exit 1
}

Set-Location $projectRoot

# Step 3: Deploy backend infrastructure
Step-Start "Deploying backend infrastructure..."
Set-Location "cdk"

try {
    if ($Quiet) {
        mvn compile -q
    } else {
        mvn compile
    }
    if ($LASTEXITCODE -ne 0) {
        throw "CDK dependencies installation failed"
    }
    Step-Done "CDK dependencies installed"
} catch {
    Step-Fail "CDK dependencies installation failed"
    exit 1
}

Write-Step "Bootstrapping CDK (if needed)..." "Blue"
try {
    cdk bootstrap aws://$Account/$Region --profile $Profile 2>$null
    Step-Done "CDK bootstrapped"
} catch {
    Write-Host "    ⚠ CDK bootstrap may have failed, continuing..." -ForegroundColor Yellow
}

# Deploy S3EventLambdaStack and SQSLambdaStack first
try {
    cdk deploy $S3EventStack $SqsStack --require-approval never --profile $Profile --context account=$Account --context region=$Region --context env=$Environment --context frontendBucketName="$FrontendBucket" --context sendBucketName="$SendBucket" --context receiveBucketName="$ReceiveBucket" --context sqsQueueName="$SqsQueue"     if ($LASTEXITCODE -ne 0) {
        throw "S3 and SQS stacks deployment failed"
    }
    Step-Done "S3 and SQS stacks deployed"
} catch {
    Step-Fail "Backend infrastructure deployment failed"
    exit 1
}

# Deploy AS2ReceiverLambdaStack
try {
    cdk deploy $AS2ReceiverStack --require-approval never --profile $Profile --context account=$Account --context region=$Region --context env=$Environment --context frontendBucketName="$FrontendBucket" --context sendBucketName="$SendBucket" --context receiveBucketName="$ReceiveBucket" --context sqsQueueName="$SqsQueue"     if ($LASTEXITCODE -ne 0) {
        throw "AS2ReceiverLambdaStack deployment failed"
    }
    Step-Done "AS2 Receiver stack deployed"
} catch {
    Step-Fail "AS2ReceiverLambdaStack deployment failed"
    exit 1
}

# Deploy ManageLambdaStack
try {
    cdk deploy $ManageStack --require-approval never --profile $Profile --context account=$Account --context region=$Region --context env=$Environment --context frontendBucketName="$FrontendBucket" --context sendBucketName="$SendBucket" --context receiveBucketName="$ReceiveBucket" --context sqsQueueName="$SqsQueue"     if ($LASTEXITCODE -ne 0) {
        throw "ManageLambdaStack deployment failed"
    }
    Step-Done "Management API stack deployed"
} catch {
    Step-Fail "ManageLambdaStack deployment failed"
    exit 1
}

Set-Location $projectRoot

# Step 4: Deploy frontend
Step-Start "Deploying frontend..."
Set-Location "cdk"

try {
    cdk deploy $FrontendStack --require-approval never --profile $Profile --context account=$Account --context region=$Region --context env=$Environment --context frontendBucketName="$FrontendBucket" --context sendBucketName="$SendBucket" --context receiveBucketName="$ReceiveBucket" --context sqsQueueName="$SqsQueue"     if ($LASTEXITCODE -ne 0) {
        throw "Frontend deployment failed"
    }
    Step-Done "Frontend and CloudFront deployed"
} catch {
    Step-Fail "Frontend deployment failed"
    exit 1
}

Set-Location $projectRoot

# Display deployment information
Write-Host ""
Write-Host "════════════════════════════════════════════════════════════" -ForegroundColor Green
Write-Host "              DEPLOYMENT COMPLETED SUCCESSFULLY" -ForegroundColor Green
Write-Host "════════════════════════════════════════════════════════════" -ForegroundColor Green

Write-Host ""

try {
    $manageLambdaOutputs = aws cloudformation describe-stacks --stack-name $ManageStack --region $Region --profile $Profile --query "Stacks[0].Outputs" --output json | ConvertFrom-Json
    $frontendOutputs = aws cloudformation describe-stacks --stack-name $FrontendStack --region $Region --profile $Profile --query "Stacks[0].Outputs" --output json | ConvertFrom-Json

    Write-Host "`n📋 Deployment Information:" -ForegroundColor Cyan
    Write-Host "═══════════════════════════" -ForegroundColor Cyan

    $ApiUrl = ""
    $UserPoolId = ""
    $ClientId = ""
    $WebsiteUrl = ""
    $CloudFrontUrl = ""
    $AS2ReceiverUrl = ""

    foreach ($output in $manageLambdaOutputs) {
        if ($output.OutputKey -eq "ManageAPIOutput") {
            $ApiUrl = $output.OutputValue
            Write-Host "🔗 Backend API URL: $ApiUrl" -ForegroundColor White
        }
        if ($output.OutputKey -eq "CognitoUserPoolIdOutput") {
            $UserPoolId = $output.OutputValue
            Write-Host "🔐 Cognito User Pool ID: $UserPoolId" -ForegroundColor White
        }
        if ($output.OutputKey -eq "CognitoUserPoolClientIdOutput") {
            $ClientId = $output.OutputValue
            Write-Host "🔑 Cognito Client ID: $ClientId" -ForegroundColor White
        }
    }

    foreach ($output in $frontendOutputs) {
        if ($output.OutputKey -eq "WebsiteURL") {
            $WebsiteUrl = $output.OutputValue
            Write-Host "🌐 Frontend Website URL: $WebsiteUrl" -ForegroundColor White
        }
        if ($output.OutputKey -eq "CloudFrontURL") {
            $CloudFrontUrl = $output.OutputValue
            Write-Host "⚡ CloudFront Distribution URL: $CloudFrontUrl" -ForegroundColor White
        }
    }

    # Get AS2 Receiver URL
    try {
        $as2ReceiverOutputs = aws cloudformation describe-stacks --stack-name $AS2ReceiverStack --region $Region --profile $Profile --query "Stacks[0].Outputs" --output json | ConvertFrom-Json
        foreach ($output in $as2ReceiverOutputs) {
            if ($output.OutputKey -eq "AS2ReceiverApiUrl") {
                $AS2ReceiverUrl = $output.OutputValue
                Write-Host "📨 AS2 Receiver API URL: $AS2ReceiverUrl" -ForegroundColor White
            }
        }
    } catch {
        # AS2 Receiver stack output retrieval failed, continue
    }

    Write-Host "`n✅ Deployment completed successfully!" -ForegroundColor Green
    Write-Host "Your AS2 application is now ready to use." -ForegroundColor Green

    Write-Host "`n📝 Next Steps:" -ForegroundColor Cyan
    Write-Host "1. Access your application at: $CloudFrontUrl" -ForegroundColor White
    Write-Host "2. Create a user account by clicking 'Sign Up' on the login page" -ForegroundColor White
    Write-Host "3. Configure your AS2 partners and certificates" -ForegroundColor White

} catch {
    Write-Warning "Could not retrieve deployment outputs, but deployment appears successful"
    Write-Host "✅ Deployment completed!" -ForegroundColor Green
}