setup.sh

#!/bin/bash

WARNING='\033[0;91m'
HIGHLIGHT='\033[0;93m'
INFO='\033[0;36m'
NC='\033[0m' # No Color

current_dir=$(pwd)
repo_dir=$(dirname "${BASH_SOURCE[0]}")

printf "${HIGHLIGHT}This script MUST be run from within the cloned repository's directory.${NC}\n"
sleep 2s

if [ "$current_dir" != "$repo_dir" ]; then
  while true; do
    printf "${HIGHLIGHT}%s${NC}${WARNING}%s${NC}\n" "DIRECTORY SCRIPT IS RUNNING FROM: " "$current_dir"
    sleep 1s
    printf "${WARNING}%s${NC}\n" "CHECK ABOVE TO ENSURE YOU ARE RUNNING THIS SCRIPT IN THE CORRECT DIRECTORY."
    sleep 1s
    read -rp "$(echo -e "\n${HIGHLIGHT}Is ${WARNING}$current_dir${HIGHLIGHT} the correct directory you cloned the PentestonBash repository to?\nType [Y] to continue with setup, or [N] to exit.\n${WARNING}[Y/n]: ${NC}")" yn
    case $yn in
    [Yy]*)
      printf "${INFO}Installing PentestonBash plugin...${NC}\n"
      break
      ;;
    [Nn]*)
      printf "${WARNING}This script must be run from the cloned repository directory.${NC}\n"
      printf "${INFO}Change to the repository's directory before attempting to run this setup.sh script again.${NC}\n"
      exit 1
      break
      ;;
    *)
      printf "${WARNING}It's a Y/n question, provide a Y/n answer.${NC}\n"
      ;;
    esac
  done
fi


if [ -f "./.pentestonbash" ]; then 
  cp ./.pentestonbash ~/.pentestonbash
  printf "${INFO}File .pentestonbash found. Copying to user's home directory.${NC}\n"
  sleep 1s
else
  printf "${WARNING}File .pentestonbash not found. Please ensure the repository is cloned correctly and that you are following the plugin's install instructions.${NC}\n" && exit 1
fi

printf "${INFO}Sourcing .pentestonbash in user's .bashrc file...${NC}\n"
# Check if file is already sourced
if grep -q "source ~/.pentestonbash" ~/.bashrc; then
  printf "${INFO}File is already sourced. No need to source a second time...${NC}\n"
else
  echo -e "\nsource ~/.pentestonbash\n" >> ~/.bashrc
fi

printf "${INFO}Install complete!${NC}\n"
sleep 1s

while true; do
  printf "\n"
  read -rp "$(echo -e "${HIGHLIGHT}Cleanup and remove the setup.sh script? [Y/n]: ${NC}")" yn
  case $yn in
  [Yy]*)
    printf "${INFO}Removing setup.sh script...${NC}\n"
    rm -f setup.sh
    break
    ;;
  [Nn]*)
    printf "${INFO}Leaving setup.sh script in place. You can run it again to repeat the setup process.${NC}\n"
    break
    ;;
  *)
    printf "${WARNING}It's a Y/n question, provide a Y/n answer.${NC}\n"
    ;;
  esac
done