pipeserver first implementation
commit
8f310cbddf
@ -0,0 +1,73 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<style type="text/css">
|
||||||
|
body {
|
||||||
|
background: #888;
|
||||||
|
}
|
||||||
|
body.connected {
|
||||||
|
background: white;
|
||||||
|
}
|
||||||
|
#shell {
|
||||||
|
white-space: pre-wrap;
|
||||||
|
height: 40em;
|
||||||
|
width: 80em;
|
||||||
|
overflow-y: auto;
|
||||||
|
font-family: monospace;
|
||||||
|
background: black;
|
||||||
|
color: green;
|
||||||
|
font-size: 14px;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>socket to me</h1>
|
||||||
|
<p>This page uses <a href="https://developer.mozilla.org/en-US/docs/Web/API/WebSockets_API/Writing_WebSocket_client_applications">WebSockets!</a>.</p>
|
||||||
|
<p><span id="connections">0</span> active connections</p>
|
||||||
|
|
||||||
|
<div id="shell"></div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
var ws_addr = 'ws://'+window.location.host+window.location.pathname,
|
||||||
|
sock = null,
|
||||||
|
shell = document.getElementById("shell"),
|
||||||
|
connections = document.getElementById("connections");
|
||||||
|
|
||||||
|
function connect () {
|
||||||
|
sock = new WebSocket(ws_addr);
|
||||||
|
sock.onopen = function (event) {
|
||||||
|
console.log("socket opened");
|
||||||
|
document.body.classList.add("connected");
|
||||||
|
// sock.send(JSON.stringify({
|
||||||
|
// src: "connect",
|
||||||
|
// }));
|
||||||
|
};
|
||||||
|
sock.onmessage = function (event) {
|
||||||
|
// console.log("message", event);
|
||||||
|
if (typeof(event.data) == "string") {
|
||||||
|
var msg = JSON.parse(event.data);
|
||||||
|
// console.log("message JSON", msg);
|
||||||
|
if (msg.src == "stdin" && msg.line) {
|
||||||
|
var line = document.createElement("div");
|
||||||
|
line.classList.add("line");
|
||||||
|
line.innerHTML = msg.line;
|
||||||
|
shell.appendChild(line);
|
||||||
|
} else if (msg.src == "connect") {
|
||||||
|
connections.innerHTML = msg.connections;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
sock.onclose = function (event) {
|
||||||
|
// console.log("socket closed");
|
||||||
|
connections.innerHTML = "?";
|
||||||
|
document.body.classList.remove("connected");
|
||||||
|
sock = null;
|
||||||
|
window.setTimeout(connect, 2500);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
connect();
|
||||||
|
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
@ -0,0 +1,91 @@
|
|||||||
|
import sys, os, asyncio, json, argparse
|
||||||
|
import aiohttp, aiohttp_jinja2, jinja2
|
||||||
|
from aiohttp import web
|
||||||
|
# from urllib.parse import urlparse, unquote as urlunquote, quote as urlquote
|
||||||
|
|
||||||
|
|
||||||
|
async def index (request):
|
||||||
|
ws = web.WebSocketResponse()
|
||||||
|
ws_ready = ws.can_prepare(request)
|
||||||
|
if not ws_ready.ok:
|
||||||
|
return aiohttp_jinja2.render_template(request.app['template'], request, {})
|
||||||
|
|
||||||
|
await ws.prepare(request)
|
||||||
|
# print('index.Websocket connection ready')
|
||||||
|
connections = request.app['websockets']
|
||||||
|
connections.append(ws)
|
||||||
|
|
||||||
|
await ws.send_str(json.dumps({'src': 'connect', 'connections': len(connections)}))
|
||||||
|
print ("{} active connections".format(len(connections)))
|
||||||
|
|
||||||
|
# incoming messages
|
||||||
|
async for msg in ws:
|
||||||
|
print ("index.got msg")
|
||||||
|
if msg.type == aiohttp.WSMsgType.TEXT:
|
||||||
|
print(msg.data)
|
||||||
|
if msg.data == 'close':
|
||||||
|
await ws.close()
|
||||||
|
# else:
|
||||||
|
# await ws.send_str(msg.data + '/answer')
|
||||||
|
|
||||||
|
# if the connection closes, the above loop simply finishes
|
||||||
|
# and we arrive here
|
||||||
|
connections.remove(ws)
|
||||||
|
print ("{} active connections".format(len(connections)))
|
||||||
|
return ws
|
||||||
|
|
||||||
|
# The following is a fusion of:
|
||||||
|
# https://stackoverflow.com/questions/31510190/aysncio-cannot-read-stdin-on-windows#36785819
|
||||||
|
# https://docs.aiohttp.org/en/stable/web_advanced.html#background-tasks
|
||||||
|
async def listen_to_stdin(app):
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
try:
|
||||||
|
while True:
|
||||||
|
line = await loop.run_in_executor(None, sys.stdin.readline)
|
||||||
|
for ws in app['websockets']:
|
||||||
|
await ws.send_str(json.dumps({'src': 'stdin', 'line': line.rstrip()}))
|
||||||
|
except asyncio.CancelledError:
|
||||||
|
pass
|
||||||
|
finally:
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def start_background_tasks(app):
|
||||||
|
app['stdin_listener'] = asyncio.create_task(listen_to_stdin(app))
|
||||||
|
|
||||||
|
async def cleanup_background_tasks(app):
|
||||||
|
app['stdin_listener'].cancel()
|
||||||
|
await app['stdin_listener']
|
||||||
|
|
||||||
|
def main ():
|
||||||
|
ap = argparse.ArgumentParser("make & serve")
|
||||||
|
ap.add_argument("--host", default="localhost")
|
||||||
|
ap.add_argument("--port", type=int, default=8080)
|
||||||
|
ap.add_argument("--static", nargs=2, default=None, action="append")
|
||||||
|
ap.add_argument("--template", default="pipeserver.html", help="Template for web page to serve")
|
||||||
|
args = ap.parse_args()
|
||||||
|
|
||||||
|
# On Windows, the default event loop is SelectorEventLoop which does not support run_in_executor. ProactorEventLoop should be used instead.
|
||||||
|
if sys.platform == 'win32':
|
||||||
|
loop = asyncio.ProactorEventLoop()
|
||||||
|
asyncio.set_event_loop(loop)
|
||||||
|
|
||||||
|
app = web.Application()
|
||||||
|
app['websockets'] = []
|
||||||
|
template_path, app['template'] = os.path.split(args.template)
|
||||||
|
jinja_env = aiohttp_jinja2.setup(
|
||||||
|
app, loader=jinja2.FileSystemLoader(template_path))
|
||||||
|
# jinja_env.filters['datetimeformat'] = format_datetime
|
||||||
|
|
||||||
|
# setup routes
|
||||||
|
app.add_routes([web.get('/', index)])
|
||||||
|
if args.static:
|
||||||
|
for name, path in args.static:
|
||||||
|
print ("Adding static route {0} -> {1}".format(name, path))
|
||||||
|
app.router.add_static(name, path)
|
||||||
|
|
||||||
|
app.on_startup.append(start_background_tasks)
|
||||||
|
app.on_cleanup.append(cleanup_background_tasks)
|
||||||
|
web.run_app(app, host=args.host, port=args.port)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
Loading…
Reference in New Issue