dotey

#!/usr/bin/env bash

# The MIT License (MIT)
# Copyright (c) 2012 Fabian Franke http://fabianfranke.de

# layout variables
GREEN="\033[0;32m"
YELLOW="\033[0;33m"
RED="\033[0;31m"
COLORRESET="\033[0m"
SPACER="   "

# filename for the private dotfiles archive
packfilename='privdots'

# the big linker
function install() {
  backupall=false
  deleteall=false
  skipall=false
  backupdir=${PWD}/backup

  # in case we want to link from a sub directory in .dotfiles, e.g. privatedots
  if [ ! -z $1 ]; then
    prefix=$1
  fi

  # we have to check if there are any linkable files at all or at strange
  # and wild links will appear
  if  [ "`ls $prefix*.symlink | wc -l`" != 0 ]; then
    echo -e "$GREEN-$COLORRESET found some .symlink files, starting symlinking"
  else
    echo -e "$RED-$COLORRESET found no .symlink files, nothing to do"
    exit
  fi

  for file in $prefix*.symlink; do

    # target is the planned home directory link
    # purefile name is needed in case we work with a prefix
    target="$HOME/.`basename ${file%.*}`"
    purefilename=`basename ${file%.*}`

    # if the target is already linked to our .dotfiles directory skip it
    if [ "`readlink $target`" ==  "${PWD}/$file" ]; then
      continue
    fi

    # we have an existing .dotfile
    if [ -e "$target" ]; then
      # go down the fast lane if the user has already wished to do something
      # to all files
      if $backupall ; then
        echo -e "$SPACER$YELLOW-$COLORRESET moving $target to backup"
        mv $target $backupdir/
        echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
        ln -s ${PWD}/$file $target
      elif $deleteall ; then
        echo -e "$SPACER$RED-$COLORRESET deleting existing $target"
        rm -r $target
        echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
        ln -s ${PWD}/$file $target
      elif $skipall ; then
        echo -e "$SPACER$YELLOW-$COLORRESET skipping ${file%.*}"
      else
        # no general decision saved, let's ask
        # lower case: decision for the current file
        # upper case: decision for the current and all following file conflicts
        echo "$target exists. What to do? (b)ackup, (d)elete, (s)kip. Or for this and further existing files (B)ackup, (S)kip, (D)elete"
        read filedecision
        case $filedecision in
         b)
           echo -e "$SPACER$YELLOW-$COLORRESET moving $target to .dotfiles/backup/.$purefilename"
           mv $target $backupdir/
           echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
           ln -s ${PWD}/$file $target
           ;;
         B)
           echo backup all
           backupall=true
           echo -e "$SPACER$YELLOW-$COLORRESET moving $target to .dotfiles/backup/.$purefilename"
           mv $target $backupdir/
           echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
           ln -s ${PWD}/$file $target
           ;;
         d)
           echo -e "$SPACER$RED-$COLORRESET deleting $target"
           rm -r $target
           echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
           ln -s ${PWD}/$file $target
           ;;
         D)
           echo deleting all
           deleteall=true
           echo -e "$SPACER$RED-$COLORRESET deleting $target"
           rm -r $target
           echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
           ln -s ${PWD}/$file $target
           ;;
         s)
           echo -e "$SPACER$YELLOW-$COLORRESET skipping ${file%.*}"
           continue
           ;;
         S)
           echo skipping all
           skipall=true
           echo -e "$SPACER$YELLOW-$COLORRESET skipping ${file%.*}"
           continue
           ;;
         *)
           echo -e "$RED-$COLORRESET wrong input, skipping"
           continue
           ;;
       esac
      fi

    # no existing .dotfile in ~, link found file
    else
      echo -e "$SPACER$GREEN+$COLORRESET linking $file to $target"
      ln -s ${PWD}/$file $target
    fi
  done
}

function uninstall() {

  # in case we want to link from a sub directory if .dotfiles. e.g. privatedots
  if [ ! -z $1 ]; then
    prefix=$1
  fi

  # loop over *.symlink file in user .dotfiles/$prefix directory if a link from
  # the home directory points to the file, delete the symlink. When a backup is
  #found restore it.
  for sym in $prefix*.symlink; do
    purefilename=`basename ${sym%.*}`
    target="$HOME/.$purefilename"
    if [ "`readlink $target`" ==  "${PWD}/$sym" ]; then
      echo -e "$RED-$COLORRESET Deleting $target"
      rm $target
      if [ -e "backup/.$purefilename" ]; then
        echo -e "$SPACER$GREEN+$COLORRESET restoring original file"
        mv backup/.$purefilename $HOME/
      fi
    fi
  done
}

# fetch updates from git repo and _hide_ the git output for a nicer user experience
function update() {
  echo -e "$GREEN-$COLORRESET Fetching updates from github"
  git pull >/dev/null
  link
}

# someday todo: merge with install()
function link() {
  echo -e "$GREEN-$COLORRESET Trying to link new files"
  install
}

# check if a private dotfiles archive with name $packfilename is present
# decrypt, and extract
# call install() with sub directory prefix
function installprivatedots() {
  echo -e "$GREEN-$COLORRESET Decrypting, unpacking and linking your private dotfiles"
  if [ -e "$packfilename.des3" ]; then
    openssl des3 -a -d -in $packfilename.des3 -out $packfilename.tbz2 && tar xfj $packfilename.tbz2 && rm $packfilename.tbz2 $packfilename.des3
  else
    echo -e "$RED-$COLORRESET ERROR: Cannot find transport container."
    echo -e "$SPACER$RED-$COLORRESET It should be named $packfilename.des3"
  fi
  install "privatedots/"
}

# someday todo: merge with link() and install()
function linkprivatedots() {
  echo -e "$GREEN-$COLORRESET Updating links from your private dotfiles"
  install "privatedots/"
}

# pack ~/.dotfiles/privatedots with tar/bzip2 and encrypt with openssl
function packprivatedots() {
  echo -e "$GREEN-$COLORRESET compressing and encrypting private dotfiles"
  echo -e "$SPACER$YELLOW-$COLORRESET prompting for a password in a little while"
  tar -jcf $packfilename.tbz2 privatedots && openssl des3 -a -in $packfilename.tbz2 -out $packfilename.des3 && rm $packfilename.tbz2
  echo
  if [ -e "$packfilename.des3" ]; then
    echo -e "$GREEN-$COLORRESET your private dotfiles are ready at $PWD/$packfilename.des3"
    echo -e "$SPACER Transfer them over to your other installations and run ./dotey installprivatedots"
  else
    echo -e "$RED-$COLORRESET something went terribly wrong"
  fi
}

function delprivatedots() {
  echo -e "$SPACER$RED-$COLORRESET unlinking and deleting your private dotfiles from this box"
  echo -e "$SPACER  CTRL+C to abort now"

  echo -en "$RED"
  for i in {1..5}; do
    echo -n .
    sleep 1
  done
  echo -e "$COLORRESET"

  uninstall "privatedots/"
  rm privatedots/*
}

function usage() {
  cat << EOF
How to use this:

./dotey install
  Symlink dotfiles to the home directory

./dotey uninstall
  Remove all dotfile symlinks from home directory and restore from backup if possible

./dotey update
  Fetch updates from the repository and update symlinks if needed

./dotey link
  Update symlinks if needed

./dotey privatedots install
  Decrypt and unpack your private dotfiles and install them at this computer

./dotey privatedots link
  Link newly added private dotfiles to your home directory

./dotey privatedots pack
  Encrypt and pack your private dotfiles to transfer them to another box

./dotey privatedots delete
  Delete your private dotfiles from this computer

EOF
}

case $1 in
  install)
    install
    ;;
  uninstall)
    uninstall
    ;;
  link)
    link
    ;;
  update)
    update
    ;;
  privatedots)
    case $2 in
      pack)
        packprivatedots
        ;;
      install)
        installprivatedots
        ;;
      link)
        linkprivatedots
        ;;
      delete)
        deleteprivatedots
        ;;
      *)
        usage
        ;;
    esac
    ;;
  *)
    usage
    ;;
esac