#!/bin/bash
set -e

echo "📦 Updating system packages..."
apt update -y
apt upgrade -y

echo "🔧 Installing dependencies..."
apt install -y apt-transport-https ca-certificates curl gnupg lsb-release

echo "🔑 Adding Docker’s official GPG key..."
curl -fsSL https://download.docker.com/linux/$(. /etc/os-release; echo "$ID")/gpg | gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg

echo "📂 Adding Docker repository..."
echo \
  "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] \
  https://download.docker.com/linux/$(. /etc/os-release; echo "$ID") \
  $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null

echo "📦 Installing Docker Engine and Compose..."
apt update -y
apt install -y docker-ce docker-ce-cli containerd.io docker-compose-plugin

echo "⚙️ Enabling and starting Docker service..."
systemctl enable docker
systemctl start docker

echo "👤 Adding current user to docker group..."
usermod -aG docker $USER

echo "✅ Docker and Docker Compose installation complete!"
echo "➡️ Log out and log back in (or run 'newgrp docker') to use Docker without sudo."
echo
echo "💡 Docker version:"
docker --version || echo "Docker not yet available in current shell"
echo "💡 Docker Compose version:"
docker compose version || echo "Docker Compose not yet available in current shell"