#!/bin/bash
#
# Worklog wrapper
#
# Alternatives to current worklog backend:
#
# https://github.com/distilledhype/node-worklog
# https://github.com/snaptortoise/worklog
# https://github.com/tormaroe/worklog
# https://github.com/winged/worklog
# https://github.com/yaccz/worklog

BASE="$HOME/file"
CODE="$HOME/code"
GROUP="$1"
ACTION="$2"
BASENAME="`basename $0`"
EDITOR=${EDITOR:=vi}

# Set window title
# http://stackoverflow.com/questions/899609/gnu-screen-run-script-that-sends-commands-to-the-screen-session-it-is-being-run
function timelog_window_title {
  if [ -n "$STY" ]; then
    screen -p $WINDOW -X title "$*"
  else
    if which xtitle &> /dev/null; then
      xtitle "$*"
    fi
  fi
}

# Setup a new worklog configuration
function timelog_setup {
  cat > $BASE/$GROUP/worklog/projects <<EOF
# Worklog project file
# note that projects appear in Worklog in REVERSE order

#H:Hosting
#I:Infrastructure
#O:Organization
EOF

  echo "First run, you should edit your project list..."
  $EDITOR $BASE/$GROUP/worklog/projects
  ( cd $BASE/$GROUP/worklog && worklog )
}

# Loop over available worklogs
function timelog_menu {
  # Basic variables
  count=0
  bases=(`find "$BASE" -maxdepth 2 -name worklog -exec dirname {} \;`)
  codes=(`find "$CODE" -maxdepth 2 -name worklog -exec dirname {} \;`)
  candidates=(${bases[@]} ${codes[@]})

  # Clear the screen everytime we enter the menu
  clear

  # Draw menu and parse options
  if [ ! -z "$candidates" ]; then
    echo "Please select one of the following projects:"
    echo ""

    for candidate in ${candidates[@]}; do
      echo -e "\t[$count] `basename $candidate`"
      let count++
    done

    echo ""
    read -p "Enter option (r to refresh, Ctrl-C or q to quit): " option

    if [ "$option" == "q" ]; then
      exit
    elif [ "$option" == "r" ]; then
      return
    elif [[ "$option" =~ ^[0-9]+$ ]] && [ ! -z "${candidates[$option]}" ]; then
      project="`basename ${candidates[$option]}`"
      timelog $project
    else
      echo "Invalid option"
      exit 1
    fi
  else
    echo "Usage: $BASENAME [group] [edit]"
    exit 1
  fi
}

# Run worklog for a given project
function timelog_run {
  # Set window title
  timelog_window_title "log: $GROUP"

  # Set base folder
  if [ -d "$CODE/$GROUP" ] && [ ! -d "$BASE/$GROUP" ]; then
    BASE="$CODE"
  fi

  # Ensure folder exist
  mkdir -p $BASE/$GROUP/worklog

  # Run
  if [ ! -e "$BASE/$GROUP/worklog/projects" ]; then
    timelog_setup
  elif [ "$ACTION" == "edit" ]; then
    $EDITOR $BASE/$GROUP/worklog/projects
  else
    ( cd $BASE/$GROUP/worklog && worklog )
  fi
}

# Main
if [ ! -z "$GROUP" ]; then
  timelog_run
else
  while true; do
    # Set window title
    timelog_window_title "log"

    timelog_menu
  done
fi

# Restore screen title
timelog_window_title terminal