# Ask for confirmation before running the script
# === Confirmation ===
$confirmationMessage = @"
This script will perform the following customizations to Windows 11:

Taskbar & UI Customizations:
- Disable Widgets, Teams Chat, Task View, and Copilot icons
- Disable Copilot via Group Policy and remove from startup
- Set the Search box to an icon
- Disable the news and interests feed
- Align the Taskbar to the left
- Enable Dark Theme for both apps and the system

Application Unpinning:
- Unpin the following apps from the Taskbar:
  - Microsoft Edge, Microsoft Store, Mail, Copilot, Microsoft 365 (Office),
    Outlook (new), Outlook, Microsoft Teams (personal)

System Changes:
- Set the desktop wallpaper to the default dark theme image
- Disable OneDrive from starting automatically and stop its process
- Clean the Start Menu layout by downloading a predefined configuration

"@
Write-Host $confirmationMessage -ForegroundColor Yellow
$confirmation = Read-Host "Do you want to continue with these changes? (Y/N)"
if ($confirmation -notin @('Y', 'y')) {
    Write-Host "Operation cancelled." -ForegroundColor Red
    exit
}

Write-Host "Starting Windows 11 customization script..."

# Define paths and application list
$advPath = 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Explorer\Advanced'
$searchPath = 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Search'
$personalPath = 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Themes\Personalize'
$appsToRemove = @("Microsoft Edge", "Microsoft Store", "Mail", "Copilot", "Microsoft 365 (Office)", "Outlook (new)", "Outlook", "Microsoft Teams (personal)", "Microsoft Teams")

# Apply Taskbar and UI customizations
Write-Host "Applying Taskbar and UI customizations..."
New-ItemProperty -Path $advPath -Name TaskbarDa -Value 0 -PropertyType DWord -Force -ErrorAction SilentlyContinue # Widgets
Write-Host "- Disabled Widgets icon."
New-ItemProperty -Path $advPath -Name TaskbarMn -Value 0 -PropertyType DWord -Force # Teams Chat
Write-Host "- Disabled Teams Chat icon."
New-ItemProperty -Path $advPath -Name ShowTaskViewButton -Value 0 -PropertyType DWord -Force # Task View
Write-Host "- Disabled Task View button."
New-ItemProperty -Path $advPath -Name ShowCopilotButton -Value 0 -PropertyType DWord -Force # Copilot
Write-Host "- Disabled Copilot button."
Write-Host "Disabling Copilot via Group Policy..."
Remove-ItemProperty 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Run' -Name 'Copilot' -ErrorAction SilentlyContinue
Write-Host "- Removed Copilot from startup."
New-Item -Path 'HKCU:\Software\Policies\Microsoft\Windows' -Name 'WindowsCopilot' -Force | Out-Null
Set-ItemProperty -Path 'HKCU:\Software\Policies\Microsoft\Windows\WindowsCopilot' -Name 'TurnOffWindowsCopilot' -Type DWord -Value 1
Write-Host "- Set Group Policy to turn off Windows Copilot."
Get-Process WebViewHost -ErrorAction SilentlyContinue | Stop-Process -Force
Write-Host "- Stopped the Copilot process (WebViewHost)."
New-ItemProperty -Path $searchPath -Name SearchboxTaskbarMode -Value 0 -PropertyType DWord -Force # Search box
Write-Host "- Set search box to icon only."
New-ItemProperty -Path $advPath -Name FeedEnabled -Value 0 -PropertyType DWord -Force # News feed
Write-Host "- Disabled news and interests feed."
New-ItemProperty -Path $advPath -Name TaskbarAl -Value 0 -PropertyType DWord -Force # Start on left
Write-Host "- Aligned Taskbar to the left."
New-ItemProperty -Path $personalPath -Name AppsUseLightTheme -Value 0 -PropertyType DWord -Force # Dark Theme Apps
Write-Host "- Enabled Dark Theme for applications."
New-ItemProperty -Path $personalPath -Name SystemUsesLightTheme -Value 0 -PropertyType DWord -Force # Dark Theme System
Write-Host "- Enabled Dark Theme for the system."

# Unpin applications from the Taskbar
Write-Host "Unpinning applications from the Taskbar..."
foreach ($app in $appsToRemove) {
    Write-Host "- Attempting to unpin '$app'..."
    $item = ((New-Object -Com Shell.Application).Namespace('shell:::{4234d49b-0245-4df3-b780-3893943456e1}').Items() | Where-Object { $_.Name -eq $app })
    if ($item) {
        $verb = $item.Verbs() | Where-Object { $_.Name.Replace('&', '') -match 'Unpin from taskbar' }
        if ($verb) {
            $verb.DoIt()
            Write-Host "  - Successfully unpinned '$app'."
        } else {
            Write-Host "  - 'Unpin from taskbar' action not found for '$app'."
        }
    } else {
        Write-Host "  - Application '$app' not found."
    }
}

# Set the desktop wallpaper
Write-Host "Setting desktop wallpaper..."
$wall = "$env:SystemRoot\Web\4K\Wallpaper\Windows\img19_1920x1200.jpg"

Add-Type '
  using System;
  using System.Runtime.InteropServices;
  public class Native {
      [DllImport("user32.dll", SetLastError=true)]
      public static extern bool SystemParametersInfo(int uAction, int uParam, string lpvParam, int fuWinIni);
  }
'

[Native]::SystemParametersInfo(0x0014, 0, $wall, 0x03) | Out-Null
Write-Host "- Wallpaper set to the default dark theme Windows 11 wallpaper."

# Disable OneDrive
Write-Host "Disabling OneDrive..."
$runKey = 'HKCU:\Software\Microsoft\Windows\CurrentVersion\Run'
Remove-ItemProperty -Path $runKey -Name 'OneDrive' -ErrorAction SilentlyContinue
Write-Host "- Removed OneDrive from startup."
Get-Process OneDrive -ErrorAction SilentlyContinue | Stop-Process -Force
Write-Host "- Stopped the OneDrive process."

# The original script had a second unpin loop. It is preserved here for consistency.
Write-Host "Running second pass to unpin applications from the Taskbar..."
foreach ($app in $appsToRemove) {
    Write-Host "- Attempting to unpin '$app'..."
    $item = ((New-Object -Com Shell.Application).Namespace('shell:::{4234d49b-0245-4df3-b780-3893943456e1}').Items() | Where-Object { $_.Name -eq $app })
    if ($item) {
        $verb = $item.Verbs() | Where-Object { $_.Name.Replace('&', '') -match 'Unpin from taskbar' }
        if ($verb) {
            $verb.DoIt()
            Write-Host "  - Successfully unpinned '$app'."
        } else {
            Write-Host "  - 'Unpin from taskbar' action not found for '$app'."
        }
    } else {
        Write-Host "  - Application '$app' not found."
    }
}

# Clean Start Menu layout
Write-Host "Cleaning Start Menu layout..."
$startMenuPath = "$env:LOCALAPPDATA\Packages\Microsoft.Windows.StartMenuExperienceHost_cw5n1h2txyewy\LocalState"
$startMenuFile = "start2.bin"
$downloadUrl = "https://raw.githubusercontent.com/Ad3t0/Ad3t0/master/powershell/bin/start2.bin"
$tempFile = Join-Path $env:TEMP $startMenuFile

# Ensure the destination directory exists
if (-not (Test-Path $startMenuPath)) {
    New-Item -Path $startMenuPath -ItemType Directory -Force | Out-Null
}

# Try downloading the file to a temporary location
try {
    Write-Host "- Attempting to download '$startMenuFile'..."
    Invoke-WebRequest -Uri $downloadUrl -OutFile $tempFile -UseBasicParsing

    # If download is successful, copy to destination
    Copy-Item -Path $tempFile -Destination (Join-Path $startMenuPath $startMenuFile) -Force
    Write-Host "- Successfully downloaded and applied custom Start Menu layout."
    Stop-Process -Name StartMenuExperienceHost -Force -ErrorAction SilentlyContinue
    Write-Host "- Start Menu process restarted to apply changes."
} catch {
    Write-Host "- Download of '$startMenuFile' failed. Skipping Start Menu cleanup." -ForegroundColor Yellow
} finally {
    # Clean up the temporary file if it exists
    if (Test-Path $tempFile) {
        Remove-Item $tempFile -Force
    }
}

Write-Host "Customization script finished successfully."