setup_instance_management.ps1

# Define installation directory and log file path
$InstallDir = "C:\CloudRigSetup"
$LogFilePath = "C:\Program Files\setup-log.txt"
$RestartFlagPath = "$InstallDir\restart-flag.txt"
$CounterFilePath = "$InstallDir\instance-counter.txt"  # File to keep track of instance numbers

# Function to log messages
function Log-Message {
    param (
        [string]$Message
    )
    $logEntry = "$Message"
    $logEntry | Out-File -FilePath $LogFilePath -Append
}

# Log starting the script
Log-Message "Script started."

# Check if the script has been restarted already
if (Test-Path -Path $RestartFlagPath) {
    Log-Message "Restart flag detected. Continuing script after restart."
    Remove-Item -Path $RestartFlagPath -Force
} else {
    # Ensure the script is running as administrator
    if (-not ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) {
        Log-Message "The script is not running as Administrator. Restarting with elevated privileges..."
        
        # Create a restart flag
        New-Item -Path $RestartFlagPath -ItemType File -Force | Out-Null
        
        Start-Process powershell "-File `"$PSCommandPath`"" -Verb RunAs
        exit
    }
}

# Create setup directory
try {
    New-Item -Path $InstallDir -ItemType Directory -Force
    Log-Message "Setup directory created at $InstallDir."
} catch {
    Log-Message "Failed to create setup directory. Error: $_"
    exit 1
}

# Initialize or read instance number
if (-not (Test-Path -Path $CounterFilePath)) {
    # If the file does not exist, create it and set the initial instance number to 1
    1 | Out-File -FilePath $CounterFilePath
    $instanceNumber = 1
    Log-Message "Counter file not found. Starting with instance number $instanceNumber."
} else {
    # Read the last instance number from the file
    $instanceNumber = Get-Content -Path $CounterFilePath
    $instanceNumber = [int]$instanceNumber + 1
    Log-Message "Read last instance number from counter file. Incrementing to $instanceNumber."
}

# Save the new instance number back to the file
$instanceNumber | Out-File -FilePath $CounterFilePath -Force

# Generate a random 10-character password
function Generate-RandomPassword {
    $chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%^&*()_-+=<>?'
    $password = -join ((65..90) + (97..122) + (48..57) + (33..47) | Get-Random -Count 10 | ForEach-Object {[char]$_})
    return $password
}

# Generate the password and store it in $AdminPassword
$AdminPassword = Generate-RandomPassword

# Define the instance name using the incremented number
$instanceName = "hb_gaming_$instanceNumber"  # Unique instance name based on the incremented number

# Install Chocolatey if not installed
if (-not (Get-Command choco.exe -ErrorAction SilentlyContinue)) {
    try {
        Log-Message "Chocolatey not found. Installing Chocolatey..."
        Set-ExecutionPolicy Bypass -Scope Process -Force;
        [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.SecurityProtocolType]::Tls12;
        iex ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
        Log-Message "Chocolatey installed."
    } catch {
        Log-Message "Chocolatey installation failed. Error: $_"
        exit 1
    }
}

# Install necessary software using Chocolatey
try {
    Log-Message "Installing Node.js, AWS CLI, Git, and Nginx using Chocolatey..."
    choco install -y nodejs awscli git nginx | Out-Null
    Log-Message "Node.js, AWS CLI, Git, and Nginx installed."
} catch {
    Log-Message "Failed to install one or more applications via Chocolatey. Error: $_"
}

$env:PATH = [System.Environment]::GetEnvironmentVariable("PATH", [System.EnvironmentVariableTarget]::Machine)

# Install Steam
try {
    Log-Message "Installing Steam..."
    Invoke-WebRequest -Uri "https://cdn.cloudflare.steamstatic.com/client/installer/SteamSetup.exe" -OutFile "$InstallDir\SteamSetup.exe"
    Start-Process "$InstallDir\SteamSetup.exe" -ArgumentList "/silent /install /S" -Wait
    Log-Message "Steam installed successfully."
} catch {
    Log-Message "Steam installation failed. Error: $_"
}

# Install Sunshine
try {
    Log-Message "Installing Sunshine..."
    Invoke-WebRequest -Uri "https://github.com/LizardByte/Sunshine/releases/download/v0.23.1/sunshine-windows-installer.exe" -OutFile "$InstallDir\sunshine-installer.exe"
    Start-Process "$InstallDir\sunshine-installer.exe" -ArgumentList '/SILENT /install /S' -Wait
    Log-Message "Sunshine installed successfully."
} catch {
    Log-Message "Sunshine installation failed. Error: $_"
}

# Download NVIDIA drivers using a manual method
try {
    Log-Message "Downloading NVIDIA drivers..."
    $request = [System.Net.WebRequest]::Create("https://github.com/Chandirasegaran/node-react-learn/releases/download/n2.0/NVIDIA.zip")
    $response = $request.GetResponse()
    $stream = $response.GetResponseStream()
    $outputFile = "$InstallDir\nvidia-drivers.zip"
    $file = New-Object System.IO.FileStream($outputFile, [System.IO.FileMode]::Create, [System.IO.FileAccess]::Write)
    
    $buffer = New-Object byte[] 4096
    while (($bytesRead = $stream.Read($buffer, 0, $buffer.Length)) -gt 0) {
        $file.Write($buffer, 0, $bytesRead)
    }

    $file.Close()
    $stream.Close()
    $response.Close()
    Log-Message "NVIDIA drivers downloaded."

    Log-Message "Installing NVIDIA drivers silently..."
    Expand-Archive -Path "$InstallDir\nvidia-drivers.zip" -DestinationPath "$InstallDir\nvidia-drivers"
    Log-Message "Nvidia drivers Extracted and Installation Started."
    Start-Process "$InstallDir\nvidia-drivers\setup.exe" -ArgumentList '-s -n' -Wait
    Log-Message "Nvidia drivers installed."
} catch {
    Log-Message "NVIDIA drivers installation failed. Error: $_"
}

# Server Setup
try {
    Log-Message "Setting up server..."
    $env:PATH = [System.Environment]::GetEnvironmentVariable("PATH", [System.EnvironmentVariableTarget]::Machine)

    # Download the server zip file from the specified URL
    Invoke-WebRequest -Uri "https://github.com/Chandirasegaran/node-react-learn/releases/download/server1.0/server.zip" -OutFile "$InstallDir\server.zip"

    # Extract the server.zip file to the specified directory
    Expand-Archive -Path "$InstallDir\server.zip" -DestinationPath "$InstallDir\server" -Force

    # Change directory to the server folder
    Set-Location -Path "$InstallDir\server"

    # Install npm dependencies
    npm install | Out-Null
    Log-Message "Server setup completed successfully."
} catch {
    Log-Message "Server setup failed. Error: $_"
}

# Install AWS CDK
try {
    Log-Message "Installing AWS CDK..."
    $env:PATH = [System.Environment]::GetEnvironmentVariable("PATH", [System.EnvironmentVariableTarget]::Machine)

    npm install -g aws-cdk | Out-Null
    Log-Message "AWS CDK installed successfully."
} catch {
    Log-Message "AWS CDK installation failed. Error: $_"
}

# Configure Windows Firewall rules for Sunshine
try {
    Log-Message "Configuring Windows Firewall rules for Sunshine..."
    New-NetFirewallRule -DisplayName "Sunshine TCP Inbound" -Direction Inbound -Protocol TCP -LocalPort 47984 -Action Allow
    New-NetFirewallRule -DisplayName "Sunshine UDP Inbound" -Direction Inbound -Protocol UDP -LocalPort 47989 -Action Allow
    New-NetFirewallRule -DisplayName "Sunshine TCP Outbound" -Direction Outbound -Protocol TCP -LocalPort 47984 -Action Allow
    New-NetFirewallRule -DisplayName "Sunshine UDP Outbound" -Direction Outbound -Protocol UDP -LocalPort 47989 -Action Allow
    Log-Message "Windows Firewall rules configured for Sunshine."
} catch {
    Log-Message "Failed to configure Windows Firewall rules. Error: $_"
}

# Install and start Node.js server using PM2
try {
    Log-Message "Installing and starting Node.js server using PM2..."

    $env:Path += ";C:\Program Files\nodejs;C:\Users\Administrator\AppData\Roaming\npm"

    $env:PATH = [System.Environment]::GetEnvironmentVariable("PATH", [System.EnvironmentVariableTarget]::Machine)

    npm install pm2 -g | Out-Null

    $env:PATH = [System.Environment]::GetEnvironmentVariable("PATH", [System.EnvironmentVariableTarget]::Machine)

    pm2 start C:\CloudRigSetup\server\server.js
    pm2 save  # Save PM2 process list and re-deploy on startup
    Log-Message "Node.js server started using PM2."
} catch {
    Log-Message "Node.js server setup failed. Error: $_"
}

# Configure Nginx as reverse proxy
try {
    Log-Message "Configuring Nginx as reverse proxy..."
    Stop-Process -Name nginx -Force -ErrorAction SilentlyContinue
    Start-Process "C:\tools\nginx-1.27.1\nginx.exe"
    Log-Message "Nginx configured as reverse proxy."
} catch {
    Log-Message "Nginx configuration failed. Error: $_"
}



# Define the API endpoint and JSON payload
$endpoint = "https://password-store-aws.onrender.com/store-data"
$body = @{
    instance_number = $instanceNumber  # Use dynamically generated instance name
    password = $AdminPassword
}
$jsonBody = $body | ConvertTo-Json
$headers = @{
    "Content-Type" = "application/json"
}

# Send the POST request to store the password and instance name
try {
    $response = Invoke-RestMethod -Uri $endpoint -Method Post -Body $jsonBody -Headers $headers
    Log-Message "Password posted successfully for instance $instanceName. Setting Windows password now."
    Log-Message $instanceNumber
    Log-Message "Admin Password $AdminPassword"
    # Set the password for the Administrator account
    Start-Process net -ArgumentList "user", "Administrator", $AdminPassword -NoNewWindow -Wait
    Log-Message "Password set for Administrator account."
} catch {
    Log-Message "An error occurred while posting the data for instance $instanceName: $_.Exception.Message"
    Pause
}

# Set up Windows AutoLogon
try {
    Log-Message "Setting up Windows AutoLogon..."
    Invoke-WebRequest -Uri "https://live.sysinternals.com/Autologon.exe" -OutFile "$InstallDir\Autologon.exe"
    Set-ItemProperty -Path "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System" -Name "DisableCAD" -Value 1
    Start-Process "$InstallDir\Autologon.exe" -ArgumentList "Administrator", ".\", $AdminPassword, "/accepteula" -NoNewWindow -Wait
    Log-Message "Windows AutoLogon set up for instance $instanceName."
} catch {
    Log-Message "Windows AutoLogon setup failed for instance $instanceName. Error: $_"
}

# Log script completion
Log-Message "Setup complete for instance $instanceName. The instance is fully configured and ready."