#!/usr/bin/env bash
set -e

echo "==============================="
echo " Installing Docker on Ubuntu "
echo "==============================="

# Ensure script is run as root
if [ "$EUID" -ne 0 ]; then
  echo "❌ Please run as root (use sudo)"
  exit 1
fi

echo "➡️ Updating system packages..."
apt update -y

echo "➡️ Installing Docker Engine (docker.io)..."
apt install -y docker.io

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

echo "➡️ Installing Docker Compose v2..."
apt install -y docker-compose-v2

echo "➡️ Adding current user to docker group..."
if [ -n "$SUDO_USER" ]; then
  usermod -aG docker "$SUDO_USER"
  echo "ℹ️ User '$SUDO_USER' added to docker group"
else
  echo "ℹ️ No sudo user detected, skipping group add"
fi

echo "==============================="
echo " ✅ Docker installation complete "
echo "==============================="

echo "Docker version:"
docker --version

echo "Docker Compose version:"
docker compose version

echo ""
echo "⚠️ Log out and log back in to use docker without sudo"