109 lines
2.9 KiB
Python
Executable File
109 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import sys
|
|
import os
|
|
p = os.path.join
|
|
pUp = os.path.dirname
|
|
s = False
|
|
if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
|
|
s = os.path.realpath(sys.executable)
|
|
else:
|
|
s = os.path.realpath(__file__)
|
|
sp = pUp(s)
|
|
|
|
# script start
|
|
import toml
|
|
import subprocess
|
|
|
|
def eprint(*args, **kwargs): print(*args, file=sys.stderr, **kwargs)
|
|
|
|
def addressStr(addr):
|
|
return addr[0] + ":" + str(addr[1])
|
|
|
|
def addressSslStr(addr):
|
|
return addr[0] + ":" + str(addr[1] + 1)
|
|
|
|
def perr(rtn,cmd = None,op = None):
|
|
if rtn == 0: return
|
|
if cmd == None: cmd = []
|
|
exc = subprocess.CalledProcessError(rtn,cmd,op)
|
|
raise exc
|
|
|
|
def pcallStr(*args,**kwargs):
|
|
proc = subprocess.Popen(*args,**kwargs, stdout=subprocess.PIPE)
|
|
response = proc.stdout.read().decode("utf-8").strip("\n")
|
|
rtn = proc.wait()
|
|
perr(rtn,args[0])
|
|
return response
|
|
|
|
def main():
|
|
print("* Parsing config ...")
|
|
configFiles = ["watch.toml"]
|
|
if len(sys.argv) > 1:
|
|
configFiles = sys.argv[1:]
|
|
|
|
length = len(configFiles)
|
|
index = 0
|
|
while index < length:
|
|
configFiles[index] = p(sp,"config",configFiles[index])
|
|
index += 1
|
|
|
|
config = toml.load(configFiles)
|
|
for key in config["args"]:
|
|
os.environ["fstream_arg_" + key] = str(config["args"][key])
|
|
if not "values" in config: config["values"] = {}
|
|
if not "python" in config["values"]: config["values"]["python"] = sys.executable
|
|
|
|
for decoder in config["decoder"]:
|
|
for i in range(len(decoder)):
|
|
for key in config["values"]:
|
|
decoder[i] = decoder[i].replace("$val:" +key+ "$",str(config["values"][key]))
|
|
|
|
for key in os.environ:
|
|
decoder[i] = decoder[i].replace("$env:" +key+ "$",os.environ[key])
|
|
|
|
aes = False
|
|
if "aespass" in config:
|
|
if not "bufsize" in config["args"]:
|
|
print("aespass is set, but args.bufsize is not.")
|
|
sys.exit(1)
|
|
os.environ["fstream_aespass"] = config["aespass"]
|
|
os.environ["fstream_aesbuffer"] = str(config["args"]["bufsize"] - 16)
|
|
aes = True
|
|
|
|
address = config["address"].rsplit(":",1)
|
|
address[1] = int(address[1])
|
|
|
|
cmds = []
|
|
cmds.append([sys.executable,p(sp,"fstream.py"),addressStr(address),"watch"])
|
|
if aes == True: cmds.append([sys.executable,p(sp,"util","fstream-util-aes_to_pipe.py")])
|
|
for decoder in config["decoder"]: cmds.append(decoder)
|
|
|
|
procs = []
|
|
procs.append(subprocess.Popen(cmds[0],stdout=subprocess.PIPE))
|
|
for i in range(1,len(cmds) - 1):
|
|
procs.append(subprocess.Popen(cmds[i],stdin=procs[i - 1].stdout,stdout=subprocess.PIPE))
|
|
procs.append(subprocess.Popen(cmds[-1],stdin=procs[-1].stdout))
|
|
|
|
# Pwetty pwease add thew funcshun owo uwu xoxoxox (<S-F10>)
|
|
import time
|
|
running = True
|
|
while running:
|
|
for proc in procs:
|
|
if proc.poll() is not None:
|
|
eprint("Process died: " + str(proc.args))
|
|
running = False
|
|
break
|
|
time.sleep(1)
|
|
|
|
eprint("Terminating process chain (" + str(len(procs)) + ")...")
|
|
|
|
for proc in procs:
|
|
eprint("Closing: " + str(proc.args))
|
|
try:
|
|
proc.terminate()
|
|
proc.wait()
|
|
except Exception as ex:
|
|
eprint("Error waiting for process!!! " + str(ex))
|
|
|
|
main()
|