$conflictFound = $false

# Helper function to report conflict
function Write-Conflict {
    param(
        [string]$Message,
        [string]$Path,
        [string]$Fix
    )
    $script:conflictFound = $true
    Write-Host "CONFLICT: $Message" -ForegroundColor Red
    Write-Host "   Path: $Path" -ForegroundColor Red
    Write-Host "   → $Fix" -ForegroundColor Red
    Write-Host ""
}

# ───────────────────────────────────────────────
#   HKCU checks
# ───────────────────────────────────────────────

$hkcuAD = Get-ItemProperty -Path "HKCU:\Software\Microsoft\Windows\CurrentVersion\Policies\ActiveDesktop" -EA SilentlyContinue
if ($hkcuAD -and $hkcuAD.NoChangingWallPaper -eq 1) {
    Write-Conflict "NoChangingWallPaper = 1" `
                   "HKCU\...\Policies\ActiveDesktop" `
                   "Remove the policy that sets NoChangingWallPaper = 1"
}

$hkcuSys = Get-ItemProperty -Path "HKCU:\Software\Microsoft\Windows\CurrentVersion\Policies\System" -EA SilentlyContinue
if ($hkcuSys) {
    if ($hkcuSys.Wallpaper) {
        Write-Conflict "Wallpaper value is set" `
                       "HKCU\...\Policies\System" `
                       "Remove the policy that sets Wallpaper value"
    }
    if ($null -ne $hkcuSys.WallpaperStyle) {
        Write-Conflict "WallpaperStyle = $($hkcuSys.WallpaperStyle)" `
                       "HKCU\...\Policies\System" `
                       "Remove the policy that sets WallpapeStyle value"
    }
}

# ───────────────────────────────────────────────
#   HKLM checks
# ───────────────────────────────────────────────

if ($regHKLM) {
    $hlkmSys = Get-ItemProperty -Path "HKLM:\Software\Microsoft\Windows\CurrentVersion\Policies\System" -EA SilentlyContinue
    if ($hlkmSys) {
        if ($hlkmSys.Wallpaper) {
            Write-Conflict "Wallpaper value is set (machine policy)" `
                           "HKLM\...\Policies\System" `
                           "Remove the policy that sets Wallpaper value"
        }
        if ($null -ne $hlkmSys.WallpaperStyle) {
            Write-Conflict "WallpaperStyle = $($hlkmSys.WallpaperStyle)" `
                           "HKLM\...\Policies\System" `
                           "Remove the policy that sets WallpapeStyle value"
        }
    }

    $hlkmAD = Get-ItemProperty -Path "HKLM:\Software\Microsoft\Windows\CurrentVersion\Policies\ActiveDesktop" -EA SilentlyContinue
    if ($hlkmAD -and $hlkmAD.NoChangingWallPaper -eq 1) {
        Write-Conflict "NoChangingWallPaper = 1 (machine policy)" `
                       "HKLM\...\Policies\ActiveDesktop" `
                       "Remove the policy that sets NoChangingWallPaper = 1"
    }

    # Personalization CSP (MDM/Intune)
    $csp = Get-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\PersonalizationCSP" -EA SilentlyContinue
    if ($csp) {
        if ($csp.DesktopImagePath) {
            Write-Conflict "DesktopImagePath is set (MDM/Intune)" `
                           "HKLM\...\PersonalizationCSP" `
                           "Remove value: DesktopImagePath"
        }
        if ($csp.DesktopImageUrl) {
            Write-Conflict "DesktopImageUrl is set (MDM/Intune)" `
                           "HKLM\...\PersonalizationCSP" `
                           "Remove value: DesktopImageUrl"
        }
    }
}

# ───────────────────────────────────────────────
#               Final Result
# ───────────────────────────────────────────────

Write-Host "`n" -NoNewline

if ($conflictFound) {
    Write-Host "RESULT: Blocking Group Policy or Intune/MDM policy detected" -ForegroundColor Red
    Write-Host "       Desktop background change will most likely be prevented." -ForegroundColor Red
} else {
    Write-Host "RESULT: No blocking policies detected" -ForegroundColor Green
    Write-Host "       Desktop background should be changeable." -ForegroundColor Green
}

# Cleanup
if ($regHKLM) { $regHKLM.Dispose() }