dot

#!/bin/bash

options="install diff status"
list=(
    '.config'
    '.local'
    '.tmux.conf'
)

printUsage()
{
    echo "Usage: dot [OPTION]"
    echo "OPTIONS:"
    echo "  install [--config]"
    echo "  diff    [--config]"
    echo "  status  [-a] [-l num]"
}

printTrackedUntracked()
{
    local dotpath=$1
    local printed=$2
    local found=0

    cd "$HOME" || return 1

    for dir in "${list[@]}"; do
        [ -d "$dotpath/$dir" ] || continue

        while IFS= read -r root; do
            [ -e "$root" ] || continue
            [ -d "$root" ] || continue

            local missing_dirs=()

            while IFS= read -r entry; do
                local skip=0

                for missing_dir in "${missing_dirs[@]}"; do
                    if [[ "$entry" == "$missing_dir/"* ]]; then
                        skip=1
                        break
                    fi
                done

                [ "$skip" == 1 ] && continue
                [ -e "$dotpath/$entry" ] && continue

                [ "$printed" == 0 ] && printed=1 && echo && echo "Untracked configuration:"
                echo -e "\e[1;31m    $entry\e[0;37m"
                found=1

                if [ -d "$entry" ]; then
                    missing_dirs+=("$entry")
                fi
            done < <(find "$root" -mindepth 1 | sort)
        done < <(cd "$dotpath" && find "$dir" -mindepth 1 -maxdepth 1 -type d | sort)
    done

    return "$found"
}

printAllUntracked()
{
    local dotpath=$1
    local printed=$2
    local maxdepth=$3
    local found=0

    cd "$HOME" || return 1

    for dir in "${list[@]}"; do
        [ -d "$dir" ] || continue
        [ -d "$dotpath/$dir" ] || continue

        while IFS= read -r root; do
            [ -e "$dotpath/$root" ] && continue

            [ "$printed" == 0 ] && printed=1 && echo && echo "Untracked configuration:"
            while IFS= read -r entry; do
                echo -e "\e[1;31m    $entry\e[0;37m"
            done < <(find "$root" -maxdepth "$maxdepth" | sort)
            found=1
        done < <(find "$dir" -mindepth 1 -maxdepth 1 -type d | sort)
    done

    return "$found"
}

dotStatus()
{
    local show_all=0
    local level=3
    flag=0
    error=0
    dotpath=$PWD
    track=0

    while [ $# -gt 0 ]; do
        case $1 in
            -a)
                show_all=1
                ;;
            -l)
                shift

                if [ -z "$1" ] || [[ ! "$1" =~ ^[0-9]+$ ]]; then
                    echo "error: -l requires a numeric depth" && echo
                    printUsage
                    return 2
                fi

                level=$1
                ;;
            *)
                echo "error: wrong option" && echo
                printUsage
                return 2
                ;;
        esac

        shift
    done

    echo "Dot status"

    # Basic Configuration File
    for dir in ${list[@]}; do
        files=`find $dir -type f`

        for file in $files; do
            if [ "$MSYSTEM" ]; then
                if [[ "$file" != *"lese"* ]]; then
                    continue
                fi
            fi

            [ ! -f "$HOME/$file" ] && echo -e "    \e[1;31mnonexistent:   $file\e[0m" && error=1 && continue
            res="`diff $file $HOME/$file`"

            if [ "$res" ]; then
                [ $flag == 0 ] && flag=1 && echo "Changes for dotfiles:"
                echo -e "\e[1;31m    modified:   $file\e[0;37m"
            fi
        done
    done

    printTrackedUntracked "$dotpath" "$track"
    if [ $? == 1 ]; then
        track=1
    fi

    if [ "$show_all" == 1 ]; then
        printAllUntracked "$dotpath" "$track" "$level"
        if [ $? == 1 ]; then
            track=1
        fi
    fi

    [ $flag == 0 -a $track == 0 -a $error == 0 ] && echo "all same"
}

diffFile()
{
    res="`diff $1 $2`"

    if [ "$res" ]; then
        if type diff-so-fancy >/dev/null 2>&1; then
            echo && (diff -ru $1 $2 | diff-so-fancy)
        else
            diff -ru $1 $2
        fi
    fi
}

dotDiff()
{
    echo "Dot diff"

    # Basic Configuration File
    for dir in ${list[@]}; do
        files=`find $dir -type f`

        for file in $files; do
            if [ "$MSYSTEM" ]; then
                if [[ "$file" != *"lese"* ]]; then
                    continue
                fi
            fi

            [ ! -f "$HOME/$file" ] && echo -e "    \e[1;31mnonexistent:   $file\e[0m" && continue
            diffFile $file $HOME/$file
        done
    done
}

dotInstall()
{
    echo "Dot install"

    # Basic Configuration File
    for dir in ${list[@]}; do
        if type rsync >/dev/null 2>&1; then
            rsync -av $dir $HOME
        else
            cp -rf $dir $HOME
        fi
        [ $? == 0 ] && echo "$dir install done"
    done

    echo && echo -e "Do you want to initialize the configuration? (y/n) \c" && read option
    if [ "$option" == "n" ]; then
        exit 0
    fi

    echo "" >> $HOME/.profile
    echo "source \"\$HOME/.config/lese/init.sh\" --file" >> $HOME/.profile
    echo "profile init successful"
}

if [ $# -lt 1 ]; then
    printUsage
    exit 1
fi

case $1 in
    status)
        dotStatus "${@:2}"
        ;;
    diff)
        dotDiff $2
        ;;
    install)
        dotInstall $2
        ;;
    *)
        echo "error: wrong option" && echo
        printUsage
        exit 2
        ;;
esac