from meshtastic import BROADCAST_NUM
from utils import get_node_list, decimal_to_hex, get_nodeNum
import globals
from curses_ui import update_packetlog_win, draw_node_list, update_messages_window, draw_channel_list, add_notification


def on_receive(packet):
    # update packet log
    globals.packet_buffer.append(packet)
    if len(globals.packet_buffer) > 20:
        # trim buffer to 20 packets
        globals.packet_buffer = globals.packet_buffer[-20:]
        
    if globals.display_log:
        update_packetlog_win()
    try:
        if 'decoded' in packet and packet['decoded']['portnum'] == 'NODEINFO_APP':
            get_node_list()
            draw_node_list()

        elif 'decoded' in packet and packet['decoded']['portnum'] == 'TEXT_MESSAGE_APP':
            message_bytes = packet['decoded']['payload']
            message_string = message_bytes.decode('utf-8')
            if packet.get('channel'):
                channel_number = packet['channel']
            else:
                channel_number = 0
                myNodeNum = get_nodeNum()
            if packet['to'] == myNodeNum:
                if packet['from'] in globals.channel_list:
                    pass
                else:
                    globals.channel_list.append(packet['from'])
                    globals.all_messages[packet['from']] = []
                    draw_channel_list()

                channel_number = globals.channel_list.index(packet['from'])

            if globals.channel_list[channel_number] != globals.channel_list[globals.selected_channel]:
                add_notification(channel_number)

            # Add received message to the messages list
            message_from_id = packet['from']
            message_from_string = ""
            for node in globals.interface.nodes.values():
                if message_from_id == node['num']:
                    message_from_string = node["user"]["longName"]  # Get the long name using the node ID
                    break
                else:
                    message_from_string = str(decimal_to_hex(message_from_id))  # If long name not found, use the ID as string
        
            if globals.channel_list[channel_number] in globals.all_messages:
                globals.all_messages[globals.channel_list[channel_number]].append((f">> {message_from_string} ", message_string))
            else:
                globals.all_messages[globals.channel_list[channel_number]] = [(f">> {message_from_string} ", message_string)]
                draw_channel_list()
            update_messages_window()

    except KeyError as e:
        print(f"Error processing packet: {e}")