From 674cf3d56349bed44380db2cb72529f797d7ee44 Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Mon, 22 Sep 2014 17:59:28 -0400
Subject: move some things around

---
 termcast_client/__init__.py | 97 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 termcast_client/__init__.py

(limited to 'termcast_client/__init__.py')

diff --git a/termcast_client/__init__.py b/termcast_client/__init__.py
new file mode 100644
index 0000000..64e6c2a
--- /dev/null
+++ b/termcast_client/__init__.py
@@ -0,0 +1,97 @@
+import argparse
+import json
+import os
+import pity
+import shutil
+import signal
+import socket
+import sys
+
+class Client(object):
+    def __init__(self, host, port, username, password):
+        self.host = host
+        self.port = port
+        self.username = username
+        self.password = password
+
+    def run(self, argv):
+        sock = socket.socket()
+        sock.connect((self.host, self.port))
+        sock.send(self._build_connection_string())
+        self.winch_set = False
+        pity.spawn(
+            argv,
+            lambda fd: self._master_read(fd, sock),
+            handle_window_size=True
+        )
+
+    def _master_read(self, fd, sock):
+        if not self.winch_set:
+            prev_handler = signal.getsignal(signal.SIGWINCH)
+            signal.signal(
+                signal.SIGWINCH,
+                lambda signum, frame: self._winch(
+                    sock,
+                    lambda: prev_handler(signum, frame)
+                )
+            )
+            self.winch_set = True
+
+        data = os.read(fd, 1024)
+        sock.send(data)
+        return data
+
+    def _winch(self, sock, prev_handler):
+        prev_handler()
+        # XXX a bit racy - should try to avoid splitting existing escape
+        # sequences
+        sock.send(self._build_winsize_metadata_string())
+
+    def _build_connection_string(self):
+        auth = (
+            b'hello ' +
+            self.username.encode('utf-8') +
+            b' ' +
+            self.password.encode('utf-8') +
+            b'\n'
+        )
+        metadata = self._build_connection_metadata_string()
+        return auth + metadata
+
+    def _build_connection_metadata_string(self):
+        # for now
+        return self._build_winsize_metadata_string()
+
+    def _build_winsize_metadata_string(self):
+        size = shutil.get_terminal_size()
+        return self._build_metadata_string({
+            "geometry": [ size.columns, size.lines ],
+        })
+
+    def _build_metadata_string(self, data):
+        return b'\033]499;' + json.dumps(data).encode('utf-8') + b'\007'
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--host', default="noway.ratry.ru")
+    parser.add_argument('--port', type=int, default=31337)
+    parser.add_argument('--username', default=os.getenv("USER"))
+    parser.add_argument('--password', default="asdf")
+    parser.add_argument(
+        'command',
+        nargs=argparse.REMAINDER,
+    )
+
+    args = parser.parse_args(sys.argv[1:])
+
+    client = Client(
+        host=args.host,
+        port=args.port,
+        username=args.username,
+        password=args.password,
+    )
+
+    command = args.command
+    if len(args.command) < 1:
+        command = os.getenv("SHELL", default="/bin/sh")
+    client.run(command)
-- 
cgit v1.2.3