Refactoring and windows support

.gitignore

@@ -4,3 +4,4 @@ logs
 *.egg-info
 build
 dist
+*~

aw_watcher_afk/__init__.py

@@ -1 +1,3 @@
 from .afk import main
+
+import __main__

aw_watcher_afk/__main__.py

@@ -0,0 +1,2 @@
+import aw_watcher_afk
+aw_watcher_afk.main()

aw_watcher_afk/afk.py

@@ -12,8 +12,25 @@
 from .listeners import KeyboardListener, MouseListener
 from .config import watcher_config
 
+if platform.system() == "Windows":
+    from .windows import time_since_last_input as _time_since_last_input_winfail
+
 settings = {}
 
+def _time_since_last_input_unix():
+    raise NotImplementedError
+
+
+def time_since_last_input():
+    system = platform.system()
+    if system == "Darwin" or system == "Linux":
+        return _time_since_last_input_unix()
+    elif system == "Windows":
+        return _time_since_last_input_winfail()
+    else:
+        raise Exception("unknown platform")
+
+
 def main() -> None:
     """ Set up argparse """
     parser = argparse.ArgumentParser("A watcher for keyboard and mouse input to detect AFK state")
@@ -44,6 +61,7 @@ def main() -> None:
     client.setup_bucket(bucketname, eventtype)
     client.connect()
 
+    # TODO: Get rid of desktop notifications, move to client instead
     """ Desktop Notifications """
     if args.desktop_notify:
         from gi.repository import Notify
@@ -73,7 +91,6 @@ def send_notification(msg):
     last_activity = now         # Last time of input activity
     second_last_activity = now  # Second last time of input activity
     last_update = now           # Last report time
-    last_check = now            # Last check/poll time
 
     def _report_state(afk, duration, timestamp, update=False):
         """
@@ -143,7 +160,7 @@ def set_state(new_afk, when):
 
             new_event = False
             if mouseListener.has_new_event() or keyboardListener.has_new_event():
-                # logger.debug("New event")
+                logger.debug("New event")
                 new_event = True
                 second_last_activity, last_activity = last_activity, now
                 # Get/clear events

aw_watcher_afk/listeners.py

@@ -21,7 +21,7 @@ class KeyboardListener(PyKeyboardEvent, EventFactory):
     def __init__(self):
         PyKeyboardEvent.__init__(self)
         self.logger = logging.getLogger("aw.watchers.afk.keyboard")
-        self.logger.setLevel(logging.INFO)
+        self.logger.setLevel(logging.DEBUG)
         self.new_event = threading.Event()
         self._reset_data()
 

aw_watcher_afk/windows.py

@@ -0,0 +1,41 @@
+import time
+
+import ctypes
+from ctypes import Structure, POINTER, WINFUNCTYPE, windll
+from ctypes.wintypes import BOOL, UINT, DWORD
+
+
+class LastInputInfo(Structure):
+    _fields_ = [
+        ("cbSize", UINT),
+        ("dwTime", DWORD)
+    ]
+
+
+def _getLastInputTick():
+    prototype = WINFUNCTYPE(BOOL, POINTER(LastInputInfo))
+    paramflags = (1, "lastinputinfo"),
+    c_GetLastInputInfo = prototype(("GetLastInputInfo", ctypes.windll.user32), paramflags)
+
+    l = LastInputInfo()
+    l.cbSize = ctypes.sizeof(LastInputInfo)
+    assert 0 != c_GetLastInputInfo(l)
+    return l.dwTime
+
+
+def _getTickCount() -> int:
+    prototype = WINFUNCTYPE(DWORD)
+    paramflags = ()
+    c_GetTickCount = prototype(("GetTickCount", ctypes.windll.kernel32), paramflags)
+    return c_GetTickCount()
+
+
+def time_since_last_input():
+    seconds_since_input = (_getTickCount() - _getLastInputTick())/1000
+    return seconds_since_input
+
+
+if __name__ == "__main__":
+    while True:
+        time.sleep(1)
+        print(time_since_last_input())