reset_db.sh

#!/bin/bash

if [[ "$(uname)" == "Darwin" ]]; then
    psql -S apfell_db<<EOF
DROP TABLE payload CASCADE;
DROP TABLE operator CASCADE;
DROP TABLE callback CASCADE;
DROP TABLE task CASCADE;
DROP TABLE response CASCADE;
DROP TABLE c2profile CASCADE;
DROP TABLE operation CASCADE;
DROP TABLE operatoroperation CASCADE;
DROP TABLE payloadtype CASCADE;
DROP TABLE command CASCADE;
DROP TABLE payloadtypec2profile CASCADE;
DROP TABLE commandparameters CASCADE;
DROP TABLE filemeta CASCADE;
DROP TABLE payloadcommand CASCADE;
DROP TABLE c2profileparameters CASCADE;
DROP TABLE c2profileparametersinstance CASCADE;
DROP TABLE attack CASCADE;
DROP TABLE credential CASCADE;
DROP TABLE keylog CASCADE;
DROP TABLE transform CASCADE;
DROP TABLE loadedcommands CASCADE;
DROP TABLE commandtransform CASCADE;
DROP TABLE attackcommand CASCADE;
DROP TABLE attacktask CASCADE;
DROP TABLE artifact CASCADE;
DROP TABLE artifacttemplate CASCADE;
DROP TABLE taskartifact CASCADE;
EOF
    pg_ctl -D /usr/local/var/postgres restart

#make sure we're root
elif [[ $EUID -ne 0 ]]; then
   echo -e "\n[-] This script must be run as root!\n"
   exit 1
else

service postgresql restart
#delete all rows in our tables
sudo -u postgres psql -q -S apfell_db<<EOF
DROP TABLE payload CASCADE;
DROP TABLE operator CASCADE;
DROP TABLE callback CASCADE;
DROP TABLE task CASCADE;
DROP TABLE response CASCADE;
DROP TABLE c2profile CASCADE;
DROP TABLE operation CASCADE;
DROP TABLE operatoroperation CASCADE;
DROP TABLE payloadtype CASCADE;
DROP TABLE command CASCADE;
DROP TABLE payloadtypec2profile CASCADE;
DROP TABLE commandparameters CASCADE;
DROP TABLE filemeta CASCADE;
DROP TABLE payloadcommand CASCADE;
DROP TABLE c2profileparameters CASCADE;
DROP TABLE c2profileparametersinstance CASCADE;
DROP TABLE attack CASCADE;
DROP TABLE credential CASCADE;
DROP TABLE keylog CASCADE;
DROP TABLE transform CASCADE;
DROP TABLE loadedcommands CASCADE;
DROP TABLE commandtransform CASCADE;
DROP TABLE attackcommand CASCADE;
DROP TABLE attacktask CASCADE;
DROP TABLE artifact CASCADE;
DROP TABLE artifacttemplate CASCADE;
DROP TABLE taskartifact CASCADE;
EOF

service postgresql restart

fi