build.ps1

# ================================
# Pyarmor 1shot Full Automation
# build.ps1
# ================================

$ErrorActionPreference = "Stop"

function Write-Step {
    param([string]$Message)
    Write-Host ""
    Write-Host ("=" * 80) -ForegroundColor DarkGray
    Write-Host $Message -ForegroundColor Cyan
    Write-Host ("=" * 80) -ForegroundColor DarkGray
}

function Fail {
    param([string]$Message)
    Write-Host "[ERROR] $Message" -ForegroundColor Red
    exit 1
}

function Assert-Exists {
    param(
        [Parameter(Mandatory = $true)][string]$Path,
        [string]$Label = "Path"
    )
    if (-not (Test-Path -LiteralPath $Path)) {
        Fail "$Label not found: $Path"
    }
}

function Ensure-Dir {
    param([Parameter(Mandatory = $true)][string]$Path)
    if (-not (Test-Path -LiteralPath $Path)) {
        New-Item -ItemType Directory -Path $Path | Out-Null
    }
}

function Invoke-NativeCommand {
    param(
        [Parameter(Mandatory = $true)][string]$FilePath,
        [Parameter()][string[]]$Arguments = @(),
        [string]$WorkingDirectory = $PWD.Path
    )

    Write-Host "[CMD] $FilePath $($Arguments -join ' ')" -ForegroundColor Yellow

    Push-Location $WorkingDirectory
    try {
        & $FilePath @Arguments
        $exitCode = $LASTEXITCODE
    }
    finally {
        Pop-Location
    }

    if ($exitCode -ne 0) {
        throw "명령 실행 실패 (exit code $exitCode): $FilePath $($Arguments -join ' ')"
    }
}

# ---- 고정 경로 ----
$BASE_DIR     = "C:\tmp\Pyarmor-Static-Unpack-1shot"
$BUILD_DIR    = Join-Path $BASE_DIR "build"
$PYCDC_DIR    = Join-Path $BASE_DIR "pycdc"

$TARGET_DIR   = "C:\tmp\v305.exe_extracted"
$RUNTIME_PYD  = Join-Path $TARGET_DIR "pyarmor_runtime_009673\pyarmor_runtime.pyd"

$ONESHOT_EXE  = Join-Path $BUILD_DIR "Release\pyarmor-1shot.exe"
$ONESHOT_DEST = Join-Path $BASE_DIR "oneshot\pyarmor-1shot.exe"

$SHOT_SCRIPT  = Join-Path $BASE_DIR "oneshot\shot.py"

$LOG_FILE     = Join-Path $BASE_DIR "1shot.log"

$WSL_DEST     = "\\wsl.localhost\Ubuntu-24.04\home\phantom\v138\test_data"
$PYC_PATTERN  = "엔잡곰자동화시스템_반자동v305_배포용_s.pyc*"

# ---- Validation ----
Write-Step "Validation"

Assert-Exists -Path $BASE_DIR -Label "BaseDir"
Assert-Exists -Path $PYCDC_DIR -Label "pycdc"
Assert-Exists -Path $TARGET_DIR -Label "TargetDir"
Assert-Exists -Path $RUNTIME_PYD -Label "RuntimePyd"
Assert-Exists -Path $SHOT_SCRIPT -Label "shot.py"

try {
    $pythonCmd = (Get-Command python -ErrorAction Stop).Source
}
catch {
    Fail "python not found in PATH"
}

try {
    $cmakeCmd = (Get-Command cmake -ErrorAction Stop).Source
}
catch {
    Fail "cmake not found in PATH"
}

Ensure-Dir -Path $WSL_DEST

# ---- Clean build ----
Write-Step "Clean build"

if (Test-Path -LiteralPath $BUILD_DIR) {
    Remove-Item -Recurse -Force $BUILD_DIR
}
New-Item -ItemType Directory -Path $BUILD_DIR | Out-Null

# 기존 로그 삭제
if (Test-Path -LiteralPath $LOG_FILE) {
    Remove-Item -Force $LOG_FILE
}

# ---- Configure ----
Write-Step "CMake configure"

Invoke-NativeCommand `
    -FilePath $cmakeCmd `
    -Arguments @("-S", $PYCDC_DIR, "-B", $BUILD_DIR, "-G", "Visual Studio 17 2022")

# ---- Build (max CPU) ----
Write-Step "CMake build"

Invoke-NativeCommand `
    -FilePath $cmakeCmd `
    -Arguments @("--build", $BUILD_DIR, "--config", "Release", "--", "/m")

Assert-Exists -Path $ONESHOT_EXE -Label "pyarmor-1shot.exe"

# ---- Copy exe ----
Write-Step "Copy exe"

Copy-Item -LiteralPath $ONESHOT_EXE -Destination $ONESHOT_DEST -Force
Assert-Exists -Path $ONESHOT_DEST -Label "oneshot exe"

# ---- Run shot ----
Write-Step "Run shot.py"

Clear-Host
Set-Location $BASE_DIR

& $pythonCmd `
    $SHOT_SCRIPT `
    $TARGET_DIR `
    -r $RUNTIME_PYD `
    --show-all `
    --no-banner `
    --export-raw-data `
    *> $LOG_FILE

if ($LASTEXITCODE -ne 0) {
    throw "Python 실행 실패 (exit code $LASTEXITCODE)"
}

Assert-Exists -Path $LOG_FILE -Label "1shot.log"

# ---- Copy artifacts ----
Write-Step "Copy artifacts"

Copy-Item -LiteralPath $LOG_FILE -Destination $WSL_DEST -Force

$pycFiles = Get-ChildItem -Path (Join-Path $TARGET_DIR $PYC_PATTERN) -ErrorAction SilentlyContinue

if (-not $pycFiles) {
    Fail "pyc 파일 없음: $(Join-Path $TARGET_DIR $PYC_PATTERN)"
}

foreach ($f in $pycFiles) {
    Copy-Item -LiteralPath $f.FullName -Destination $WSL_DEST -Force
    Write-Host "Copied: $($f.FullName)" -ForegroundColor Green
}

Write-Step "Done"
Write-Host "Done: Full pipeline success" -ForegroundColor Green