1 files changed, 80 insertions, 0 deletions
diff --git a/rwa/support/sessionservice/trigger.py b/rwa/support/sessionservice/trigger.py
new file mode 100644
index 0000000..fa9734a
--- /dev/null
+++ b/rwa/support/sessionservice/trigger.py
@@ -0,0 +1,80 @@
+# This file is part of Remote Support Desktop
+# https://gitlab.das-netzwerkteam.de/RemoteWebApp/rwa.support.sessionservice
+# Copyright 2020, 2021 Jonathan Weth <dev@jonathanweth.de>
+# Copyright 2020 Daniel Teichmann <daniel.teichmann@das-netzwerkteam.de>
+# Copyright 2020 Mike Gabriel <mike.gabriel@das-netzwerkteam.de>
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the
+# Free Software Foundation, Inc.,
+# 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import threading
+from typing import Any, Callable, Optional, Union
+from wsgiref.simple_server import make_server
+
+import port_for
+from flask import Flask, abort, request, jsonify
+
+
+class TriggerServerThread(threading.Thread):
+    """Simple Flask server (wrapped as thread) for triggering actions on sessions."""
+
+    def __init__(self, trigger_method: Callable[[int, dict, Optional[str]], Union[dict, bool]]):
+        super().__init__()
+        self.port = port_for.select_random()
+
+        app = Flask(__name__)
+
+        @app.route("/", methods=["POST"])
+        def trigger():
+            json = request.json
+            token = json.get("token", "")
+            try:
+                session_id = int(json.get("session_id"))
+                r = trigger_method(session_id, {"token": token}, "trigger")
+                if r:
+                    return "Successful triggered"
+                else:
+                    return abort(403)
+            except (ValueError, TypeError):
+                return abort(404)
+
+        @app.route("/authenticate/", methods=["POST"])
+        def authenticate():
+            json = request.json
+            try:
+                session_id = int(json.get("session_id"))
+                pin = int(json.get("pin", ""))
+                r = trigger_method(session_id, {"pin": pin}, "authenticate")
+                if r:
+                    return jsonify(r)
+                else:
+                    return abort(403)
+            except (ValueError, TypeError):
+                return abort(404)
+
+
+        self.srv = make_server("0.0.0.0", self.port, app)
+        self.ctx = app.app_context()
+        self.ctx.push()
+
+    def run(self):
+        self.srv.serve_forever()
+
+    def shutdown(self):
+        self.srv.shutdown()