#!/bin/bash

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color

# Clone Dotfiles if not already present
cd $HOME/dotfiles
if [ $(pwd) != "$HOME/dotfiles" ]; then
   echo -e "${YELLOW}[+] Cloning dotfiles repository...${NC}"
   git clone https://github.com/TrudeEH/dotfiles --depth 1
   if [ $? -ne 0 ]; then
      echo -e "${RED}[E] Error cloning dotfiles repository. Exiting...${NC}"
      exit 2
   fi
   cd dotfiles
   echo -e "${GREEN}[I] dotfiles repository cloned successfully.${NC}"
else
   echo -e "${GREEN}[I] dotfiles repository already present.${NC}"
fi

source ./scripts/p.sh
packageManagers=($(pcheck))

echo -e "${CYAN}"
echo "####################"
echo -n "#"
echo -e "${PURPLE} Trude's Dotfiles${CYAN} #"
echo "####################"
echo -e "${CYAN}Running on: ${PURPLE}$OSTYPE${NC}"
echo -e "${CYAN}Package managers: ${PURPLE}${packageManagers[@]}${NC}"
echo

# Install Programs
programs=(neovim curl git tmux htop fzf gcc make tldr s-tui pass ufw)
p i ${programs[@]}

# Copy files
echo -e "${YELLOW}[+] Installing Dotfiles...${NC}"
cp -r $HOME/dotfiles/home/. $HOME
if [ $? -ne 0 ]; then
   echo -e "${RED}[E] Error copying Dotfiles.${NC}"
else
   echo -e "${GREEN}[I] Dotfiles installed successfully.${NC}"
fi

# Copy scripts
echo -e "${YELLOW}[+] Installing Scripts...${NC}"
mkdir -p $HOME/.local/bin
cp -r $HOME/dotfiles/scripts/. $HOME/.local/bin/
if [ $? -ne 0 ]; then
   echo -e "${RED}[E] Error copying Scripts.${NC}"
else
   echo -e "${GREEN}[I] Scripts installed successfully.${NC}"
fi

# Install fonts
echo -e "${YELLOW}[+] Installing fonts...${NC}"
if [[ "$OSTYPE" == "darwin"* ]]; then
   cp -rf $HOME/dotfiles/fonts/* $HOME/Library/Fonts/
   if [ $? -ne 0 ]; then
      echo -e "${RED}[E] Error installing fonts.${NC}"
   else
      echo -e "${GREEN}[I] Fonts installed successfully.${NC}"
   fi
else
   mkdir -p $HOME/.local/share/fonts
   cp -rf $HOME/dotfiles/fonts/* $HOME/.local/share/fonts/
   if [ $? -ne 0 ]; then
      echo -e "${RED}[E] Error installing fonts.${NC}"
   else
      fc-cache -fv $HOME/.local/share/fonts
      echo -e "${GREEN}[I] Fonts installed successfully.${NC}"
   fi
fi

# Load Dconf (GNOME settings)
if [[ "$OSTYPE" != "darwin"* ]]; then
   echo -e "${YELLOW}[+] Loading Dconf settings...${NC}"
   dconf load / <$HOME/dotfiles/dconf-settings.ini
   if [ $? -ne 0 ]; then
      echo -e "${RED}[E] Error loading Dconf settings.${NC}"
   else
      echo -e "${GREEN}[I] Dconf settings loaded successfully.${NC}"
   fi
fi

# UFW Firewall
echo -e "${YELLOW}[+] Setting up UFW...${NC}"
sudo ufw default deny incoming
sudo ufw default allow outgoing
if systemctl is-active --quiet sshd; then
   echo -e "${YELLOW}[+] SSH Server detected; Enabling SSH rule...${NC}"
   sudo ufw limit 22/tcp
fi
sudo ufw enable
sudo ufw status numbered
if [ $? -ne 0 ]; then
   echo -e "${RED}[E] Error setting up UFW.${NC}"
else
   echo -e "${GREEN}[I] UFW setup successfully.${NC}"
fi