init.sh

#!/bin/bash

set -e

# Variables
# ----------------------------------------------------------------------------------------
## Directories
NEO_HOME_DIR="${NEO_HOME_DIR:-$HOME}"
DOTFILES_DIR="$NEO_HOME_DIR/Developer/dotfiles"
NVM_VERSION="v0.40.1"
WHO_AM_I_VALUE=${WHO_AM_I_VALUE:-$(whoami)}

# Check if running in non-interactive mode
if [[ "$1" == "--non-interactive" ]] || [[ "$NON_INTERACTIVE" == "true" ]] || [[ "$ANSIBLE_MANAGED" == "true" ]]; then
  echo -e "${YELLOW}Running in non-interactive mode${NC}"
  NON_INTERACTIVE=true
else
  NON_INTERACTIVE=false
fi

## Colors
WHITE="\033[1;37m"
GRAY="\033[0;90m"
GREEN="\033[0;32m"
YELLOW="\033[0;33m"
INDIGO="\033[0;94m"
RED="\033[0;31m"
NC="\033[0m"

## OS checks
IS_MACOS=$(uname -s | grep -i "darwin" | wc -l | tr -d '[:space:]')
IS_LINUX=$(uname -s | grep -i "linux" | wc -l | tr -d '[:space:]')


# Safety checks
# ----------------------------------------------------------------------------------------
## Prevent running script as root but only if not running in non-interactive mode
if [[ "$NON_INTERACTIVE" == "false" ]]; then
    if [ "$EUID" -eq 0 ]; then
        echo -e "${RED}===> Please don't run as root.${NC}"
        exit 1
    fi
fi

## Make sure ZSH is installed
if [[ ! -f "/bin/zsh" ]]; then
    if [[ $IS_LINUX -eq 1 ]]; then
        echo -e "${YELLOW}===> Installing ZSH...${NC}"
        sudo apt install -y zsh
    else
        echo -e "${RED}===> ZSH is not installed. Please install ZSH first.${NC}"
        exit 1
    fi
fi

# If zsh is not main shell, switch to it
if [[ "$NON_INTERACTIVE" == "false" ]]; then
    if [[ $SHELL != *"zsh"* ]]; then
        echo -e "${YELLOW}===> Switching to zsh...${NC}"
        chsh -s $(which zsh)
    fi
fi


# Helper functions
# ----------------------------------------------------------------------------------------
## Link and backup files
link_and_backup() {
    local SKIP_LINKING=0
    local DEFAULT_FILE="$NEO_HOME_DIR/${2:-$1}"
    local LINK_FILE="$DOTFILES_DIR/$1"
    local SUDO=0

    if [[ $3 == "--realpath" ]]; then
        DEFAULT_FILE="$2"
    fi

    if [[ $4 == "--sudo" ]]; then
        SUDO=1
    fi

    if [ ! -f "$LINK_FILE" ] && [ ! -d "$LINK_FILE" ]; then
        echo -e "${RED}==> $LINK_FILE does not exist. Skipping!${NC}"
        return
    fi

    if [[ "$LINK_FILE" == "$DOTFILES_DIR/.gitignore.work" ]] && [ ! -f "$LINK_FILE" ]; then
        if [[ $SUDO -ne 1 ]]; then
            touch "$LINK_FILE"
        else
            sudo touch "$LINK_FILE"
        fi

        echo -e "${GREEN}==> Created empty file for $LINK_FILE.${NC}"
    fi

    if [ -f "$DEFAULT_FILE" ] && [ ! -L "$DEFAULT_FILE" ]; then
        if [[ $SUDO -ne 1 ]]; then
            mv "$DEFAULT_FILE" "$DEFAULT_FILE.backup"
        else
            sudo mv "$DEFAULT_FILE" "$DEFAULT_FILE.backup"
        fi
        echo -e "${GREEN}==> Created backup for $DEFAULT_FILE.${NC}"
    else
        if [[ -L $DEFAULT_FILE ]] && [[ $LINK_FILE == $(readlink "$DEFAULT_FILE") ]]; then
            SKIP_LINKING=1
        elif [[ -f "$DEFAULT_FILE" ]]; then
            if [[ $SUDO -ne 1 ]]; then
                mv "$DEFAULT_FILE" "$DEFAULT_FILE.bak"
            else
                sudo mv "$DEFAULT_FILE" "$DEFAULT_FILE.bak"
            fi
            echo -e "${GREEN}==> Created backup for $DEFAULT_FILE.${NC}"
        fi
    fi

    if [[ $SKIP_LINKING -eq 0 ]]; then
        if [[ $SUDO -ne 1 ]]; then
            ln -s "$LINK_FILE" "$DEFAULT_FILE"
        else
            sudo ln -s "$LINK_FILE" "$DEFAULT_FILE"
        fi
        echo -e "${GREEN}==> Created symlink for $DEFAULT_FILE from $LINK_FILE.${NC}"
    else
        DEFAULT_FILE_BASENAME=$(basename "$DEFAULT_FILE")
        echo -e "${GRAY}==> Symlink already exists for $DEFAULT_FILE_BASENAME. Skipping!${NC}"
    fi
}

# Begin Script
# ---------------------------------------------------------------------------------------------------

cd $NEO_HOME_DIR
echo -e "${WHITE}==> Initializing...${NC}"

# Install NVM
# ---------------------------------------------------------------------------------------------------
if [ -f "$NVM_DIR/nvm.sh" ]; then
    echo -e "${GRAY}==> NVM is already installed. Skipping...${NC}"
else
    echo -e "${WHITE}==> Installing NVM...${NC}"
    curl -o- "https://raw.githubusercontent.com/nvm-sh/nvm/${NVM_VERSION}/install.sh" | bash
fi

# Prepare & link dotfiles
# ---------------------------------------------------------------------------------------------------
## Make necessary directories
mkdir -p "$NEO_HOME_DIR/.config" "$NEO_HOME_DIR/.config/lazygit"

## Link the dotfiles...
link_and_backup "zellij" ".config/zellij"
link_and_backup "zsh/zshrc.sh" ".zshrc"
link_and_backup "zsh/aliases.sh" ".zshrc_aliases"
link_and_backup "zsh/functions.sh" ".zshrc_functions"
link_and_backup "zsh/paths.sh" ".zshrc_paths"
link_and_backup "zsh/private.sh" ".zshrc_scripts"
link_and_backup "starship/starship.toml" ".config/starship.toml"
link_and_backup "configs/curlrc" ".curlrc"
link_and_backup "configs/hushlogin" ".hushlogin"
link_and_backup "configs/wgetrc" ".wgetrc"
link_and_backup "configs/screenrc" ".screenrc"
link_and_backup "lazygit/lazygit.yml" ".config/lazygit/config.yml"
link_and_backup "git/global-gitignore" ".global-gitignore"
link_and_backup "git/githooks" ".githooks"
link_and_backup "git/base.cfg" ".gitconfig"
[[ $IS_LINUX -eq 1 ]] && link_and_backup "git/linux.cfg" ".gitconfig.extended"
[[ $IS_MACOS -eq 1 ]] && link_and_backup "git/mac.cfg" ".gitconfig.extended"
[[ $IS_MACOS -eq 1 ]] && link_and_backup "configs/mackup.cfg" ".mackup.cfg"
[[ -f "$DOTFILES_DIR/git/private.cfg" ]] && link_and_backup "git/private.cfg" ".gitconfig.private"
[[ -f "$DOTFILES_DIR/git/work.cfg" ]] && link_and_backup "git/work.cfg" ".gitconfig.work"

# Generate SSH Keys
# ------------------------------------------------------------------------------
SSH_CHECK_RUN_FILE="/tmp/dotfiles__ssh-skip-check"
if [[ ! -f "$NEO_HOME_DIR/.ssh/id_ed25519.pub" ]] && [[ ! -f "$SSH_CHECK_RUN_FILE" ]]; then
    touch "$SSH_CHECK_RUN_FILE"

    if [[ "$NON_INTERACTIVE" == "true" ]]; then
        echo -e "${YELLOW}==> Automatically generating SSH key in non-interactive mode${NC}"
        mkdir -p "$NEO_HOME_DIR/.ssh"
        ssh-keygen -t ed25519 -C "public@neoi.sh" -f "$NEO_HOME_DIR/.ssh/id_ed25519" -N "" -q
    else
        echo -e "${WHITE}==> Would you like to generate an SSH key? [y/N]${NC}"
        read -r answer

        if [[ $answer == "y" || $answer == "Y" ]]; then
            echo -e "${WHITE}==> Generating SSH key...${NC}"
            ssh-keygen -t ed25519 -C "public@neoi.sh" -f $NEO_HOME_DIR/.ssh/id_ed25519
        else
            echo -e "${WHITE}==> Skipping SSH key generation...${NC}"
        fi
    fi
fi

## Additional set up for SSH
if [[ -f "$NEO_HOME_DIR/.ssh/id_ed25519.pub" ]]; then
    link_and_backup "ssh/ssh-config" ".ssh/config"
    link_and_backup "ssh/config.d" ".ssh/config.d"
fi

## Add SSH key to keychain (MacOS only)
if [[ -f "$NEO_HOME_DIR/.ssh/id_ed25519" ]]; then
    [[ $IS_MACOS -eq 1 ]] && ssh-add --apple-use-keychain $NEO_HOME_DIR/.ssh/id_ed25519
    # [[ $IS_LINUX -eq 1 ]] && eval `ssh-agent`
    # [[ $IS_LINUX -eq 1 ]] && ssh-add -k $NEO_HOME_DIR/.ssh/id_ed25519
fi


# Run platform specific scripts...
# ------------------------------------------------------------------------------
if [[ $IS_MACOS -eq 1 ]]; then
    export HOMEBREW_CASK_OPTS="--appdir=/Applications"
    source "$DOTFILES_DIR/init-mac.sh"
elif [[ $IS_LINUX -eq 1 ]]; then
    source "$DOTFILES_DIR/init-linux.sh"
fi

# Generate GPG key
# ------------------------------------------------------------------------------
GPG_CHECK_RUN_FILE="/tmp/dotfiles__gpg-skip-check"
if [[ -z "$(gpg --list-keys 2>/dev/null)" ]] && [[ ! -f "$GPG_CHECK_RUN_FILE" ]]; then
    touch "$GPG_CHECK_RUN_FILE"

    if [[ "$NON_INTERACTIVE" == "true" ]]; then
        echo -e "${YELLOW}==> Automatically generating GPG key in non-interactive mode${NC}"
        # Create batch file for unattended GPG key generation
        cat > /tmp/gpg-batch <<EOF
%echo Generating GPG key
Key-Type: RSA
Key-Length: 4096
Subkey-Type: RSA
Subkey-Length: 4096
Name-Real: Neo Automated
Name-Email: public@neoi.sh
Expire-Date: 0
%no-protection
%commit
%echo Done
EOF
        gpg --batch --generate-key /tmp/gpg-batch
        rm -f /tmp/gpg-batch
    else
        echo -e "${WHITE}==> Would you like to generate a GPG key? [y/N]${NC}"
        read -r answer

        if [[ $answer == "y" || $answer == "Y" ]]; then
            echo -e "${WHITE}==> Generating GPG key...${NC}"
            gpg --full-generate-key
        else
            echo -e "${WHITE}==> Skipping GPG key generation...${NC}"
        fi
    fi
fi

# Vim customisations...
# if [[ -f "/usr/bin/vim" ]]; then
#     _link_and_backup ".vimrc"
#     _link_and_backup ".vim"
#     $DOTFILES_DIR/.vim/install.sh
# else
#     echo -e "${YELLOW}===> Vim is not installed. Skipping.${NC}"
# fi

echo -e "${GREEN}==> Initialization complete.${NC}"
exit 0