diff --git a/main.py b/main.py index 4988df2..b676ca9 100644 --- a/main.py +++ b/main.py @@ -12,8 +12,27 @@ from asyncio_mqtt import Client import settings + +STORAGE_FILE = 'data/data.json' +storage = dict(nodes={}) +def save_data(): + with open(STORAGE_FILE, 'w') as f: + json.dump(storage, f, sort_keys=True, indent=4) +try: + storage = json.load(open(STORAGE_FILE)) +except FileNotFoundError: + save_data() + + bot = TelegramClient('data/bot', settings.API_ID, settings.API_HASH).start(bot_token=settings.API_TOKEN) +def int_to_mesh_id(num): + # converts signed 32-bit ints to unsigned hex meshtastic IDs + # example: -697240346 to !d670f4e6 + + hex_str = hex(num & 0xFFFFFFFF) + return '!' + hex_str[2:] + async def process_text(data): msg_to = data.get('to', None) @@ -23,12 +42,40 @@ async def process_text(data): try: msg_text = data['payload']['text'] + sender = data['from'] except KeyError: logging.info('Invalid payload: %s', str(data)) return - logging.info('Channel message: %s', msg_text) - await bot.send_message(settings.CHAT_ID, msg_text) + sender_id = int_to_mesh_id(sender) + + try: + from_name = storage['nodes'][sender_id]['long_name'] + except KeyError: + from_name = 'Unknown sender' + + logging.info('Channel message from %s: %s', from_name, msg_text) + + msg = from_name + ': ' + msg_text + await bot.send_message(settings.CHAT_ID, msg) + +async def process_info(data): + try: + node_id = data['payload']['id'] + long_name = data['payload']['longname'] + short_name = data['payload']['shortname'] + except KeyError: + logging.info('Invalid payload: %s', str(data)) + return + + logging.info('Node %s info: %s (%s)', node_id, long_name, short_name) + + storage['nodes'][node_id] = dict( + node_id=node_id, + long_name=long_name, + short_name=short_name, + ) + save_data() async def process_mqtt(message): @@ -55,6 +102,8 @@ async def process_mqtt(message): if msg_type == 'text': await process_text(data) + if msg_type == 'nodeinfo': + await process_info(data) else: logging.debug('Ignored message type %s: %s', msg_type, text) return @@ -67,4 +116,5 @@ async def fetch_mqtt(): await process_mqtt(message) with bot: - bot.loop.run_until_complete(fetch_mqtt()) + bot.loop.create_task(fetch_mqtt()) + bot.run_until_disconnected()