From 0867df1e4b568bf255606e3d20a3b6cfec19969a Mon Sep 17 00:00:00 2001
From: Ian McEwen <ianmcorvidae@ianmcorvidae.net>
Date: Mon, 7 Oct 2024 20:37:40 -0700
Subject: [PATCH] add simple argparse args for connection

---
 curses-client.py | 55 ++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 51 insertions(+), 4 deletions(-)

diff --git a/curses-client.py b/curses-client.py
index 69a4b03..96ea0cb 100644
--- a/curses-client.py
+++ b/curses-client.py
@@ -7,7 +7,7 @@ V 0.1.7
 '''
 
 import curses
-import meshtastic.serial_interface, meshtastic.tcp_interface
+import meshtastic.serial_interface, meshtastic.tcp_interface, meshtastic.ble_interface
 from pubsub import pub
 import textwrap  # Import the textwrap module
 
@@ -19,9 +19,55 @@ except ImportError:
 
 from settings import settings
 
+import argparse
+
+parser = argparse.ArgumentParser(
+        add_help=False,
+        epilog="If no connection arguments are specified, we attempt a serial connection and then a TCP connection to localhost.")
+
+connOuter = parser.add_argument_group('Connection', 'Optional arguments to specify a device to connect to and how.')
+conn = connOuter.add_mutually_exclusive_group()
+conn.add_argument(
+    "--port",
+    "--serial",
+    "-s",
+    help="The port to connect to via serial, e.g. `/dev/ttyUSB0`.",
+    nargs="?",
+    default=None,
+    const=None,
+)
+conn.add_argument(
+    "--host",
+    "--tcp",
+    "-t",
+    help="The hostname or IP address to connect to using TCP.",
+    nargs="?",
+    default=None,
+    const="localhost",
+)
+conn.add_argument(
+    "--ble",
+    "-b",
+    help="The BLE device MAC address or name to connect to.",
+    nargs="?",
+    default=None,
+    const="any"
+)
+
+args = parser.parse_args()
+
 # Initialize Meshtastic interface
-interface = meshtastic.serial_interface.SerialInterface()
-# interface = meshtastic.tcp_interface.TCPInterface(hostname='192.168.xx.xx')
+if args.ble:
+    interface = meshtastic.ble_interface.BLEInterface(args.ble if args.ble != "any" else None)
+elif args.host:
+    interface = meshtastic.tcp_interface.TCPInterface(args.host)
+else:
+    try:
+        interface = meshtastic.serial_interface.SerialInterface(args.port)
+    except PermissionError as ex:
+        print("You probably need to add yourself to the `dialout` group to use a serial connection.")
+    if interface.devPath is None:
+        interface = meshtastic.tcp_interface.TCPInterface("localhost")
 
 myinfo = interface.getMyNodeInfo()
 
@@ -447,4 +493,5 @@ def main(stdscr):
 pub.subscribe(on_receive, 'meshtastic.receive')
 
 if __name__ == "__main__":
-    curses.wrapper(main)
\ No newline at end of file
+    curses.wrapper(main)
+