Refactor AWS Python Client

Unofficial/Python/remote-control-client/aws-based-rc/client/.gitignore

@@ -0,0 +1 @@
+__pycache__/*

Unofficial/Python/remote-control-client/aws-based-rc/client/client.py

@@ -0,0 +1,76 @@
+# Written for Python 3.6.x
+
+import json
+import os
+import random
+import string
+import subprocess
+import time
+
+from urllib import request
+from urllib.error import URLError
+
+from winapi import WinApi
+
+
+class Client:
+
+    def __init__(self, api_base, api_key):
+        self.api_base = api_base
+        self.api_key = api_key
+
+        self.client_id = self.generate_id()
+        self.current_path = os.path.dirname(os.path.realpath(__file__))
+
+        opener = request.build_opener()
+        opener.addheaders = [('User-Agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; x64; rv:60.0) Gecko/20100101 Firefox/60.0')]
+        request.install_opener(opener)
+
+    def generate_id(self):
+        client_id = ''.join(random.choices(string.ascii_letters + string.digits, k=random.randint(12, 14)))
+
+        WinApi.show_message('Client ID', client_id, WinApi.BoxType.MB_OK, WinApi.IconType.INFO)
+
+        return client_id
+
+    def get_response(self, client_id):
+        try:
+            with request.urlopen(f'{self.api_base}{client_id}?key={self.api_key}') as response:
+                return json.loads(response.read())
+        except URLError as e:
+            print(f'Failed to open url: {e}')
+            return None
+
+    def loop(self):
+        while True:
+            response = self.get_response(self.client_id)
+
+            if response is None:
+                continue
+
+            for item in response:
+                # Check if the server echos back the client id
+                if item.get('clientID') == self.client_id:
+                    action = item.get('action')
+
+                    if action and item.get('data'):
+                        data = json.loads(item['data'])
+
+                        if action == 'messagebox':
+                            WinApi.show_message(
+                                data.get('title') or '',
+                                data.get('message') or '',
+                                WinApi.box_type(data.get('type') or ''),
+                                data.get('icon') and WinApi.IconType[data['icon'].upper()] or None
+                            )
+
+                        elif action == 'showimage':
+                            try:
+                                request.urlretrieve(data.get('url') or '', 'image.tmp')
+                            except URLError as e:
+                                print('Failed to retrieve image: {e}')
+                                continue
+
+                            subprocess.call(f'rundll32 "C:\Program Files\Windows Photo Viewer\PhotoViewer.dll", ImageView_Fullscreen {self.current_path}\\image.tmp')
+
+            time.sleep(20)

Unofficial/Python/remote-control-client/aws-based-rc/client/run.py

@@ -0,0 +1,26 @@
+# A file for you to run the client with ease.
+from optparse import OptionParser
+import os
+import sys
+
+from client import Client
+
+parser = OptionParser()
+parser.add_option('-k', '--key', dest='api_key', help='The Api Key', metavar='api_key', default=None)
+parser.add_option('-b', '--base', dest='api_base', help='Base url of the api', metavar='base_url', default=None)
+
+(options, args) = parser.parse_args()
+
+API_KEY = os.environ.get('AWS_CONTROL_API_KEY') or options.api_key
+API_BASE = os.environ.get('AWS_CONTROL_API_BASE') or options.api_base
+
+if not API_KEY or not API_BASE:
+    print('Invalid API_KEY/API_BASE set, please use `file.py -k <KEY> -b <BASE>` or set the `AWS_CONTROL_API_KEY/AWS_CONTROL_API_BASE` env vars!')
+    sys.exit(0)
+
+client = Client(API_BASE, API_KEY)
+
+# loop blocks until it's exited
+client.loop()
+
+print('Exiting program...')

Unofficial/Python/remote-control-client/aws-based-rc/client/winapi.py

@@ -0,0 +1,30 @@
+from enum import Enum
+import ctypes
+
+
+class WinApi:
+    class BoxType(Enum):
+        MB_OK = 0x0
+        MB_OKCXL = 0x01
+        MB_YESNOCXL = 0x03
+        MB_YESNO = 0x04
+        MB_HELP = 0x4000
+
+    class IconType(Enum):
+        EXCLAIM = 0x30
+        INFO = 0x40
+        QUESTION = 0x20
+        STOP = 0x10
+
+    READABLE_MAP = {
+        'okonly':       BoxType.MB_OK,
+        'okcancel':     BoxType.MB_OKCXL,
+        'yesnocancel':  BoxType.MB_YESNOCXL,
+        'yesno':        BoxType.MB_YESNO
+    }
+
+    def box_type(string):
+        return WinApi.READABLE_MAP.get(string) or WinApi.BoxType.MB_OK
+
+    def show_message(title, content, box_type=BoxType.MB_OK, icon=None):
+        return icon and ctypes.windll.user32.MessageBoxW(0, content, box_type.value | icon.value) or ctypes.windll.user32.MessageBoxW(0, content, title, box_type.value)