mirror of
https://git.ugnet.gay/CrossTalk/azul.git
synced 2026-05-27 22:59:49 +00:00
Athena Funderburg
122 lines
3.1 KiB
Python
122 lines
3.1 KiB
Python
from typing import Dict, Set, Tuple, Any, List, Optional
|
|
import asyncio, sys, io, cProfile, pstats, traceback, util.misc
|
|
|
|
from aiohttp import web
|
|
from aiohttp.web import WSMsgType
|
|
|
|
from core.backend import Backend
|
|
from core.http import render
|
|
|
|
def register(loop: asyncio.AbstractEventLoop, backend: Backend, http_app: web.Application) -> None:
|
|
util.misc.add_to_jinja_env(http_app, 'dev', 'dev/tmpl')
|
|
|
|
http_app['webconsole'] = Webconsole(loop, backend)
|
|
|
|
http_app.router.add_get('/dev', handle_index)
|
|
http_app.router.add_get('/dev/ws', handle_websocket)
|
|
|
|
async def handle_index(req: web.Request) -> web.Response:
|
|
return render(req, 'dev:index.html', { 'wsurl': 'ws://localhost/dev/ws' })
|
|
|
|
async def handle_websocket(req: web.Request) -> web.StreamResponse:
|
|
webconsole: Webconsole = req.app['webconsole']
|
|
|
|
ws = web.WebSocketResponse()
|
|
await ws.prepare(req)
|
|
|
|
webconsole.websocks.add(ws)
|
|
await ws.send_str("Webconsole on.")
|
|
|
|
async for msg in ws:
|
|
type = msg.type
|
|
data = msg.data
|
|
if type == WSMsgType.ERROR:
|
|
print("ws error with exception {}".format(ws.exception()))
|
|
continue
|
|
if type == WSMsgType.TEXT:
|
|
data = data.strip()
|
|
if data:
|
|
ret = webconsole.run(data)
|
|
if ret:
|
|
await ws.send_str(ret)
|
|
continue
|
|
print("ws unknown", type, data[:30])
|
|
|
|
webconsole.websocks.remove(ws)
|
|
|
|
return ws
|
|
|
|
class Webconsole:
|
|
__slots__ = ('loop', 'locals', 'objs', 'websocks', 'i')
|
|
|
|
loop: asyncio.AbstractEventLoop
|
|
locals: Dict[str, object]
|
|
objs: Dict[object, str]
|
|
websocks: Set[web.WebSocketResponse]
|
|
i: int
|
|
|
|
def __init__(self, loop: asyncio.AbstractEventLoop, backend: Backend) -> None:
|
|
self.loop = loop
|
|
self.locals = {
|
|
'_': None,
|
|
'be': backend,
|
|
'dir': useful_dir,
|
|
'dirfull': dir,
|
|
'prof': profile,
|
|
}
|
|
self.objs = {}
|
|
self.websocks = set()
|
|
self.i = 0
|
|
|
|
backend._dev = self
|
|
|
|
def run(self, cmd: str) -> str:
|
|
tmp = io.StringIO()
|
|
sys.stdout = tmp
|
|
try:
|
|
self.locals['_'] = exec(compile(cmd + '\n', '<stdin>', 'single'), None, self.locals) # type: ignore
|
|
except:
|
|
(exctype, excvalue, tb) = sys.exc_info() # type: Tuple[Any, Any, Any]
|
|
ret = '\n'.join(traceback.format_exception(exctype, excvalue, tb))
|
|
else:
|
|
ret = tmp.getvalue()
|
|
finally:
|
|
sys.stdout = sys.__stdout__
|
|
return ret
|
|
|
|
def connect(self, obj: object) -> None:
|
|
varname = 'k{}'.format(self.i)
|
|
self.i += 1
|
|
self.objs[obj] = varname
|
|
self.locals[varname] = obj
|
|
msg = "# Connect: `{}`".format(varname)
|
|
for ws in self.websocks:
|
|
self.loop.create_task(ws.send_str(msg))
|
|
|
|
def disconnect(self, obj: object) -> None:
|
|
varname = self.objs.pop(obj)
|
|
msg = "# Disconnect: `{}`".format(varname)
|
|
self.locals.pop(varname, None)
|
|
for ws in self.websocks:
|
|
self.loop.create_task(ws.send_str(msg))
|
|
|
|
_PROFILE: Optional[cProfile.Profile] = None
|
|
|
|
def profile(*restrictions: Any) -> None:
|
|
global _PROFILE
|
|
if _PROFILE is None:
|
|
_PROFILE = cProfile.Profile()
|
|
_PROFILE.enable()
|
|
print("Profiling ON")
|
|
return
|
|
_PROFILE.disable()
|
|
ps = pstats.Stats(_PROFILE).sort_stats('cumulative')
|
|
ps.print_stats(*restrictions)
|
|
_PROFILE = None
|
|
|
|
def useful_dir(*args: Any) -> List[str]:
|
|
return [
|
|
x for x in dir(*args)
|
|
if not x.endswith('__')
|
|
]
|