#!/bin/bash

# Update PoshC2
echo ""

echo """__________            .__.     _________  ________  
 \_______  \____  _____|  |__   \_   ___ \ \_____  \ 
  |     ___/  _ \/  ___/  |  \  /    \  \/  /  ____/ 
  |    |  (  <_> )___ \|   Y  \ \     \____/       \ 
  |____|   \____/____  >___|  /  \______  /\_______ \  
                     \/     \/          \/         \/
  =============== v4.0 www.PoshC2.co.uk ============="""

echo ""
echo "[+] Updating PoshC2_Python"
echo ""

ROOTDIR=`dirname "$0"`
if [ ! -d "$ROOTDIR" ]; then
  ROOTDIR="/opt/PoshC2_Python/"
fi
pushd "$ROOTDIR" > /dev/null

# Backup config
echo "[+] Backup Config"
git stash > /dev/null

# Install requirements for PoshC2_Python
echo ""
echo "[+] Performing git pull on $ROOTDIR"
git pull 

# Restore config
echo "[+] Restore Config"
git stash pop > /dev/null
echo ""
echo "[+] Update complete"
echo ""

popd > /dev/null