From cc182059bd2a089b68b690eaa8065c47dc2428b0 Mon Sep 17 00:00:00 2001
From: Pedro <ptsmonteiro@gmail.com>
Date: Mon, 6 Nov 2023 13:49:37 +0100
Subject: [PATCH] Add rest API server

---
 modem/server.py      | 76 ++++++++++++++++++++++++++++++++++++++++++++
 tests/test_server.py | 35 ++++++++++++++++++++
 2 files changed, 111 insertions(+)
 create mode 100644 modem/server.py
 create mode 100644 tests/test_server.py

diff --git a/modem/server.py b/modem/server.py
new file mode 100644
index 00000000..6c376b91
--- /dev/null
+++ b/modem/server.py
@@ -0,0 +1,76 @@
+from flask import Flask, request, jsonify
+from flask_sock import Sock
+import argparse, json
+from config import CONFIG
+
+app = Flask(__name__)
+sock = Sock(app)
+
+# CLI arguments
+def get_config_filename_from_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-c", "--config", help = "Specifiy config file", default = 'config.ini') 
+    args, unknown = parser.parse_known_args()
+    return args.config
+
+config = CONFIG(get_config_filename_from_args())
+
+def api_response(data, status = 'ok'):
+    response = {
+        'status': status,
+        'data': data
+    }
+    return jsonify(response)
+
+## REST API
+@app.route('/', methods=['GET'])
+def index():
+    return jsonify({'name': 'FreeDATA API',
+                    'description': '',
+                    'api_version': 1,
+                    'license': 'GPL3.0',
+                    'documentation': 'https://wiki.freedata.app',
+                    })
+
+# get and set config
+@app.route('/config', methods=['GET', 'POST'])
+def config():
+    if request.method == 'POST':
+        new_config = json.loads(request.form['config'])
+        set_config = config.write(new_config)
+        if not set_config:
+            response = api_response(None, 'error writing config')
+        else:
+            response = api_response(set_config)
+        return response
+    elif request.method == 'GET':
+        return api_response(config.read())
+
+# get activity
+@app.route('/activity', methods=['GET'])
+def activity():
+    return "Not implemented yet."
+
+# get received messages
+@app.route('/messages', methods=['GET'])
+def messages():
+    return "Not implemented yet."
+
+# new message / delete message
+""" @app.route('/message', methods=['POST', 'DELETE'])
+def message():
+    if request.method == 'POST':
+      message = new Message(request.form['message'])
+      status = modem.send_message(message)
+    elif request.method == 'DELETE':
+      status = messageDb.deleteMessage(request.form['id'])
+    return status
+ """
+# Event websocket
+@sock.route('/events')
+def echo(sock):
+    while True:
+        data = sock.receive()
+        sock.send(data)
+
+
diff --git a/tests/test_server.py b/tests/test_server.py
new file mode 100644
index 00000000..62e22981
--- /dev/null
+++ b/tests/test_server.py
@@ -0,0 +1,35 @@
+import unittest
+from subprocess import Popen, PIPE
+import shlex
+import requests
+import time
+import json
+
+class TestConfigAPI(unittest.TestCase):
+
+    process = None
+    url = "http://127.0.0.1:5000"
+
+    @classmethod
+    def setUpClass(cls):
+        cmd = "flask --app modem/server run"
+        cls.process = Popen(shlex.split(cmd), stdin=PIPE)
+        time.sleep(1)
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.process.stdin.close()
+        cls.process.terminate()
+        if cls.process.wait() != 0:
+            print("There were some errors closing the process.")
+
+    def test_index(self):
+        r = requests.get(self.url)
+        self.assertEqual(r.status_code, 200)
+
+        data = r.json()
+        self.assertEqual(data['api_version'], 1)
+
+
+if __name__ == '__main__':
+    unittest.main()