#!/bin/sh

#Populate root's home folder from /etc/skel
cp -rv /etc/skel/. /root/

#Time to git stuff...
cd /root/
git clone https://github.com/Veil-Framework/Veil.git /opt/Veil
git clone https://github.com/leebaird/discover.git /opt/discover
git clone https://github.com/trustedsec/ptf /opt/ptf

#ssh-keygen -t rsa -N "" -f ~/.ssh/id_rsa

#Enable/disable our desired services
systemctl enable ssh
systemctl enable cups
systemctl enable apache2
systemctl enable mysql
systemctl enable postgresql
systemctl disable tor
systemctl disable cockpit.socket
systemctl enable xrdp
systemctl disable firewalld

#Allow VLC to run as root
sed -i 's/geteuid/getppid/' /usr/bin/vlc

#Get FTK Imager CLI
wget https://ad-zip.s3.amazonaws.com/ftkimager.3.1.1_ubuntu64.tar.gz
tar xvf ftkimager.3.1.1_ubuntu64.tar.gz
mv ftkimager /usr/bin/
rm ftkimager.3.1.1_ubuntu64.tar.gz

#Install VS Code
echo 'deb [signed-by=/etc/apt/trusted.gpg.d/vscodium-archive-keyring.gpg] https://paulcarroty.gitlab.io/vscodium-deb-rpm-repo/debs/ vscodium main' | sudo tee --append /etc/apt/sources.list.d/vscodium.list
wget -qO - https://gitlab.com/paulcarroty/vscodium-deb-rpm-repo/raw/master/pub.gpg | gpg --dearmor | sudo dd of=/etc/apt/trusted.gpg.d/vscodium-archive-keyring.gpg
sudo apt update && sudo apt install codium -y

#Install VS Code extensions
#codium --install-extension dbaeumer.vscode-eslint --user-data-dir=~/.vscode-oss
#codium --install-extension ecmel.vscode-html-css --user-data-dir=~/.vscode-oss
#codium --install-extension hookyqr.beautify --user-data-dir=~/.vscode-oss
#codium --install-extension ms-azuretools.vscode-docker --user-data-dir=~/.vscode-oss
#codium --install-extension ms-python.python --user-data-dir=~/.vscode-oss
#codium --install-extension ms-vscode.cpptools --user-data-dir=~/.vscode-oss
#codium --install-extension ms-vscode.csharp --user-data-dir=~/.vscode-oss
#codium --install-extension ms-vscode.go --user-data-dir=~/.vscode-oss
#codium --install-extension ms-vscode.powershell --user-data-dir=~/.vscode-oss
#codium --install-extension rebornix.ruby --user-data-dir=~/.vscode-oss
#codium --install-extension zignd.html-css-class-completion --user-data-dir=~/.vscode-oss

#Make Nemo the defualt file manager
xdg-mime default nemo.desktop inode/directory application/x-gnome-saved-search

#Distro-specific stuff
#DISTRO=$(cat /etc/lsb-release | grep DISTRIB_ID | cut -c 12-30)
DISTRO=$(lsb_release -is)

case $DISTRO in
Parrot)
#Stop ParrotSec from overwriting all our desktop customizations every time parrot-skel updates. >:-(
apt-mark hold parrot-skel
;;
Kali)
#Remove CPU-hungry MATE menu applet
apt remove mate-applet-brisk-menu -y
;;
esac