| 1 | #!/usr/bin/env python |
| 2 | import sys |
| 3 | import os |
| 4 | import signal |
| 5 | import pathlib |
| 6 | import contextvars |
| 7 | |
| 8 | from aiohttp import web |
| 9 | |
| 10 | |
| 11 | _ws = "_websockets" |
| 12 | |
| 13 | |
| 14 | async def websocket(request): |
| 15 | ws = web.WebSocketResponse() |
| 16 | await ws.prepare(request) |
| 17 | request.app[_ws].append(ws) |
| 18 | try: |
| 19 | async for msg in ws: |
| 20 | print("echo: ", msg.data) |
| 21 | ws.send_str(msg.data) |
| 22 | finally: |
| 23 | request.app[_ws].pop() |
| 24 | return ws |
| 25 | |
| 26 | |
| 27 | async def reloader(request): |
| 28 | for socket in request.app[_ws]: |
| 29 | await socket.send_str("reload") |
| 30 | return web.Response() |
| 31 | |
| 32 | |
| 33 | def load_middleware(static_path="gen"): |
| 34 | async def serve_indexes(request: web.Request, handler): |
| 35 | if request.path == "/ws": |
| 36 | return await handler(request) |
| 37 | if request.path == "/reload": |
| 38 | return await handler(request) |
| 39 | relative_file_path = pathlib.Path(request.path).relative_to('/') |
| 40 | file_path = static_path / relative_file_path |
| 41 | if not file_path.exists(): |
| 42 | return web.HTTPNotFound() |
| 43 | if file_path.is_dir(): |
| 44 | file_path /= "index.html" |
| 45 | if not file_path.exists(): |
| 46 | return web.HTTPNotFound() |
| 47 | return web.FileResponse(file_path) |
| 48 | return [web.middleware(serve_indexes)] |
| 49 | |
| 50 | |
| 51 | app = web.Application(middlewares=load_middleware(static_path="gen")) |
| 52 | app.add_routes([web.get("/ws", websocket)]) |
| 53 | app.add_routes([web.post("/reload", reloader)]) |
| 54 | app.add_routes([web.static("/", "gen", show_index=False)]) |
| 55 | app[_ws] = [] |
| 56 | |
| 57 | web.run_app(app, port=8888) |