setup.ps1

[CmdletBinding(SupportsShouldProcess)]
param(
	[switch]$SkipRecommendedPackages,
	[switch]$SkipClaudeCode,
	[switch]$SkipProfileBootstrap
)

$ErrorActionPreference = 'Stop'
Set-StrictMode -Version Latest

$requiredPackages = @(
	@{ Id = 'CoreyButler.NVMforWindows'; Description = 'Node version management' }
	@{ Id = 'Ditto.Ditto'; Description = 'Clipboard manager' }
	@{ Id = 'Git.Git'; Description = 'Git CLI' }
	@{ Id = 'GitHub.cli'; Description = 'GitHub CLI' }
	@{ Id = 'GitHub.Copilot'; Description = 'GitHub Copilot CLI tooling' }
	@{ Id = 'GitHub.GitHubDesktop'; Description = 'GitHub Desktop' }
	@{ Id = 'Google.Chrome'; Description = 'Primary browser' }
	@{ Id = 'Microsoft.PowerShell'; Description = 'PowerShell 7' }
	@{ Id = 'Microsoft.PowerToys'; Description = 'Windows productivity tools' }
	@{ Id = 'Microsoft.VisualStudioCode.Insiders'; Description = 'VS Code Insiders' }
	@{ Id = 'Microsoft.VisualStudio.Community.Insiders'; Description = 'Visual Studio Community Insiders' }
	@{ Id = 'Anthropic.Claude'; Description = 'Claude desktop app' }
	@{ Id = 'Microsoft.Teams'; Description = 'Teams client' }
	@{ Id = 'Microsoft.WindowsTerminal'; Description = 'Windows Terminal' }
	@{ Id = 'Microsoft.WSL'; Description = 'Windows Subsystem for Linux' }
	@{ Id = 'Canonical.Ubuntu'; Description = 'Ubuntu for WSL' }
)

$recommendedPackages = @(
	@{ Id = '7zip.7zip'; Description = 'Archive tooling' }
	@{ Id = 'BurntSushi.ripgrep.MSVC'; Description = 'Fast code search' }
	@{ Id = 'sharkdp.fd'; Description = 'Fast file finder' }
	@{ Id = 'jqlang.jq'; Description = 'JSON CLI tooling' }
	@{ Id = 'JesseDuffield.lazygit'; Description = 'TUI for Git' }
	@{ Id = 'Docker.DockerDesktop'; Description = 'Local containers' }
	@{ Id = 'Python.Python.3.13'; Description = 'Python runtime' }
	@{ Id = 'Microsoft.AzureCLI'; Description = 'Azure CLI' }
	@{ Id = 'Bruno.Bruno'; Description = 'API client' }
	@{ Id = 'Neovim.Neovim'; Description = 'Terminal editor' }
)

function Test-CommandAvailable {
	param(
		[Parameter(Mandatory)]
		[string]$Name
	)

	return [bool](Get-Command -Name $Name -ErrorAction SilentlyContinue)
}

function Sync-PathEnvironment {
	$machinePath = [Environment]::GetEnvironmentVariable('Path', 'Machine')
	$userPath = [Environment]::GetEnvironmentVariable('Path', 'User')
	$combinedPaths = @($machinePath, $userPath) -join ';'

	$env:Path = (
		$combinedPaths -split ';' |
		Where-Object { -not [string]::IsNullOrWhiteSpace($_) } |
		Select-Object -Unique
	) -join ';'

	foreach ($name in 'NVM_HOME', 'NVM_SYMLINK') {
		$userValue = [Environment]::GetEnvironmentVariable($name, 'User')
		$machineValue = [Environment]::GetEnvironmentVariable($name, 'Machine')
		$value = if ($userValue) { $userValue } else { $machineValue }

		if ($value) {
			Set-Item -Path "Env:$name" -Value $value
		}
	}
}

function Test-WingetPackageInstalled {
	param(
		[Parameter(Mandatory)]
		[string]$Id
	)

	$output = & winget list --id $Id --exact --accept-source-agreements 2>$null | Out-String
	return $LASTEXITCODE -eq 0 -and $output -match [regex]::Escape($Id)
}

function Install-WingetPackage {
	param(
		[Parameter(Mandatory)]
		[hashtable]$Package
	)

	if (Test-WingetPackageInstalled -Id $Package.Id) {
		Write-Host "Skipping $($Package.Id); already installed." -ForegroundColor DarkGray
		return
	}

	if ($PSCmdlet.ShouldProcess($Package.Id, 'Install package with winget')) {
		Write-Host "Installing $($Package.Id) ($($Package.Description))..." -ForegroundColor Yellow
		& winget install --id $Package.Id --exact --source winget --accept-package-agreements --accept-source-agreements --silent

		if ($LASTEXITCODE -ne 0) {
			throw "winget install failed for $($Package.Id)."
		}
	}
}

function Install-LatestNode {
	if (-not (Test-CommandAvailable -Name 'nvm')) {
		throw 'nvm is not available in the current session. Re-run this script in a new PowerShell window if the refresh step did not pick it up.'
	}

	if ($PSCmdlet.ShouldProcess('Node.js', 'Install latest version via nvm')) {
		Write-Host 'Installing latest Node.js via nvm...' -ForegroundColor Yellow
		& nvm install latest

		if ($LASTEXITCODE -ne 0) {
			throw 'nvm install latest failed.'
		}

		Write-Host 'Activating latest Node.js via nvm...' -ForegroundColor Yellow
		& nvm use latest

		if ($LASTEXITCODE -ne 0) {
			throw 'nvm use latest failed.'
		}

		Sync-PathEnvironment

		if (Test-CommandAvailable -Name 'node') {
			$nodeVersion = (& node --version).Trim()
			Write-Host "Active Node.js version: $nodeVersion" -ForegroundColor Green
		} else {
			Write-Warning 'Node.js was installed, but node is not yet visible in this session. Open a new PowerShell window if needed.'
		}
	}
}

function Install-ClaudeCode {
	if (Test-CommandAvailable -Name 'claude') {
		Write-Host 'Skipping Claude Code CLI; command is already available.' -ForegroundColor DarkGray
		return
	}

	if ($PSCmdlet.ShouldProcess('Claude Code CLI', 'Install from official script')) {
		Write-Host 'Installing Claude Code CLI...' -ForegroundColor Yellow
		Invoke-RestMethod https://claude.ai/install.ps1 | Invoke-Expression
	}
}

if (-not (Test-CommandAvailable -Name 'winget')) {
	throw 'winget is required but was not found on PATH.'
}

Write-Host 'Installing required winget packages...' -ForegroundColor Cyan
foreach ($package in $requiredPackages) {
	Install-WingetPackage -Package $package
}

if ($SkipRecommendedPackages) {
	Write-Host 'Skipping recommended developer packages.' -ForegroundColor Cyan
	Write-Host 'Recommended package set:' -ForegroundColor DarkGray
	foreach ($package in $recommendedPackages) {
		Write-Host "  $($package.Id) - $($package.Description)" -ForegroundColor DarkGray
	}
} else {
	Write-Host 'Installing recommended developer packages...' -ForegroundColor Cyan
	foreach ($package in $recommendedPackages) {
		Install-WingetPackage -Package $package
	}
}

Sync-PathEnvironment
Install-LatestNode

if (-not $SkipClaudeCode) {
	Install-ClaudeCode
}

if (-not $SkipProfileBootstrap) {
	$setupProfileScript = Join-Path $PSScriptRoot 'setup-profile.ps1'
	if (Test-Path $setupProfileScript) {
		if ($PSCmdlet.ShouldProcess($setupProfileScript, 'Bootstrap PowerShell profile')) {
			& $setupProfileScript -WhatIf:$WhatIfPreference
		}
	} else {
		Write-Warning "Could not find $setupProfileScript"
	}
}

Write-Host 'Bootstrap complete.' -ForegroundColor Green