Add trigger option and API methods for updating and getting session status

1 files changed, 55 insertions, 0 deletions

diff --git a/session.py b/session.py
index 3c8eeb9..f7ca9fe 100644
--- a/session.py
+++ b/session.py
@@ -1,10 +1,13 @@
 import os
 import secrets
 import signal
+from multiprocessing import Process
 
 import psutil
 import requests
 
+import port_for
+from flask import Flask, abort, request
 from vnc import run_vnc, save_password
 
 API_SERVER = "http://127.0.0.1:8000"
@@ -12,9 +15,16 @@ REGISTER_URL = API_SERVER + "/app/rwa/api/register/"
 
 
 class Session:
+    #: Session is running
+    STATUS_RUNNING = "running"
+
+    #: Remote has joined the session
+    STATUS_JOINED = "joined"
+
     def __init__(self):
         self._generate_password()
         self._start_vnc()
+        self._start_trigger_service()
         self._register_session()
 
     @property
@@ -47,6 +57,8 @@ class Session:
                 "port": self.ws_port,
                 "password": self.password,
                 "pid": self.vnc_pid,
+                "trigger_port": self.trigger_port,
+                "trigger_token": self.trigger_token,
             },
         )
         print(r)
@@ -56,6 +68,39 @@ class Session:
         self.api_token = self.meta["token"]
         self.pin = self.meta["pin"]
 
+    def _start_trigger_service(self):
+        self.trigger_port = port_for.select_random()
+        self.trigger_token = secrets.token_urlsafe(20)
+
+        app = Flask(__name__)
+
+        @app.route("/", methods=["POST"])
+        def trigger():
+            json = request.json
+            if json.get("token", "") == self.trigger_token:
+                self._trigger()
+                return "Successful triggered"
+            else:
+                return abort(403)
+
+        self.trigger_thread = Process(
+            target=lambda: app.run("0.0.0.0", port=self.trigger_port)
+        )
+        self.trigger_thread.start()
+
+    def _trigger(self):
+        """Event triggered by Django."""
+        print("Triggered")
+        self.pull()
+
+    def pull(self):
+        """Update status: Get status from Django."""
+        pass
+
+    def push(self):
+        """Update status: Push status to Django."""
+        pass
+
     def stop(self):
         """Stop session and clean up."""
         # Kill VNC
@@ -73,6 +118,12 @@ class Session:
             print("Delete password file")
             os.remove(self.pw_filename)
 
+        if hasattr(self, "trigger_thread"):
+            print("Kill trigger service.")
+            self.trigger_thread.terminate()
+
+        self.push()
+
         # Delete self
         del self
 
@@ -89,3 +140,7 @@ class Session:
     @property
     def client_meta(self):
         return {"id": self.pid, "url": self.web_url, "pin": self.pin}
+
+    @property
+    def status(self):
+        return {"id": self.pid, "status": self.STATUS_RUNNING}