Cleanup for stow ---STOW

[dotfiles/.git] / .local / lib / python2.7 / site-packages / pynvim / msgpack_rpc / msgpack_stream.py

diff --git a/.local/lib/python2.7/site-packages/pynvim/msgpack_rpc/msgpack_stream.py b/.local/lib/python2.7/site-packages/pynvim/msgpack_rpc/msgpack_stream.py
new file mode 100644 (file)
index 0000000..1a192f2
--- /dev/null
+++ b/.local/lib/python2.7/site-packages/pynvim/msgpack_rpc/msgpack_stream.py
@@ -0,0 +1,68 @@
+"""Msgpack handling in the event loop pipeline."""
+import logging
+
+from msgpack import Packer, Unpacker
+
+from pynvim.compat import unicode_errors_default
+
+logger = logging.getLogger(__name__)
+debug, info, warn = (logger.debug, logger.info, logger.warning,)
+
+
+class MsgpackStream(object):
+
+    """Two-way msgpack stream that wraps a event loop byte stream.
+
+    This wraps the event loop interface for reading/writing bytes and
+    exposes an interface for reading/writing msgpack documents.
+    """
+
+    def __init__(self, event_loop):
+        """Wrap `event_loop` on a msgpack-aware interface."""
+        self.loop = event_loop
+        self._packer = Packer(unicode_errors=unicode_errors_default)
+        self._unpacker = Unpacker(unicode_errors=unicode_errors_default)
+        self._message_cb = None
+
+    def threadsafe_call(self, fn):
+        """Wrapper around `BaseEventLoop.threadsafe_call`."""
+        self.loop.threadsafe_call(fn)
+
+    def send(self, msg):
+        """Queue `msg` for sending to Nvim."""
+        pass # replaces next logging statement
+        #debug('sent %s', msg)
+        self.loop.send(self._packer.pack(msg))
+
+    def run(self, message_cb):
+        """Run the event loop to receive messages from Nvim.
+
+        While the event loop is running, `message_cb` will be called whenever
+        a message has been successfully parsed from the input stream.
+        """
+        self._message_cb = message_cb
+        self.loop.run(self._on_data)
+        self._message_cb = None
+
+    def stop(self):
+        """Stop the event loop."""
+        self.loop.stop()
+
+    def close(self):
+        """Close the event loop."""
+        self.loop.close()
+
+    def _on_data(self, data):
+        self._unpacker.feed(data)
+        while True:
+            try:
+                pass # replaces next logging statement
+                #debug('waiting for message...')
+                msg = next(self._unpacker)
+                pass # replaces next logging statement
+                #debug('received message: %s', msg)
+                self._message_cb(msg)
+            except StopIteration:
+                pass # replaces next logging statement
+                #debug('unpacker needs more data...')
+                break