# Apply missing company_workflow_lobster columns on tenant DB(s)
param(
    [string]$TenantIds = "33"
)

$ErrorActionPreference = "Stop"
$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
$mysqlJar = "$env:USERPROFILE\.m2\repository\com\mysql\mysql-connector-j\8.0.33\mysql-connector-j-8.0.33.jar"
$javaHome = if ($env:JAVA_HOME) { $env:JAVA_HOME } else { "D:\jdk-17.0.2" }

if (-not (Test-Path $mysqlJar)) {
    Write-Host "MySQL JDBC jar missing: $mysqlJar" -ForegroundColor Red
    exit 1
}

Push-Location $scriptDir
try {
    & "$javaHome\bin\javac.exe" -encoding UTF-8 ApplyLobsterWorkflowLobsterColumnsMigration.java
    if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
    $ids = @($TenantIds -split '[,\s]+' | Where-Object { $_ -match '^\d+$' })
    foreach ($id in $ids) {
        & "$javaHome\bin\java.exe" -cp "$mysqlJar;." ApplyLobsterWorkflowLobsterColumnsMigration $id
        if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
    }
} finally {
    Pop-Location
}