First commit

server/backend/main.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from flask import Flask, render_template, send_from_directory, jsonify, redirect
+from app.decorators import auth
+from app.blueprints.ioc import ioc_bp
+from app.blueprints.whitelist import whitelist_bp
+from app.blueprints.config import config_bp
+import datetime
+import secrets
+import jwt
+from OpenSSL import SSL
+from app.utils import read_config
+from sys import path
+
+app = Flask(__name__, template_folder="../../app/backend/dist")
+app.config["SECRET_KEY"] = secrets.token_bytes(32)
+
+
+@app.route("/", methods=["GET"])
+@auth.login_required
+def main():
+    """
+        Return the index.html generated by Vue
+    """
+    return render_template("index.html")
+
+
+@app.route("/api/get-token", methods=["GET"])
+@auth.login_required
+def get_token():
+    """
+        Return the JWT token for API requests.
+    """
+    token = jwt.encode({"exp": datetime.datetime.now() +
+                        datetime.timedelta(hours=24)}, app.config["SECRET_KEY"])
+    return jsonify({"token": token.decode("utf8")})
+
+
+@app.route("/<p>/<path:path>", methods=["GET"])
+@auth.login_required
+def get_file(p, path):
+    """
+        Return the backend assets (css, js files, fonts etc.)
+    """
+    rp = "../../app/backend/dist/{}".format(p)
+    return send_from_directory(rp, path) if p in ["css", "fonts", "js", "img"] else redirect("/")
+
+
+@app.errorhandler(404)
+def page_not_found(e):
+    return redirect("/")
+
+
+# API Blueprints.
+app.register_blueprint(ioc_bp, url_prefix='/api/ioc')
+app.register_blueprint(whitelist_bp, url_prefix='/api/whitelist')
+app.register_blueprint(config_bp, url_prefix='/api/config')
+
+if __name__ == '__main__':
+    ssl_cert = "{}/{}".format(path[0], 'cert.pem')
+    ssl_key = "{}/{}".format(path[0], 'key.pem')
+
+    if read_config(("backend", "remote_access")):
+        app.run(host="0.0.0.0", debug=True, port=443,
+                ssl_context=(ssl_cert, ssl_key))
+    else:
+        app.run(port=443, debug=True, ssl_context=(ssl_cert, ssl_key))