From 5f124806d86808eeb5a9f0663589690a0fa4ec36 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Damir=20Jeli=C4=87?= <poljar@termina.org.uk>
Date: Mon, 22 Apr 2019 21:28:15 +0200
Subject: [PATCH] pantalaimon: Add a Cli dbus based client.

---
 pantalaimon/panctl.py | 197 ++++++++++++++++++++++++++++++++++++++++++
 setup.py              |   6 +-
 2 files changed, 201 insertions(+), 2 deletions(-)
 create mode 100644 pantalaimon/panctl.py

diff --git a/pantalaimon/panctl.py b/pantalaimon/panctl.py
new file mode 100644
index 0000000..9a075f7
--- /dev/null
+++ b/pantalaimon/panctl.py
@@ -0,0 +1,197 @@
+"""Cli utility to control pantalaimon."""
+
+import attr
+import asyncio
+import sys
+
+from typing import List
+
+from prompt_toolkit import PromptSession
+from prompt_toolkit.eventloop.defaults import use_asyncio_event_loop
+from prompt_toolkit.patch_stdout import patch_stdout
+from prompt_toolkit.completion import Completer, Completion
+
+import dbus
+from gi.repository import GLib
+from dbus.mainloop.glib import DBusGMainLoop
+from janus import Queue
+from queue import Empty
+
+
+use_asyncio_event_loop()
+
+
+@attr.s
+class PanCompleter(Completer):
+    """Completer for panctl commands."""
+    commands = attr.ib(type=List[str])
+
+    def complete_commands(self, last_word):
+        """Complete the available commands."""
+        compl_words = []
+
+        for command in self.commands:
+            if last_word in command:
+                compl_words.append(command)
+
+        for compl_word in compl_words:
+            yield Completion(compl_word, -len(last_word))
+
+    def get_completions(self, document, complete_event):
+        """Build the completions."""
+        text_before_cursor = document.text_before_cursor
+        text_before_cursor = str(text_before_cursor)
+        words = text_before_cursor.split(" ")
+
+        last_word = words[-1]
+
+        if len(words) == 1:
+            return self.complete_commands(last_word)
+
+        return ""
+
+
+@attr.s
+class DbusT:
+    loop = attr.ib()
+    receive_queue = attr.ib()
+    send_queue = attr.ib()
+
+    async def send(self, message):
+        self.receive_queue.async_q.put(message)
+        GLib.idle_add(self._message_callback)
+
+    def quit(self):
+        self.loop.quit()
+
+    def _message_callback(self):
+        print("Message")
+        try:
+            message = self.receive_queue.get_nowait()
+        except Empty:
+            return True
+
+        return True
+
+    def dbus_loop(self):
+        DBusGMainLoop(set_as_default=True)
+        # TODO register to signals here
+        # bus = dbus.SessionBus()
+        self.loop.run()
+
+
+@attr.s
+class PanCtl:
+    bus_thread = attr.ib()
+    bus = attr.ib(init=False)
+    ctl = attr.ib(init=False)
+
+    commands = ["list-users", "export-keys", "import-keys"]
+
+    def __attrs_post_init__(self):
+        self.bus = dbus.SessionBus()
+        self.ctl = self.bus.get_object(
+            "org.pantalaimon",
+            "/org/pantalaimon/Control",
+            introspect=True
+        )
+
+    def list_users(self):
+        """List the daemons users."""
+        users = self.ctl.list(
+            dbus_interface="org.pantalaimon.control.list_users"
+        )
+        print("pantalaimon users:")
+        for user, devic in users:
+            print(" ", user, devic)
+
+    def import_keys(self, args):
+        try:
+            user, filepath, passphrase = args
+        except ValueError:
+            print("Invalid arguments for command")
+            return
+
+        self.ctl.import_keys(
+            user,
+            filepath,
+            passphrase,
+            dbus_interface="org.pantalaimon.control.import_keys"
+        )
+
+
+    def export_keys(self, args):
+        try:
+            user, filepath, passphrase = args
+        except ValueError:
+            print("Invalid arguments for command")
+            return
+
+        self.ctl.export_keys(
+            user,
+            filepath,
+            passphrase,
+            dbus_interface="org.pantalaimon.control.export_keys"
+        )
+
+    async def loop(self):
+        """Event loop for panctl."""
+        completer = PanCompleter(self.commands)
+        promptsession = PromptSession("panctl> ", completer=completer)
+
+        while True:
+            with patch_stdout():
+                try:
+                    result = await promptsession.prompt(async_=True)
+                except EOFError:
+                    break
+
+            words = result.split(" ")
+
+            if not words:
+                continue
+
+            command = words[0]
+
+            if command == "list-users":
+                self.list_users()
+
+            elif command == "export-keys":
+                args = words[1:]
+                self.export_keys(args)
+
+            elif command == "import-keys":
+                args = words[1:]
+                self.import_keys(args)
+
+
+def main():
+    loop = asyncio.get_event_loop()
+    main_queue = Queue()
+    glib_queue = Queue()
+    glib_loop = GLib.MainLoop()
+
+    bus_thread = DbusT(glib_loop, main_queue.sync_q, glib_queue.sync_q)
+
+    try:
+        panctl = PanCtl(bus_thread)
+    except dbus.exceptions.DBusException:
+        print("Error, no pantalaimon bus found")
+        sys.exit(-1)
+
+    fut = loop.run_in_executor(
+        None,
+        bus_thread.dbus_loop
+    )
+
+    try:
+        loop.run_until_complete(panctl.loop())
+    except KeyboardInterrupt:
+        pass
+
+    GLib.idle_add(bus_thread.quit)
+    loop.run_until_complete(fut)
+
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
index 61eed7f..865c405 100644
--- a/setup.py
+++ b/setup.py
@@ -21,12 +21,14 @@ setup(
         "logbook",
         "peewee",
         "dbus-python",
-        "janus"
+        "janus",
+        "prompt_toolkit",
         "typing;python_version<'3.5'",
         "matrix-nio"
     ],
     entry_points={
-        "console_scripts": ["pantalaimon=pantalaimon.daemon:main"],
+        "console_scripts": ["pantalaimon=pantalaimon.daemon:main",
+                            "panctl=pantalaimon.panctl:main"],
     },
     zip_safe=False
 )