set -e

# Get Ubuntu information
source /etc/os-release

# Require Ubuntu 24.04+
if [[ "$ID" != "ubuntu" ]] || ! dpkg --compare-versions "$VERSION_ID" ge "24.04"; then
    echo "Ubuntu 24.04 or newer is required."
    exit 1
fi

echo "Ubuntu $VERSION_ID detected."

# Install wget if missing
if ! command -v wget >/dev/null 2>&1; then
    sudo apt update
    sudo apt install -y wget
fi

# Download Microsoft package repository configuration
wget \
    "https://packages.microsoft.com/config/ubuntu/$VERSION_ID/packages-microsoft-prod.deb" \
    -O /tmp/packages-microsoft-prod.deb

# Add Microsoft repository
sudo dpkg -i /tmp/packages-microsoft-prod.deb

# Remove downloaded installer
rm /tmp/packages-microsoft-prod.deb

# Refresh repositories
sudo apt update

# Install PowerShell
sudo apt install -y powershell

# Verify installation
pwsh --version