src: add user session detection implementation

Detect user login and logout for Linux and Windows.

Poll the session change in 5 second intervals in a thread. Use the
same event socket previously used by the old session detection
mechanism to notify a session change.

Report an active interactive session through the /refresh response
so a new ogserver instance can update the session status.

1 files changed, 30 insertions, 1 deletions

diff --git a/src/windows/ogOperations.py b/src/windows/ogOperations.py
index bc7e954..ef4e308 100644
--- a/src/windows/ogOperations.py
+++ b/src/windows/ogOperations.py
@@ -8,11 +8,13 @@
 
 import os
 import ctypes
+import pythoncom
 import subprocess
 from subprocess import CalledProcessError
 import multiprocessing as mp
 from multiprocessing import Process, freeze_support
 from src.log import OgError
+import wmi
 
 from PIL import Image, ImageDraw
 from pystray import Icon, Menu, MenuItem
@@ -64,6 +66,7 @@ def create_systray():
 class OgWindowsOperations:
 
     def __init__(self):
+        self.session = False
         freeze_support()
         mp.set_start_method('spawn')
         self.systray_p = Process(target=create_systray, daemon=True)
@@ -125,4 +128,30 @@ class OgWindowsOperations:
         raise OgError('Function not implemented')
 
     def refresh(self, ogRest):
-        return {"status": "WIN"}
+        if self.session:
+            session_value = 'WINS'
+        else:
+            session_value = 'WIN'
+        return {"status": session_value}
+
+    def check_interactive_session_change(self):
+        old_status = self.session
+        pythoncom.CoInitialize()
+        has_logged_user = False
+        try:
+            c = wmi.WMI()
+            sessions = c.Win32_LogonSession(LogonType=2)
+
+            for session in sessions:
+                if has_logged_user:
+                    break
+                for user in session.associators("Win32_LoggedOnUser"):
+                    has_logged_user = True
+                    break
+        finally:
+            pythoncom.CoUninitialize()
+        self.session = has_logged_user
+
+        if self.session != old_status:
+            return self.session
+        return None