Write-Host "==== WINDOWS 11 FULL START MENU & UI REPAIR ===="

# 1. Stop UI processes
Write-Host "Stopping Explorer and UI processes..."
Stop-Process -Name explorer -Force -ErrorAction SilentlyContinue
Stop-Process -Name StartMenuExperienceHost -Force -ErrorAction SilentlyContinue
Stop-Process -Name ShellExperienceHost -Force -ErrorAction SilentlyContinue
Stop-Process -Name SearchHost -Force -ErrorAction SilentlyContinue

# 2. Stop dependent services
Write-Host "Stopping services..."
net stop StateRepository /y
net stop AppXSvc /y
net stop ClipSVC /y
net stop WSearch /y

# 3. Fix permissions on AppX folders (if locked)
Write-Host "Fixing folder permissions..."
$folders = @(
    "$env:LOCALAPPDATA\Packages\Microsoft.Windows.StartMenuExperienceHost_cw5n1h2txyewy",
    "$env:LOCALAPPDATA\Packages\Microsoft.Windows.ShellExperienceHost_cw5n1h2txyewy",
    "$env:LOCALAPPDATA\Microsoft\Windows\Caches"
)
foreach ($f in $folders) {
    if (Test-Path $f) {
        takeown /f $f /r /d Y
        icacls $f /reset /t
        icacls $f /grant Administrators:F /t
        icacls $f /grant "$env:USERNAME":F /t
    }
}

# 4. Clear UI caches
Write-Host "Cleaning UI caches..."
foreach ($f in $folders) {
    if (Test-Path $f) { Remove-Item "$f\*" -Recurse -Force -ErrorAction SilentlyContinue }
}

# 5. Restart services
Write-Host "Starting services..."
net start AppXSvc
net start StateRepository
net start ClipSVC
net start WSearch

# 6. Re-register critical AppX packages
Write-Host "Re-registering StartMenuExperienceHost..."
Add-AppxPackage -DisableDevelopmentMode -Register "C:\Windows\SystemApps\Microsoft.Windows.StartMenuExperienceHost_cw5n1h2txyewy\AppxManifest.xml"

Write-Host "Re-registering ShellExperienceHost..."
Add-AppxPackage -DisableDevelopmentMode -Register "C:\Windows\SystemApps\ShellExperienceHost_cw5n1h2txyewy\AppxManifest.xml"

Write-Host "Re-registering all system apps..."
Get-AppxPackage -AllUsers | ForEach-Object {
    if (Test-Path "$($_.InstallLocation)\AppXManifest.xml") {
        Add-AppxPackage -DisableDevelopmentMode -Register "$($_.InstallLocation)\AppXManifest.xml" -ErrorAction SilentlyContinue
    }
}

# 7. System integrity check
Write-Host "Running SFC (system file check)..."
sfc /scannow

Write-Host "Running DISM image repair..."
DISM /Online /Cleanup-Image /RestoreHealth

# 8. Restart Explorer
Write-Host "Restarting Explorer..."
Start-Process explorer.exe

Write-Host "==== FULL REPAIR COMPLETED ===="
Write-Host "Recommended: Reboot the PC for all changes to take effect."