#requires -RunAsAdministrator
$ErrorActionPreference = 'Stop'
$ProgressPreference = 'SilentlyContinue'

function Select-InstallItems {
  $choices = @(
    @{ Id = "everything"; Name = "Everything (with service)" },
    @{ Id = "neovim"; Name = "Neovim" },
    @{ Id = "fd"; Name = "fd" },
    @{ Id = "ripgrep"; Name = "ripgrep" },
    @{ Id = "font"; Name = "JetBrainsMono Nerd Font" },
    @{ Id = "lazyvim"; Name = "LazyVim configuration" }
  )

  if (-not [Environment]::UserInteractive) {
    return [pscustomobject]@{ Cancelled = $false; Selected = @($choices.Id) }
  }

  try {
    Add-Type -AssemblyName System.Windows.Forms -ErrorAction Stop
    Add-Type -AssemblyName System.Drawing -ErrorAction Stop
  } catch {
    Write-Host "Interactive prompt unavailable; defaulting to installing all items."
    return [pscustomobject]@{ Cancelled = $false; Selected = @($choices.Id) }
  }

  $form = New-Object System.Windows.Forms.Form
  $form.Text = "Windows development tools setup"
  $form.StartPosition = "CenterScreen"
  $form.ClientSize = New-Object System.Drawing.Size(430, 310)
  $form.FormBorderStyle = "FixedDialog"
  $form.MaximizeBox = $false
  $form.MinimizeBox = $false
  $form.TopMost = $true

  $label = New-Object System.Windows.Forms.Label
  $label.Text = "Select the items to install:"
  $label.AutoSize = $true
  $label.Location = New-Object System.Drawing.Point(15, 15)
  $form.Controls.Add($label)

  $list = New-Object System.Windows.Forms.CheckedListBox
  $list.CheckOnClick = $true
  $list.Location = New-Object System.Drawing.Point(15, 42)
  $list.Size = New-Object System.Drawing.Size(400, 210)
  for ($i = 0; $i -lt $choices.Count; $i++) {
    [void]$list.Items.Add($choices[$i].Name, $true)
  }
  $form.Controls.Add($list)

  $installButton = New-Object System.Windows.Forms.Button
  $installButton.Text = "Install"
  $installButton.DialogResult = [System.Windows.Forms.DialogResult]::OK
  $installButton.Location = New-Object System.Drawing.Point(255, 267)
  $form.AcceptButton = $installButton
  $form.Controls.Add($installButton)

  $cancelButton = New-Object System.Windows.Forms.Button
  $cancelButton.Text = "Cancel"
  $cancelButton.DialogResult = [System.Windows.Forms.DialogResult]::Cancel
  $cancelButton.Location = New-Object System.Drawing.Point(340, 267)
  $form.CancelButton = $cancelButton
  $form.Controls.Add($cancelButton)

  $dialogResult = $form.ShowDialog()
  if ($dialogResult -ne [System.Windows.Forms.DialogResult]::OK) {
    $form.Dispose()
    return [pscustomobject]@{ Cancelled = $true; Selected = @() }
  }

  $selected = @()
  for ($i = 0; $i -lt $choices.Count; $i++) {
    if ($list.GetItemChecked($i)) {
      $selected += $choices[$i].Id
    }
  }
  $form.Dispose()
  return [pscustomobject]@{ Cancelled = $false; Selected = $selected }
}

$selection = Select-InstallItems
if ($selection.Cancelled) {
  Write-Host "Setup cancelled."
  exit 0
}

$selectedIds = @($selection.Selected)
if (($selectedIds -contains "lazyvim") -and ($selectedIds -notcontains "neovim")) {
  Write-Host "LazyVim requires Neovim; adding Neovim to the installation selection."
  $selectedIds += "neovim"
}
if ($selectedIds.Count -eq 0) {
  Write-Host "No items selected. Nothing to install."
  exit 0
}

try {
  [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12 -bor [Net.SecurityProtocolType]::Tls13
} catch {
  [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
}

$workDir = Join-Path $env:TEMP ("vm-setup-" + (Get-Date -Format "yyyyMMdd-HHmmss"))
New-Item -ItemType Directory -Path $workDir -Force | Out-Null

function Download-File($url, $outFile) {
  Write-Host "Downloading: $url"
  Invoke-WebRequest -Uri $url -OutFile $outFile -UseBasicParsing
}

function Run-Installer($path, $arguments) {
  Write-Host "Installing: $path $arguments"
  $p = Start-Process -FilePath $path -ArgumentList $arguments -Wait -PassThru
  if ($p.ExitCode -ne 0) { throw "Installer failed: $path (ExitCode=$($p.ExitCode))" }
}

function Exists-Any($patterns) {
  foreach ($p in $patterns) {
    if (Test-Path $p) { return $true }
    $m = Get-ChildItem -Path $p -ErrorAction SilentlyContinue
    if ($m) { return $true }
  }
  return $false
}

function Install-WingetPackage($name, $id) {
  Write-Host "`n=== $name ==="
  & winget list --id $id --exact --accept-source-agreements 2>$null | Out-Null
  if ($LASTEXITCODE -eq 0) {
    Write-Host "Already installed. Skipping."
    return
  }

  & winget install --id $id --exact --silent --accept-package-agreements --accept-source-agreements --disable-interactivity
  if ($LASTEXITCODE -ne 0) {
    throw "winget failed to install $name (ExitCode=$LASTEXITCODE)"
  }
}

$apps = @(
  @{
    Id    = "everything"
    Name  = "Everything (with service)"
    Url   = "https://www.voidtools.com/Everything-1.4.1.1030.x64-Setup.exe"
    File  = "Everything-1.4.1.1030.x64-Setup.exe"
    Args  = '/S -install-options "-app-data -install-service -install-start-menu-shortcuts -install-efu-association -install-run-on-system-startup -install-language 1033"'
    Check = @("$env:ProgramFiles\Everything\Everything.exe")
  }
)

try {
  foreach ($app in $apps) {
    if ($selectedIds -notcontains $app.Id) {
      continue
    }

    Write-Host "`n=== $($app.Name) ==="
    if (Exists-Any $app.Check) {
      Write-Host "Already installed. Skipping."
      continue
    }

    $installerPath = Join-Path $workDir $app.File
    Download-File $app.Url $installerPath
    Run-Installer $installerPath $app.Args

    if (-not (Exists-Any $app.Check)) {
      Write-Host "Installer exited OK but detection didn't confirm. Verify manually."
    } else {
      Write-Host "Installed OK."
    }
  }

  $wingetApps = @(
    @{ SelectionId = "neovim"; Name = "Neovim";                 Id = "Neovim.Neovim" },
    @{ SelectionId = "fd";     Name = "fd";                     Id = "sharkdp.fd" },
    @{ SelectionId = "ripgrep"; Name = "ripgrep";               Id = "BurntSushi.ripgrep.MSVC" },
    @{ SelectionId = "font";   Name = "JetBrainsMono Nerd Font"; Id = "DEVCOM.JetBrainsMonoNerdFont" }
  )

  $selectedWingetApps = @($wingetApps | Where-Object { $selectedIds -contains $_.SelectionId })
  if ($selectedWingetApps.Count -gt 0) {
    if (-not (Get-Command winget -ErrorAction SilentlyContinue)) {
      throw "winget is required. Install or update App Installer, then run this script again."
    }

    foreach ($app in $selectedWingetApps) {
      Install-WingetPackage $app.Name $app.Id
    }
  }

  if ($selectedIds -contains "lazyvim") {
    $nvimConfig = Join-Path $env:LOCALAPPDATA "nvim"
    Write-Host "`n=== LazyVim ==="
    $starterZip = Join-Path $workDir "lazyvim-starter.zip"
    $starterDir = Join-Path $workDir "starter"
    Download-File "https://github.com/LazyVim/starter/archive/refs/heads/main.zip" $starterZip
    Expand-Archive -LiteralPath $starterZip -DestinationPath $starterDir -Force

    $starterRoot = Join-Path $starterDir "starter-main"
    if (-not (Test-Path -LiteralPath $starterRoot)) {
      throw "LazyVim starter archive did not contain the expected directory."
    }

    if (Test-Path -LiteralPath $nvimConfig) {
      $backup = "$nvimConfig.backup-$(Get-Date -Format 'yyyyMMdd-HHmmssfff')"
      Copy-Item -LiteralPath $nvimConfig -Destination $backup -Recurse
      Remove-Item -LiteralPath $nvimConfig -Recurse -Force
      Write-Host "Backed up the existing configuration to $backup."
    }

    Copy-Item -LiteralPath $starterRoot -Destination $nvimConfig -Recurse
    Remove-Item -LiteralPath (Join-Path $nvimConfig ".git") -Recurse -Force -ErrorAction SilentlyContinue

    $pluginsDir = Join-Path $nvimConfig "lua\plugins"
    New-Item -ItemType Directory -Path $pluginsDir -Force | Out-Null
    @'
return {
  { import = "lazyvim.plugins.extras.editor.neo-tree" },
  { import = "lazyvim.plugins.extras.editor.telescope" },
  {
    "NeogitOrg/neogit",
    dependencies = {
      "nvim-lua/plenary.nvim",
      "sindrets/diffview.nvim",
    },
    opts = {
      kind = "tab",
    },
    keys = {
      { "<leader>gg", "<cmd>Neogit<cr>", desc = "Git status (Neogit)" },
    },
  },
  {
    "sindrets/diffview.nvim",
    keys = {
      { "<leader>gd", "<cmd>DiffviewOpen<cr>", desc = "Git diff (Diffview)" },
    },
  },
}
'@ | Set-Content -LiteralPath (Join-Path $pluginsDir "git.lua") -Encoding UTF8

    Write-Host "Installed LazyVim starter in $nvimConfig."
  }

  Write-Host "`nDone."
}
finally {
  if (Test-Path $workDir) {
    Remove-Item -LiteralPath $workDir -Recurse -Force -ErrorAction SilentlyContinue
  }
}