mybash/setup.sh

#!/bin/sh

RC='\033[0m'
RED='\033[31m'
YELLOW='\033[33m'
GREEN='\033[32m'

# Check if the home directory and linuxtoolbox folder exist, create them if they don't
LINUXTOOLBOXDIR="$HOME/linuxtoolbox"

if [ ! -d "$LINUXTOOLBOXDIR" ]; then
    echo "${YELLOW}Creating linuxtoolbox directory: $LINUXTOOLBOXDIR${RC}"
    mkdir -p "$LINUXTOOLBOXDIR"
    echo "${GREEN}linuxtoolbox directory created: $LINUXTOOLBOXDIR${RC}"
fi

if [ ! -d "$LINUXTOOLBOXDIR/mybash" ]; then
    echo "${YELLOW}Cloning mybash repository into: $LINUXTOOLBOXDIR/mybash${RC}"
    git clone https://github.com/ChrisTitusTech/mybash "$LINUXTOOLBOXDIR/mybash"
    if [ $? -eq 0 ]; then
        echo "${GREEN}Successfully cloned mybash repository${RC}"
    else
        echo "${RED}Failed to clone mybash repository${RC}"
        exit 1
    fi
fi

cd "$LINUXTOOLBOXDIR/mybash"

command_exists() {
    command -v $1 >/dev/null 2>&1
}

checkEnv() {
    ## Check for requirements.
    REQUIREMENTS='curl groups sudo'
    for req in $REQUIREMENTS; do
        if ! command_exists $req; then
            echo "${RED}To run me, you need: $REQUIREMENTS${RC}"
            exit 1
        fi
    done

    ## Check Package Handler
    PACKAGEMANAGER='apt yum dnf pacman zypper'
    for pgm in $PACKAGEMANAGER; do
        if command_exists $pgm; then
            PACKAGER=$pgm
            echo "Using $pgm"
            break
        fi
    done

    if [ -z "$PACKAGER" ]; then
        echo "${RED}Can't find a supported package manager${RC}"
        exit 1
    fi

    ## Check if the current directory is writable.
    GITPATH=`dirname \`realpath $0\``
    if [ ! -w $GITPATH ]; then
        echo "${RED}Can't write to $GITPATH${RC}"
        exit 1
    fi

    ## Check SuperUser Group
    SUPERUSERGROUP='wheel sudo root'
    for sug in $SUPERUSERGROUP; do
        if groups | grep $sug >/dev/null; then
            SUGROUP=$sug
            echo "Super user group $SUGROUP"
            break
        fi
    done

    ## Check if member of the sudo group.
    if ! groups | grep $SUGROUP >/dev/null; then
        echo "${RED}You need to be a member of the sudo group to run me!${RC}"
        exit 1
    fi
}

installDepend() {
    ## Check for dependencies.
    DEPENDENCIES='bash bash-completion tar tree multitail fastfetch tldr trash-cli'
    echo "${YELLOW}Installing dependencies...${RC}"
    if [ "$PACKAGER" = "pacman" ]; then
        if ! command_exists yay && ! command_exists paru; then
            echo "Installing yay as AUR helper..."
            sudo $PACKAGER --noconfirm -S base-devel
            cd /opt && sudo git clone https://aur.archlinux.org/yay-git.git && sudo chown -R $USER:$USER ./yay-git
            cd yay-git && makepkg --noconfirm -si
        else
            echo "Aur helper already installed"
        fi
        if command_exists yay; then
            AUR_HELPER="yay"
        elif command_exists paru; then
            AUR_HELPER="paru"
        else
            echo "No AUR helper found. Please install yay or paru."
            exit 1
        fi
        $AUR_HELPER --noconfirm -S $DEPENDENCIES
    else
        sudo $PACKAGER install -yq $DEPENDENCIES
    fi
}

installStarship() {
    if command_exists starship; then
        echo "Starship already installed"
        return
    fi

    if ! curl -sS https://starship.rs/install.sh | sh; then
        echo "${RED}Something went wrong during starship install!${RC}"
        exit 1
    fi
    if command_exists fzf; then
        echo "Fzf already installed"
    else
        git clone --depth 1 https://github.com/junegunn/fzf.git ~/.fzf
        ~/.fzf/install
    fi
}

installZoxide() {
    if command_exists zoxide; then
        echo "Zoxide already installed"
        return
    fi

    if ! curl -sS https://raw.githubusercontent.com/ajeetdsouza/zoxide/main/install.sh | sh; then
        echo "${RED}Something went wrong during zoxide install!${RC}"
        exit 1
    fi
}

install_additional_dependencies() {
    case `command -v apt || command -v zypper || command -v dnf || command -v pacman` in
        *apt)
            curl -LO https://github.com/neovim/neovim/releases/latest/download/nvim.appimage
            chmod u+x nvim.appimage
            ./nvim.appimage --appimage-extract
            sudo mv squashfs-root /opt/neovim
            sudo ln -s /opt/neovim/AppRun /usr/bin/nvim
            ;;
        *zypper)
            sudo zypper refresh
            sudo zypper install -y neovim 
            ;;
        *dnf)
            sudo dnf check-update
            sudo dnf install -y neovim 
            ;;
        *pacman)
            sudo pacman -Syu
            sudo pacman -S --noconfirm neovim 
            ;;
        *)
            echo "No supported package manager found. Please install neovim manually."
            exit 1
            ;;
    esac
}

linkConfig() {
    ## Get the correct user home directory.
    USER_HOME=`getent passwd ${SUDO_USER:-$USER} | cut -d: -f6`
    ## Check if a bashrc file is already there.
    OLD_BASHRC="$USER_HOME/.bashrc"
    if [ -e $OLD_BASHRC ]; then
        echo "${YELLOW}Moving old bash config file to $USER_HOME/.bashrc.bak${RC}"
        if ! mv $OLD_BASHRC $USER_HOME/.bashrc.bak; then
            echo "${RED}Can't move the old bash config file!${RC}"
            exit 1
        fi
    fi

    if [ ! -d $USER_HOME/.config/fastfetch ]; then
        mkdir -p $USER_HOME/.config/fastfetch
    fi
    echo "${YELLOW}Linking new bash config file...${RC}"
    ## Make symbolic link.
    ln -svf $GITPATH/config.jsonc $USER_HOME/.config/fastfetch/config.jsonc
    ln -svf $GITPATH/.bashrc $USER_HOME/.bashrc
    ln -svf $GITPATH/starship.toml $USER_HOME/.config/starship.toml
}

checkEnv
installDepend
installStarship
installZoxide
install_additional_dependencies

if linkConfig; then
    echo "${GREEN}Done!\nrestart your shell to see the changes.${RC}"
else
    echo "${RED}Something went wrong!${RC}"
fi