Setup-Environment.ps1

param
(
    [Parameter(Mandatory = $true)]
    [string]$DeploymentName,
    [Parameter(Mandatory = $true)]
    [string]$Location,
    [switch]$IsLocal,
    [switch]$SkipInfrastructure,
    [switch]$WhatIf
)

Write-Host "Starting environment setup..."

if ($SkipInfrastructure) {
    Write-Host "Skipping infrastructure deployment..."
    $InfrastructureOutputs = Get-Content -Path './infra/InfrastructureOutputs.json' -Raw | ConvertFrom-Json
}
else {
    Write-Host "Deploying infrastructure..."
    $InfrastructureOutputs = (./infra/Deploy-Infrastructure.ps1 `
            -DeploymentName $DeploymentName `
            -Location $Location `
            -WhatIf:$WhatIf)

    if ($WhatIf) {
        Write-Host "WhatIf mode is enabled. Exiting without deploying."
        exit 0
    }
}

if (-not $InfrastructureOutputs) {
    Write-Error "Failed to deploy infrastructure."
    exit 1
}

$AzureAIServicesEndpoint = $InfrastructureOutputs.environmentInfo.value.azureAIServicesEndpoint
$AzureOpenAIEndpoint = $InfrastructureOutputs.environmentInfo.value.azureOpenAIEndpoint
$AzureOpenAIChatDeployment = $InfrastructureOutputs.environmentInfo.value.azureOpenAIChatDeployment
$AzureStorageAccount = $InfrastructureOutputs.environmentInfo.value.azureStorageAccount

Write-Host "Updating ./src/AIDocumentPipeline/local.settings.json..."

$LocalSettingsPath = './src/AIDocumentPipeline/local.settings.json'
$LocalSettings = Get-Content -Path $LocalSettingsPath -Raw | ConvertFrom-Json
$LocalSettings.Values.AZURE_AISERVICES_ENDPOINT = $AzureAIServicesEndpoint
$LocalSettings.Values.AZURE_OPENAI_ENDPOINT = $AzureOpenAIEndpoint
$LocalSettings.Values.AZURE_OPENAI_CHAT_DEPLOYMENT = $AzureOpenAIChatDeployment
$LocalSettings.Values.AZURE_STORAGE_ACCOUNT = $AzureStorageAccount
$LocalSettings | ConvertTo-Json | Out-File -FilePath $LocalSettingsPath -Encoding utf8

if ($IsLocal) {
    Write-Host "Starting local environment..."

    docker-compose up
}
else {
    Write-Host "Deploying AI Document Pipeline app to Azure..."

    $AppOutputs = (./infra/apps/AIDocumentPipeline/Deploy-App.ps1 `
            -InfrastructureOutputsPath './infra/InfrastructureOutputs.json')
}