serial chat

examples/serial_chat.py

@@ -0,0 +1,108 @@
+#!/usr/bin/python
+
+import asyncio
+import sys
+import argparse
+import json
+from meshcore import MeshCore
+from meshcore import SerialConnection
+from meshcore import EventType
+
+PORT = "/dev/ttyACM0"
+BAUDRATE = 115200
+DEST = "t114_fdl"
+
+# Subscribe to incoming messages
+async def handle_message(event):
+    data = event.payload
+
+    contact = mc.get_contact_by_key_prefix(data['pubkey_prefix'])
+
+    print(f"{contact['adv_name']}: {data['text']}")
+
+async def main () :
+    global mc
+    # Parse command line arguments
+    parser = argparse.ArgumentParser(description="Print incoming messages and send message to destination")
+    parser.add_argument("-b", "--baudrate", default=BAUDRATE, help="Baudrate")
+    parser.add_argument("-p", "--port", default = PORT, help="Serial port")
+    parser.add_argument("-d", "--dest", default=DEST, help="default destination")
+    parser.add_argument("--timeout", type=float, default=10.0, help="ACK timeout in seconds")
+    args = parser.parse_args()
+
+    mc = await MeshCore.create_serial(args.port, args.baudrate)
+
+    await mc.ensure_contacts()
+
+    # Subscribe to private messages
+    private_subscription = mc.subscribe(EventType.CONTACT_MSG_RECV, handle_message)
+    
+    # Subscribe to channel messages
+    channel_subscription = mc.subscribe(EventType.CHANNEL_MSG_RECV, handle_message)
+    
+    await mc.start_auto_message_fetching()
+
+    contact = mc.get_contact_by_name(args.dest)
+    if contact is None:
+        print(f"Contact '{DEST}' not found in contacts.")
+        return
+
+    try:
+        while True:
+            line = (await asyncio.to_thread(sys.stdin.readline)).rstrip('\n')
+
+            if line.startswith("to ") :
+                dest = line[3:]
+                nc = mc.get_contact_by_name(dest)
+                if mc is None:
+                    print(f"Contact '{DEST}' not found in contacts.")
+                    return
+                else :
+                    contact = nc
+
+            elif line == "quit" or line == "q" :
+                break
+
+            elif line == "contacts" :
+                print (json.dumps(mc.contacts,indent=2))
+
+            elif line == "" :
+                pass
+
+            else :
+                if line.startswith("send") :
+                    line = line[5:]
+                ret = await mc.commands.send_msg(contact , line)
+                exp_ack = ret["expected_ack"].hex()
+                print(" Sent ... ", end="", flush=True)
+                res = await mc.wait_for_event(EventType.ACK, attribute_filters={"code": exp_ack}, timeout=5)
+                if res is None :
+                    print ("No ack !!!")
+                else :
+                    print ("Ack")
+            
+    except KeyboardInterrupt:
+        meshcore.stop()
+        print("\nExiting...")
+    except asyncio.CancelledError:
+        # Handle task cancellation from KeyboardInterrupt in asyncio.run()
+        print("\nTask cancelled - cleaning up...")
+    finally:
+        # Clean up subscriptions
+        mc.unsubscribe(private_subscription)
+        mc.unsubscribe(channel_subscription)
+    
+        # Stop auto-message fetching
+        await mc.stop_auto_message_fetching()
+    
+        # Disconnect
+        await mc.disconnect()
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        # This prevents the KeyboardInterrupt traceback from being shown
+        print("\nExited cleanly")
+    except Exception as e:
+        print(f"Error: {e}")