Update init.sh

init.sh

@@ -1,33 +1,116 @@
-#!/bin/bash
+#!/usr/bin/env bash
 
-DEBIAN_FRONTEND=noninteractive apt-get update
+set -e
-DEBIAN_FRONTEND=noninteractive apt-get upgrade -y
-DEBIAN_FRONTEND=noninteractive apt-get install build-essential curl make git wget qrencode python-3 python3-pip -y
 
+colorized_echo() {
+    local color=$1
+    local text=$2
+    
+    case $color in
+        "red")
+        printf "\e[91m${text}\e[0m\n";;
+        "green")
+        printf "\e[92m${text}\e[0m\n";;
+        "yellow")
+        printf "\e[93m${text}\e[0m\n";;
+        "blue")
+        printf "\e[94m${text}\e[0m\n";;
+        "magenta")
+        printf "\e[95m${text}\e[0m\n";;
+        "cyan")
+        printf "\e[96m${text}\e[0m\n";;
+        *)
+            echo "${text}"
+        ;;
+    esac
+}
+
+installing() {
+    check_running_as_root
+    detect_os
+    detect_and_update_package_manager
+    install_package
+    install_go
+    install_awg
+    install_awg_tools
+}
+check_running_as_root() {
+    if [ "$(id -u)" != "0" ]; then
+        colorized_echo red "This command must be run as root."
+        exit 1
+    fi
+}
+detect_os() {
+    # Detect the operating system
+    if [ -f /etc/lsb-release ]; then
+        OS=$(lsb_release -si)
+        elif [ -f /etc/os-release ]; then
+        OS=$(awk -F= '/^NAME/{print $2}' /etc/os-release | tr -d '"')
+        elif [ -f /etc/redhat-release ]; then
+        OS=$(cat /etc/redhat-release | awk '{print $1}')
+        elif [ -f /etc/arch-release ]; then
+        OS="Arch"
+    else
+        colorized_echo red "Unsupported operating system"
+        exit 1
+    fi
+}
+detect_and_update_package_manager() {
+    colorized_echo blue "Updating package manager"
+    if [[ "$OS" == "Ubuntu"* ]] || [[ "$OS" == "Debian"* ]]; then
+        PKG_MANAGER="apt-get"
+        $PKG_MANAGER update
+    else
+        colorized_echo red "Unsupported operating system"
+        exit 1
+    fi
+}
+install_package () {
+    if [ -z $PKG_MANAGER ]; then
+        detect_and_update_package_manager
+    fi
+    colorized_echo blue "Installing Package"
+    if [[ "$OS" == "Ubuntu"* ]] || [[ "$OS" == "Debian"* ]]; then
+        $PKG_MANAGER -y install build-essential \
+        curl \
+        make \
+        git \
+        wget \
+        qrencode \
+        python3 \
+        python3-pip
+    else
+        colorized_echo red "Unsupported operating system"
+        exit 1
+    fi
+}
+
+
+install_go() {
     mkdir -p /opt/go
     cd /opt/go
-wget https://go.dev/dl/go1.22.0.linux-amd64.tar.gz
+    curl -O https://go.dev/dl/go1.22.0.linux-amd64.tar.gz
     rm -rf /usr/local/go && tar -C /usr/local -xzf go1.22.0.linux-amd64.tar.gz
     echo "export PATH=$PATH:/usr/local/go/bin" >> /etc/profile
-sleep 2
+    source ~/.profile
-source /etc/profile
+    if go version >/dev/null 2>&1; then
-# if the go version does not show, then update the session
+        echo "Golang installed"
+    else
+        colorized_echo red "golang not found"
+        exit 1
+    fi
+}
+install_awg() {
 
-#Install amnezia-go
+}
-git clone https://github.com/amnezia-vpn/amneziawg-go.git /opt/amnezia-go
+install_awg_tools() {
-cd /opt/amnezia-go
-sleep 1
-source /etc/profile
-sleep 1
-make
-#copy to amneziawg-go
-cp /opt/amnezia-go/amneziawg-go /usr/bin/amneziawg-go
 
-#Install amnezia-tools
+}
-git clone https://github.com/amnezia-vpn/amneziawg-tools.git /opt/amnezia-tools
-cd /opt/amnezia-tools/src
-make
-make install
 
-#Install PyQt6
+
-pip3 install PyQt6
+case "$1" in
+    install)
+    shift; installing "$@";;
+    *)
+    usage;;
+esac