# Restore backup and properly fix the sync script
$password = ConvertTo-SecureString "Paper123!@#" -AsPlainText -Force
$cred = New-Object System.Management.Automation.PSCredential("INTRANET\sysadmin", $password)

Write-Host "=== Restoring and Fixing Sync Script ===" -ForegroundColor Cyan
Write-Host ""

Invoke-Command -ComputerName 192.168.0.6 -Credential $cred -ScriptBlock {
    $scriptPath = "C:\Shares\test\scripts\Sync-FromNAS.ps1"
    $scriptsDir = "C:\Shares\test\scripts"

    Write-Host "[1] Finding most recent backup" -ForegroundColor Yellow
    Write-Host "=" * 80 -ForegroundColor Gray

    $backups = Get-ChildItem "$scriptsDir\Sync-FromNAS.ps1.backup-*" | Sort-Object LastWriteTime -Descending
    $latestBackup = $backups | Select-Object -First 1

    if ($latestBackup) {
        Write-Host "[OK] Found backup: $($latestBackup.Name)" -ForegroundColor Green
        Write-Host "     Created: $($latestBackup.LastWriteTime)" -ForegroundColor Gray

        # Restore the backup
        Copy-Item -Path $latestBackup.FullName -Destination $scriptPath -Force
        Write-Host "[OK] Restored backup" -ForegroundColor Green
    } else {
        Write-Host "[WARNING] No backup found - using current script" -ForegroundColor Yellow
    }

    Write-Host ""
    Write-Host "[2] Reading original script to understand structure" -ForegroundColor Yellow
    Write-Host "=" * 80 -ForegroundColor Gray

    $lines = Get-Content $scriptPath

    # Find the PSCP/PLINK lines and the actual function implementations
    for ($i = 0; $i -lt $lines.Count; $i++) {
        if ($lines[$i] -match '^\$PSCP\s*=') {
            Write-Host "Line $($i+1): $($lines[$i])" -ForegroundColor Cyan
        }
        if ($lines[$i] -match '^\$PLINK\s*=') {
            Write-Host "Line $($i+1): $($lines[$i])" -ForegroundColor Cyan
        }
        if ($lines[$i] -match 'PSCP.*-ssh.*-pw') {
            Write-Host "Line $($i+1) [PSCP call]: ...$(($lines[$i] -replace '.*(\$PSCP.*)', '$1'))" -ForegroundColor Yellow
        }
    }

    Write-Host ""
    Write-Host "[3] Applying targeted fixes" -ForegroundColor Yellow
    Write-Host "=" * 80 -ForegroundColor Gray

    # Just replace the tool paths and add better error logging
    # Don't try to rewrite the functions - just improve what's there

    for ($i = 0; $i -lt $lines.Count; $i++) {
        # Replace tool paths
        if ($lines[$i] -match '^\$PSCP\s*=\s*".*pscp\.exe"') {
            $lines[$i] = '$SCP = "C:\Program Files\OpenSSH\scp.exe"  # Changed from PSCP to OpenSSH'
            Write-Host "[UPDATED] Line $($i+1): Tool path PSCP -> SCP" -ForegroundColor Green
        }

        if ($lines[$i] -match '^\$PLINK\s*=\s*".*plink\.exe"') {
            $lines[$i] = '$SSH = "C:\Program Files\OpenSSH\ssh.exe"  # Changed from PLINK to OpenSSH'
            Write-Host "[UPDATED] Line $($i+1): Tool path PLINK -> SSH" -ForegroundColor Green
        }

        # Replace PSCP calls with SCP calls and add verbose flag
        if ($lines[$i] -match '\$result\s*=\s*&\s*\$PSCP\s+-ssh\s+-pw\s+\$NAS_PASSWORD\s+-hostkey\s+\$NAS_HOSTKEY') {
            # This is a pscp call - replace with scp
            if ($lines[$i] -match 'LocalPath.*RemotePath') {
                # This is Copy-ToNAS (local first, then remote)
                $lines[$i] = '    $result = & $SCP -o StrictHostKeyChecking=accept-new -o UserKnownHostsFile="$SCRIPTS_DIR\.ssh\known_hosts" -o PreferredAuthentications=password -o PubkeyAuthentication=no -o PasswordAuthentication=yes $LocalPath "${NAS_USER}@${NAS_IP}:$RemotePath" 2>&1'
                Write-Host "[UPDATED] Line $($i+1): Copy-ToNAS PSCP -> SCP" -ForegroundColor Green
            } else {
                # This is Copy-FromNAS (remote first, then local)
                $lines[$i] = '    $result = & $SCP -o StrictHostKeyChecking=accept-new -o UserKnownHostsFile="$SCRIPTS_DIR\.ssh\known_hosts" -o PreferredAuthentications=password -o PubkeyAuthentication=no -o PasswordAuthentication=yes "${NAS_USER}@${NAS_IP}:$RemotePath" $LocalPath 2>&1'
                Write-Host "[UPDATED] Line $($i+1): Copy-FromNAS PSCP -> SCP" -ForegroundColor Green
            }

            # Add error logging on the next non-empty line after "return $LASTEXITCODE"
            for ($j = $i + 1; $j -lt $lines.Count; $j++) {
                if ($lines[$j] -match 'return \$LASTEXITCODE') {
                    # Insert error logging before the return
                    $lines[$j] = "    if (`$LASTEXITCODE -ne 0) { Write-Log `"  SCP ERROR (exit $LASTEXITCODE): `$result`" }`n" + $lines[$j]
                    Write-Host "[ADDED] Line $($j+1): Error logging" -ForegroundColor Green
                    break
                }
            }
        }

        # Replace PLINK calls with SSH calls
        if ($lines[$i] -match '\&\s*\$PLINK\s+-batch') {
            $lines[$i] = $lines[$i] -replace '\$PLINK\s+-batch', '$SSH -o StrictHostKeyChecking=accept-new -o UserKnownHostsFile="$SCRIPTS_DIR\.ssh\known_hosts" -o PreferredAuthentications=password -o PubkeyAuthentication=no -o PasswordAuthentication=yes'
            Write-Host "[UPDATED] Line $($i+1): PLINK -> SSH" -ForegroundColor Green
        }
    }

    Write-Host ""
    Write-Host "[4] Saving updated script" -ForegroundColor Yellow
    Write-Host "=" * 80 -ForegroundColor Gray

    # Save the modified script
    $lines | Out-File -FilePath $scriptPath -Encoding UTF8 -Force

    Write-Host "[OK] Script saved" -ForegroundColor Green

    Write-Host ""
    Write-Host "[5] Testing script syntax" -ForegroundColor Yellow
    Write-Host "=" * 80 -ForegroundColor Gray

    # Test if the script has valid syntax
    try {
        $null = [System.Management.Automation.PSParser]::Tokenize((Get-Content $scriptPath -Raw), [ref]$null)
        Write-Host "[OK] Script syntax is valid" -ForegroundColor Green
    } catch {
        Write-Host "[ERROR] Script has syntax errors: $($_.Exception.Message)" -ForegroundColor Red
    }
}

Write-Host ""
Write-Host "=== Fix Complete ===" -ForegroundColor Cyan
