#!/bin/bash

function nmap_soft_install() {
	goodecho "[+] Installing Nmap from package manager"
	install_dependencies "nmap"
}

function wireshark_soft_install() {
	goodecho "[+] Installing Wireshark from package manager"
	echo "wireshark-common wireshark-common/install-setuid boolean true" | sudo debconf-set-selections
	install_dependencies "wireshark"
}

function metasploit_soft_install() {
    # Get system architecture
    ARCH=$(uname -m)
    
    # Check if architecture is supported
    if [[ "$ARCH" == "x86_64" ]] || [[ "$ARCH" == "amd64" ]] || [[ "$ARCH" == "aarch64" ]] || [[ "$ARCH" == "arm64" ]]; then
        goodecho "[+] Installing Metasploit for $ARCH architecture"
        [ -d /root/thirdparty ] || mkdir /root/thirdparty
        cd /root/thirdparty
        curl https://raw.githubusercontent.com/rapid7/metasploit-omnibus/master/config/templates/metasploit-framework-wrappers/msfupdate.erb > msfinstall
        chmod 755 msfinstall
        sudo ./msfinstall
    else
        criticalecho-noexit "[-] Unsupported architecture: $ARCH"
        criticalecho-noexit "[-] Metasploit installation is only supported on amd64/x86_64 and arm64/aarch64"
    fi
}

function tshark_soft_install() {
	goodecho "[+] Installing TShark from package manager"
	install_dependencies "tshark"
}

function impacket_soft_install() {
	goodecho "[+] Installing Impacket from package manager"
	install_dependencies "python3-impacket"
}

function autorecon_soft_install() {
	goodecho "[+] Installing Autorecon from GitHub with PIP"
	pipx install git+https://github.com/Tib3rius/AutoRecon.git
}

function responder_soft_install() {
	goodecho "[+] Installing Autorecon from GitHub"
	[ -d /opt/network ] || mkdir -p /opt/network
	cd /opt/network
	gitinstall "https://github.com/lgandx/Responder.git" "responder_soft_install"
	cd Responder
	pip3install -r requirements.txt
}

function kismet_soft_install() {
	goodecho "[+] Installing Kismet dependencies"
	[ -d /rftools ] || mkdir -p /rftools
	cd /rftools
	check_and_install_lib "librtlsdr-dev librtlsdr0" "librtlsdr"
	install_dependencies "libsqlite3-dev ubertooth libprelude-dev build-essential git libwebsockets-dev pkg-config zlib1g-dev libnl-3-dev libnl-genl-3-dev libcap-dev libpcap-dev libnm-dev libdw-dev libsqlite3-dev libprotobuf-dev libprotobuf-c-dev protobuf-compiler protobuf-c-compiler libsensors-dev libusb-1.0-0-dev python3 python3-setuptools python3-protobuf python3-requests python3-serial python3-usb python3-dev python3-websockets libubertooth-dev libbtbb-dev libmosquitto-dev"
	goodecho "[+] Installing Kismet"
	installfromnet "git clone https://www.kismetwireless.net/git/kismet.git"
	cd kismet
	./configure --enable-bladerf --enable-wifi-coconut --enable-btgeiger --enable-prelude --enable-python-tools
	make
	make -j$(nproc)
	make suidinstall
	make forceconfigs
	make install
}

function gowitnes_soft_install() {
    goodecho "[+] Installing webcopilot"
    export GOPROXY=https://proxy.golang.org,direct
    export GOSUMDB=sum.golang.org
    go install github.com/sensepost/gowitness@latest
}

function webcopilot_soft_install() {
    goodecho "[+] Installing gowitness"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/FlUxIuS/webcopilot.git" "webcopilot_soft_install"
    cd webcopilot
    chmod +x install.sh
	./install.sh
}

function subenum_soft_install() {
    goodecho "[+] Installing SubEnum"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/FlUxIuS/SubEnum.git" "subenum_soft_install"
    cd SubEnum
    chmod +x setup.sh
	./setup.sh
	ln -s $(pwd)/subenum.sh /usr/sbin/subenum.sh
}

function mbtget_soft_install() {
    goodecho "[+] Installing mbtget"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/sourceperl/mbtget.git" "mbtget_soft_install"
    cd mbtget
    perl Makefile.PL
    make
	make install
}

function bettercap_soft_install() {
	goodecho "[+] Installing bettercap"
	rm -rf ~/.cache/go-build #TODO: trying to solve build exit for ARM on GitHub
	export GOPROXY=direct
	install_dependencies "libnetfilter-queue-dev"
	[ -d /rftools/bluetooth ] || mkdir -p /rftools/bluetooth
	cd /rftools/bluetooth
	gitinstall "https://github.com/bettercap/bettercap.git"
	cd bettercap
	make build
	make install
	ln -s /rftools/bluetoot/bettercap/bettercap /usr/bin/bettercap
}

function sipvicious_soft_install() {
    goodecho "[+] Installing SIP Vicious"
    install_dependencies "libxml2-dev libxslt1-dev"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/EnableSecurity/sipvicious.git" "sipvicious_soft_install"
    cd sipvicious
    pip3install .
}

function voipire_soft_install() {
    goodecho "[+] Installing VoIPire"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/CR-DMcDonald/voipire.git" "voipire_soft_install"
    cd voipire
    cargo build --release
    ln -s $(pwd)/target/release/voipire /usr/bin/voipire
}

function sippts_soft_install() {
    goodecho "[+] Installing SIPPTS"
    [ -d /opt/network ] || mkdir -p /opt/network
    cd /opt/network
    gitinstall "https://github.com/Pepelux/sippts.git" "sippts_soft_install"
    cd sippts
    pip3install .
}