meshtastic

- Unnamed repository; edit this file 'description' to name the repository.
git clone git://git.acid.vegas/-c.git
Log | Files | Refs | Archive | README | LICENSE

commit fe2a3e4bb55bf41d7e2326b530039bad38963a1f
parent aca1e287038f7ca3a550ba7771635332de5a7648
Author: acidvegas <acid.vegas@acid.vegas>
Date: Thu, 25 Apr 2024 15:39:14 -0400

Updated README documentation and improved overall code-baae of the serial interfacing tool

Diffstat:
MREADME.md | 12+++++++++++-
Mmeshtastic_serial.py | 181++++++++++++++++++++++++++++++++++++++++++++++++-------------------------------

2 files changed, 121 insertions(+), 72 deletions(-)

diff --git a/README.md b/README.md
@@ -3,7 +3,17 @@
 
 ## WORK-IN-PROGRESS
 
-Here I will be just throw up random bits of code I write as I experiment more with these Meshtastic devices. Currently using a Lilygo T-Deck & a Heltec Lora v3 for testing.
+Here I will be just throw up random bits of code I write as I experiment more with these Meshtastic devices.
+
+Currently using a [Lilygo T-Deck](https://www.lilygo.cc/products/t-deck) & a [Heltec Lora 32 v3](https://heltec.org/project/wifi-lora-32-v3/) for testing.
+
+I am waiting on a [Lilygo T-Beam](https://www.lilygo.cc/products/t-beam-v1-1-esp32-lora-module) & a [RAK Wireless 4631](https://store.rakwireless.com/products/wisblock-core-modules?variant=42440631419078) to arrive for expanding my network & conducting more testing.
+
+Reading device packets over serial or TCP allows you to see the decoded data easily & fast. I have plans to add MQTT interfacing to this repository, but note that this requires you to decrypt incomming packet payloads using the channels PSK. We will get to that in the future.
+
+The goal is to experiment with the possibilities of Python as a means of interfacing with a Meshtastic device, playing with basic I/O operations, etc. My first project is going to be a relay for IRC & Meshtastic to communicate.
+
+#### Stay tuned!
 
 ___
 
diff --git a/meshtastic_serial.py b/meshtastic_serial.py
@@ -19,9 +19,6 @@ except ImportError:
 	raise ImportError('pubsub library not found (pip install pypubsub)') # Confirm this Pypi package name...
 
 
-# Global variables
-node_long_names = {}
-
 # Initialize logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s | %(levelname)9s | %(funcName)s | %(message)s')
 
@@ -32,73 +29,131 @@ def now():
 	return time.strftime('%Y-%m-%d %H:%M:%S')
 
 
-def on_connect(interface, topic=pub.AUTO_TOPIC):
-	'''
-	Callback function for connection established
+class Meshtastic(object):
+	def __init__(self, serial: str):
+		self.interface = None   # We will define the interface in the run() function
+		self.nodes     = {}     # Nodes will populate with the on_node() callback
+		self.serial    = serial # Serial device to use for the Meshtastic interface
+		
 
-	:param interface: Meshtastic interface
-	:param topic:     PubSub topic
-	'''
+	def disconnect(self):
+		'''Disconnect from the Meshtastic interface'''
 
-	logging.info('Connection established')
+		if pub.getDefaultTopicMgr().hasSubscribers(): 
+			pub.unsubAll()
+			logging.info('Unsubscribed from all Meshtastic topics')
+		else:
+			logging.warning('No Meshtastic topics to unsubscribe from')
 
+		if self.interface:
+			self.interface.close()
+			logging.info('Meshtastic interface closed')
+		else:
+			logging.warning('No Meshtastic interface to close')
 
-def on_disconnect(interface, topic=pub.AUTO_TOPIC):
-	'''
-	Callback function for connection lost
 
-	:param interface: Meshtastic interface
-	:param topic:     PubSub topic
-	'''
+	def run(self):
+		'''Start the Meshtastic interface and subscribe to the callback functions'''
+		
+		if devices := findPorts():
+			if not os.path.exists(args.serial) or not args.serial in devices:
+				raise SystemExit(f'Invalid serial device: {args.serial} (Available: {devices})') # Show available devices if the specified device is invalid
+		else:
+			raise SystemExit('No serial devices found')
 
-	logging.error('Connection lost')
+		# Initialize the Meshtastic interface
+		self.interface = SerialInterface(self.serial)
 
+		logging.info('Meshtastic interface started over serial on {self.serial}')
 
-def on_packet(packet: dict):
-	'''
-	Callback function for received packets
+		# Get the current node information
+		me = self.interface.nodes[self.interface.myInfo.my_node_num]
+		logging.debug(me)
+		
+		# Create the Meshtastic callback subscriptions
+		pub.subscribe(self.event_connect,    'meshtastic.connection.established')
+		pub.subscribe(self.event_disconnect, 'meshtastic.connection.lost')
+		pub.subscribe(self.on_node,          'meshtastic.node.updated')
+		pub.subscribe(self.on_packet,        'meshtastic.receive')
 
-	:param packet: Packet received
-	'''
+		logging.debug('Listening for Meshtastic events...')
 
-	if packet['decoded']['portnum'] == 'TEXT_MESSAGE_APP':
-		sender_id = str(packet['from'])
-		message = packet['decoded']['payload'].decode('utf-8')
+		# The meshtastic.receive topics can be broken down further:
+		# pub.subscribe(self.on_text,      'meshtastic.receive.text')
+		# pub.subscribe(self.on_position,  'meshtastic.receive.position')
+		# pub.subscribe(self.on_user,      'meshtastic.receive.user')
+		# pub.subscribe(self.on_data,      'meshtastic.receive.data.portnum')
 
-		# Message from self
-		if sender_id == str(interface.myInfo.my_node_num):
-			print(f'{now()} {node_long_names[sender_id]}: {message}')
+		
+	def event_connect(self, interface, topic=pub.AUTO_TOPIC):
+		'''
+		Callback function for connection established
 
-		# Message from others
-		if sender_id in node_long_names:
-			print(f'{now()} {node_long_names[sender_id]}: {message}')
+		:param interface: Meshtastic interface
+		:param topic:     PubSub topic
+		'''
 
-		# Unknown message (maybe trigger for rescanning the nodes if we dont find the sender in the list)
-		else:
-			print(f'{now()} UNK: {message}')
+		logging.info('Connection established')
+
+
+	def event_disconnect(self, interface, topic=pub.AUTO_TOPIC):
+		'''
+		Callback function for connection lost
+
+		:param interface: Meshtastic interface
+		:param topic:     PubSub topic
+		'''
+
+		logging.warning('Connection lost')
 
 
-def on_node(interface, topic=pub.AUTO_TOPIC):
-	'''
-	Callback function for node updates
+	def on_packet(self, packet: dict):
+		'''
+		Callback function for received packets
 
-	:param interface: Meshtastic interface
-	:param topic:     PubSub topic
-	'''
+		:param packet: Packet received
+		'''
 
-	if not interface.nodes:
-		logging.warning('No nodes found')
-		return
+		# Handle incoming text messages
+		if packet['decoded']['portnum'] == 'TEXT_MESSAGE_APP':
+			sender = packet['from']
+			msg    = packet['decoded']['payload'].decode('utf-8')
 
-	for node in interface.nodes.values():
-		short = node['user']['shortName']
-		long  = node['user']['longName'].encode('ascii', 'ignore').decode().rstrip()
-		num   = str(node['num'])
-		id    = node['user']['id']
-		mac   = node['user']['macaddr']
-		hw    = node['user']['hwModel']
+			# Message from self
+			if sender == self.interface.myInfo.my_node_num:
+				print(f'{now()} {self.nodes[sender]}: {msg}') # Can do custom formatting here or ignore the message, just an example
 
-		node_long_names[num] = long # we store the node updates in a dictionary so we can parse the names of who sent incomming messages
+			# Message from others
+			if sender in self.nodes:
+				print(f'{now()} {self.nodes[sender]}: {msg}')
+
+			# Unknown sender
+			else:
+				# TODO: Trigger request for node update here
+				print(f'{now()} UNK: {msg}')
+
+
+	def on_node(self, interface, topic=pub.AUTO_TOPIC):
+		'''
+		Callback function for node updates
+
+		:param interface: Meshtastic interface
+		:param topic:     PubSub topic
+		'''
+
+		if not interface.nodes:
+			logging.warning('No nodes found')
+			return
+
+		for node in interface.nodes.values():
+			short = node['user']['shortName']
+			long  = node['user']['longName'].encode('ascii', 'ignore').decode().rstrip()
+			num   = node['num']
+			id    = node['user']['id']
+			mac   = node['user']['macaddr']
+			hw    = node['user']['hwModel']
+
+			self.nodes[num] = long # we store the node updates in a dictionary so we can parse the names of who sent incomming messages
 
 
 
@@ -107,27 +162,11 @@ if __name__ == '__main__':
 	parser.add_argument('--serial', default='/dev/ttyACM0', help='Use serial interface')
 	args = parser.parse_args()
 
-	# Check if the serial device exists
-	if available_devices := findPorts():
-		if not os.path.exists(args.serial) or not args.serial in available_devices:
-			raise SystemExit(f'Invalid serial device: {args.serial} (Available: {available_devices})')
-	else:
-		raise SystemExit('No serial devices found')
+	# Define the Meshtastic client
+	mesh = Meshtastic(args.serial)
 
 	# Initialize the Meshtastic interface
-	interface = SerialInterface(args.serial)
-
-	# Create the Meshtastic callback subscriptions
-	pub.subscribe(on_connect,    'meshtastic.connection.established')
-	pub.subscribe(on_disconnect, 'meshtastic.connection.lost')
-	pub.subscribe(on_node,       'meshtastic.node.updated')
-	pub.subscribe(on_packet,     'meshtastic.receive')
-
-	# The meshtastic.receive topics can be broken down further:
-	# pub.subscribe(on_text,      'meshtastic.receive.text')
-	# pub.subscribe(on_position,  'meshtastic.receive.position')
-	# pub.subscribe(on_user,      'meshtastic.receive.user')
-	# pub.subscribe(on_data,      'meshtastic.receive.data.portnum')
+	mesh.run()
 
 	# Keep-alive loop
 	try:
@@ -136,4 +175,4 @@ if __name__ == '__main__':
 	except KeyboardInterrupt:
 		pass
 	finally:
-		interface.close()
+		mesh.disconnect()