1 files changed, 101 insertions, 0 deletions
diff --git a/autoplanner.py b/autoplanner.py
new file mode 100755
index 0000000..0ce7886
--- /dev/null
+++ b/autoplanner.py
@@ -0,0 +1,101 @@
+#!/bin/python3
+import http.client
+import urllib.request
+import urllib.parse
+import urllib.error
+import json
+import dateutil.parser
+import re
+from uuid import uuid4
+from icalendar import Calendar, Event
+from shared import *
+import sys
+from datetime import datetime
+
+DAY = 24 * 60 * 60
+# don't calculate trip for events further than a week in the future
+FUTURE_MAX = DAY * 7
+
+cal = Calendar()
+KEY = ""
+CFG = {}
+
+def get_events():
+  c = Calendar.from_ical(sys.stdin.read())
+  return list(c.walk('vevent'))
+
+def read_file(filename):
+  f = open(filename, "r")
+  r = str(f.read())
+  f.close()
+  return r
+
+def fake_trip(date):
+  return read_file("./api_response.json")
+
+def real_trip(date):
+  headers = {
+    'Ocp-Apim-Subscription-Key': KEY,
+    'X-Request-Id': str(uuid4()),
+  }
+  param_dict = {
+    'searchForArrival': True,
+    'dateTime': date,
+  }
+  param_dict.update(CFG)
+  params = urllib.parse.urlencode(param_dict)
+  conn = http.client.HTTPSConnection('gateway.apiportal.ns.nl')
+  conn.request("GET", f"/rio-mlab-proxy-api/reisinfo/api/v3/trips?{params}", "", headers)
+  response = conn.getresponse()
+  data = response.read()
+  conn.close()
+  return data
+
+def get_trip(date):
+  return real_trip(date)
+
+def leg2desc(leg):
+  return "desc"
+
+def trip2ical(trip):
+  actual_trip = next(t for t in trip['trips'] if t['status'] != 'CANCELLED')
+  if actual_trip == None: return
+
+  for leg in actual_trip['legs']:
+    if leg['travelType'] == 'WALK': continue
+
+    ev = Event()
+    ev.add('summary', f"{leg['name']} -> {leg['direction']}")
+    ev.add('description', leg2desc(leg))
+    ev.add('dtstart', dateutil.parser.parse(leg['origin']['plannedDateTime']))
+    ev.add('dtend', dateutil.parser.parse(leg['destination']['plannedDateTime']))
+    cal.add_component(ev)
+
+def main():
+  cal.add('prodid', 'trein')
+  cal.add('version', '2.0')
+
+  global KEY, CFG
+  KEY = get_public_key()
+  CFG = json.loads(read_file("./autoplanner.json"))
+
+  now = datetime.now().timestamp()
+
+  # this is garbage code, but gets a list containing the date/time for each
+  # first event of every day after today
+  times = list(map(lambda x: x.decoded('dtstart').timestamp(), get_events())) # get event times (epoch)
+  times = [(x // DAY, x % DAY) for x in times] # split into days and seconds
+  times = [x for x in times if x[1] == min([y[1] for y in times if y[0] == x[0]])] # only leave smallest seconds in day
+  times = [x for x in times if now < (x[0] * DAY + x[1]) < (now + FUTURE_MAX)] # only leave after today and until FUTURE_MAX
+  times = [datetime.fromtimestamp(x[0] * DAY + x[1]) for x in times] # convert back to datetime
+
+  for time in times:
+    time_s = time.isoformat()
+    trip = get_trip(time_s)
+    trip2ical(json.loads(trip))
+
+  print(str(cal.to_ical(), 'utf-8'))
+
+if __name__ == "__main__":
+  main()
+