setup

#!/bin/sh
#
# Dotfiles setup script from https://github.com/Joushou/dotfiles
#
# Use as you please, but give some credit, will ya'?
#

GITPATH=https://github.com/Joushou/dotfiles.git
USERHOME=~
TARGET=$USERHOME/.dotfiles
LOGFILE=$TARGET/setup_log

# Toggles for command-line options
OPT_UPGRADE=0
OPT_UPDATE=0
OPT_VERIFY=0
OPT_DRYRUN=1
OPT_CASK=0
OPT_CONF=0
OPT_TOOLCHAIN=0
OPT_SHELL=0
OPT_GITCLONE=0
OPT_DUPES=0
OPT_VERBOSE=1

# Color stuff
case $TERM in
    xterm*)
        RED_START="\033[1;31m"
        GREEN_START="\033[1;32m"
        YELLOW_START="\033[1;33m"
        WHITE_START="\033[1;37m"
        COLOR_END="\033[0m"
        ;;
    screen*)
        RED_START="\033[1;31m"
        GREEN_START="\033[1;32m"
        YELLOW_START="\033[1;33m"
        WHITE_START="\033[1;37m"
        COLOR_END="\033[0m"
        ;;
esac

###############################################################################
#                                                                             #
# Utility functions                                                           #
#                                                                             #
###############################################################################

# This is stupid, but it works
is_set() {
    return $1
}

# Checks if the tool is available
in_path() {
    which $1 1>/dev/null 2>/dev/null
    ret=$?
    printf '%b' "[tool availability check for $1: $ret]\n" >> $LOGFILE
    return $ret
}

# Log error and exit
error() {
    printf '%b' "$RED_START! $1$COLOR_END\n"
    printf '%b' "! $1\n" >> $LOGFILE
    exit 1
}

# Log
log() {
    printf '%b' "$GREEN_START> $WHITE_START$1$COLOR_END\n"
    printf '%b' "> $1\n" >> $LOGFILE
}

# Run, log and handle error
run() {
    printf '%b' "$GREEN_START>    $YELLOW_START$1$COLOR_END\n"
    printf '%b' ">    $1\n" >> $LOGFILE
    cmd=${@#$1}
    success=1

    printf '%b' "[$1: $cmd]\n" >> $LOGFILE

    # TODO Fix logging in verbose mode

    if is_set $OPT_VERBOSE
    then
        is_set $OPT_DRYRUN || $cmd && success=0
    else
        is_set $OPT_DRYRUN || $cmd >> $LOGFILE 2>&1 && success=0
    fi

    printf '%b' "\n" >> $LOGFILE

    if [ "$success" = "1" ]
    then
        echo $output
        error "FAILED, check $LOGFILE for details"
    fi
}

###############################################################################
#                                                                             #
# Generic Setup                                                               #
#                                                                             #
###############################################################################

# Setup the dotfile git repo
setup_git() {
    log "Setting up dotfiles repo"
    # Hacky "DIR IS NOT EMPTY!" workaround
    run "Initialising git" git init
    run "Adding remote" git remote add origin $GITPATH
    run "Pulling master" git pull --force origin master
    run "Initialising submodules" git submodule update --init
}

# Setup symlinks
setup_symlinks() {
    log "Setting up symlinks"
    run "zshrc" ln -s $TARGET/zshrc $USERHOME/.zshrc
    run "vimrc" ln -s $TARGET/vimrc $USERHOME/.vimrc
    run "gvimrc" ln -s $TARGET/gvimrc $USERHOME/.gvimrc
    run "nvimrc" ln -s $TARGET/vimrc $USERHOME/.nvimrc
    run "nvim" ln -s $USERHOME/.vim $USERHOME/.nvim
}

# Prepare vim
setup_vim() {
    log "Setting up VIM"
    run "Make extra .vim subdirectories" mkdir -p $USERHOME/.vim/backup $USERHOME/.vim/swap $USERHOME/.vim/colors
    run "Clone Vundle" git clone https://github.com/gmarik/Vundle.vim.git $USERHOME/.vim/bundle/Vundle.vim
    run "Download Molokai" eval "curl https://raw.githubusercontent.com/tomasr/molokai/master/colors/molokai.vim > $USERHOME/.vim/colors/molokai.vim"
    run "Install plugins" eval "vim +PluginInstall +qall > /dev/null" # Avoid garbage
    sleep 0.5
    run "Ensure xon isn't set" eval "stty -ixon || true"
    run "Make tty sane" eval "stty sane || true"
}

# Prepare antigen
setup_zsh() {
    log "Setting up zsh"
    run "Downloading zgen" eval "curl https://raw.githubusercontent.com/tarjoilija/zgen/master/zgen.zsh > $USERHOME/.zgen.zsh"
    run "Initialising zsh" eval "zsh $USERHOME/.zshrc > /dev/null" # Avoid garbage
    run "Changing user shell" sudo chsh -s $(which zsh) $USER
}

# Setup dispatcher
setup() {
    is_set $OPT_GITCLONE && setup_git
    setup_symlinks
    setup_vim
    is_set $OPT_SHELL && setup_zsh
}

###############################################################################
#                                                                             #
# OS X specific stuff                                                         #
#                                                                             #
# TODO setup powerline fonts                                                  #
###############################################################################

# Set defaults
osx_conf() {
    log "Configuring OS X"
    run "Be silent on boot" sudo nvram SystemAudioVolume=" "
    run "Increasing window resize speed" defaults write NSGlobalDomain NSWindowResizeTime -float 0.001
    run "Increasing mission control animation speed" defaults write com.apple.dock expose-animation-duration -float 0.1
    run "Setting fast key repeat" defaults write NSGlobalDomain KeyRepeat -int 0
    run "Always use sub-pixel font rendering" defaults write NSGlobalDomain AppleFontSmoothing -int 2
    run "Setting Dock scaling effect" defaults write com.apple.dock mineffect -string "scale"
    run "Setting Terminal.app UTF-8 mode" defaults write com.apple.terminal StringEncodings -array 4
    run "Unhiding ~/Library" chflags nohidden $USERHOME/Library
}

# Install brew
brew_install() {
    run "Install brew" ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
    run "Making paths file" sudo tee /etc/paths <<EOS
/usr/local/bin
/usr/local/opt/coreutils/libexec/gnubin
/usr/bin
/bin
/usr/local/sbin
/usr/sbin
/sbin
EOS
    eval $(/usr/libexec/path_helper -s)
}

# Setup brew and install packages
pre_osx() {
    is_set $OPT_CONF &&      osx_conf

                             log "Installing packages"
    in_path brew ||          brew_install

    is_set $OPT_UPDATE &&    run "Updating brew" brew update
    is_set $OPT_UPGRADE &&   run "Upgrading brew" brew upgrade
    is_set $OPT_VERIFY &&    run "Verifying brew" brew doctor

    is_set $OPT_DUPES &&     run "Installing dupe tap" brew tap homebrew/dupes

    is_set $OPT_CASK &&      run "Setting up cask" brew install caskroom/cask/brew-cask

                             # TODO zsh is technically a dupe, as zsh is bundled in old version
    is_set $OPT_SHELL &&     run "Shell" brew install tmux zsh
    is_set $OPT_DUPES &&     run "Newer shell utils" brew install coreutils findutils grep
                             run "Shell utils" brew install ascii
                             run "Version control" brew install git mercurial
                             run "Code assist" brew install ctags
    is_set $OPT_TOOLCHAIN && run "Toolchain" brew install autoconf automake go

                             # editor stuff
    is_set $OPT_DUPES &&     run "VIM" brew install vim
    is_set $OPT_CASK &&      run "MacVIM" brew cask install macvim flashlight iterm2 arduino
}

# Cleanup after brew and git
post_osx() {
    log "Cleaning up"
    run "Git" git gc --aggressive
    run "Brew" brew cleanup
}

###############################################################################
#                                                                             #
# Linux                                                                       #
#                                                                             #
# TODO: Add pacman support.                                                   #
###############################################################################

# Install packages
pre_linux() {
    if in_path apt-get
    then
                                    log "Installing packages with APT"
        is_set $OPT_UPDATE &&       run "Updating APT" sudo apt-get -y update
        is_set $OPT_UPGRADE &&      run "Upgrading packages" sudo apt-get -y upgrade

        is_set $OPT_SHELL &&        run "Shell" sudo apt-get -y install tmux zsh
                                    run "Shell utils" sudo apt-get -y install ascii
                                    run "Version control" sudo apt-get -y install git mercurial
                                    run "Code assist" sudo apt-get -y install ctags
        is_set $OPT_TOOLCHAIN &&    run "Toolchain" sudo apt-get -y install binutils autoconf automake gdb clang golang

                                    # editor stuff
                                    run "VIM" sudo apt-get -y install vim
    elif in_path yum
    then
        log "Installing packages with YUM"
        is_set $OPT_UPGRADE &&      run "Updating YUM" sudo yum -y update
        is_set $OPT_SHELL &&        run "Shell" sudo yum -y install tmux zsh
                                    run "Shell utils" sudo yum -y install ascii
                                    run "Version control" sudo yum -y install git mercurial
                                    run "Code assist" sudo yum -y install ctags
        is_set $OPT_TOOLCHAIN &&    run "Toolchain" sudo yum -y install binutils autoconf automake gdb clang golang
                                    run "VIM" sudo yum -y install vim
   elif in_path pacman
	then
	   log "Installing packages with Pacman"
	   is_set $OPT_UPDATE &&		   run "Updating pacman" sudo pacman --noconfirm -Sy
	   is_set $OPT_UPGRADE && 		   run "Upgrading pacman" sudo pacman --noconfirm -Su
		                  			   run "Shell" sudo pacman --noconfirm -S tmux zsh
					                     run "Shell utils" sudo pacman --noconfirm -S ascii
		                     			run "Version control" sudo pacman --noconfirm -S git mercurial
					                     run "Code assist" sudo pacman --noconfirm -S ctags
	   is_set $OPT_TOOLCHAIN &&	   run "Toolchain" sudo pacman --noconfirm -S base-devel go
					                     run "VIM" sudo pacman --noconfirm -S vim
   else
        error "Package manager unknown"
    fi
}

# Cleanup apt and git
post_linux() {
                        log "Cleaning up"
                        run "Git" git gc --aggressive
    in_path apt-get &&  run "APT" sudo apt-get clean
}

###############################################################################
#                                                                             #
# BSD                                                                         #
#                                                                             #
###############################################################################

# Install packages
pre_bsd() {
    if in_path pkg
    then
                                    log "Installing packages with pkg"

        is_set $OPT_SHELL &&        run "Shell" sudo ASSUME_ALWAYS_YES=true pkg install tmux zsh
                                    run "Version control" sudo ASSUME_ALWAYS_YES=true pkg install git mercurial
                                    run "Code assist" sudo ASSUME_ALWAYS_YES=true pkg install ctags
        is_set $OPT_TOOLCHAIN &&    run "Toolchain" sudo ASSUME_ALWAYS_YES=true pkg install binutils autoconf automake gdb clang-devel llvm-devel go

                                    # editor stuff
                                    run "VIM" sudo ASSUME_ALWAYS_YES=true pkg install vim
    else
        error "Package manager unknown"
    fi
}

# Cleanup apt and git
post_bsd() {
                        log "Cleaning up"
                        run "Git" git gc --aggressive
    in_path pkg &&      run "pkg" sudo ASSUME_ALWAYS_YES=true pkg clean
}

###############################################################################
#                                                                             #
# Main                                                                        #
#                                                                             #
###############################################################################

#
# Handle parameters
#
# TODO: Add force OS option
mkdir -p $TARGET
for i in $@
do
    case $i in
        --no-upgrade)
            OPT_UPGRADE=1
            ;;
        --no-update)
            OPT_UPDATE=1
            ;;
        --no-verify)
            OPT_VERIFY=1
            ;;
        --no-cask)
            OPT_CASK=1
            ;;
        --no-conf)
            OPT_CONF=1
            ;;
        --no-toolchain)
            OPT_TOOLCHAIN=1
            ;;
        --no-shell)
            OPT_SHELL=1
            ;;
        --no-dupes)
            OPT_DUPES=1
            ;;
        --no-gitclone)
            OPT_GITCLONE=1
            ;;
        --dry-run)
            OPT_DRYRUN=0
            ;;
        --verbose)
            OPT_VERBOSE=0
            ;;
        *)
            error "Unknown parameter: $i"
            ;;
    esac
done

log "Starting setup"
run "Entering directory" cd $TARGET
run "Requesting sudo rights" sudo -v

#
# OS detect
#

darwin() {
   log "OS X detected"
   pre_osx
   setup
   post_osx
}

linux() {
   log "Linux detected"
   pre_linux
   setup
   post_linux
}

bsd() {
   log "BSD detected"
   pre_bsd
   setup
   post_bsd
}

os=$(uname)
case "$os" in
    Darwin)
        darwin
        ;;
    Linux)
        linux
        ;;
    FreeBSD)
        bsd
        ;;
    *)
        # We might just be on a dash-based OS which doesn't give us OSTYPE (ubuntu/debian)
        if [ -e /etc/debian_version ]
        then
            pre_linux
            setup
            post_linux
        else

            error "Platform of type $OSTYPE not supported"
        fi
        ;;
esac

log "All done"