onix/install/scripts/package_manager.sh

#!/bin/bash
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
source $SCRIPT_DIR/variables.sh

#Required packages list as below.
package_list=("docker" "docker-compose" "jq")

command_exists() {
  command -v "$1"  &>/dev/null
}

# Redirect input from /dev/null to silence prompts
export DEBIAN_FRONTEND=noninteractive
export APT_KEY_DONT_WARN_ON_DANGEROUS_USAGE=1


#Install Package
install_package() {
    if [ -x "$(command -v apt-get)" ]; then
        # APT (Debian/Ubuntu)
        if [ "$1" == "docker" ]; then
            docker > /dev/null 2>&1
            if [ $? -ne 0 ]; then
                curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /usr/share/keyrings/docker-archive-keyring.gpg
                echo "deb [signed-by=/usr/share/keyrings/docker-archive-keyring.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
                sudo apt update >/dev/null 2>&1
                sudo apt install -y docker-ce docker-ce-cli containerd.io >/dev/null 2>&1
                sudo usermod -aG docker $USER
                source ~/.bashrc
                command_exists docker
                if [ $? -eq 0 ]; then
                    sleep 10
                    sudo systemctl enable docker.service
                    sudo systemctl restart docker.service
                fi
            else
                echo "Docker is already installed."
            fi
        else
            if ! dpkg -l | grep -q "^ii  $1 "; then
                sudo apt-get update >/dev/null 2>&1
                sudo apt-get install -y $1 > /dev/null 2>&1
            else
                echo "$1 is already installed."
            fi
        fi
    elif [ -x "$(command -v yum)" ]; then
        # YUM (Red Hat/CentOS)
        if [ "$1" == "docker" ]; then
            if ! rpm -qa | grep -q docker-ce; then
                sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
                # Install Docker
                sudo yum install -y docker >/dev/null 2>&1
                sudo systemctl enable docker.service
                sudo systemctl start docker.service
                sudo usermod -aG docker $USER
                source ~/.bashrc
            else
                echo "Docker is already installed."
            fi
        else
            if ! rpm -qa | grep -q "^$1-"; then
                sudo yum install -y $1 >/dev/null 2>&1
            else
                echo "$1 is already installed."
            fi
        fi
    elif [ -x "$(command -v amazon-linux-extras)" ]; then
        # Amazon Linux 2
        if ! amazon-linux-extras list | grep -q "^$1 "; then
            sudo amazon-linux-extras install $1 >/dev/null 2>&1
        else
            echo "$1 is already installed."
        fi
    else
        echo "${RED}Unsupported package manager. Please install $1 manually.${NC}"
        exit 1
    fi
}


remove_package(){
    if [ -x "$(command -v apt-get)" ]; then
        # APT (Debian/Ubuntu)
        sudo apt-get purge -y $1  >/dev/null 2>&1
        sudo apt autoremove -y  >/dev/null 2>&1
    elif [ -x "$(command -v yum)" ]; then
        # YUM (Red Hat/CentOS)
        sudo yum remove -y $1  >/dev/null 2>&1
        sudo yum autoremove -y >/dev/null 2>&1
    fi
}

# Function to install Docker
install_docker_bash() {
    # Install Docker Bash completion
    echo "Installing Docker Bash completion..."
    sudo curl -L https://raw.githubusercontent.com/docker/cli/master/contrib/completion/bash/docker -o /etc/bash_completion.d/docker
}

# Function to install Docker Compose
install_docker_compose() {
    command_exists docker-compose
    if [ $? -eq 0 ]; then
        echo "docker-compose is already installed."
        return
    else
        echo "Installing Docker Compose..."
        sudo curl -L "https://github.com/docker/compose/releases/latest/download/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose
        sudo chmod +x /usr/local/bin/docker-compose
    fi

    # Check if Docker Compose installation was successful
    if [ $? -eq 0 ]; then
        echo "Docker Compose installed successfully."
    else
        echo "${RED}Failed to install Docker Compose. Exiting.${NC}"
        exit 1
    fi

    if [ -f /etc/bash_completion.d/docker-compose ]; then
        echo "Docker Compose Bash completion is already installed."
    else
        # Install Docker Compose Bash completion
        echo "Installing Docker Compose Bash completion..."
        sudo curl -L https://raw.githubusercontent.com/docker/compose/master/contrib/completion/bash/docker-compose -o /etc/bash_completion.d/docker-compose
    fi
}


# Check if package is already installed

for package in "${package_list[@]}"; do
    if ! command_exists $package; then
	install_package "$package"
    fi
    if [ "$package" == "docker" ]; then
        if [[ $(uname -s ) == 'Linux' ]];then        
            if [ -f /etc/bash_completion.d/docker ]; then
                echo "Docker Bash completion is already installed."
            else
                install_docker_bash
            fi
        fi
    fi
    if [ "$package" == "docker-compose" ]; then
        if [[ $(uname -s ) == 'Linux' ]];then
            install_docker_compose
        fi
    fi
done