#!/bin/bash

# === Temporary directory check ===
if [[ -w /tmp ]]; then
    TMPDIR="/tmp"
else
    TMPDIR="$(pwd)/tmp_local"
    mkdir -p "$TMPDIR"
    if [[ ! -w "$TMPDIR" ]]; then
        echo "Error: no write permission in /tmp or current directory."
        exit 1
    fi
fi
echo "Using temporary directory: $TMPDIR"

if [[ $1 == "" ]]; then
    echo "First argument is empty!"
    echo "Type help for more information"
    exit 1
fi

if [[ $1 == "help" ]]; then
    echo "More information is available at github.com/ZoLuSs/rmmagent-script"
    echo ""
    echo "INSTALL arguments:"
    echo "Arg 1: 'install'"
    echo "Arg 2: Mesh agent URL"
    echo "Arg 3: API URL"
    echo "Arg 4: Client ID"
    echo "Arg 5: Site ID"
    echo "Arg 6: Auth Key"
    echo "Arg 7: Agent Type 'server' or 'workstation'"
    echo ""
    echo "UPDATE arguments:"
    echo "Arg 1: 'update'"
    echo ""
    echo "UNINSTALL arguments:"
    echo "Arg 1: 'uninstall'"
    echo "Arg 2: Mesh agent FQDN (e.g. mesh.example.com)"
    echo "Arg 3: Mesh agent ID (ID must be wrapped in single quotes)"
    echo ""
    exit 0
fi

if [[ $1 != "install" && $1 != "update" && $1 != "uninstall" ]]; then
    echo "First argument can only be 'install', 'update' or 'uninstall'!"
    echo "Type help for more information"
    exit 1
fi

## Detect system architecture
system=$(uname -m)
case $system in
    x86_64) system="amd64" ;;
    i386|i686) system="x86" ;;
    aarch64) system="arm64" ;;
    armv6l) system="armv6" ;;
    *) echo "Unsupported architecture: $system"; exit 1 ;;
esac

## Variables
mesh_url=$2
rmm_url=$3
rmm_client_id=$4
rmm_site_id=$5
rmm_auth=$6
rmm_agent_type=$7
mesh_fqdn=$2
mesh_id=$3

go_version="1.21.6"
go_url_amd64="https://go.dev/dl/go$go_version.linux-amd64.tar.gz"
go_url_x86="https://go.dev/dl/go$go_version.linux-386.tar.gz"
go_url_arm64="https://go.dev/dl/go$go_version.linux-arm64.tar.gz"
go_url_armv6="https://go.dev/dl/go$go_version.linux-armv6l.tar.gz"

function go_install() {
    if ! command -v go &> /dev/null; then
        case $system in
            amd64) wget -O "$TMPDIR/golang.tar.gz" "$go_url_amd64" ;;
            x86) wget -O "$TMPDIR/golang.tar.gz" "$go_url_x86" ;;
            arm64) wget -O "$TMPDIR/golang.tar.gz" "$go_url_arm64" ;;
            armv6) wget -O "$TMPDIR/golang.tar.gz" "$go_url_armv6" ;;
        esac
        rm -rf /usr/local/go/
        tar -xvzf "$TMPDIR/golang.tar.gz" -C /usr/local/
        rm "$TMPDIR/golang.tar.gz"
        export PATH=$PATH:/usr/local/go/bin
        echo "Go installed successfully."
    fi
}

function agent_compile() {
    echo "Compiling agent..."
    wget -O "$TMPDIR/rmmagent.tar.gz" "https://github.com/amidaware/rmmagent/archive/refs/heads/master.tar.gz"
    tar -xf "$TMPDIR/rmmagent.tar.gz" -C "$TMPDIR/"
    rm "$TMPDIR/rmmagent.tar.gz"
    cd "$TMPDIR/rmmagent-master"
    case $system in
        amd64) env CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags "-s -w" -o "$TMPDIR/temp_rmmagent" ;;
        x86) env CGO_ENABLED=0 GOOS=linux GOARCH=386 go build -ldflags "-s -w" -o "$TMPDIR/temp_rmmagent" ;;
        arm64) env CGO_ENABLED=0 GOOS=linux GOARCH=arm64 go build -ldflags "-s -w" -o "$TMPDIR/temp_rmmagent" ;;
        armv6) env CGO_ENABLED=0 GOOS=linux GOARCH=arm go build -ldflags "-s -w" -o "$TMPDIR/temp_rmmagent" ;;
    esac
    cd "$TMPDIR"
    rm -R "$TMPDIR/rmmagent-master"
}

function update_agent() {
    systemctl stop tacticalagent
    cp "$TMPDIR/temp_rmmagent" /usr/local/bin/rmmagent
    rm "$TMPDIR/temp_rmmagent"
    systemctl start tacticalagent
}

function install_agent() {
    cp "$TMPDIR/temp_rmmagent" /usr/local/bin/rmmagent
    "$TMPDIR/temp_rmmagent" -m install -api $rmm_url -client-id $rmm_client_id -site-id $rmm_site_id -agent-type $rmm_agent_type -auth $rmm_auth
    rm "$TMPDIR/temp_rmmagent"
    cat << "EOF" > /etc/systemd/system/tacticalagent.service
[Unit]
Description=Tactical RMM Linux Agent
[Service]
Type=simple
ExecStart=/usr/local/bin/rmmagent -m svc
User=root
Group=root
Restart=always
RestartSec=5s
LimitNOFILE=1000000
KillMode=process
[Install]
WantedBy=multi-user.target
EOF
    systemctl daemon-reload
    systemctl enable --now tacticalagent
    systemctl start tacticalagent
}

function install_mesh() {
    wget -O "$TMPDIR/meshagent" $mesh_url
    chmod +x "$TMPDIR/meshagent"
    mkdir -p /opt/tacticalmesh
    "$TMPDIR/meshagent" -install --installPath="/opt/tacticalmesh"
    rm "$TMPDIR/meshagent"
    rm -f "$TMPDIR/meshagent.msh"
}

function uninstall_agent() {
    systemctl stop tacticalagent
    systemctl disable tacticalagent
    rm /etc/systemd/system/tacticalagent.service
    systemctl daemon-reload
    rm /usr/local/bin/rmmagent
    rm -rf /etc/tacticalagent
}

function uninstall_mesh() {
    wget "https://$mesh_fqdn/meshagents?script=1" -O "$TMPDIR/meshinstall.sh" || wget "https://$mesh_fqdn/meshagents?script=1" --no-proxy -O "$TMPDIR/meshinstall.sh"
    chmod 755 "$TMPDIR/meshinstall.sh"
    "$TMPDIR/meshinstall.sh" uninstall https://$mesh_fqdn $mesh_id || "$TMPDIR/meshinstall.sh" uninstall uninstall uninstall https://$mesh_fqdn $mesh_id
    rm "$TMPDIR/meshinstall.sh"
}

case $1 in
    install)
        go_install
        install_mesh
        agent_compile
        install_agent
        echo "Tactical Agent installation completed."
        exit 0;;
    update)
        go_install
        agent_compile
        update_agent
        echo "Tactical Agent update completed."
        exit 0;;
    uninstall)
        uninstall_agent
        uninstall_mesh
        echo "Tactical Agent uninstall completed."
        exit 0;;
esac