wizard.ps1

<#
.SYNOPSIS
  Interactive wizard — deploys the Global Azure Kit from zero to dashboard.

.DESCRIPTION
  Guides you step-by-step through the entire setup:
    1. Azure login
    2. Terraform backend (storage account for state)
    3. Terraform deploy (12 Azure resources)
    4. Firmware configuration (WiFi + IoT Hub credentials)
    5. Monitor dashboard (.env + launch)

  Every step shows expected output and asks for confirmation.
  Run from the repository root folder.

.EXAMPLE
  .\wizard.ps1
#>

$ErrorActionPreference = "Stop"
$repoRoot = $PSScriptRoot

# ═══════════════════════════════════════════════════════════════════════════
# Helper functions
# ═══════════════════════════════════════════════════════════════════════════

function Write-Step {
    param([string]$Number, [string]$Title)
    Write-Host ""
    Write-Host "═══════════════════════════════════════════════════════════════" -ForegroundColor Cyan
    Write-Host "  STEP $Number — $Title" -ForegroundColor Cyan
    Write-Host "═══════════════════════════════════════════════════════════════" -ForegroundColor Cyan
    Write-Host ""
}

function Write-OK    { param([string]$Msg) Write-Host "  OK  $Msg" -ForegroundColor Green }
function Write-Fail  { param([string]$Msg) Write-Host "  FAIL  $Msg" -ForegroundColor Red }
function Write-Info  { param([string]$Msg) Write-Host "  $Msg" -ForegroundColor Gray }
function Write-Warn  { param([string]$Msg) Write-Host "  WARNING  $Msg" -ForegroundColor Yellow }

function Request-Continue {
    param([string]$Prompt = "Continue?")
    Write-Host ""
    $answer = Read-Host "  $Prompt (Y/n)"
    if ($answer -and $answer -notmatch '^[yYsS]') {
        Write-Host ""
        Write-Host "  Wizard stopped. You can re-run it anytime with: .\wizard.ps1" -ForegroundColor Yellow
        Write-Host ""
        exit 0
    }
}

function Read-UserInput {
    param([string]$Prompt, [string]$Default)
    if ($Default) {
        $val = Read-Host "  $Prompt [$Default]"
        if (-not $val) { $val = $Default }
    } else {
        do {
            $val = Read-Host "  $Prompt"
            if (-not $val) { Write-Warn "This field is required." }
        } while (-not $val)
    }
    return $val
}

# ═══════════════════════════════════════════════════════════════════════════
# Welcome
# ═══════════════════════════════════════════════════════════════════════════

Clear-Host
Write-Host ""
Write-Host "  ================================================================" -ForegroundColor Magenta
Write-Host "     Global Azure Kit — Interactive Wizard" -ForegroundColor Magenta
Write-Host "     Global Azure Torino 2026 — by ExtraNet Srl" -ForegroundColor Magenta
Write-Host "  ================================================================" -ForegroundColor Magenta
Write-Host ""
Write-Host "  This wizard will guide you through the complete setup:" -ForegroundColor White
Write-Host "    1. Azure login" -ForegroundColor White
Write-Host "    2. Terraform backend (state storage)" -ForegroundColor White
Write-Host "    3. Terraform deploy (12 Azure resources)" -ForegroundColor White
Write-Host "    4. Firmware configuration (WiFi + credentials)" -ForegroundColor White
Write-Host "    5. Monitor dashboard (launch)" -ForegroundColor White
Write-Host ""
Write-Host "  You can stop at any point and re-run the wizard later." -ForegroundColor Gray
Write-Host ""

# ═══════════════════════════════════════════════════════════════════════════
# Prerequisites check
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "0" "Checking prerequisites"

$prereqOk = $true

# Terraform
$tfVer = $null
try { $tfVer = (terraform version -json 2>$null | ConvertFrom-Json).terraform_version } catch {}
if ($tfVer) {
    Write-OK "Terraform $tfVer"
} else {
    Write-Fail "Terraform not found — install from https://terraform.io/downloads"
    $prereqOk = $false
}

# Azure CLI
$azVer = $null
try { $azVer = (az version 2>$null | ConvertFrom-Json).'azure-cli' } catch {}
if ($azVer) {
    Write-OK "Azure CLI $azVer"
} else {
    Write-Fail "Azure CLI not found — install from https://aka.ms/installazurecliwindows"
    $prereqOk = $false
}

# azure-iot extension
if ($azVer) {
    $iotExt = az extension list 2>$null | ConvertFrom-Json | Where-Object { $_.name -eq 'azure-iot' }
    if ($iotExt) {
        Write-OK "Azure IoT extension $($iotExt.version)"
    } else {
        Write-Info "Azure IoT extension not found — installing..."
        az extension add --name azure-iot --output none 2>$null
        if ($LASTEXITCODE -eq 0) {
            Write-OK "Azure IoT extension installed"
        } else {
            Write-Fail "Could not install azure-iot extension"
            $prereqOk = $false
        }
    }
}

# Python
$pyVer = $null
try { $pyVer = (python --version 2>&1).ToString().Replace("Python ", "") } catch {}
if ($pyVer) {
    Write-OK "Python $pyVer"
} else {
    Write-Fail "Python not found — install from https://python.org"
    $prereqOk = $false
}

if (-not $prereqOk) {
    Write-Host ""
    Write-Fail "Some prerequisites are missing. Install them and re-run the wizard."
    Write-Host ""
    exit 1
}

Write-Host ""
Write-OK "All prerequisites found!"

# ═══════════════════════════════════════════════════════════════════════════
# STEP 1 — Azure login
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "1" "Azure login"

# Check if already logged in
$account = $null
try { $account = az account show 2>$null | ConvertFrom-Json } catch {}

if ($account) {
    $tenantName = $account.tenantDisplayName
    if (-not $tenantName) { $tenantName = $account.tenantId }
    Write-OK "Already logged in"
    Write-Info "Subscription: $($account.name)"
    Write-Info "Tenant:       $tenantName"
    Write-Info "ID:           $($account.id)"
    $answer = Read-Host "`n  Use this account? (Y/n)"
    if ($answer -and $answer -notmatch '^[yYsS]') {
        Write-Info "Logging out of current session..."
        az logout 2>$null
        Write-Info "Opening browser for Azure login..."
        az login --output none 2>$null
        if ($LASTEXITCODE -ne 0) {
            Write-Warn "Browser login failed, trying device code..."
            az login --use-device-code --output none
        }
        $account = az account show | ConvertFrom-Json
    }
} else {
    Write-Info "Opening browser for Azure login..."
    az login --output none 2>$null
    if ($LASTEXITCODE -ne 0) {
        Write-Warn "Browser login failed, trying device code..."
        az login --use-device-code --output none
    }
    $account = az account show | ConvertFrom-Json
}

if (-not $account) {
    Write-Fail "Login failed. Run 'az login' manually and re-run the wizard."
    exit 1
}

$subscriptionId = $account.id
Write-Host ""
Write-OK "Logged in!"
Write-Info "Subscription: $($account.name)"
Write-Info "ID:           $subscriptionId"

# Preflight: IoT Hub F1 limit (1 Free tier per subscription).
Write-Host ""
Write-Info "Checking Free tier IoT Hub quota..."
$existingF1 = az iot hub list --query "[?sku.name=='F1'].{name:name,rg:resourceGroup}" 2>$null | ConvertFrom-Json
if ($existingF1) {
    Write-Warn "Existing F1 (Free tier) IoT Hub found in this subscription:"
    foreach ($h in $existingF1) {
        Write-Info "  - $($h.name)  (resource group: $($h.rg))"
    }
    Write-Warn "Azure allows only ONE F1 IoT Hub per subscription."
    Write-Info "Options:"
    Write-Info "  (a) Delete the existing F1 hub"
    Write-Info "  (b) Edit terraform.auto.tfvars later and set iothub_sku = ""S1"""
    Write-Info "      (~EUR 21/month pro-rata — pennies for a workshop)"
    Request-Continue "Continue anyway?"
} else {
    Write-OK "No existing F1 IoT Hub — safe to create one"
}

# ═══════════════════════════════════════════════════════════════════════════
# STEP 2 — Terraform backend
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "2" "Terraform backend (state storage)"

$backendFile = Join-Path $repoRoot "terraform\backend.tfvars"

if (Test-Path $backendFile) {
    Write-OK "backend.tfvars already exists — skipping backend setup"
    Write-Info "File: $backendFile"
    Request-Continue "Continue to Terraform deploy?"
} else {
    Write-Info "This creates an Azure Storage Account to store Terraform state."
    Write-Info "It runs the script: terraform\scripts\setup-tfstate-backend.ps1"
    Request-Continue "Create the backend storage?"

    & "$repoRoot\terraform\scripts\setup-tfstate-backend.ps1"

    if (-not (Test-Path $backendFile)) {
        Write-Fail "backend.tfvars was not created. Check errors above."
        exit 1
    }
    Write-OK "Backend ready!"
}

# ═══════════════════════════════════════════════════════════════════════════
# STEP 3 — Terraform variables
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "3" "Terraform variables"

$tfvarsFile = Join-Path $repoRoot "terraform\terraform.auto.tfvars"

if (Test-Path $tfvarsFile) {
    Write-OK "terraform.auto.tfvars already exists"
    Write-Info "File: $tfvarsFile"
    # Read and show current subscription_id
    $currentContent = Get-Content $tfvarsFile -Raw
    if ($currentContent -match 'subscription_id\s*=\s*"([^"]*)"') {
        $currentSubId = $Matches[1]
        Write-Info "Current subscription_id: $currentSubId"
        if ($currentSubId -eq $subscriptionId) {
            Write-OK "Matches your current subscription"
        } elseif ($currentSubId -match '^(ENTER-YOUR-SUBSCRIPTION-ID-HERE)?$') {
            Write-Warn "subscription_id is still the placeholder!"
            Write-Info "Updating it to: $subscriptionId"
            $currentContent = $currentContent -replace 'subscription_id\s*=\s*"[^"]*"', "subscription_id = `"$subscriptionId`""
            $currentContent | Set-Content -Path $tfvarsFile -Encoding UTF8 -NoNewline
            Write-OK "Updated!"
        } else {
            Write-Warn "subscription_id is different from your current login"
            $answer = Read-Host "  Update to $subscriptionId ? (Y/n)"
            if (-not $answer -or $answer -match '^[yYsS]') {
                $currentContent = $currentContent -replace 'subscription_id\s*=\s*"[^"]*"', "subscription_id = `"$subscriptionId`""
                $currentContent | Set-Content -Path $tfvarsFile -Encoding UTF8 -NoNewline
                Write-OK "Updated!"
            }
        }
    }
} else {
    Write-Info "Creating terraform.auto.tfvars from the example template..."

    $exampleFile = Join-Path $repoRoot "terraform\terraform.auto.tfvars.example"
    $content = Get-Content $exampleFile -Raw
    $content = $content -replace 'subscription_id\s*=\s*"[^"]*"', "subscription_id = `"$subscriptionId`""
    $content | Set-Content -Path $tfvarsFile -Encoding UTF8 -NoNewline

    Write-OK "Created terraform.auto.tfvars"
    Write-Info "subscription_id = $subscriptionId"
    Write-Info ""
    Write-Info "Other settings (defaults work fine for most people):"
    Write-Info "  project_prefix = globalazure-kit"
    Write-Info "  location       = italynorth"
    Write-Info "  iothub_location = westeurope"
    Write-Info "  device_count   = 1"

    $answer = Read-Host "`n  Want to change any of these? (y/N)"
    if ($answer -match '^[yYsS]') {
        Write-Info "Opening the file in Notepad..."
        Start-Process notepad $tfvarsFile -Wait
        Write-OK "Saved!"
    }
}

# ═══════════════════════════════════════════════════════════════════════════
# STEP 4 — Terraform init + apply
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "4" "Terraform deploy"

Push-Location "$repoRoot\terraform"
try {
    # Init
    Write-Info "Running: terraform init ..."
    terraform init "-backend-config=backend.tfvars"
    if ($LASTEXITCODE -ne 0) {
        Write-Fail "terraform init failed. Check errors above."
        exit 1
    }
    Write-OK "Terraform initialized"

    # Plan
    Write-Host ""
    Write-Info "Running: terraform plan ..."
    terraform plan -out="tfplan"
    if ($LASTEXITCODE -ne 0) {
        Write-Fail "terraform plan failed. Check errors above."
        exit 1
    }
    Write-OK "Plan ready — review the changes above"

    Request-Continue "Deploy these resources to Azure? (type Y to confirm)"

    # Apply
    Write-Info "Running: terraform apply ..."
    terraform apply "tfplan"
    if ($LASTEXITCODE -ne 0) {
        Write-Fail "terraform apply failed. Check errors above."
        exit 1
    }

    # Clean up plan file
    Remove-Item "tfplan" -ErrorAction SilentlyContinue

    Write-Host ""
    Write-OK "Azure infrastructure deployed!"
    Write-OK "12 resources created"
} finally {
    Pop-Location
}

# ═══════════════════════════════════════════════════════════════════════════
# STEP 5 — Firmware configuration
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "5" "Firmware configuration"

Write-Info "This generates the Arduino sketch with your WiFi and IoT Hub credentials."
Write-Host ""

$wifiSsid     = Read-UserInput "WiFi network name (SSID)"
$wifiPassword  = Read-Host "WiFi password" -AsSecureString

Write-Host ""
Write-Info "Sensor type:"
Write-Info "  1) BMP280 — temperature + pressure"
Write-Info "  2) BME280 — temperature + pressure + humidity"
Write-Info "  3) auto  — auto-detect at runtime (requires both Arduino libraries)"
$sensorChoice = Read-UserInput "Choose sensor [1/2/3]"

$sensorType = switch ($sensorChoice) {
    "1"      { "BMP280" }
    "2"      { "BME280" }
    "3"      { "auto" }
    default  { Write-Warn "Invalid choice, please enter 1, 2, or 3."; exit 1 }
}

Write-Host ""
$ledChoice = Read-UserInput "Enable LED control via dashboard? [y/n]" "y"
$enableLed = $ledChoice -match '^[yY]'

Write-Host ""
Write-Info "Generating sketch..."

$extraArgs = @{}
if ($enableLed) { $extraArgs['EnableLed'] = $true }

& "$repoRoot\firmware\configure-sketch.ps1" `
    -WifiSsid $wifiSsid `
    -WifiPassword $wifiPassword `
    -SensorType $sensorType @extraArgs

$sketchPath = Join-Path $repoRoot "firmware\output\GlobalAzureKit\GlobalAzureKit.ino"
if (Test-Path $sketchPath) {
    Write-OK "Sketch ready!"
    Write-Info "File: $sketchPath"
    Write-Host ""
    Write-Info "Next: open this file in Arduino IDE, select your board, and upload."
    Write-Info "Board settings:"
    Write-Info "  Board:            ESP32C3 Dev Module"
    Write-Info "  USB CDC On Boot:  Enabled  (critical!)"
    Write-Info "  Upload Speed:     921600"
    Write-Info "  Port:             (your ESP32 COM port)"
} else {
    Write-Fail "Sketch was not generated. Check errors above."
    exit 1
}

# ═══════════════════════════════════════════════════════════════════════════
# STEP 6 — Monitor dashboard
# ═══════════════════════════════════════════════════════════════════════════

Write-Step "6" "Monitor dashboard"

# Generate .env
Write-Info "Generating .env for the monitor..."

Push-Location "$repoRoot\terraform"
try {
    $dotEnv = terraform output -raw dot_env_content
    if ($LASTEXITCODE -ne 0) { throw "Could not read Terraform output" }
} finally {
    Pop-Location
}

$envFile = Join-Path $repoRoot "monitor\.env"
$dotEnv | Set-Content -Path $envFile -Encoding UTF8
Write-OK ".env created"
Write-Info "File: $envFile"

# Python venv + dependencies
Write-Host ""
Write-Info "Setting up Python environment..."

Push-Location "$repoRoot\monitor"
try {
    if (-not (Test-Path ".venv")) {
        python -m venv .venv
        Write-OK "Virtual environment created"
    } else {
        Write-OK "Virtual environment already exists"
    }

    & .\.venv\Scripts\Activate.ps1

    Write-Info "Installing dependencies (this may take a minute)..."
    pip install -r requirements.txt --quiet
    if ($LASTEXITCODE -ne 0) {
        Write-Fail "pip install failed. Check errors above."
        exit 1
    }
    Write-OK "Dependencies installed"

    # Launch
    Write-Host ""
    Write-Host "  ================================================================" -ForegroundColor Green
    Write-Host "     SETUP COMPLETE!" -ForegroundColor Green
    Write-Host "  ================================================================" -ForegroundColor Green
    Write-Host ""
    Write-Info "What you've done:"
    Write-Info "  1. Azure login"
    Write-Info "  2. Terraform backend (state in Azure Storage)"
    Write-Info "  3. 12 Azure resources deployed"
    Write-Info "  4. Firmware sketch generated (ready to flash)"
    Write-Info "  5. Monitor dashboard ready to launch"
    Write-Host ""
    Write-Info "Next steps:"
    Write-Info "  - Flash the firmware to your ESP32-C3 via Arduino IDE"
    Write-Info "  - Upload the sketch from: firmware\output\GlobalAzureKit\"
    Write-Host ""

    Request-Continue "Launch the monitor dashboard now?"

    Write-Info "Starting dashboard on http://127.0.0.1:5000 ..."
    Write-Info "Press Ctrl+C to stop."
    Write-Host ""

    python app.py
} finally {
    Pop-Location
}