From 6c4d2b1eaadfe7083b465d9c43b85dbdc2ed70de Mon Sep 17 00:00:00 2001
From: Innovation <katieofgeeks@gmail.com>
Date: Fri, 29 Mar 2024 09:14:16 -0500
Subject: [PATCH] Begin API-side implementation of hardware tracking.

---
 src/nightserver.py | 80 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 78 insertions(+), 2 deletions(-)

diff --git a/src/nightserver.py b/src/nightserver.py
index 26e471a..302a113 100644
--- a/src/nightserver.py
+++ b/src/nightserver.py
@@ -1,4 +1,5 @@
 from flask import Flask, render_template, jsonify, request
+from copy import copy
 app = Flask(__name__)
 
 # Vitals
@@ -110,8 +111,83 @@ def setFitness():
 
 
 # Cyberware management
-#@app.route('/api/cyberware/add', methods=['POST'])
-#def addCyberware
+currentId = 1
+cyberware = []
+newCyberwareTemplate = { "id": -1, "name": "", "hotpluggable": False, "lastMalfunction": "", "frontendAwknowledge": False }
+
+
+# VERY descriptive variable names, huh?
+#cyberwareIdIndex = 0
+#cyberwareNameIndex = 1
+#cyberwareHotpluggableIndex = 2
+#cyberwareLastMalfunction = 3
+#cyberwareFrontwareAwknowledge = 4
+
+# This makes the system aware of a new piece of hardware.
+# While, for the most part, not required due to the design of this project,
+#  it's handy for error reporting and communication between hardware and
+#  the end user.
+#
+# Arguments: { name: STRING, hotpluggable: BOOL }
+# Returns:   { id: INT }
+# name: A human-readable name
+# hotpluggable argument: Used for removable modules known as shards.
+#
+# id: Returns an id of the hardware.
+@app.route('/api/cyberware/add', methods=['POST'])
+def addCyberware():
+	global cyberware
+	global currentId
+	json = request.get_json()
+
+	try:
+		tempName = json['name']
+		tempHotpluggable = json['hotpluggable']
+		tempNewCyberware = copy(newCyberwareTemplate)
+
+		tempNewCyberware['id'] = currentId
+		tempNewCyberware['name'] = tempName
+		tempNewCyberware['hotpluggable'] = tempHotpluggable
+
+		currentId = currentId + 1
+
+		cyberware.append(tempNewCyberware)
+	except:
+		return 'Incorrect usage.\nUsage: { name: STRING, hotpluggable: BOOL }\n', 400
+	return jsonify([ { "id": currentId-1 } ]), 204
+
+# Arguments: { id: INT }
+@app.route('/api/cyberware/remove', methods=['POST'])
+def removeCyberware():
+	global cyberware
+	json = request.get_json()
+
+	try:
+		desiredId = json['id']
+		i = 0
+		for c in cyberware:
+			if c['id'] == desiredId:
+				del cyberware[i]
+				return 'Cyberware removed', 204
+			i = i + 1
+	except:
+		return 'Incorrect usage.\nUsage: { id: INT }\n', 400
+	return 'ID Invalid\n', 400
+
+# Returns: { id: INT, name: STRING, hotpluggable: BOOL, lastMalfunction: STRING, frontendAwknowledge: BOOL }
+# id:                 Unique identifier of the hardware
+# name:               Human-readable name
+# hotpluggable:       Hardware can be removed during runtime.
+# lastMalfunction:    A string with information on the last malfunction.
+# frontendAwknowlege: A tool for the frontend to keep track of whether it's aware of the hardware.
+@app.route('/api/cyberware')
+def getCyberware():
+	return jsonify(cyberware)
+
+# Arguments { id: INT }
+@app.route('/api/cyberware/reset', methods=['POST'])
+def resetCyberwareMalfunction():
+	return 'Not Implemented', 501
 
 
 @app.route('/')