From d560acac3ab89ce76a0fda5095eb6e8914f9be30 Mon Sep 17 00:00:00 2001
From: guruswarupa <msgswarupa@gmail.com>
Date: Fri, 13 Sep 2024 14:08:30 +0530
Subject: [PATCH] docker setup added

---
 tabs/system-setup/6-docker-setup.sh | 87 +++++++++++++++++++++++++++++
 tabs/system-setup/tab_data.toml     |  4 ++
 2 files changed, 91 insertions(+)
 create mode 100755 tabs/system-setup/6-docker-setup.sh

diff --git a/tabs/system-setup/6-docker-setup.sh b/tabs/system-setup/6-docker-setup.sh
new file mode 100755
index 00000000..7f2a0b3f
--- /dev/null
+++ b/tabs/system-setup/6-docker-setup.sh
@@ -0,0 +1,87 @@
+#!/bin/sh -e
+
+. ../common-script.sh
+
+# Function to prompt the user for installation choice
+choose_installation() {
+    clear
+    printf "%b\n" "${YELLOW}Choose what to install:${RC}"
+    printf "%b\n" "1. ${YELLOW}Docker${RC}"
+    printf "%b\n" "2. ${YELLOW}Docker Compose${RC}"
+    printf "%b\n" "3. ${YELLOW}Both${RC}"
+    read -p "Enter your choice [1-3]: " CHOICE
+
+    case "$CHOICE" in
+        1) INSTALL_DOCKER=1; INSTALL_COMPOSE=0 ;;
+        2) INSTALL_DOCKER=0; INSTALL_COMPOSE=1 ;;
+        3) INSTALL_DOCKER=1; INSTALL_COMPOSE=1 ;;
+        *) echo "Invalid choice. Exiting."; exit 1 ;;
+    esac
+}
+
+install_docker() {
+    printf "%b\n" "${YELLOW}Installing Docker...${RC}"
+    case $PACKAGER in
+        apt-get | yum)
+            curl -fsSL https://get.docker.com | sh 
+            ;;
+        zypper)
+            $ESCALATION_TOOL ${PACKAGER} --non-interactive install docker
+            $ESCALATION_TOOL systemctl enable docker
+            $ESCALATION_TOOL systemctl start docker
+            ;;
+        pacman)
+            $ESCALATION_TOOL ${PACKAGER} -S --noconfirm docker
+            $ESCALATION_TOOL systemctl enable docker
+            $ESCALATION_TOOL systemctl start docker
+            ;;
+        *)
+            printf "${RED}Unsupported package manager. Please install Docker manually.${RC}\n"
+            exit 1
+            ;;
+    esac
+}
+
+install_docker_compose() {
+    printf "%b\n" "${YELLOW}Installing Docker Compose...${RC}"
+    case $PACKAGER in
+        apt-get | yum)
+            $ESCALATION_TOOL ${PACKAGER} update
+            $ESCALATION_TOOL ${PACKAGER} install -y docker-compose-plugin
+            ;;
+        zypper)
+            $ESCALATION_TOOL ${PACKAGER} --non-interactive install docker-compose
+            ;;
+        pacman)
+            $ESCALATION_TOOL ${PACKAGER} -S --noconfirm docker-compose
+            ;;
+        *)
+            printf "${RED}Unsupported package manager. Please install Docker Compose manually.${RC}\n"
+            exit 1
+            ;;
+    esac
+}
+
+install_components() {
+    choose_installation 
+
+    if [ "$INSTALL_DOCKER" -eq 1 ]; then
+        if ! command_exists docker; then
+            install_docker
+        else
+            printf "%b\n" "${GREEN}Docker is already installed.${RC}"
+        fi
+    fi
+
+    if [ "$INSTALL_COMPOSE" -eq 1 ]; then
+        if ! command_exists docker-compose || ! command_exists docker compose version; then
+            install_docker_compose
+        else
+            printf "%b\n" "${GREEN}Docker Compose is already installed.${RC}"
+        fi
+    fi
+}
+
+checkEnv
+checkEscalationTool
+install_components
diff --git a/tabs/system-setup/tab_data.toml b/tabs/system-setup/tab_data.toml
index 6743da3c..c1c8f02b 100644
--- a/tabs/system-setup/tab_data.toml
+++ b/tabs/system-setup/tab_data.toml
@@ -51,3 +51,7 @@ script = "3-global-theme.sh"
 [[data]]
 name = "Remove Snaps"
 script = "4-remove-snaps.sh"
+
+[[data]]
+name = "Docker Setup"
+script = "6-docker-setup.sh"