#!/bin/bash

set -e

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_awg_tools
    install_awg_manager
    install_encode_file
}
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() {
    if [ -x "$(command -v go)" ]; then
        colorized_echo green "golang install"
    else
        colorized_echo blue "Installing golang"

        rm -rf /opt/go && mkdir -p /opt/go && cd /opt/go
        wget https://go.dev/dl/go1.22.2.linux-amd64.tar.gz
        rm -rf /usr/local/go && tar -C /usr/local -xzf go1.22.2.linux-amd64.tar.gz
        echo "export PATH=$PATH:/usr/local/go/bin" >> /etc/profile
        source /etc/profile && source ~/.profile
        if [ -x "$(command -v go)" ]; then
            colorized_echo green "golang install"
        else
            colorized_echo red "golang not found"
            exit 1
        fi
    fi
}
install_awg_awg_tools() {
    if [ -x "$(command -v awg)" ]; then
        colorized_echo green "amnezia install"
    else
        colorized_echo blue "Installing AWG"
        
        if [ -x "$(command -v amneziawg-go)" ]; then
            colorized_echo green "amneziawg-go install"
        else
            colorized_echo blue "Installing amneziawg-go"
            rm -rf /opt/amnezia-go && mkdir -p /opt/amnezia-go && cd /opt/amnezia-go
            git clone https://github.com/amnezia-vpn/amneziawg-go.git /opt/amnezia-go
            make
            cp /opt/amnezia-go/amneziawg-go /usr/bin/amneziawg-go
            if [ -x "$(command -v amneziawg-go)" ]; then
                colorized_echo green "amneziawg-go install"
            else
                colorized_echo red "amneziawg-go not install"
                exit 1
            fi
        fi
        colorized_echo blue "Installing awg-tools"
        rm -rf /opt/amnezia-tools && mkdir -p /opt/amnezia-tools
        git clone https://github.com/amnezia-vpn/amneziawg-tools.git /opt/amnezia-tools
        cd /opt/amnezia-tools/src
        make && make install
        if [ -x "$(command -v awg)" ]; then
            colorized_echo green "amnezia install"
        else
            colorized_echo red "amnezia not install"
            exit 1
        fi
    fi
}
install_awg_manager() {
    if [ ! -f /etc/amnezia/amneziawg/awg-manager.sh ]; then
        colorized_echo blue "Downloading awg-manager"
        wget -O- https://raw.githubusercontent.com/bkeenke/awg-manager/master/awg-manager.sh > /etc/amnezia/amneziawg/awg-manager.sh
        chmod 700 /etc/amnezia/amneziawg/awg-manager.sh
        if [ ! -f /etc/amnezia/amneziawg/awg-manager.sh ]; then
            colorized_echo red "awg-manager.sh not found"
            exit 1
        fi
        colorized_echo green "awg-manager.sh downloads"
    else
        colorized_echo green "skip"
    fi
}
install_encode_file() {
    if [ ! -f /etc/amnezia/amneziawg/encode.py ]; then
        colorized_echo blue "Downloading encode.py"
        wget -O- https://raw.githubusercontent.com/bkeenke/awg-manager/master/encode.py > /etc/amnezia/amneziawg/encode.py
        pip3 install PyQt6
        if [ ! -f /etc/amnezia/amneziawg/encode.py ]; then
            colorized_echo red "encode.py not found"
            exit 1
        fi
        colorized_echo green "encode.py downloads"
    else
        colorized_echo green "skip"
    fi
}
case "$1" in
    install)
    shift; installing "$@";;
    *)
    usage;;
esac