sysrefresh.ps1

# ▓▓▓ SYSREFRESH TOOLKIT ▓▓▓
# Version-aware diagnostics + driver recovery
# Compatible with PowerShell Core and Windows PowerShell

# ─────────────────────────────────────────────────
# Helper: Log Writer
Checkpoint-Computer -Description "SysRefresh Toolkit" -RestorePointType "MODIFY_SETTINGS"
function Write-Log {
    param([string]$Message)
    $stamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
    Add-Content -Path "$env:SystemDrive\SysRefresh.log" -Value "$stamp → $Message"
}

# ░ Check NuGet Provider ░
function Test-NuGetProvider {
    if (Get-PackageProvider -ListAvailable | Where-Object { $_.Name -eq 'NuGet' }) {
        Write-Host "[✓] NuGet provider is present."
        Write-Log "NuGet provider verified."
    } else {
        Write-Host "[✗] NuGet provider missing."
        Write-Log "NuGet provider missing."
    }
}

# ░ Refresh Drivers ░
function Update-AllDrivers {
    Write-Host "[↻] Reinstalling published drivers..."
    pnputil /enum-drivers | ForEach-Object {
        if ($_ -match "Published Name\s*:\s*(.+)") {
            $publishedName = $matches[1]
            pnputil /install-driver $publishedName /force
        }
    }
    Write-Log "All drivers reinstalled from local store."
}

# ░ Run Windows Update ░
function Invoke-WindowsUpdate {
    Import-Module PSWindowsUpdate -ErrorAction SilentlyContinue
    Write-Host "[🔄] Windows Update execution started..."
    Get-WindowsUpdate -MicrosoftUpdate -AcceptAll -Install -Verbose
    Write-Log "Windows Update executed."
}

# ░ Performance Scan (Core-safe) ░
function Invoke-PerformanceScan {
    Write-Host "[📊] Running memory and system checks..."
    $sysMetrics = Get-CimInstance Win32_OperatingSystem |
        Select-Object FreePhysicalMemory, TotalVisibleMemorySize, LastBootUpTime
    $cpuMetrics = Get-CimInstance Win32_Processor |
        Select-Object Name, LoadPercentage, NumberOfLogicalProcessors
    Write-Log "Performance scan completed."
    $sysMetrics
    $cpuMetrics
}

# ░ WMI Repair ░
function Repair-WMI {
    Write-Host "[🧼] Repairing WMI repository..."
    net stop winmgmt /y
    Rename-Item "$env:windir\System32\wbem\Repository" -NewName "Repository.old" -Force
    net start winmgmt
    $mofList = @("cimwin32.mof", "mgmt.mof", "wmiprov.mof", "rsop.mof")
    foreach ($mof in $mofList) {
        mofcomp "$env:windir\System32\wbem\$mof"
    }
    Write-Log "WMI repair routine completed."
}

# ░ Kernel Trace (ETW) ░
function Trace-KernelPerformance {
    $sessionName = "PostKaliTrace"
    Write-Host "[⏱️] Starting kernel trace for 60s..."
    wpr -start CPU -start DiskIO -start Registry -filemode
    Start-Sleep -Seconds 60
    wpr -stop "$env:SystemDrive\SysRefresh-$sessionName.etl"
    Write-Log "Kernel trace saved to SysRefresh-$sessionName.etl"
    Write-Host "[🧠] ETL trace file saved."
}

# ░ Master Run ░
function Start-Toolkit {
    Write-Host "[🚀] Starting SysRefresh session..."
    Test-NuGetProvider
    Update-AllDrivers
    Invoke-WindowsUpdate
    Invoke-PerformanceScan
    Trace-KernelPerformance
    Write-Host "`n[✔] SysRefresh tasks completed. Review logs at C:\SysRefresh.log"
}
# ─────────────────────────────────────────────────

# Optional: Run the full sequence
Start-Toolkit