prototype the que script

1 files changed, 96 insertions, 0 deletions

diff --git a/que b/que
new file mode 100755
index 0000000..6e117b9
--- /dev/null
+++ b/que
@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+"""
+simple client for que.run
+"""
+
+import argparse
+import urllib.request as request
+import urllib.parse
+import time
+import subprocess
+import sys
+
+"""
+example desktop notifications in linux:
+
+terminal 1:
+$ que --poll bs/desktop --then "notify-send '\que' '\msg'"
+
+terminal 2:
+$ echo "hey there" | que bs/desktop -
+
+When terminal 2 succeeds, terminal 1 will print "hey there", then call the
+`notify-send` command, which displays a notification toast in linux with title
+`bs/desktop` and content `hey there`.
+
+On macOS you could use:
+
+    osascript -e 'display notification "\msg" with title "\que"'
+
+in place of notify-send."""
+
+# set to something ridiculously high so we don't run into timeouts while polling
+# or waiting for a message
+MAX_TIMEOUT = 100000000
+
+
+def main(argv=None):
+    cli = argparse.ArgumentParser(description=__doc__)
+    cli.add_argument(
+        "--host",
+        default="http://dev.simatime.com:3000",
+        help="where que-server is running",
+    )
+    cli.add_argument(
+        "--poll", default=False, action="store_true", help="stream data from the que"
+    )
+    cli.add_argument(
+        "--then",
+        help="when polling, run this shell command after each response, replacing '{}' in the command with the body of the response",
+    )
+    cli.add_argument(
+        "target", help="namespace and path of the que, like 'ns/path/subpath'"
+    )
+    cli.add_argument(
+        "infile",
+        nargs="?",
+        type=argparse.FileType("r"),
+        help="data to put on the que. Use '-' for stdin, otherwise should be a readable file",
+    )
+
+    if argv is None:
+        args = cli.parse_args()
+    else:
+        args = cli.parse_args(argv)
+
+    if args.infile:
+        # send input data
+        data = args.infile.read().encode("utf-8").strip()
+        with request.urlopen(
+            f"{args.host}/{args.target}", data=data, timeout=MAX_TIMEOUT
+        ) as req:
+            print(req.status)
+    else:
+        # no input data, do a read instead
+        params = urllib.parse.urlencode({"poll": args.poll})
+        url = f"{args.host}/{args.target}?{params}"
+        with request.urlopen(url) as req:
+            if args.poll:
+                while not time.sleep(1):
+                    msg = req.readline().decode("utf-8").strip()
+                    print(msg)
+                    if args.then:
+                        subprocess.run(
+                            args.then.replace("\msg", msg).replace("\que", args.target),
+                            shell=True,
+                        )
+            else:
+                print(req.read().decode("utf-8").strip())
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except KeyboardInterrupt:
+        print("Interrupted")
+        sys.exit(0)