From e79cdb3e24baacb504b50f5c5edae74d8a8de5f5 Mon Sep 17 00:00:00 2001
From: Oskar <o.winkels@fasttube.de>
Date: Mon, 30 Nov 2020 23:44:47 +0100
Subject: [PATCH] Add departure handling and consistency checks

---
 ftracker/core.py | 46 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 44 insertions(+), 2 deletions(-)

diff --git a/ftracker/core.py b/ftracker/core.py
index a89df6b..c777805 100644
--- a/ftracker/core.py
+++ b/ftracker/core.py
@@ -6,7 +6,7 @@ from .config import Config
 config = Config()
 
 
-from tinydb import TinyDB
+from tinydb import TinyDB, Query, operations
 dbfile = config['db_file'] or '/tmp/ftracker-db.json'
 db = TinyDB(dbfile, indent=4)
 
@@ -24,8 +24,9 @@ app = Flask(__name__)
 def get_root():
 	return "Error: No Endpoint selected. See docs/API.md for reference.", 404
 
+
 @app.route('/arrival', methods=['POST'])
-def post_time():
+def post_arrival():
 
 	try:
 		payload = request.data.decode('UTF-8')
@@ -44,6 +45,12 @@ def post_time():
 	if not name in namelist:
 		return "Error: Name not in permitted list.", 401
 
+	Entry = Query()
+	if db.contains((Entry.name == name) & (Entry.departure == None)):
+		# Did not depart last time
+		# TODO: Return structured request to resend departure
+		return "Error: Undeparted arrival exists", 406
+
 	now = datetime.utcnow()
 	db.insert({
 		'name': name,
@@ -52,3 +59,38 @@ def post_time():
 	})
 
 	return 'OK', 200
+
+
+@app.route('/departure', methods=['POST'])
+def post_departure():
+
+	try:
+		payload = request.data.decode('UTF-8')
+		data = json.loads(payload)
+	except ValueError as e:
+		return 'Error: JSON decode error:\n' + str(e), 400
+
+	if not ('name' in data and 'cleanedworkspace' in data):
+		return "Error: Key missing. See docs/API.md for reference.", 400
+
+	if not data['cleanedworkspace']:
+		return "Error: Didn't clean workspace.", 406
+
+	name = slugify(data['name'])
+
+	if not name in namelist:
+		return "Error: Name not in permitted list.", 401
+
+	Entry = Query()
+	if not db.contains((Entry.name == name) & (Entry.departure == None)):
+		# Did not arrive before
+		# TODO: Return structured request to resend arrival
+		return "Error: No arrival exists", 406
+
+	now = datetime.utcnow()
+	db.update(
+		operations.set('departure', now.isoformat()),
+		(Entry.name == name) & (Entry.departure == None)
+	)
+
+	return 'OK', 200