Merge branch 'master' of git.yggdrasil.li:utils

6 files changed, 272 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..41fbeb0
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+**/result
diff --git a/nix/default.nix b/nix/default.nix
index cef8800..097d092 100644
--- a/nix/default.nix
+++ b/nix/default.nix
@@ -8,4 +8,6 @@ self: super:
   rolling-directory = self.callPackage ./rolling-directory.nix {};
   recv = self.callPackage ./recv.nix {};
   notmuch-tcp = self.callPackage ./notmuch-tcp.nix {};
+  persistent-nix-shell = self.callPackage ./persistent-nix-shell.nix {};
+  worktime = self.callPackage ./worktime.nix {};
 }
diff --git a/nix/persistent-nix-shell.nix b/nix/persistent-nix-shell.nix
new file mode 100644
index 0000000..7b371c6
--- /dev/null
+++ b/nix/persistent-nix-shell.nix
@@ -0,0 +1,22 @@
+{ stdenv
+, zsh
+}:
+
+stdenv.mkDerivation rec {
+  pname = "persistent-nix-shell";
+  version = "0.1";
+  src = ../persistent-nix-shell;
+
+  phases = [ "buildPhase" "installPhase" ];
+
+  inherit zsh;
+
+  buildPhase = ''
+    substituteAll $src persistent-nix-shell
+  '';
+
+  installPhase = ''
+    install -m 0755 -D -t $out/bin \
+      persistent-nix-shell
+  '';
+}
diff --git a/nix/worktime.nix b/nix/worktime.nix
new file mode 100644
index 0000000..ddab8b2
--- /dev/null
+++ b/nix/worktime.nix
@@ -0,0 +1,22 @@
+{ stdenv
+, python37
+}:
+
+stdenv.mkDerivation rec {
+  pname = "worktime";
+  version = "3";
+  src = ../worktime.py;
+
+  phases = [ "buildPhase" "installPhase" ];
+
+  python = python37.withPackages (ps: with ps; [pyxdg dateutil uritools requests configparser]);
+
+  buildPhase = ''
+    substituteAll $src worktime
+  '';
+
+  installPhase = ''
+    install -m 0755 -D -t $out/bin \
+      worktime
+  '';
+}
diff --git a/persistent-nix-shell b/persistent-nix-shell
new file mode 100755
index 0000000..a17f6de
--- /dev/null
+++ b/persistent-nix-shell
@@ -0,0 +1,20 @@
+#!@zsh@/bin/zsh
+
+set -e
+
+gcrootsDir=${PWD}/.nix-gc-roots
+
+if [[ ${#@} -ge 1 ]]; then
+  shellFile=${1}
+  shift
+else
+  shellFile=${PWD}/shell.nix
+fi
+
+set -x
+mkdir -p ${gcrootsDir}
+nix-instantiate ${shellFile} --indirect --add-root ${gcrootsDir}/shell.drv
+nix-store --indirect --add-root ${gcrootsDir}/shell.dep --realise $(nix-store --query --references ${gcrootsDir}/shell.drv)
+set +x
+
+exec nix-shell $(readlink ${gcrootsDir}/shell.drv) ${@}
diff --git a/worktime.py b/worktime.py
new file mode 100755
index 0000000..9c72d30
--- /dev/null
+++ b/worktime.py
@@ -0,0 +1,205 @@
+#!@python@/bin/python
+
+import requests
+from requests.auth import HTTPBasicAuth
+from datetime import *
+from xdg import (BaseDirectory)
+import configparser
+from uritools import uricompose
+
+from dateutil.easter import *
+from dateutil.tz import *
+from dateutil.parser import isoparse
+
+from enum import Enum
+
+from math import (copysign, ceil)
+
+class TogglAPISection(Enum):
+    TOGGL = '/api/v8'
+    REPORTS = '/reports/api/v2'
+
+class TogglAPI(object):
+    def __init__(self, api_token, workspace_id):
+        self._api_token = api_token
+        self._workspace_id = workspace_id
+
+    def _make_url(self, api=TogglAPISection.TOGGL, section=['time_entries', 'current'], params={}):
+        if api is TogglAPISection.REPORTS:
+          params.update({'user_agent': 'worktime', 'workspace_id': self._workspace_id})
+
+        api_path = api.value
+        section_path = '/'.join(section)
+        return uricompose(scheme='https', host='www.toggl.com', path=f"{api_path}/{section_path}", query=params)
+
+    def _query(self, url, method):
+
+        headers = {'content-type': 'application/json'}
+
+        if method == 'GET':
+            return requests.get(url, headers=headers, auth=HTTPBasicAuth(self._api_token, 'api_token'))
+        elif method == 'POST':
+            return requests.post(url, headers=headers, auth=HTTPBasicAuth(self._api_token, 'api_token'))
+        else:
+            raise ValueError(f"Undefined HTTP method “{method}”")
+
+    def get_billable_hours(self, start_date, end_date=datetime.now(timezone.utc)):
+        url = self._make_url(api = TogglAPISection.REPORTS, section = ['summary'], params={'since': start_date.astimezone(timezone.utc).isoformat(), 'until': end_date.astimezone(timezone.utc).isoformat()})
+        r = self._query(url = url, method='GET')
+        return timedelta(milliseconds=r.json()['total_billable'])
+
+    def get_running_clock(self, now=datetime.now(timezone.utc)):
+        url = self._make_url(api = TogglAPISection.TOGGL, section = ['time_entries', 'current'])
+        r = self._query(url = url, method='GET').json()
+
+        if not r or not r['data'] or not r['data']['billable']:
+            return None
+
+        start = isoparse(r['data']['start'])
+
+        return now - start if start <= now else None
+
+class Worktime(object):
+    time_to_work = timedelta()
+    time_worked = timedelta()
+    running_entry = None
+    now = datetime.now(tzlocal())
+    time_pulled_forward = timedelta()
+    is_workday = False
+    
+    def __init__(self, start_datetime=None, end_datetime=None, now=None):
+      if now:
+        self.now = now
+        
+      config = configparser.ConfigParser()
+      config_dir = BaseDirectory.load_first_config('worktime')
+      config.read(f"{config_dir}/worktime.ini")
+      api = TogglAPI(api_token=config['TOGGL']['ApiToken'], workspace_id=config['TOGGL']['Workspace'])
+      date_format = config.get('WORKTIME', 'DateFormat', fallback='%Y-%m-%d')
+
+      start_date = start_datetime or datetime.strptime(config['WORKTIME']['StartDate'], date_format).replace(tzinfo=tzlocal())
+      end_date = end_datetime or self.now
+
+      hours_per_week = float(config.get('WORKTIME', 'HoursPerWeek', fallback=40))
+      workdays = set([int(d.strip()) for d in config.get('WORKTIME', 'Workdays', fallback='1,2,3,4,5').split(',')])
+      hours_per_day = hours_per_week / len(workdays)
+
+      holidays = set()
+
+      for year in range(start_date.year, end_date.year + 1):
+          y_easter = datetime.combine(easter(year), time(), tzinfo=tzlocal())
+
+          # Legal holidays in munich, bavaria
+          holidays.add(datetime(year, 1, 1, tzinfo=tzlocal()).date())
+          holidays.add(datetime(year, 1, 6, tzinfo=tzlocal()).date())
+          holidays.add((y_easter+timedelta(days=-2)).date())
+          holidays.add((y_easter+timedelta(days=+1)).date())
+          holidays.add(datetime(year, 5, 1, tzinfo=tzlocal()).date())
+          holidays.add((y_easter+timedelta(days=+39)).date())
+          holidays.add((y_easter+timedelta(days=+50)).date())
+          holidays.add((y_easter+timedelta(days=+60)).date())
+          holidays.add(datetime(year, 8, 15, tzinfo=tzlocal()).date())
+          holidays.add(datetime(year, 10, 3, tzinfo=tzlocal()).date())
+          holidays.add(datetime(year, 11, 1, tzinfo=tzlocal()).date())
+          holidays.add(datetime(year, 12, 25, tzinfo=tzlocal()).date())
+          holidays.add(datetime(year, 12, 26, tzinfo=tzlocal()).date())
+
+      try:
+          with open(f"{config_dir}/excused", 'r') as excused:
+              for line in excused:
+                  stripped_line = line.strip()
+                  if stripped_line:
+                      holidays.add(datetime.strptime(stripped_line, date_format).replace(tzinfo=tzlocal()).date())
+      except IOError as e:
+          if e.errno != 2:
+              raise e
+
+      pull_forward = dict()
+
+      try:
+          with open(f"{config_dir}/pull-forward", 'r') as excused:
+              for line in excused:
+                  stripped_line = line.strip()
+                  if stripped_line:
+                      [hours, date] = stripped_line.split(' ')
+                      day = datetime.strptime(date, date_format).replace(tzinfo=tzlocal()).date()
+                      if day > end_date.date():
+                        pull_forward[day] = timedelta(hours = float(hours))
+      except IOError as e:
+          if e.errno != 2:
+              raise e
+      
+
+      days_to_work = set()
+
+      start_day = start_date.date()
+      end_day = end_date.date()
+      if pull_forward:
+          end_day = max(end_day, max(list(pull_forward)))
+          
+      for day in [start_day + timedelta(days = x) for x in range(0, (end_day - start_day).days + 1)]:
+          if day.isoweekday() in workdays and not day in holidays:
+              days_to_work.add(day)
+
+      self.is_workday = self.now.date() in days_to_work
+
+      self.time_to_work = timedelta(hours = len([day for day in days_to_work if day <= end_date.date()]) * hours_per_day)
+      for day in list(pull_forward):
+          days_forward = set([d for d in days_to_work if d >= end_date.date() and d < day and not d in pull_forward])
+          hours_per_day_forward = pull_forward[day] / len(days_forward) if len(days_forward) > 0 else timedelta()
+          days_forward.discard(end_date.date())
+          self.time_pulled_forward += pull_forward[day] - hours_per_day_forward * len(days_forward)
+      self.time_to_work += self.time_pulled_forward
+
+      self.time_worked = api.get_billable_hours(start_date, self.now)
+      self.running_entry = api.get_running_clock(self.now)
+
+      if self.running_entry:
+          self.time_worked += self.running_entry
+          
+def main():
+    worktime = Worktime()
+
+    def difference_string(difference):
+      total_minutes_difference = round(difference / timedelta(minutes = 1))
+      (hours_difference, minutes_difference) = divmod(abs(total_minutes_difference), 60)
+      sign = '' if total_minutes_difference >= 0 else '-'
+
+      difference_string = f"{sign}"
+      if hours_difference != 0:
+          difference_string += f"{hours_difference}h"
+      if hours_difference == 0 or minutes_difference != 0:
+          difference_string += f"{minutes_difference}m"
+
+      return difference_string
+     
+
+    difference = worktime.time_to_work - worktime.time_worked
+    total_minutes_difference = 5 * ceil(difference / timedelta(minutes = 5))
+
+    if worktime.running_entry and abs(difference) < timedelta(days = 1) and (total_minutes_difference > 0 or abs(worktime.running_entry) >= abs(difference)) :
+        clockout_time = worktime.now + difference
+        clockout_time += (5 - clockout_time.minute % 5) * timedelta(minutes = 1)
+        clockout_time = clockout_time.replace(second = 0, microsecond = 0)
+
+        if total_minutes_difference >= 0:
+          difference_string = difference_string(total_minutes_difference * timedelta(minutes = 1))
+          print("{difference_string}/{clockout_time}".format(difference_string = difference_string, clockout_time = clockout_time.strftime("%H:%M")))
+        else:
+          difference_string = difference_string(abs(total_minutes_difference) * timedelta(minutes = 1))
+          print("{clockout_time}/{difference_string}".format(difference_string = difference_string, clockout_time = clockout_time.strftime("%H:%M")))
+    else:
+        if worktime.running_entry:
+            difference_string = difference_string(abs(total_minutes_difference) * timedelta(minutes = 1))
+            indicator = '↘' if total_minutes_difference >= 0 else '↗' # '\u25b6'
+                
+            print(f"{indicator} {difference_string}")
+        else:
+            difference_string = difference_string(total_minutes_difference * timedelta(minutes = 1))
+            if worktime.is_workday:
+                print(difference_string)
+            else:
+                print(f"({difference_string})")
+
+if __name__ == "__main__":
+    sys.exit(main())