discord-game-price-bot/install.ps1

param(
  [switch]$LinkSteamProfiles,
  [string]$BulkOwn,
  [switch]$SyncRoles
)

$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
Set-Location $ScriptDir

if ($LinkSteamProfiles) {
  if (-not (Test-Path ".env")) {
    Write-Host "No .env found yet. Run install.ps1 (with no flag) first to set up your keys."
    exit 1
  }
  Write-Host "Installing dependencies..."
  npm install --silent
  node link-steam-profiles.js
  exit 0
}

if ($BulkOwn) {
  if (-not (Test-Path ".env")) {
    Write-Host "No .env found yet. Run install.ps1 (with no flag) first to set up your keys."
    exit 1
  }
  Write-Host "Installing dependencies..."
  npm install --silent
  node bulk-own.js $BulkOwn
  exit 0
}

if ($SyncRoles) {
  if (-not (Test-Path ".env")) {
    Write-Host "No .env found yet. Run install.ps1 (with no flag) first to set up your keys."
    exit 1
  }
  Write-Host "Installing dependencies..."
  npm install --silent
  node sync-roles.js
  exit 0
}

function Test-CommandExists($name) {
  return $null -ne (Get-Command $name -ErrorAction SilentlyContinue)
}

function Get-EnvValue($key) {
  if (-not (Test-Path ".env")) { return "" }
  $line = Select-String -Path ".env" -Pattern "^$key=" -ErrorAction SilentlyContinue | Select-Object -First 1
  if ($line) {
    return ($line.Line -replace "^$key=", "")
  }
  return ""
}

function Set-EnvValue($key, $value) {
  $lines = @()
  if (Test-Path ".env") {
    $lines = Get-Content ".env" | Where-Object { $_ -notmatch "^$key=" }
  }
  $lines += "$key=$value"
  $lines | Set-Content ".env"
}

if (-not (Test-CommandExists node)) {
  Write-Host "Node.js was not found."
  if (Test-CommandExists winget) {
    $answer = Read-Host "Install it now with winget? [y/N]"
    if ($answer -eq "y" -or $answer -eq "Y") {
      winget install --id OpenJS.NodeJS.LTS -e
      Write-Host "Node.js was installed. Close and reopen this terminal, then run install.ps1 again."
      exit 0
    } else {
      Write-Host "Install Node.js from https://nodejs.org and run this script again."
      exit 1
    }
  } else {
    Write-Host "Install Node.js from https://nodejs.org and run this script again."
    exit 1
  }
}

Write-Host "Installing dependencies..."
npm install --silent

if (-not (Test-Path ".env")) {
  Copy-Item ".env.example" ".env"
}

if (-not (Test-Path "games.json")) {
  Copy-Item "games.json.example" "games.json"
}

if (-not (Test-Path "owned.json")) {
  Copy-Item "owned.json.example" "owned.json"
}

if (-not (Test-Path "steamlinks.json")) {
  Copy-Item "steamlinks.json.example" "steamlinks.json"
}

if (-not (Test-Path "memory.yaml")) {
  Copy-Item "memory.yaml.example" "memory.yaml"
}

if (-not (Test-Path "scores.json")) {
  Copy-Item "scores.json.example" "scores.json"
}

if (-not (Test-Path "roles.yaml")) {
  Copy-Item "roles.yaml.example" "roles.yaml"
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "DISCORD_BOT_TOKEN"))) {
  Write-Host ""
  Write-Host "Need a Discord bot token:"
  Write-Host "1. Go to https://discord.com/developers/applications and sign in"
  Write-Host "2. Click New Application, name it anything, create it"
  Write-Host "3. Click Bot in the left sidebar"
  Write-Host "4. IMPORTANT - scroll to Privileged Gateway Intents, turn ON both Message Content Intent and Server Members Intent, then click the Save Changes button at the bottom."
  Write-Host "   The toggles by themselves do not save. If you skip Save Changes, the bot will fail to log in with a disallowed intents error."
  Write-Host "5. Click Reset Token (or Copy) and copy the token"
  Write-Host "6. Click OAuth2 > URL Generator, check bot under Scopes, then check Send Messages, Read Message History, and Manage Roles under Bot Permissions"
  Write-Host "7. IMPORTANT - a URL appears at the bottom of that page. Copy it, paste it into your web browser, press Enter, pick your server, and click Authorize."
  Write-Host "   Creating the bot does NOT add it to your server. It will not show up in your server until you open that URL and authorize it there."
  Write-Host ""
  $tokenInput = Read-Host "Paste your Discord bot token"
  Set-EnvValue "DISCORD_BOT_TOKEN" $tokenInput
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "ITAD_API_KEY"))) {
  Write-Host ""
  Write-Host "Need a free IsThereAnyDeal API key:"
  Write-Host "1. Register a free user account at https://isthereanydeal.com"
  Write-Host "2. Go to https://isthereanydeal.com/apps/"
  Write-Host "3. Click Register App, name it anything, submit"
  Write-Host "4. Copy the API key it shows you"
  Write-Host ""
  $keyInput = Read-Host "Paste your IsThereAnyDeal API key"
  Set-EnvValue "ITAD_API_KEY" $keyInput
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "STEAM_API_KEY"))) {
  Write-Host ""
  Write-Host "Optional - only needed for !my-steam-profile and automatic Steam ownership counts on !info:"
  Write-Host "1. Go to https://steamcommunity.com/dev/apikey and sign in"
  Write-Host "2. Enter any domain name (e.g. localhost), agree to the terms, and submit"
  Write-Host "3. Copy the API key it shows you"
  Write-Host ""
  $steamKeyInput = Read-Host "Paste your Steam Web API key, or press Enter to skip"
  Set-EnvValue "STEAM_API_KEY" $steamKeyInput
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "TAVILY_API_KEY"))) {
  Write-Host ""
  Write-Host "Optional - lets the AI chat look things up on the web (like exact item locations or boss strategies) instead of only relying on what the AI model already knows. Free, no credit card required:"
  Write-Host "1. Go to https://app.tavily.com/ and sign up"
  Write-Host "2. Copy the API key it shows you (starts with tvly-)"
  Write-Host ""
  $tavilyKeyInput = Read-Host "Paste your Tavily API key, or press Enter to skip"
  Set-EnvValue "TAVILY_API_KEY" $tavilyKeyInput
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "ITAD_COUNTRY"))) {
  Write-Host ""
  Write-Host "Optional - sets which country's prices and currency to use (2-letter code, e.g. US, GB, DE, CA, AU). Defaults to US if you skip this."
  Write-Host ""
  $itadCountryInput = Read-Host "Country code, or press Enter for US"
  if ([string]::IsNullOrWhiteSpace($itadCountryInput)) { $itadCountryInput = "US" }
  Set-EnvValue "ITAD_COUNTRY" $itadCountryInput
}

if ([string]::IsNullOrWhiteSpace((Get-EnvValue "DOCKER_SSH_HOST"))) {
  Write-Host ""
  Write-Host "Optional - lets people with a specific Discord role run !docker ps and !docker restart CONTAINER to manage Docker containers on a remote server over SSH (e.g. an Unraid box running your game servers). Requires SSH key access to that server already set up:"
  Write-Host ""
  $dockerHostInput = Read-Host "SSH host for the Docker server (e.g. chappie.local), or press Enter to skip"
  if (-not [string]::IsNullOrWhiteSpace($dockerHostInput)) {
    Set-EnvValue "DOCKER_SSH_HOST" $dockerHostInput
    $dockerUserInput = Read-Host "SSH username (e.g. root)"
    Set-EnvValue "DOCKER_SSH_USER" $dockerUserInput
    $dockerKeyInput = Read-Host "Path to the SSH private key to use"
    Set-EnvValue "DOCKER_SSH_KEY_PATH" $dockerKeyInput
    $dockerRoleInput = Read-Host "Discord role ID allowed to use !docker (right-click the role in Discord, Copy Role ID)"
    Set-EnvValue "DOCKER_ROLE_ID" $dockerRoleInput
    $dockerProtectedInput = Read-Host "Container names to protect from !docker restart, comma-separated (e.g. Plex-Media-Server), or press Enter for none"
    Set-EnvValue "DOCKER_PROTECTED_CONTAINERS" $dockerProtectedInput
  } else {
    Write-Host "Skipping - !docker won't be available. Run this script again any time to set it up."
  }
}

if (-not (Test-CommandExists ollama)) {
  Write-Host ""
  $aiAnswer = Read-Host "Want the bot to use AI to give smarter, in-character answers when people chat with it? This is free - it installs Ollama and runs a small AI model locally on this machine, no API key or cost involved. [y/N]"
  if ($aiAnswer -eq "y" -or $aiAnswer -eq "Y") {
    if (Test-CommandExists winget) {
      Write-Host "Installing Ollama..."
      winget install --id Ollama.Ollama -e
    } else {
      Write-Host "Install it yourself from https://ollama.com/download, then run this script again to finish setting it up."
    }
  } else {
    Write-Host "Skipping - the bot won't reply to chat mentions. Run this script again any time to set it up."
  }
}

if (Test-CommandExists ollama) {
  $currentModel = Get-EnvValue "OLLAMA_MODEL"
  if ([string]::IsNullOrWhiteSpace($currentModel)) {
    $currentModel = "llama3.2"
    Set-EnvValue "OLLAMA_MODEL" $currentModel
  }

  $modelList = & ollama list 2>$null
  $modelFound = $false
  foreach ($line in $modelList) {
    if ($line -match [regex]::Escape($currentModel)) {
      $modelFound = $true
      break
    }
  }

  if (-not $modelFound) {
    Write-Host ""
    Write-Host "Downloading the $currentModel model for the bot to use (one-time, may take a few minutes)..."
    ollama pull $currentModel
    if ($LASTEXITCODE -ne 0) {
      Write-Host "Could not download the model automatically - make sure Ollama is running, then run: ollama pull $currentModel"
    }
  }
}

if (-not (Test-CommandExists pm2)) {
  Write-Host ""
  Write-Host "Installing pm2 to keep the bot running in the background..."
  npm install -g pm2
}

$AppName = "discord-bot"

pm2 delete $AppName 2>$null | Out-Null
pm2 start index.js --name $AppName --cwd $ScriptDir --output "$ScriptDir\bot.log" --error "$ScriptDir\bot.error.log"
pm2 save

$taskName = "PM2 Resurrect - $AppName"
$existingTask = Get-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
if (-not $existingTask) {
  $pm2Command = Get-Command pm2 -ErrorAction SilentlyContinue
  if ($pm2Command) {
    $action = New-ScheduledTaskAction -Execute $pm2Command.Source -Argument "resurrect"
    $trigger = New-ScheduledTaskTrigger -AtLogOn
    Register-ScheduledTask -TaskName $taskName -Action $action -Trigger $trigger -Description "Restores PM2-managed apps (including $AppName) at logon" | Out-Null
  }
}

Write-Host ""
Write-Host "Installed and running in the background."
Write-Host "View logs: pm2 logs $AppName"
Write-Host "Check status: pm2 status"
Write-Host "To stop it: pm2 stop $AppName"
Write-Host "To restart it after editing index.js: powershell -ExecutionPolicy Bypass -File install.ps1"