2 files changed, 36 insertions, 4 deletions

diff --git a/bin/task/context_spend_time.sh b/bin/task/context_spend_time.sh
new file mode 100755
index 0000000..cbadc47
--- /dev/null
+++ b/bin/task/context_spend_time.sh
@@ -0,0 +1,31 @@
+#! /bin/bash
+
+LOG_FILE=~/log/context
+context="$(cat ~/.task/context)"
+count="$1"
+
+if [ -z "$context" ] || [ "$context" = none ]; then 
+  exit 1
+fi
+
+if [ -n "$1" ]; then
+  # Update Log file
+  while read -r ctx sec; do
+    if [ "$ctx" = "$context" ]; then
+      summary=$(( "$sec" + "$count" ))
+      update=true
+      break
+    fi
+  done <$LOG_FILE
+  if [ "$update" = true ]; then
+    sed -i -E "s/^$context.*/$context\t$summary" $LOG_FILE
+  else
+    echo -e "$context\t$summary" >>$LOG_FILE
+  fi
+else
+  # Print times for each context
+  while read -r ctx sec; do
+    echo -ne "$ctx\t"
+    date -u -d @"$sec" +%H:%M:%S
+  done <$LOG_FILE
+fi
diff --git a/bin/unix/timer.sh b/bin/unix/timer.sh
index 152d828..b4f48c5 100755
--- a/bin/unix/timer.sh
+++ b/bin/unix/timer.sh
@@ -5,7 +5,8 @@ exec 3>&1
 exec 1>/dev/tty
 
 SIGNAL=${1:-SIGTERM}
-COMMAND="$2"
+COMMAND_START="$2"
+COMMAND_EXIT="$3"
 
 # If SIGNAL is received, switch to next display
 trap 'next_display' "$SIGNAL"
@@ -34,8 +35,8 @@ toggle_timer() {
 read -p '? ' -r input
 # Disable input on terminal
 stty -echo 
-# If COMMAND is given, run it after timer is set
-result="$([ -n "$COMMAND" ] && eval "$COMMAND" 2>&1)"
+# If COMMAND_START is given, run it after timer is set
+result="$([ -n "$COMMAND_START" ] && eval "$COMMAND_START" 2>&1)"
 notify-send "$result" &>/tmp/openbox
 
 
@@ -78,7 +79,7 @@ timer() {
   done
 }
 
-trap 'exec 1>&3; echo $count; send-notify foo' EXIT KILL TERM STOP
+trap 'exec 1>&3; [ -n "COMMAND_EXIT" ] && eval "$COMMAND_EXIT"' EXIT QUIT HUP
 
 while [ $count -lt $SET ]; do
   [ $stop = true ] && sleep 0.3 && continue