# Factory CLI Windows Installer
# Usage: 
# curl.exe -b 'session=cookie_value' -fsSL https://app.factory.ai/cli/windows -o install.ps1
# powershell -ExecutionPolicy Bypass -File install.ps1
# del install.ps1

$ErrorActionPreference = 'Stop'

# Function to write colored output
function Write-Info($message) {
    Write-Host "$message" -ForegroundColor Green
}

function Write-Warning($message) {
    Write-Host "$message" -ForegroundColor Yellow
}

function Write-Error($message) {
    Write-Host "$message" -ForegroundColor Red
    exit 1
}

# Detect architecture
$arch = $env:PROCESSOR_ARCHITECTURE
if ($arch -eq "AMD64" -or $arch -eq "X64") {
    $architecture = "x64"
} elseif ($arch -eq "ARM64") {
    $architecture = "arm64"
} else {
    Write-Error "Unsupported architecture: $arch"
}

$version = "0.22.3"
$baseUrl = "https://downloads.factory.ai"
$binaryName = "droid.exe"
$rgBinaryName = "rg.exe"
$url = "$baseUrl/factory-cli/releases/$version/windows/$architecture/$binaryName"
$shaUrl = "$baseUrl/factory-cli/releases/$version/windows/$architecture/$binaryName.sha256"
$rgUrl = "$baseUrl/ripgrep/windows/$architecture/$rgBinaryName"
$rgShaUrl = "$baseUrl/ripgrep/windows/$architecture/$rgBinaryName.sha256"

Write-Info "Downloading Factory CLI v$version for Windows-$architecture"

# Create temporary directory
$tempDir = New-TemporaryFile | ForEach-Object { Remove-Item $_; New-Item -ItemType Directory -Path $_ }
$binaryPath = Join-Path $tempDir $binaryName
$rgBinaryPath = Join-Path $tempDir $rgBinaryName

try {

# Check for curl
if (!(Get-Command curl.exe -ErrorAction SilentlyContinue)) {
    Write-Error "curl is required but not found. Please install curl or upgrade to Windows 10 1803+ where it's built-in."
}

& curl.exe -fsSL -o $binaryPath $url
if ($LASTEXITCODE -ne 0) {
    Write-Error "Download failed"
}

Write-Info "Fetching and verifying checksum"
$expectedHash = (& curl.exe -fsSL $shaUrl).Trim()
if ($LASTEXITCODE -ne 0) {
    Write-Error "Failed to fetch checksum"
}
    
    # Verify checksum
    try {
        $actualHash = (Get-FileHash -Path $binaryPath -Algorithm SHA256).Hash.ToLower()
        
        if ($actualHash -ne $expectedHash.ToLower()) {
            Write-Error "Checksum verification failed"
        }
        Write-Info "Checksum verification passed"
    } catch {
        Write-Warning "Could not verify checksum, continuing anyway"
    }
    
    # Download and install ripgrep
    Write-Info "Downloading ripgrep for Windows-$architecture"
    
& curl.exe -fsSL -o $rgBinaryPath $rgUrl
if ($LASTEXITCODE -ne 0) {
    Write-Error "Ripgrep download failed"
}

Write-Info "Fetching and verifying ripgrep checksum"
$rgExpectedHash = (& curl.exe -fsSL $rgShaUrl).Trim()
if ($LASTEXITCODE -ne 0) {
    Write-Error "Failed to fetch ripgrep checksum"
}
    
    # Verify ripgrep checksum
    try {
        $rgActualHash = (Get-FileHash -Path $rgBinaryPath -Algorithm SHA256).Hash.ToLower()
        
        if ($rgActualHash -ne $rgExpectedHash.ToLower()) {
            Write-Error "Ripgrep checksum verification failed"
        }
        Write-Info "Ripgrep checksum verification passed"
    } catch {
        Write-Warning "Could not verify ripgrep checksum, continuing anyway"
    }
    
    # Determine installation directories
    $installDir = $null
    
    # Try user's local bin directory first
    $userBin = Join-Path $env:USERPROFILE "bin"
    if (Test-Path $userBin) {
        $installDir = $userBin
    } else {
        # Create user bin directory
        New-Item -ItemType Directory -Path $userBin -Force | Out-Null
        $installDir = $userBin
    }
    
    $installPath = Join-Path $installDir $binaryName
    
    # Create factory bin directory for ripgrep
    $factoryDir = Join-Path $env:USERPROFILE ".factory"
    $factoryBinDir = Join-Path $factoryDir "bin"
    New-Item -ItemType Directory -Path $factoryBinDir -Force | Out-Null
    
    $rgInstallPath = Join-Path $factoryBinDir $rgBinaryName
    
    # Stop any running droid processes
    $droidProcesses = Get-Process -Name "droid" -ErrorAction SilentlyContinue
    if ($droidProcesses) {
        Write-Info "Stopping old droid process(es)"
        Stop-Process -Name "droid" -Force -ErrorAction SilentlyContinue
        Start-Sleep -Seconds 1
    }
    
    # Copy binaries to installation directories
    Copy-Item -Path $binaryPath -Destination $installPath -Force
    Copy-Item -Path $rgBinaryPath -Destination $rgInstallPath -Force
    
    Write-Info "Factory CLI v$version installed successfully to $installPath"
    Write-Info "Ripgrep installed successfully to $rgInstallPath"
    
    # Check if installation directory is in persistent PATH
    $userPath = [Environment]::GetEnvironmentVariable('Path','User')
    if ($userPath -notlike "*$installDir*") {
        # Add to persistent PATH
        [Environment]::SetEnvironmentVariable('Path', $userPath + ";$installDir", 'User')
        # Update current session
        $env:PATH = "$installDir;$env:PATH"
        Write-Info "Added $installDir to your PATH permanently"
    } else {
        Write-Info "PATH already configured"
    }
    Write-Info "Run 'droid' to get started!"
    
} finally {
    # Clean up temporary directory
    Remove-Item $tempDir -Recurse -Force -ErrorAction SilentlyContinue
}