Add command line interface

Signed-off-by: Denis 'GNUtoo' Carikli <GNUtoo@cyberdimension.org>

1 files changed, 32 insertions, 14 deletions

diff --git a/xmm.py b/xmm.py
index b97cce8..6352b1f 100755
--- a/xmm.py
+++ b/xmm.py
@@ -24,10 +24,17 @@ import sys
 EX_USAGE = 64
 
 def usage(progname):
-    print("{} <port>".format(progname))
-    print("Example: {} /dev/ttyACM0".format(progname))
+    print("Usage:")
+    print("{} commands list # list available AT commands".format(progname))
+    print("{} commands help ".format(progname)
+          + "# list usage of available AT commands")
     sys.exit(EX_USAGE)
 
+def print_header(string):
+    print('+' + '-' * (len(string) + 2) + '+')
+    print('| ' + string + ' |')
+    print('+' + '-' * (len(string) + 2) + '+')
+
 class USB(object):
     def __init__(self):
         self._modem_path = None
@@ -67,8 +74,7 @@ class USB(object):
     def get_ttys(self):
         all_ttys = self._get_ttys()
         all_ttys.sort()
-        # return [all_ttys[0], all_ttys[2]]
-        return [all_ttys[0]]
+        return [all_ttys[0], all_ttys[3]]
 
     def route_modem_to_usb(self):
         route_command = 'echo MODEM > /sys/devices/virtual/sec/switch/usb_sel'
@@ -187,11 +193,6 @@ class UART(object):
         return lines
 
     def get_cmds_usage(self):
-        def print_cmd_header(command):
-            print('+' + '-' * (len(command) + 2) + '+')
-            print('| ' + command + ' |')
-            print('+' + '-' * (len(command) + 2) + '+')
-
         for command in self.list_commands():
             # Don't reset the modem
             if command == 'ATE':
@@ -201,19 +202,36 @@ class UART(object):
                     continue
                 usage = self.get_cmd_usage(command)
                 if len(usage) > 0:
-                    print_cmd_header(command)
+                    print_header(command)
                     for line in usage:
                         print (line)
 
     def close(self):
         self.port.close()
 
-if __name__ == '__main__':
+def get_ttys():
     usb = USB()
     if not usb.has_modem():
         usb.route_modem_to_usb()
         print('Please unplug and replug the device')
         sys.exit(0)
-    for tty in usb.get_ttys():
-        uart = UART(tty)
-        uart.get_cmds_usage()
+    return usb.get_ttys()
+
+if __name__ == '__main__':
+    if len(sys.argv) == 3 and \
+       sys.argv[1] == "commands" and sys.argv[2] == "list":
+        for tty in get_ttys():
+            print_header(tty)
+            uart = UART(tty)
+            for command in uart.list_commands():
+                print(command)
+    elif len(sys.argv) == 3 and \
+       sys.argv[1] == "commands" and sys.argv[2] == "help":
+        for tty in get_ttys():
+            tty_header = int((78 - len(tty)) / 2) * " " \
+                + tty + " " *int((78 - len(tty)) / 2)
+            print_header(tty_header)
+            uart = UART(tty)
+            uart.get_cmds_usage()
+    else:
+        usage(sys.argv[0])