Beginning of emq-cli administrative tool

1 files changed, 120 insertions, 0 deletions

diff --git a/bin/emq-cli b/bin/emq-cli
new file mode 100755
index 0000000..389bdbd
--- /dev/null
+++ b/bin/emq-cli
@@ -0,0 +1,120 @@
+#!/usr/bin/env python
+# -*- mode: python -*-
+#
+# This file is part of eventmq.
+#
+# eventmq is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# eventmq is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with eventmq.  If not, see <http://www.gnu.org/licenses/>.
+from __future__ import print_function
+
+import argparse
+import cmd
+import sys
+
+from eventmq import exceptions
+from eventmq.client.messages import send_emqp_message
+from eventmq.constants import PROTOCOL_VERSION
+from eventmq.sender import Sender
+from eventmq.utils.messages import generate_msgid
+
+class Shell(cmd.Cmd):
+    """
+    Interactive EventMQ Shell
+    """
+    def __init__(self, addr=None):
+        cmd.Cmd.__init__(self)
+
+        self.prompt = "> "
+
+        self.addr = addr
+        if self.addr:
+            self.do_connect(addr)
+
+        self.socket = Sender()
+
+    def do_connect(self, line):
+        """
+        Connect to a service
+        """
+        if not line:
+            sys.stderr.write('Error: No address provided\n')
+            return
+
+        try:
+            self.socket.connect(line)
+        except exceptions.EventMQError as e:
+            if "status=201" in e.message:
+                sys.stderr.write('Error: Already connected to {} Disconnect '
+                                 'first\n'.format(self.addr))
+                return
+            raise
+
+        self.addr = line
+        print("Connecting to {}...".format(line))
+        self.send_message('HELLO')
+
+    def do_disconnect(self, line):
+        """
+        Disconnect from a service
+        """
+        print('Closing connection to {}'.format(self.addr))
+        self.socket.rebuild()
+        self.addr = None
+
+    def do_status(self, line):
+        """
+        Request the status of the connected component
+        """
+        raise NotImplementedError('TODO: STATUS command')
+        self.send_message(('STATUS',))
+        print(self.socket.recv_multipart())
+
+    def do_shutdown(self, line):
+        """
+        Request the connected component shutdown
+        """
+        self.send_message('DISCONNECT')
+        import time; time.sleep(1)
+        print(self.socket.recv_multipart())
+
+    def do_send_cmd(self, line):
+        """
+        Send a raw command to the connected host.
+        """
+
+    def send_message(self, command, message=()):
+        if not self.addr:
+            sys.stderr.write('Error: Not connected\n')
+            return
+
+        message = (command, generate_msgid('admin:')) + message
+
+        self.socket.send_multipart(message, PROTOCOL_VERSION)
+
+    def do_quit(self, line):
+        return True
+
+    def do_EOF(self, line):
+        return self.do_quit(line)
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(description='Utility for interacting with '
+                                     'an EventMQ cluster')
+
+    parser.add_argument('--addr', '-a', type=str, nargs='?',
+                        help='specify address to connect to')
+
+    args = parser.parse_args()
+
+    app = Shell(addr=args.addr)
+    app.cmdloop()