WIP mux prompt and socket input

This commit is contained in:
lash 2024-10-28 00:45:07 +00:00
parent f803256ee9
commit fa64acac1d
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746

View File

@ -4,6 +4,8 @@ import enum
import uuid import uuid
import os import os
import logging import logging
import sys
import select
from multiprocessing import Process from multiprocessing import Process
from multiprocessing import Value from multiprocessing import Value
from multiprocessing import Pipe from multiprocessing import Pipe
@ -271,6 +273,7 @@ class Router:
def exec(self, cmd): def exec(self, cmd):
logg.debug("router exec {}".format(cmd))
if cmd.c & 0xa0 > 0: if cmd.c & 0xa0 > 0:
return self.sync(cmd.i) return self.sync(cmd.i)
fn = self.__routes[cmd.c] fn = self.__routes[cmd.c]
@ -280,10 +283,17 @@ class Router:
po.start() po.start()
def __del__(self): def finish(self):
print("syncing")
for k, v in self.__r.items(): for k, v in self.__r.items():
print("syncing key " + k)
r = self.sync(k) r = self.sync(k)
logging.debug("synced " + k + ": " + r) logg.debug("synced " + k + ": " + r)
print("synced " + k + ": " + r)
def __del__(self):
self.finish()
def noop_handler(cmd): def noop_handler(cmd):
@ -296,21 +306,76 @@ def foo_handler(cmd):
parser = yacc.yacc() parser = yacc.yacc()
cmds = [ running = True
"KEY_CREATE foo-bar Foo BAr",
"KEY_CREATE bar-baz Foo BAr 0xabcdef0",
"VOUCHER_MINT - 0xdeadbeef 13 0xfdaf", class FileGet:
"VOUCHER_MINT baz-baz 0xdeadbeef 13 0xfdaf", def __init__(self, o, fp):
"VOUCHER_MINT bar-bar 0xdeadbeef 42", self.__running = True
"VOUCHER_CREATE foo-foo Foo Bar 42", self.__o = o
"VOUCHER_CREATE inky-pinky-blinky Foo Bar 42 44 233", self.__f = open(fp, 'r')
"VOUCHER_TRANSFER clyde-sue 0xbeeffeed0123 666 foo 0x0fdc",
] def run(self):
for v in cmds: while self.__running:
v = ifc.get()
if v == None:
break
v = v.strip('\n')
if len(v) == 0:
break
r = parser.parse(v)
self.__o.exec(r)
def get(self):
return self.__f.readline()
class WaitGet:
def __init__(self, o, *r):
self.__running = True
self.__o = o
self.__f = r
# TODO: router copy results in missing keys to sync when closing down
def __process(self, f):
while self.__running:
r = select.select([f], [], [])
v = r[0][0].recv()
if v == None:
break
v = v.strip('\n')
if len(v) == 0:
break
r = parser.parse(v)
self.__o.exec(r)
def run(self):
(fo, fi,) = Pipe()
p = Process(target=self.__process, args=(fo,))
p.start()
while self.__running:
v = input("> ")
if v == "":
fi.send(None)
break
fi.send(v)
p.join()
logg.debug('waitget run end')
if __name__ == '__main__':
ifc = None
o = Router() o = Router()
o.register(CmdId.KEY_CREATE, noop_handler) o.register(CmdId.KEY_CREATE, noop_handler)
o.register(CmdId.VOUCHER_MINT, noop_handler) o.register(CmdId.VOUCHER_MINT, noop_handler)
o.register(CmdId.VOUCHER_CREATE, noop_handler) o.register(CmdId.VOUCHER_CREATE, noop_handler)
o.register(CmdId.VOUCHER_TRANSFER, foo_handler) o.register(CmdId.VOUCHER_TRANSFER, foo_handler)
r = parser.parse(v)
o.exec(r) if len(sys.argv) > 1:
ifc = FileGet(o, sys.argv[1])
else:
ifc = WaitGet(o, sys.stdin)
ifc.run()
o.finish()