RepoHub/src/lib/scriptGenerator.ts

import { GeneratedScript, Platform, SelectedPackage } from '@/types'

export function generateScript(packages: SelectedPackage[], platform: Platform): GeneratedScript {
  const script = createScriptContent(packages, platform)

  return {
    platform: platform.id,
    script,
    packages,
    generatedAt: new Date().toISOString()
  }
}

function createScriptContent(packages: SelectedPackage[], platform: Platform): string {
  switch (platform.id) {
    case 'debian':
    case 'ubuntu':
      return generateDebianBasedScript(packages, platform.name)
    case 'fedora':
      return generateFedoraScript(packages)
    case 'arch':
      return generateArchScript(packages)
    case 'windows':
      return generateWindowsScript(packages)
    case 'macos':
      return generateMacOSScript(packages)
    default:
      throw new Error(`Unsupported platform: ${platform.id}`)
  }
}

function generateDebianBasedScript(packages: SelectedPackage[], platformName: string): string {
  const packageNames = packages.map(p => p.name).join(' ')

  return `#!/bin/bash

# RepoHub Installation Script for ${platformName}
# Generated on ${new Date().toISOString()}
# This script is idempotent and safe to run multiple times

set -e

echo "Starting package installation for ${platformName}..."

# Update package lists
echo "Updating package lists..."
sudo apt update

# Install packages
echo "Installing packages: ${packageNames}"
sudo apt install -y ${packageNames}

# Verify installation
echo "Verifying installation..."
for package in ${packageNames}; do
    if dpkg -l | grep -q "^ii  $package "; then
        echo "✓ $package installed successfully"
    else
        echo "✗ $package installation failed"
    fi
done

echo "Installation completed!"`
}

function generateFedoraScript(packages: SelectedPackage[]): string {
  const packageNames = packages.map(p => p.name).join(' ')

  return `#!/bin/bash

# RepoHub Installation Script for Fedora
# Generated on ${new Date().toISOString()}
# This script is idempotent and safe to run multiple times

set -e

echo "Starting package installation for Fedora..."

# Update package lists
echo "Updating package lists..."
sudo dnf update -y

# Install packages
echo "Installing packages: ${packageNames}"
sudo dnf install -y ${packageNames}

# Verify installation
echo "Verifying installation..."
for package in ${packageNames}; do
    if rpm -q $package >/dev/null 2>&1; then
        echo "✓ $package installed successfully"
    else
        echo "✗ $package installation failed"
    fi
done

echo "Installation completed!"`
}

function generateArchScript(packages: SelectedPackage[]): string {
  const packageNames = packages.map(p => p.name).join(' ')

  return `#!/bin/bash

# RepoHub Installation Script for Arch Linux
# Generated on ${new Date().toISOString()}
# This script is idempotent and safe to run multiple times

set -e

echo "Starting package installation for Arch Linux..."

AUR_HELPER=""

# Check for paru or yay
if command -v paru &> /dev/null; then
    echo "paru detected."
    AUR_HELPER="paru"
elif command -v yay &> /dev/null; then
    echo "yay detected."
    AUR_HELPER="yay"
else
    echo "Neither paru nor yay is installed. Installing yay..."

    # Install dependencies
    echo "Installing dependencies..."
    sudo pacman -S --needed --noconfirm git base-devel

    # Clone and build yay
    echo "Building yay..."
    git clone https://aur.archlinux.org/yay-bin.git
    cd yay-bin
    makepkg -si --noconfirm
    cd ..
    rm -rf yay-bin

    echo "yay installed successfully!"
    AUR_HELPER="yay"
fi

# Update package lists
echo "Updating package lists with $AUR_HELPER..."
$AUR_HELPER -Sy --noconfirm

# Install packages
echo "Installing packages: ${packageNames}"
$AUR_HELPER -S --noconfirm ${packageNames}

# Verify installation
echo "Verifying installation..."
for package in ${packageNames}; do
    if pacman -Qi $package >/dev/null 2>&1; then
        echo "✓ $package installed successfully"
    else
        echo "✗ $package installation failed"
    fi
done

echo "Installation completed!"`
}

function generateWindowsScript(packages: SelectedPackage[]): string {
  const packageNames = packages.map(p => p.name).join(' ')

  return `# RepoHub Installation Script for Windows
# Generated on ${new Date().toISOString()}
# This script is idempotent and safe to run multiple times

Write-Host 'Starting package installation for Windows...'

# Try to unblock this script (no-op if not needed)
try { Unblock-File -Path $MyInvocation.MyCommand.Path -ErrorAction SilentlyContinue } catch {}

# Warn if not running as Administrator (some installs may require elevation)
$isAdmin = ([Security.Principal.WindowsPrincipal] [Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
if (-not $isAdmin) {
    Write-Warning "It's recommended to run this script as Administrator for best results."
}

# Ensure winget is available
if (-not (Get-Command winget -ErrorAction SilentlyContinue)) {
    Write-Warning 'Windows Package Manager (winget) is not installed.'
    Write-Host "We'll open the Microsoft Store page for 'App Installer' (includes winget)."
    Write-Host 'After installation completes, please re-run this script.'
    try {
        Start-Process "ms-windows-store://pdp/?productId=9NBLGGH4NNS1"
    } catch {
        Write-Host "If the Store didn't open, install 'App Installer' manually from: https://aka.ms/getwinget"
    }
    exit 1
}

# Install packages using winget
Write-Host 'Installing packages: ${packageNames}'

$packages = @(${packages.map(p => `'${p.name}'`).join(', ')})

foreach ($package in $packages) {
    Write-Host "Installing $package..."
    try {
        winget install --id $package --accept-package-agreements --accept-source-agreements -e
        if ($LASTEXITCODE -ne 0) {
            Write-Host "[ERROR] $package installation failed with exit code $LASTEXITCODE"
        } else {
            Write-Host "[OK] $package installed successfully"
        }
    } catch {
        Write-Host "[ERROR] $package installation failed: $_"
    }
}

Write-Host 'Installation completed!'`
}

function generateMacOSScript(packages: SelectedPackage[]): string {
  const packageNames = packages.map(p => p.name).join(' ')

  return `#!/bin/bash

# RepoHub Installation Script for macOS
# Generated on ${new Date().toISOString()}
# This script is idempotent and safe to run multiple times

set -e

echo "Starting package installation for macOS..."

# Check if Homebrew is installed
if ! command -v brew &> /dev/null; then
    echo "Homebrew is not installed. Installing Homebrew..."
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
else
    echo "Homebrew is already installed. Updating..."
    brew update
fi

# Install packages
echo "Installing packages: ${packageNames}"
brew install ${packageNames}

# Verify installation
echo "Verifying installation..."
for package in ${packageNames}; do
    if brew list --formula | grep -q "^$package$"; then
        echo "✓ $package installed successfully"
    else
        echo "✗ $package installation failed"
    fi
done

echo "Installation completed!"
}`
}