#!/bin/sh

# A shell script to use as plugin that includes several pentesting helpers, such as:
# * Wordlists
# * Env. Variables
# * aliases

# base variables
WORDLIST_PATH="$HOME/documents/tools/lists"


function pentest {
    echo "Pentest Shortuts Plugins"
    echo "[*] Aliases: "
    echo "    pentest: display help for the commands and shortcuts"
    echo ""
    echo "[*] Env. Variables: "
    echo "    ROCKYOU: $ROCKYOU"
    echo "    WLIST_DIRS: $WLIST_DIRS"
    echo "    WLIST_FILES: $WLIST_FILES"
    echo "[*] Functions"
    echo "    set_ip <ip>: set global env variable \$IP/\$ip to <ip>"
    echo "    get_ip: prints \$ip"
}

function set_ip {
    if [ "$1" != "" ]; then
        echo "[*] Setting IP to $1"
        echo "$1" > $HOME/.pentest_target
        # export the scope IP as global variable
        export ip=$(cat $HOME/.pentest_target)
        export IP=$ip
    fi
}

function get_ip {
    # print the ip, for using this function
    echo $ip
}

export ROCKYOU=$WORDLIST_PATH/SecLists/Discovery/Web-Content/raft-medium-directories.txt
export WLIST_DIRS=$WORDLIST_PATH/SecLists/Discovery/Web-Content/raft-medium-directories.txt
export WLIST_FILES=$WORDLIST_PATH/SecLists/Discovery/Web-Content/raft-medium-files.txt