#!/usr/bin/env zsh

set -euo pipefail

function main() {
    default_system_pass="kali"
    default_go_version="1.26.3"
    default_password="redeye6"

    printf "\n"
    cprint "------------------------------------------------------------"
    cprint "\t\t\tredeye"
    cprint "------------------------------------------------------------"

    read "update_toggle?do you want to preform a full-upgrade? (y/n): "
    read -s "custom_password?do you want to update your password? (default: $default_password): "
    if [[ -n "$custom_password" ]]; then
        default_password="$custom_password"
    fi
    printf "\n"
    read "custom_go_version?what go version should be installed? (default: $default_go_version): "
    printf "\n"
    if [[ -n "$custom_go_version" ]]; then
        default_go_version="$custom_go_version"
    fi

    updateSystem "$default_system_pass" "$update_toggle"

    cprint "setting aliases"
    echo "alias g='go'" >> /home/kali/.zshrc
    echo "alias p='python3'" >> /home/kali/.zshrc
    echo "alias s='subl'" >> /home/kali/.zshrc

    installGo "$default_go_version"
    installTools

    echo -n "kali:$default_password" | sudo chpasswd

    cprint "done"
}

function installGo() {
    cprint "installing golang v$1"
    wget https://go.dev/dl/go$1.linux-amd64.tar.gz
    sudo rm -rf /usr/local/go
    sudo tar -C /usr/local -xzf go$1.linux-amd64.tar.gz
    echo "export PATH=\$PATH:/usr/local/go/bin" >> /home/kali/.zshrc
    export PATH=$PATH:/usr/local/go/bin
    rm ./go$1.linux-amd64.tar.gz
    source /home/kali/.zshrc
    cprint "golang installed"
}

function installTools() {
    cprint "installing credslayer"
    pipx install credslayer
    cprint "installing ffuf"
    go install github.com/ffuf/ffuf/v2@latest

    cprint "installing sublime text"
    wget -qO - https://download.sublimetext.com/sublimehq-pub.gpg | sudo tee /etc/apt/keyrings/sublimehq-pub.asc > /dev/null
    echo -e 'Types: deb\nURIs: https://download.sublimetext.com/\nSuites: apt/stable/\nSigned-By: /etc/apt/keyrings/sublimehq-pub.asc' | sudo tee /etc/apt/sources.list.d/sublime-text.sources
    #re-update after adding sublime
    sudo apt update -y
    sudo apt install sublime-text
    cprint "sublime installed"

    cprint "installing firewall"
    sudo apt install ufw -y
    sudo  ufw enable
    cprint "firewall installed"
}

function updateSystem() {
    cprint "updating system, this may take a while"
    echo "$1" | sudo -Sv
    sudo apt update -y
    if [[ "$2" == "y" ]]; then
        sudo apt full-upgrade -y
    fi
    sudo apt autoremove -y
    cprint "system updated"
}

#cyan printing
function cprint() {
	printf "\e[0;36m$1\e[0m\n"
}

main "$@"