# Register-MachineEnforce.ps1 - One-time setup for the Standard-Machine
# logon-enforce scheduled task. Called by Run-ShopfloorSetup.ps1 on
# Standard-Machine PCs only (Timeclocks skip). Idempotent: re-running
# refreshes staged scripts and re-registers the task.
#
# Parallel to CMM\09-Setup-CMM.ps1 steps 3-4 (stage Install-FromManifest +
# Machine-Enforce, register the task) with no imaging-time install step -
# initial UDC/eDNC/NTLARS install is already handled by the preinstall
# phase on the PXE server.

$ErrorActionPreference = 'Continue'

$installRoot    = 'C:\Program Files\GE\MachineApps'
$runtimeLib     = Join-Path $installRoot 'lib\Install-FromManifest.ps1'
$runtimeEnforce = Join-Path $installRoot 'Machine-Enforce.ps1'
$logDir         = 'C:\Logs\MachineApps'
$setupLog       = Join-Path $logDir 'setup.log'

$sourceLib     = Join-Path $PSScriptRoot 'lib\Install-FromManifest.ps1'
$sourceEnforce = Join-Path $PSScriptRoot 'Machine-Enforce.ps1'

if (-not (Test-Path $logDir))       { New-Item -Path $logDir -ItemType Directory -Force | Out-Null }
if (-not (Test-Path $installRoot))  { New-Item -Path $installRoot -ItemType Directory -Force | Out-Null }
if (-not (Test-Path (Join-Path $installRoot 'lib'))) {
    New-Item -Path (Join-Path $installRoot 'lib') -ItemType Directory -Force | Out-Null
}

function Write-SetupLog {
    param([string]$Message, [string]$Level = 'INFO')
    $line = "[{0}] [{1}] {2}" -f (Get-Date -Format 'yyyy-MM-dd HH:mm:ss'), $Level, $Message
    Write-Host $line
    Add-Content -Path $setupLog -Value $line -ErrorAction SilentlyContinue
}

Write-SetupLog "=== Register-MachineEnforce start ==="

foreach ($pair in @(
    @{ Src = $sourceLib;     Dst = $runtimeLib },
    @{ Src = $sourceEnforce; Dst = $runtimeEnforce }
)) {
    if (-not (Test-Path $pair.Src)) {
        Write-SetupLog "Source not found: $($pair.Src) - cannot stage" "ERROR"
        continue
    }
    Copy-Item -Path $pair.Src -Destination $pair.Dst -Force
    Write-SetupLog "Staged $($pair.Src) -> $($pair.Dst)"
}

$taskName = 'GE Shopfloor Machine Apps Enforce'
$existing = Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
if ($existing) {
    Write-SetupLog "Removing existing scheduled task '$taskName'"
    Unregister-ScheduledTask -TaskName $taskName -Confirm:$false -ErrorAction SilentlyContinue
}

Write-SetupLog "Registering scheduled task '$taskName' (logon trigger, SYSTEM)"
try {
    $action    = New-ScheduledTaskAction `
        -Execute 'powershell.exe' `
        -Argument "-NoProfile -ExecutionPolicy Bypass -File `"$runtimeEnforce`""

    $trigger   = New-ScheduledTaskTrigger -AtLogOn
    $principal = New-ScheduledTaskPrincipal -UserId 'SYSTEM' -LogonType ServiceAccount -RunLevel Highest
    # ExecutionTimeLimit 1 hour; UDC/eDNC/NTLARS combined shouldn't exceed that.
    $settings  = New-ScheduledTaskSettingsSet `
        -AllowStartIfOnBatteries `
        -DontStopIfGoingOnBatteries `
        -StartWhenAvailable `
        -ExecutionTimeLimit (New-TimeSpan -Hours 1) `
        -MultipleInstances IgnoreNew

    Register-ScheduledTask `
        -TaskName $taskName `
        -Action $action `
        -Trigger $trigger `
        -Principal $principal `
        -Settings $settings `
        -Description 'GE Shopfloor Machine: enforce UDC/eDNC/NTLARS version against tsgwp00525 SFLD share on user logon' | Out-Null

    Write-SetupLog "Scheduled task registered"
} catch {
    Write-SetupLog "Failed to register scheduled task: $_" "ERROR"
}

Write-SetupLog "=== Register-MachineEnforce end ==="
