Merge with store

store/server.py

@@ -0,0 +1,105 @@
+from datetime import datetime
+
+from db import execute, get_first
+from flask import Flask, Response, request
+
+app = Flask(__name__)
+
+
+def getUserId():
+    try:
+        header = request.headers["Authorization"]
+        return header.split()[1]
+    except:
+        return None
+
+
+@app.route("/", methods=["POST"])
+def register():
+    try:
+        id = request.json["token"]
+        execute("insert into users(id, enabled) values(%s, true)", (id,))
+        return "", 201
+    except:
+        return "", 400
+
+
+@app.route("/me", methods=["POST"])
+def track():
+    try:
+        execute(
+            "insert into logins(id, info, time) values(%s, %s, %s)",
+            (getUserId(), request.user_agent.string, datetime.now()),
+        )
+        return "", 204
+    except:
+        return "", 400
+
+
+@app.route("/me/root", methods=["PUT"])
+def set_root():
+    try:
+        root_id = request.json["root_id"]
+        execute(
+            "insert into latest(id, userID, timestamp) values(%s, %s, %s)",
+            (root_id, getUserId(), datetime.now()),
+        )
+        return "", 201
+    except Exception as e:
+        return str(e), 400
+
+
+@app.route("/me/root", methods=["GET"])
+def get_root():
+    try:
+        print(f"get root")
+        result = get_first(
+            "select id from latest where userID = %s order by timestamp desc",
+            (getUserId(),),
+        )
+        if result is None:
+            return "", 404
+        return result[0], 200
+    except Exception as e:
+        return str(e), 400
+
+
+@app.route("/me/<string:id>", methods=["POST"])
+def create(id):
+    try:
+        payload = request.json["data"]
+        execute(
+            "insert into objects(id, userID, serialized) values(%s, %s, %s)",
+            (id, getUserId(), payload),
+        )
+        return "", 201
+    except:
+        return "", 400
+
+
+@app.route("/me/<string:id>", methods=["GET"])
+def read(id):
+    try:
+        result = get_first(
+            "select serialized from objects where userID = %s and id = %s",
+            (getUserId(), id),
+        )
+        if result is None:
+            return "", 404
+        return result[0], 200
+    except:
+        return "", 400
+
+
+@app.after_request
+def postprocess(response):
+    response.headers["Access-Control-Allow-Origin"] = "*"
+    response.headers["Access-Control-Allow-Methods"] = "GET,POST,PUT,DELETE,OPTIONS"
+    response.headers[
+        "Access-Control-Allow-Headers"
+    ] = "Content-Type, Access-Control-Allow-Headers, Authorization, X-Requested-With"
+    return response
+
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0")