1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#
# Copyright (C) 2021 Soleta Networks <info@soleta.eu>
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the
# Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
import os
import ctypes
import subprocess
from subprocess import CalledProcessError
import multiprocessing as mp
from multiprocessing import Process, freeze_support
from PIL import Image, ImageDraw
from pystray import Icon, Menu, MenuItem
from src.ogRest import ThreadState
EWX_POWEROFF=0x00000008
EWX_REBOOT=0x00000002
def _create_default_image():
"""
Creates a default image for the tray icon. Use in case
no favicon.ico is found.
"""
width = height = 250
color1 = (255, 255, 255)
color2 = (255, 0, 255)
image = Image.new('RGB', (width, height), color1)
dc = ImageDraw.Draw(image)
dc.rectangle(
(width // 2, 0, width, height // 2),
fill=color2)
dc.rectangle(
(0, height // 2, width // 2, height),
fill=color2)
return image
def create_image():
try:
image = Image.open(r'./favicon.ico')
image = Image.composite(image, Image.new('RGB', image.size, 'white'), image)
except:
image = _create_default_image()
return image
def create_systray():
menu = Menu(MenuItem('Powered by Soleta Networks!',
lambda icon, item: 1))
icon = Icon('ogClient', create_image(), menu=menu)
assert icon.icon
icon.run()
systray_p = Process(target=create_systray)
class OgWindowsOperations:
def __init__(self):
freeze_support()
mp.set_start_method('spawn')
systray_p.start()
def _restartBrowser(self, url):
raise NotImplementedError
def poweroff(self):
user32 = ctypes.WinDLL('user32')
systray_p.terminate()
user32.ExitWindowsEx(EWX_POWEROFF, 0x0)
def reboot(self):
user32 = ctypes.WinDLL('user32')
systray_p.terminate()
user32.ExitWindowsEx(EWX_REBOOT, 0x0)
def shellrun(self, request, ogRest):
cmd = request.getrun()
try:
result = subprocess.run(cmd,
shell=True,
stdin=subprocess.DEVNULL,
capture_output=True,
text=True,
check=True)
except CalledProcessError as error:
if error.stderr:
return error.stderr
if error.stdout:
return error.stdout
return "{Non zero exit code and empty output}"
return result.stdout
def session(self, request, ogRest):
raise NotImplementedError
def hardware(self, path, ogRest):
raise NotImplementedError
def setup(self, request, ogRest):
raise NotImplementedError
def image_restore(self, request, ogRest):
raise NotImplementedError
def image_create(self, path, request, ogRest):
raise NotImplementedError
def refresh(self, ogRest):
return {"status": "WIN"}
def probe(self, ogRest):
return {'status': 'WIN' if ogRest.state != ThreadState.BUSY else 'BSY'}
|
