setupenv

#!/usr/bin/env bash

LINKED=0
FAILED=0
ERRORS=()

show_usage ()
{
    echo
    echo "Links dotfiles to your $HOME directory"
    echo
    echo "USAGE: $0 [-f] [--force]"
    echo
    echo "  force        Prevents confirmation prompt to override existing files"
    echo

    [ ! -z "$1" ] && exit "$1"
}

link_dotfiles ()
{
    local FILE

    if [ -z "$1" ]; then
        return
    fi

    pushd "$1" > /dev/null || return

    for FILE in .*
    do
        if [[ "$FILE" == "." ]] || [[ "$FILE" == ".." ]] || [[ "$FILE" == ".git" ]]; then
            continue
        fi

        if [[ "$FILE" == ".ssh" ]]; then
            # make our ~/.ssh if it doesn't exist
            if ! mkdir -p "$HOME/.ssh"; then
                ERRORS+=("Failed to create $HOME/.ssh")
                ((++FAILED))
                continue
            fi
            chmod 700 "$HOME/.ssh"

            # setup proper permissions on existing keys
            [[ -e "$HOME/.ssh/id_rsa" ]] && chmod 600 "$HOME/.ssh/id_rsa"
            [[ -e "$HOME/.ssh/id_rsa.pub" ]] && chmod 644 "$HOME/.ssh/id_rsa.pub"
            [[ -e "$HOME/.ssh/id_ed25519" ]] && chmod 600 "$HOME/.ssh/id_ed25519"
            [[ -e "$HOME/.ssh/id_ed25519.pub" ]] && chmod 644 "$HOME/.ssh/id_ed25519.pub"

            touch "$HOME/.ssh/authorized_keys"
            chmod 644 "$HOME/.ssh/authorized_keys"
            touch "$HOME/.ssh/known_hosts"
            chmod 644 "$HOME/.ssh/known_hosts"

            # Generate allowed_signers from local signing key for git signature verification
            local EMAIL
            EMAIL=$(git config user.email 2>/dev/null)
            if [ -n "$EMAIL" ] && [ -f "$HOME/.ssh/id_ed25519.pub" ]; then
                echo "$EMAIL $(cat "$HOME/.ssh/id_ed25519.pub")" > "$HOME/.ssh/allowed_signers"
                chmod 644 "$HOME/.ssh/allowed_signers"
            fi

            # copy any files
            if cp -rfv "$PWD/$FILE/"* "$HOME/$FILE/"; then
                ((++LINKED))
            else
                ERRORS+=("Failed to copy $FILE")
                ((++FAILED))
            fi

            continue
        fi

        if [ -e "$FILE" ]; then
            if [ -d "$HOME/$FILE" ]; then
                rm -rf "$HOME/$FILE"
            fi
            if ln -fsv "$PWD/$FILE" "$HOME/$FILE"; then
                ((++LINKED))
            else
                ERRORS+=("Failed to link $FILE")
                ((++FAILED))
            fi
        fi
    done

    popd > /dev/null || return
}

if [ "$1" == "--help" ]; then
    show_usage 64
fi

if [[ "$1" != "--force" && "$1" != "-f" ]]; then
    echo
    read -p "This may overwrite existing files in your home directory. Are you sure? (Y/n) " -n 1
    echo
    case $REPLY in
        [Nn]*)
            exit 64
            ;;
    esac
fi

# Link dotfiles from this dir
link_dotfiles "${PWD}"

# Copy bin files from this dir
if [ -d "./bin" ]; then
    cp -rfv "${PWD}/bin" "$HOME"
fi

# Source platform specific profile
UNAME=$(uname)

case $UNAME in
    "")
        ;;
    "Darwin")
        PLATFORM=mac
        ;;
    *)
        PLATFORM="$(tr '[:upper:]' '[:lower:]' <<< "$UNAME")"
        ;;
esac

if [ -d "$PLATFORM" ]; then
    pushd "$PLATFORM" > /dev/null || exit 1

    # Link dotfiles from platform dir
    link_dotfiles "${PWD}"

    # Run setupenv if exists
    if [ -x "./setupenv" ]; then
        ./setupenv
    fi

    # Copy bin files from platform dir
    if [ -d "./bin" ]; then
        cp -rfv "${PWD}/bin" "$HOME"
    fi

    popd > /dev/null || true
fi

# Summary
echo
echo "Done: $LINKED linked, $FAILED failed"

if [ ${#ERRORS[@]} -gt 0 ]; then
    echo
    echo "Errors:"
    for err in "${ERRORS[@]}"; do
        echo "  - $err"
    done
    exit 1
fi