1
0
forked from iarv/contact

Compare commits

...

2 Commits

Author SHA1 Message Date
pdxlocations
8c2f0429ac probably not working changes 2025-04-05 20:55:33 -07:00
pdxlocations
425ee85ef0 rename state 2025-04-05 20:06:48 -07:00
13 changed files with 424 additions and 399 deletions

View File

@@ -29,7 +29,11 @@ from contact.utilities.arg_parser import setup_parser
from contact.utilities.interfaces import initialize_interface from contact.utilities.interfaces import initialize_interface
from contact.utilities.input_handlers import get_list_input from contact.utilities.input_handlers import get_list_input
from contact.utilities.utils import get_channels, get_node_list, get_nodeNum from contact.utilities.utils import get_channels, get_node_list, get_nodeNum
import contact.globals as globals import contact.globals as globals
from contact.ui.ui_state import NodeState
node_state = NodeState()
# Set ncurses compatibility settings # Set ncurses compatibility settings
os.environ["NCURSES_NO_UTF8_ACS"] = "1" os.environ["NCURSES_NO_UTF8_ACS"] = "1"
@@ -65,25 +69,25 @@ def main(stdscr):
logging.info("Initializing interface %s", args) logging.info("Initializing interface %s", args)
with globals.lock: with globals.lock:
globals.interface = initialize_interface(args) node_state.interface = initialize_interface(args)
if globals.interface.localNode.localConfig.lora.region == 0: if node_state.interface.localNode.localConfig.lora.region == 0:
confirmation = get_list_input("Your region is UNSET. Set it now?", "Yes", ["Yes", "No"]) confirmation = get_list_input("Your region is UNSET. Set it now?", "Yes", ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
set_region(globals.interface) set_region(node_state)
globals.interface.close() node_state.interface.close()
globals.interface = initialize_interface(args) node_state.interface = initialize_interface(args)
logging.info("Interface initialized") logging.info("Interface initialized")
globals.myNodeNum = get_nodeNum() get_nodeNum(node_state)
globals.channel_list = get_channels() globals.channel_list = get_channels()
globals.node_list = get_node_list() globals.node_list = get_node_list()
pub.subscribe(on_receive, 'meshtastic.receive') pub.subscribe(on_receive, 'meshtastic.receive')
init_nodedb() init_nodedb(node_state)
load_messages_from_db() load_messages_from_db(node_state)
logging.info("Starting main UI") logging.info("Starting main UI")
main_ui(stdscr) main_ui(stdscr, node_state)
except Exception as e: except Exception as e:
console_output = output_capture.getvalue() console_output = output_capture.getvalue()

View File

@@ -1,4 +1,4 @@
interface = None # interface = None
lock = None lock = None
display_log = False display_log = False
all_messages = {} all_messages = {}
@@ -6,7 +6,7 @@ channel_list = []
notifications = [] notifications = []
packet_buffer = [] packet_buffer = []
node_list = [] node_list = []
myNodeNum = 0 # myNodeNum = 0
selected_channel = 0 selected_channel = 0
selected_message = 0 selected_message = 0
selected_node = 0 selected_node = 0

View File

@@ -10,7 +10,7 @@ import contact.globals as globals
from datetime import datetime from datetime import datetime
def on_receive(packet, interface): def on_receive(packet, node_state):
with globals.lock: with globals.lock:
# Update packet log # Update packet log
@@ -20,7 +20,7 @@ def on_receive(packet, interface):
globals.packet_buffer = globals.packet_buffer[-20:] globals.packet_buffer = globals.packet_buffer[-20:]
if globals.display_log: if globals.display_log:
draw_packetlog_win() draw_packetlog_win(node_state)
try: try:
if 'decoded' not in packet: if 'decoded' not in packet:
return return
@@ -32,7 +32,7 @@ def on_receive(packet, interface):
if packet['decoded']['portnum'] == 'NODEINFO_APP': if packet['decoded']['portnum'] == 'NODEINFO_APP':
if "user" in packet['decoded'] and "longName" in packet['decoded']["user"]: if "user" in packet['decoded'] and "longName" in packet['decoded']["user"]:
maybe_store_nodeinfo_in_db(packet) maybe_store_nodeinfo_in_db(packet, node_state)
elif packet['decoded']['portnum'] == 'TEXT_MESSAGE_APP': elif packet['decoded']['portnum'] == 'TEXT_MESSAGE_APP':
message_bytes = packet['decoded']['payload'] message_bytes = packet['decoded']['payload']
@@ -66,7 +66,7 @@ def on_receive(packet, interface):
# Add received message to the messages list # Add received message to the messages list
message_from_id = packet['from'] message_from_id = packet['from']
message_from_string = get_name_from_database(message_from_id, type='short') + ":" message_from_string = get_name_from_database(message_from_id, node_state, type='short') + ":"
if globals.channel_list[channel_number] not in globals.all_messages: if globals.channel_list[channel_number] not in globals.all_messages:
globals.all_messages[globals.channel_list[channel_number]] = [] globals.all_messages[globals.channel_list[channel_number]] = []
@@ -95,9 +95,9 @@ def on_receive(packet, interface):
globals.all_messages[globals.channel_list[channel_number]].append((f"{config.message_prefix} {message_from_string} ", message_string)) globals.all_messages[globals.channel_list[channel_number]].append((f"{config.message_prefix} {message_from_string} ", message_string))
if refresh_channels: if refresh_channels:
draw_channel_list() draw_channel_list(node_state)
if refresh_messages: if refresh_messages:
draw_messages_window(True) draw_messages_window(node_state, True)
save_message_to_db(globals.channel_list[channel_number], message_from_id, message_string) save_message_to_db(globals.channel_list[channel_number], message_from_id, message_string)

View File

@@ -11,7 +11,7 @@ ack_naks = {}
# Note "onAckNak" has special meaning to the API, thus the nonstandard naming convention # Note "onAckNak" has special meaning to the API, thus the nonstandard naming convention
# See https://github.com/meshtastic/python/blob/master/meshtastic/mesh_interface.py#L462 # See https://github.com/meshtastic/python/blob/master/meshtastic/mesh_interface.py#L462
def onAckNak(packet): def onAckNak(packet, node_state):
from contact.ui.contact_ui import draw_messages_window from contact.ui.contact_ui import draw_messages_window
request = packet['decoded']['requestId'] request = packet['decoded']['requestId']
if(request not in ack_naks): if(request not in ack_naks):
@@ -35,13 +35,13 @@ def onAckNak(packet):
globals.all_messages[acknak['channel']][acknak['messageIndex']] = (config.sent_message_prefix + confirm_string + ": ", message) globals.all_messages[acknak['channel']][acknak['messageIndex']] = (config.sent_message_prefix + confirm_string + ": ", message)
update_ack_nak(acknak['channel'], acknak['timestamp'], message, ack_type) update_ack_nak(acknak['channel'], acknak['timestamp'], message, ack_type, node_state)
channel_number = globals.channel_list.index(acknak['channel']) channel_number = globals.channel_list.index(acknak['channel'])
if globals.channel_list[channel_number] == globals.channel_list[globals.selected_channel]: if globals.channel_list[channel_number] == globals.channel_list[globals.selected_channel]:
draw_messages_window() draw_messages_window(node_state)
def on_response_traceroute(packet): def on_response_traceroute(packet, node_state):
"""on response for trace route""" """on response for trace route"""
from contact.ui.contact_ui import draw_channel_list, draw_messages_window, add_notification from contact.ui.contact_ui import draw_channel_list, draw_messages_window, add_notification
@@ -56,18 +56,18 @@ def on_response_traceroute(packet):
msg_str = "Traceroute to:\n" msg_str = "Traceroute to:\n"
route_str = get_name_from_database(packet["to"], 'short') or f"{packet['to']:08x}" # Start with destination of response route_str = get_name_from_database(packet["to"], node_state, 'short') or f"{packet['to']:08x}" # Start with destination of response
# SNR list should have one more entry than the route, as the final destination adds its SNR also # SNR list should have one more entry than the route, as the final destination adds its SNR also
lenTowards = 0 if "route" not in msg_dict else len(msg_dict["route"]) lenTowards = 0 if "route" not in msg_dict else len(msg_dict["route"])
snrTowardsValid = "snrTowards" in msg_dict and len(msg_dict["snrTowards"]) == lenTowards + 1 snrTowardsValid = "snrTowards" in msg_dict and len(msg_dict["snrTowards"]) == lenTowards + 1
if lenTowards > 0: # Loop through hops in route and add SNR if available if lenTowards > 0: # Loop through hops in route and add SNR if available
for idx, node_num in enumerate(msg_dict["route"]): for idx, node_num in enumerate(msg_dict["route"]):
route_str += " --> " + (get_name_from_database(node_num, 'short') or f"{node_num:08x}") \ route_str += " --> " + (get_name_from_database(node_num, node_state, 'short') or f"{node_num:08x}") \
+ " (" + (str(msg_dict["snrTowards"][idx] / 4) if snrTowardsValid and msg_dict["snrTowards"][idx] != UNK_SNR else "?") + "dB)" + " (" + (str(msg_dict["snrTowards"][idx] / 4) if snrTowardsValid and msg_dict["snrTowards"][idx] != UNK_SNR else "?") + "dB)"
# End with origin of response # End with origin of response
route_str += " --> " + (get_name_from_database(packet["from"], 'short') or f"{packet['from']:08x}") \ route_str += " --> " + (get_name_from_database(packet["from"], node_state, 'short') or f"{packet['from']:08x}") \
+ " (" + (str(msg_dict["snrTowards"][-1] / 4) if snrTowardsValid and msg_dict["snrTowards"][-1] != UNK_SNR else "?") + "dB)" + " (" + (str(msg_dict["snrTowards"][-1] / 4) if snrTowardsValid and msg_dict["snrTowards"][-1] != UNK_SNR else "?") + "dB)"
msg_str += route_str + "\n" # Print the route towards destination msg_str += route_str + "\n" # Print the route towards destination
@@ -77,15 +77,15 @@ def on_response_traceroute(packet):
backValid = "hopStart" in packet and "snrBack" in msg_dict and len(msg_dict["snrBack"]) == lenBack + 1 backValid = "hopStart" in packet and "snrBack" in msg_dict and len(msg_dict["snrBack"]) == lenBack + 1
if backValid: if backValid:
msg_str += "Back:\n" msg_str += "Back:\n"
route_str = get_name_from_database(packet["from"], 'short') or f"{packet['from']:08x}" # Start with origin of response route_str = get_name_from_database(packet["from"], node_state, 'short') or f"{packet['from']:08x}" # Start with origin of response
if lenBack > 0: # Loop through hops in routeBack and add SNR if available if lenBack > 0: # Loop through hops in routeBack and add SNR if available
for idx, node_num in enumerate(msg_dict["routeBack"]): for idx, node_num in enumerate(msg_dict["routeBack"]):
route_str += " --> " + (get_name_from_database(node_num, 'short') or f"{node_num:08x}") \ route_str += " --> " + (get_name_from_database(node_num, node_state, 'short') or f"{node_num:08x}") \
+ " (" + (str(msg_dict["snrBack"][idx] / 4) if msg_dict["snrBack"][idx] != UNK_SNR else "?") + "dB)" + " (" + (str(msg_dict["snrBack"][idx] / 4) if msg_dict["snrBack"][idx] != UNK_SNR else "?") + "dB)"
# End with destination of response (us) # End with destination of response (us)
route_str += " --> " + (get_name_from_database(packet["to"], 'short') or f"{packet['to']:08x}") \ route_str += " --> " + (get_name_from_database(packet["to"], node_state, 'short') or f"{packet['to']:08x}") \
+ " (" + (str(msg_dict["snrBack"][-1] / 4) if msg_dict["snrBack"][-1] != UNK_SNR else "?") + "dB)" + " (" + (str(msg_dict["snrBack"][-1] / 4) if msg_dict["snrBack"][-1] != UNK_SNR else "?") + "dB)"
msg_str += route_str + "\n" # Print the route back to us msg_str += route_str + "\n" # Print the route back to us
@@ -94,7 +94,7 @@ def on_response_traceroute(packet):
globals.channel_list.append(packet['from']) globals.channel_list.append(packet['from'])
refresh_channels = True refresh_channels = True
if(is_chat_archived(packet['from'])): if(is_chat_archived(packet['from']), node_state):
update_node_info_in_db(packet['from'], chat_archived=False) update_node_info_in_db(packet['from'], chat_archived=False)
channel_number = globals.channel_list.index(packet['from']) channel_number = globals.channel_list.index(packet['from'])
@@ -105,21 +105,21 @@ def on_response_traceroute(packet):
add_notification(channel_number) add_notification(channel_number)
refresh_channels = True refresh_channels = True
message_from_string = get_name_from_database(packet['from'], type='short') + ":\n" message_from_string = get_name_from_database(packet['from'], node_state, type='short') + ":\n"
if globals.channel_list[channel_number] not in globals.all_messages: if globals.channel_list[channel_number] not in globals.all_messages:
globals.all_messages[globals.channel_list[channel_number]] = [] globals.all_messages[globals.channel_list[channel_number]] = []
globals.all_messages[globals.channel_list[channel_number]].append((f"{config.message_prefix} {message_from_string}", msg_str)) globals.all_messages[globals.channel_list[channel_number]].append((f"{config.message_prefix} {message_from_string}", msg_str))
if refresh_channels: if refresh_channels:
draw_channel_list() draw_channel_list(node_state)
if refresh_messages: if refresh_messages:
draw_messages_window(True) draw_messages_window(node_state, True)
save_message_to_db(globals.channel_list[channel_number], packet['from'], msg_str) save_message_to_db(globals.channel_list[channel_number], packet['from'], msg_str)
def send_message(message, destination=BROADCAST_NUM, channel=0): def send_message(message, node_state, destination=BROADCAST_NUM, channel=0):
myid = globals.myNodeNum myid = node_state.myNodeNum
send_on_channel = 0 send_on_channel = 0
channel_id = globals.channel_list[channel] channel_id = globals.channel_list[channel]
if isinstance(channel_id, int): if isinstance(channel_id, int):
@@ -128,7 +128,7 @@ def send_message(message, destination=BROADCAST_NUM, channel=0):
elif isinstance(channel_id, str): elif isinstance(channel_id, str):
send_on_channel = channel send_on_channel = channel
sent_message_data = globals.interface.sendText( sent_message_data = node_state.interface.sendText(
text=message, text=message,
destinationId=destination, destinationId=destination,
wantAck=True, wantAck=True,
@@ -168,9 +168,9 @@ def send_message(message, destination=BROADCAST_NUM, channel=0):
ack_naks[sent_message_data.id] = {'channel': channel_id, 'messageIndex': len(globals.all_messages[channel_id]) - 1, 'timestamp': timestamp} ack_naks[sent_message_data.id] = {'channel': channel_id, 'messageIndex': len(globals.all_messages[channel_id]) - 1, 'timestamp': timestamp}
def send_traceroute(): def send_traceroute(node_state):
r = mesh_pb2.RouteDiscovery() r = mesh_pb2.RouteDiscovery()
globals.interface.sendData( node_state.interface.sendData(
r, r,
destinationId=globals.node_list[globals.selected_node], destinationId=globals.node_list[globals.selected_node],
portNum=portnums_pb2.PortNum.TRACEROUTE_APP, portNum=portnums_pb2.PortNum.TRACEROUTE_APP,

View File

@@ -25,17 +25,17 @@ def main(stdscr):
parser = setup_parser() parser = setup_parser()
args = parser.parse_args() args = parser.parse_args()
interface = initialize_interface(args) node_state.interface = initialize_interface(args)
if interface.localNode.localConfig.lora.region == 0: if node_state.interface.localNode.localConfig.lora.region == 0:
confirmation = get_list_input("Your region is UNSET. Set it now?", "Yes", ["Yes", "No"]) confirmation = get_list_input("Your region is UNSET. Set it now?", "Yes", ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
set_region(interface) set_region(node_state.interface)
interface.close() node_state.interface.close()
interface = initialize_interface(args) node_state.interface = initialize_interface(args)
stdscr.clear() stdscr.clear()
stdscr.refresh() stdscr.refresh()
settings_menu(stdscr, interface) settings_menu(stdscr, node_state)
except Exception as e: except Exception as e:
console_output = output_capture.getvalue() console_output = output_capture.getvalue()
@@ -52,6 +52,9 @@ logging.basicConfig( # Run `tail -f client.log` in another terminal to view live
) )
if __name__ == "__main__": if __name__ == "__main__":
from contact.ui.ui_state import NodeState
node_state = NodeState()
log_file = config.log_file_path log_file = config.log_file_path
log_f = open(log_file, "a", buffering=1) # Enable line-buffering for immediate log writes log_f = open(log_file, "a", buffering=1) # Enable line-buffering for immediate log writes

View File

@@ -13,7 +13,7 @@ import contact.ui.default_config as config
import contact.ui.dialog import contact.ui.dialog
import contact.globals as globals import contact.globals as globals
def handle_resize(stdscr, firstrun): def handle_resize(stdscr, firstrun, node_state):
global messages_pad, messages_win, nodes_pad, nodes_win, channel_pad, channel_win, function_win, packetlog_win, entry_win global messages_pad, messages_win, nodes_pad, nodes_win, channel_pad, channel_win, function_win, packetlog_win, entry_win
# Calculate window max dimensions # Calculate window max dimensions
@@ -80,17 +80,17 @@ def handle_resize(stdscr, firstrun):
curses.curs_set(1) curses.curs_set(1)
try: try:
draw_function_win() draw_function_win(node_state)
draw_channel_list() draw_channel_list(node_state)
draw_messages_window(True) draw_messages_window(node_state, True)
draw_node_list() draw_node_list(node_state)
except: except:
# Resize events can come faster than we can re-draw, which can cause a curses error. # Resize events can come faster than we can re-draw, which can cause a curses error.
# In this case we'll see another curses.KEY_RESIZE in our key handler and draw again later. # In this case we'll see another curses.KEY_RESIZE in our key handler and draw again later.
pass pass
def main_ui(stdscr): def main_ui(stdscr, node_state):
global input_text global input_text
input_text = "" input_text = ""
stdscr.keypad(True) stdscr.keypad(True)
@@ -111,7 +111,7 @@ def main_ui(stdscr):
elif globals.current_window == 1: elif globals.current_window == 1:
scroll_messages(-1) scroll_messages(-1)
elif globals.current_window == 2: elif globals.current_window == 2:
scroll_nodes(-1) scroll_nodes(-1, node_state)
elif char == curses.KEY_DOWN: elif char == curses.KEY_DOWN:
if globals.current_window == 0: if globals.current_window == 0:
@@ -119,11 +119,11 @@ def main_ui(stdscr):
elif globals.current_window == 1: elif globals.current_window == 1:
scroll_messages(1) scroll_messages(1)
elif globals.current_window == 2: elif globals.current_window == 2:
scroll_nodes(1) scroll_nodes(1, node_state)
elif char == curses.KEY_HOME: elif char == curses.KEY_HOME:
if globals.current_window == 0: if globals.current_window == 0:
select_channel(0) select_channel(0, node_state)
elif globals.current_window == 1: elif globals.current_window == 1:
globals.selected_message = 0 globals.selected_message = 0
refresh_pad(1) refresh_pad(1)
@@ -132,7 +132,7 @@ def main_ui(stdscr):
elif char == curses.KEY_END: elif char == curses.KEY_END:
if globals.current_window == 0: if globals.current_window == 0:
select_channel(len(globals.channel_list) - 1) select_channel(len(globals.channel_list) - 1, node_state)
elif globals.current_window == 1: elif globals.current_window == 1:
msg_line_count = messages_pad.getmaxyx()[0] msg_line_count = messages_pad.getmaxyx()[0]
globals.selected_message = max(msg_line_count - get_msg_window_lines(), 0) globals.selected_message = max(msg_line_count - get_msg_window_lines(), 0)
@@ -142,7 +142,7 @@ def main_ui(stdscr):
elif char == curses.KEY_PPAGE: elif char == curses.KEY_PPAGE:
if globals.current_window == 0: if globals.current_window == 0:
select_channel(globals.selected_channel - (channel_win.getmaxyx()[0] - 2)) # select_channel will bounds check for us select_channel(globals.selected_channel - (channel_win.getmaxyx()[0] - 2), node_state) # select_channel will bounds check for us
elif globals.current_window == 1: elif globals.current_window == 1:
globals.selected_message = max(globals.selected_message - get_msg_window_lines(), 0) globals.selected_message = max(globals.selected_message - get_msg_window_lines(), 0)
refresh_pad(1) refresh_pad(1)
@@ -151,7 +151,7 @@ def main_ui(stdscr):
elif char == curses.KEY_NPAGE: elif char == curses.KEY_NPAGE:
if globals.current_window == 0: if globals.current_window == 0:
select_channel(globals.selected_channel + (channel_win.getmaxyx()[0] - 2)) # select_channel will bounds check for us select_channel(globals.selected_channel + (channel_win.getmaxyx()[0] - 2), node_state) # select_channel will bounds check for us
elif globals.current_window == 1: elif globals.current_window == 1:
msg_line_count = messages_pad.getmaxyx()[0] msg_line_count = messages_pad.getmaxyx()[0]
globals.selected_message = min(globals.selected_message + get_msg_window_lines(), msg_line_count - get_msg_window_lines()) globals.selected_message = min(globals.selected_message + get_msg_window_lines(), msg_line_count - get_msg_window_lines())
@@ -169,7 +169,7 @@ def main_ui(stdscr):
channel_win.attrset(get_color("window_frame")) channel_win.attrset(get_color("window_frame"))
channel_win.box() channel_win.box()
channel_win.refresh() channel_win.refresh()
highlight_line(False, 0, globals.selected_channel) highlight_line(False, 0, globals.selected_channel, node_state)
refresh_pad(0) refresh_pad(0)
if old_window == 1: if old_window == 1:
messages_win.attrset(get_color("window_frame")) messages_win.attrset(get_color("window_frame"))
@@ -177,11 +177,11 @@ def main_ui(stdscr):
messages_win.refresh() messages_win.refresh()
refresh_pad(1) refresh_pad(1)
elif old_window == 2: elif old_window == 2:
draw_function_win() draw_function_win(node_state)
nodes_win.attrset(get_color("window_frame")) nodes_win.attrset(get_color("window_frame"))
nodes_win.box() nodes_win.box()
nodes_win.refresh() nodes_win.refresh()
highlight_line(False, 2, globals.selected_node) highlight_line(False, 2, globals.selected_node, node_state)
refresh_pad(2) refresh_pad(2)
if globals.current_window == 0: if globals.current_window == 0:
@@ -189,7 +189,7 @@ def main_ui(stdscr):
channel_win.box() channel_win.box()
channel_win.attrset(get_color("window_frame")) channel_win.attrset(get_color("window_frame"))
channel_win.refresh() channel_win.refresh()
highlight_line(True, 0, globals.selected_channel) highlight_line(True, 0, globals.selected_channel, node_state)
refresh_pad(0) refresh_pad(0)
elif globals.current_window == 1: elif globals.current_window == 1:
messages_win.attrset(get_color("window_frame_selected")) messages_win.attrset(get_color("window_frame_selected"))
@@ -198,12 +198,12 @@ def main_ui(stdscr):
messages_win.refresh() messages_win.refresh()
refresh_pad(1) refresh_pad(1)
elif globals.current_window == 2: elif globals.current_window == 2:
draw_function_win() draw_function_win(node_state)
nodes_win.attrset(get_color("window_frame_selected")) nodes_win.attrset(get_color("window_frame_selected"))
nodes_win.box() nodes_win.box()
nodes_win.attrset(get_color("window_frame")) nodes_win.attrset(get_color("window_frame"))
nodes_win.refresh() nodes_win.refresh()
highlight_line(True, 2, globals.selected_node) highlight_line(True, 2, globals.selected_node, node_state)
refresh_pad(2) refresh_pad(2)
# Check for Esc # Check for Esc
@@ -212,7 +212,7 @@ def main_ui(stdscr):
# Check for Ctrl + t # Check for Ctrl + t
elif char == chr(20): elif char == chr(20):
send_traceroute() send_traceroute(node_state)
curses.curs_set(0) # Hide cursor curses.curs_set(0) # Hide cursor
contact.ui.dialog.dialog(stdscr, "Traceroute Sent", "Results will appear in messages window.\nNote: Traceroute is limited to once every 30 seconds.") contact.ui.dialog.dialog(stdscr, "Traceroute Sent", "Results will appear in messages window.\nNote: Traceroute is limited to once every 30 seconds.")
curses.curs_set(1) # Show cursor again curses.curs_set(1) # Show cursor again
@@ -229,20 +229,20 @@ def main_ui(stdscr):
globals.selected_channel = globals.channel_list.index(node_list[globals.selected_node]) globals.selected_channel = globals.channel_list.index(node_list[globals.selected_node])
if(is_chat_archived(globals.channel_list[globals.selected_channel])): if(is_chat_archived(globals.channel_list[globals.selected_channel], node_state)):
update_node_info_in_db(globals.channel_list[globals.selected_channel], chat_archived=False) update_node_info_in_db(globals.channel_list[globals.selected_channel], chat_archived=False)
globals.selected_node = 0 globals.selected_node = 0
globals.current_window = 0 globals.current_window = 0
draw_node_list() draw_node_list(node_state)
draw_channel_list() draw_channel_list(node_state)
draw_messages_window(True) draw_messages_window(node_state, True)
elif len(input_text) > 0: elif len(input_text) > 0:
# Enter key pressed, send user input as message # Enter key pressed, send user input as message
send_message(input_text, channel=globals.selected_channel) send_message(input_text, node_state, channel=globals.selected_channel)
draw_messages_window(True) draw_messages_window(node_state, True)
# Clear entry window and reset input text # Clear entry window and reset input text
input_text = "" input_text = ""
@@ -259,7 +259,7 @@ def main_ui(stdscr):
elif char == "`": # ` Launch the settings interface elif char == "`": # ` Launch the settings interface
curses.curs_set(0) curses.curs_set(0)
settings_menu(stdscr, globals.interface) settings_menu(stdscr, node_state.interface)
curses.curs_set(1) curses.curs_set(1)
refresh_node_list() refresh_node_list()
handle_resize(stdscr, False) handle_resize(stdscr, False)
@@ -268,11 +268,11 @@ def main_ui(stdscr):
# Display packet log # Display packet log
if globals.display_log is False: if globals.display_log is False:
globals.display_log = True globals.display_log = True
draw_messages_window(True) draw_messages_window(node_state, True)
else: else:
globals.display_log = False globals.display_log = False
packetlog_win.erase() packetlog_win.erase()
draw_messages_window(True) draw_messages_window(node_state, True)
elif char == curses.KEY_RESIZE: elif char == curses.KEY_RESIZE:
input_text = "" input_text = ""
@@ -291,29 +291,29 @@ def main_ui(stdscr):
del globals.channel_list[globals.selected_channel] del globals.channel_list[globals.selected_channel]
globals.selected_channel = min(globals.selected_channel, len(globals.channel_list) - 1) globals.selected_channel = min(globals.selected_channel, len(globals.channel_list) - 1)
select_channel(globals.selected_channel) select_channel(globals.selected_channel, node_state)
draw_channel_list() draw_channel_list(node_state)
draw_messages_window() draw_messages_window(node_state)
if(globals.current_window == 2): if(globals.current_window == 2):
curses.curs_set(0) curses.curs_set(0)
confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node])} from nodedb?", "No", ["Yes", "No"]) confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node], node_state)} from nodedb?", "No", ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
globals.interface.localNode.removeNode(globals.node_list[globals.selected_node]) node_state.interface.localNode.removeNode(globals.node_list[globals.selected_node])
# Directly modifying the interface from client code - good? Bad? If it's stupid but it works, it's not supid? # Directly modifying the interface from client code - good? Bad? If it's stupid but it works, it's not supid?
del(globals.interface.nodesByNum[globals.node_list[globals.selected_node]]) del(node_state.interface.nodesByNum[globals.node_list[globals.selected_node]])
# Convert to "!hex" representation that interface.nodes uses # Convert to "!hex" representation that interface.nodes uses
hexid = f"!{hex(globals.node_list[globals.selected_node])[2:]}" hexid = f"!{hex(globals.node_list[globals.selected_node])[2:]}"
del(globals.interface.nodes[hexid]) del(node_state.interface.nodes[hexid])
globals.node_list.pop(globals.selected_node) globals.node_list.pop(globals.selected_node)
draw_messages_window() draw_messages_window(node_state)
draw_node_list() draw_node_list(node_state)
else: else:
draw_messages_window() draw_messages_window(node_state)
curses.curs_set(1) curses.curs_set(1)
continue continue
@@ -325,25 +325,25 @@ def main_ui(stdscr):
# ^F # ^F
elif char == chr(6): elif char == chr(6):
if globals.current_window == 2: if globals.current_window == 2:
selectedNode = globals.interface.nodesByNum[globals.node_list[globals.selected_node]] selectedNode = node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]
curses.curs_set(0) curses.curs_set(0)
if 'isFavorite' not in selectedNode or selectedNode['isFavorite'] == False: if 'isFavorite' not in selectedNode or selectedNode['isFavorite'] == False:
confirmation = get_list_input(f"Set {get_name_from_database(globals.node_list[globals.selected_node])} as Favorite?", None, ["Yes", "No"]) confirmation = get_list_input(f"Set {get_name_from_database(globals.node_list[globals.selected_node])} as Favorite?", node_state, None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
globals.interface.localNode.setFavorite(globals.node_list[globals.selected_node]) node_state.interface.localNode.setFavorite(globals.node_list[globals.selected_node])
# Maybe we shouldn't be modifying the nodedb, but maybe it should update itself # Maybe we shouldn't be modifying the nodedb, but maybe it should update itself
globals.interface.nodesByNum[globals.node_list[globals.selected_node]]['isFavorite'] = True node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]['isFavorite'] = True
refresh_node_list() refresh_node_list()
else: else:
confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node])} from Favorites?", None, ["Yes", "No"]) confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node])} from Favorites?", node_state, None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
globals.interface.localNode.removeFavorite(globals.node_list[globals.selected_node]) node_state.interface.localNode.removeFavorite(globals.node_list[globals.selected_node])
# Maybe we shouldn't be modifying the nodedb, but maybe it should update itself # Maybe we shouldn't be modifying the nodedb, but maybe it should update itself
globals.interface.nodesByNum[globals.node_list[globals.selected_node]]['isFavorite'] = False node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]['isFavorite'] = False
refresh_node_list() refresh_node_list()
@@ -351,20 +351,20 @@ def main_ui(stdscr):
elif char == chr(7): elif char == chr(7):
if globals.current_window == 2: if globals.current_window == 2:
selectedNode = globals.interface.nodesByNum[globals.node_list[globals.selected_node]] selectedNode = node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]
curses.curs_set(0) curses.curs_set(0)
if 'isIgnored' not in selectedNode or selectedNode['isIgnored'] == False: if 'isIgnored' not in selectedNode or selectedNode['isIgnored'] == False:
confirmation = get_list_input(f"Set {get_name_from_database(globals.node_list[globals.selected_node])} as Ignored?", "No", ["Yes", "No"]) confirmation = get_list_input(f"Set {get_name_from_database(globals.node_list[globals.selected_node])} as Ignored?", node_state, "No", ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
globals.interface.localNode.setIgnored(globals.node_list[globals.selected_node]) node_state.interface.localNode.setIgnored(globals.node_list[globals.selected_node])
globals.interface.nodesByNum[globals.node_list[globals.selected_node]]['isIgnored'] = True node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]['isIgnored'] = True
else: else:
confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node])} from Ignored?", "No", ["Yes", "No"]) confirmation = get_list_input(f"Remove {get_name_from_database(globals.node_list[globals.selected_node])} from Ignored?", node_state, "No", ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
globals.interface.localNode.removeIgnored(globals.node_list[globals.selected_node]) node_state.interface.localNode.removeIgnored(globals.node_list[globals.selected_node])
globals.interface.nodesByNum[globals.node_list[globals.selected_node]]['isIgnored'] = False node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]['isIgnored'] = False
handle_resize(stdscr, False) handle_resize(stdscr, False)
@@ -377,7 +377,7 @@ def main_ui(stdscr):
def draw_channel_list(): def draw_channel_list(node_state):
channel_pad.erase() channel_pad.erase()
win_height, win_width = channel_win.getmaxyx() win_height, win_width = channel_win.getmaxyx()
start_index = max(0, globals.selected_channel - (win_height - 3)) # Leave room for borders start_index = max(0, globals.selected_channel - (win_height - 3)) # Leave room for borders
@@ -388,9 +388,9 @@ def draw_channel_list():
for channel in globals.channel_list: for channel in globals.channel_list:
# Convert node number to long name if it's an integer # Convert node number to long name if it's an integer
if isinstance(channel, int): if isinstance(channel, int):
if is_chat_archived(channel): if is_chat_archived(channel, node_state):
continue continue
channel_name = get_name_from_database(channel, type='long') channel_name = get_name_from_database(channel, node_state, type='long')
if channel_name is None: if channel_name is None:
continue continue
channel = channel_name channel = channel_name
@@ -418,7 +418,7 @@ def draw_channel_list():
refresh_pad(0) refresh_pad(0)
def draw_messages_window(scroll_to_bottom = False): def draw_messages_window(node_state, scroll_to_bottom = False):
"""Update the messages window based on the selected channel and scroll position.""" """Update the messages window based on the selected channel and scroll position."""
messages_pad.erase() messages_pad.erase()
@@ -459,9 +459,9 @@ def draw_messages_window(scroll_to_bottom = False):
refresh_pad(1) refresh_pad(1)
draw_packetlog_win() draw_packetlog_win(node_state)
def draw_node_list(): def draw_node_list(node_state):
global nodes_pad global nodes_pad
# This didn't work, for some reason an error is thown on startup, so we just create the pad every time # This didn't work, for some reason an error is thown on startup, so we just create the pad every time
@@ -479,9 +479,9 @@ def draw_node_list():
logging.error("Traceback: %s", traceback.format_exc()) logging.error("Traceback: %s", traceback.format_exc())
for i, node_num in enumerate(globals.node_list): for i, node_num in enumerate(globals.node_list):
node = globals.interface.nodesByNum[node_num] node = node_state.interface.nodesByNum[node_num]
secure = 'user' in node and 'publicKey' in node['user'] and node['user']['publicKey'] secure = 'user' in node and 'publicKey' in node['user'] and node['user']['publicKey']
node_str = f"{'🔐' if secure else '🔓'} {get_name_from_database(node_num, 'long')}".ljust(box_width - 2)[:box_width - 2] node_str = f"{'🔐' if secure else '🔓'} {get_name_from_database(node_num, node_state, 'long')}".ljust(box_width - 2)[:box_width - 2]
color = "node_list" color = "node_list"
if 'isFavorite' in node and node['isFavorite']: if 'isFavorite' in node and node['isFavorite']:
color = "node_favorite" color = "node_favorite"
@@ -501,21 +501,21 @@ def draw_node_list():
curses.curs_set(1) curses.curs_set(1)
entry_win.refresh() entry_win.refresh()
def select_channel(idx): def select_channel(idx, node_state):
old_selected_channel = globals.selected_channel old_selected_channel = globals.selected_channel
globals.selected_channel = max(0, min(idx, len(globals.channel_list) - 1)) globals.selected_channel = max(0, min(idx, len(globals.channel_list) - 1))
draw_messages_window(True) draw_messages_window(node_state, True)
# For now just re-draw channel list when clearing notifications, we can probably make this more efficient # For now just re-draw channel list when clearing notifications, we can probably make this more efficient
if globals.selected_channel in globals.notifications: if globals.selected_channel in globals.notifications:
remove_notification(globals.selected_channel) remove_notification(globals.selected_channel)
draw_channel_list() draw_channel_list(node_state)
return return
highlight_line(False, 0, old_selected_channel) highlight_line(False, 0, old_selected_channel, node_state)
highlight_line(True, 0, globals.selected_channel) highlight_line(True, 0, globals.selected_channel, node_state)
refresh_pad(0) refresh_pad(0)
def scroll_channels(direction): def scroll_channels(direction, node_state):
new_selected_channel = globals.selected_channel + direction new_selected_channel = globals.selected_channel + direction
if new_selected_channel < 0: if new_selected_channel < 0:
@@ -523,7 +523,7 @@ def scroll_channels(direction):
elif new_selected_channel >= len(globals.channel_list): elif new_selected_channel >= len(globals.channel_list):
new_selected_channel = 0 new_selected_channel = 0
select_channel(new_selected_channel) select_channel(new_selected_channel, node_state)
def scroll_messages(direction): def scroll_messages(direction):
globals.selected_message += direction globals.selected_message += direction
@@ -533,17 +533,17 @@ def scroll_messages(direction):
refresh_pad(1) refresh_pad(1)
def select_node(idx): def select_node(idx, node_state):
old_selected_node = globals.selected_node old_selected_node = globals.selected_node
globals.selected_node = max(0, min(idx, len(globals.node_list) - 1)) globals.selected_node = max(0, min(idx, len(globals.node_list) - 1))
highlight_line(False, 2, old_selected_node) highlight_line(False, 2, old_selected_node, node_state)
highlight_line(True, 2, globals.selected_node) highlight_line(True, 2, globals.selected_node, node_state)
refresh_pad(2) refresh_pad(2)
draw_function_win() draw_function_win(node_state)
def scroll_nodes(direction): def scroll_nodes(direction, node_state):
new_selected_node = globals.selected_node + direction new_selected_node = globals.selected_node + direction
if new_selected_node < 0: if new_selected_node < 0:
@@ -551,9 +551,9 @@ def scroll_nodes(direction):
elif new_selected_node >= len(globals.node_list): elif new_selected_node >= len(globals.node_list):
new_selected_node = 0 new_selected_node = 0
select_node(new_selected_node) select_node(new_selected_node, node_state)
def draw_packetlog_win(): def draw_packetlog_win(node_state):
columns = [10,10,15,30] columns = [10,10,15,30]
span = 0 span = 0
@@ -574,10 +574,10 @@ def draw_packetlog_win():
break break
# Format each field # Format each field
from_id = get_name_from_database(packet['from'], 'short').ljust(columns[0]) from_id = get_name_from_database(packet['from'], node_state, 'short').ljust(columns[0])
to_id = ( to_id = (
"BROADCAST".ljust(columns[1]) if str(packet['to']) == "4294967295" "BROADCAST".ljust(columns[1]) if str(packet['to']) == "4294967295"
else get_name_from_database(packet['to'], 'short').ljust(columns[1]) else get_name_from_database(packet['to'], node_state, 'short').ljust(columns[1])
) )
if 'decoded' in packet: if 'decoded' in packet:
port = packet['decoded']['portnum'].ljust(columns[2]) port = packet['decoded']['portnum'].ljust(columns[2])
@@ -645,10 +645,10 @@ def search(win):
entry_win.erase() entry_win.erase()
def draw_node_details(): def draw_node_details(node_state):
node = None node = None
try: try:
node = globals.interface.nodesByNum[globals.node_list[globals.selected_node]] node = node_state.interface.nodesByNum[globals.node_list[globals.selected_node]]
except KeyError: except KeyError:
return return
@@ -702,9 +702,9 @@ def draw_help():
draw_centered_text_field(function_win, function_str, 0, get_color("commands")) draw_centered_text_field(function_win, function_str, 0, get_color("commands"))
def draw_function_win(): def draw_function_win(node_state):
if(globals.current_window == 2): if(globals.current_window == 2):
draw_node_details() draw_node_details(node_state)
else: else:
draw_help() draw_help()
@@ -746,7 +746,7 @@ def refresh_pad(window):
box.getbegyx()[0] + 1, box.getbegyx()[1] + 1, box.getbegyx()[0] + 1, box.getbegyx()[1] + 1,
box.getbegyx()[0] + lines, box.getbegyx()[1] + box.getmaxyx()[1] - 2) box.getbegyx()[0] + lines, box.getbegyx()[1] + box.getmaxyx()[1] - 2)
def highlight_line(highlight, window, line): def highlight_line(highlight, window, line, node_state):
pad = nodes_pad pad = nodes_pad
color = get_color("node_list") color = get_color("node_list")
@@ -754,7 +754,7 @@ def highlight_line(highlight, window, line):
if window == 2: if window == 2:
node_num = globals.node_list[line] node_num = globals.node_list[line]
node = globals.interface.nodesByNum[node_num] node = node_state.interface.nodesByNum[node_num]
if 'isFavorite' in node and node['isFavorite']: if 'isFavorite' in node and node['isFavorite']:
color = get_color("node_favorite") color = get_color("node_favorite")
if 'isIgnored' in node and node['isIgnored']: if 'isIgnored' in node and node['isIgnored']:

View File

@@ -15,7 +15,7 @@ from contact.utilities.control_utils import parse_ini_file, transform_menu_path
from contact.ui.user_config import json_editor from contact.ui.user_config import json_editor
from contact.ui.ui_state import MenuState from contact.ui.ui_state import MenuState
state = MenuState() menu_state = MenuState()
# Constants # Constants
@@ -39,10 +39,10 @@ config_folder = os.path.join(locals_dir, "node-configs")
field_mapping, help_text = parse_ini_file(translation_file) field_mapping, help_text = parse_ini_file(translation_file)
def display_menu(state): def display_menu(menu_state):
min_help_window_height = 6 min_help_window_height = 6
num_items = len(state.current_menu) + (1 if state.show_save_option else 0) num_items = len(menu_state.current_menu) + (1 if menu_state.show_save_option else 0)
# Determine the available height for the menu # Determine the available height for the menu
max_menu_height = curses.LINES max_menu_height = curses.LINES
@@ -62,18 +62,18 @@ def display_menu(state):
menu_win.border() menu_win.border()
menu_win.keypad(True) menu_win.keypad(True)
menu_pad = curses.newpad(len(state.current_menu) + 1, width - 8) menu_pad = curses.newpad(len(menu_state.current_menu) + 1, width - 8)
menu_pad.bkgd(get_color("background")) menu_pad.bkgd(get_color("background"))
header = " > ".join(word.title() for word in state.menu_path) header = " > ".join(word.title() for word in menu_state.menu_path)
if len(header) > width - 4: if len(header) > width - 4:
header = header[:width - 7] + "..." header = header[:width - 7] + "..."
menu_win.addstr(1, 2, header, get_color("settings_breadcrumbs", bold=True)) menu_win.addstr(1, 2, header, get_color("settings_breadcrumbs", bold=True))
transformed_path = transform_menu_path(state.menu_path) transformed_path = transform_menu_path(menu_state.menu_path)
for idx, option in enumerate(state.current_menu): for idx, option in enumerate(menu_state.current_menu):
field_info = state.current_menu[option] field_info = menu_state.current_menu[option]
current_value = field_info[1] if isinstance(field_info, tuple) else "" current_value = field_info[1] if isinstance(field_info, tuple) else ""
full_key = '.'.join(transformed_path + [option]) full_key = '.'.join(transformed_path + [option])
display_name = field_mapping.get(full_key, option) display_name = field_mapping.get(full_key, option)
@@ -82,41 +82,41 @@ def display_menu(state):
display_value = f"{current_value}"[:width // 2 - 4] display_value = f"{current_value}"[:width // 2 - 4]
try: try:
color = get_color("settings_sensitive" if option in sensitive_settings else "settings_default", reverse=(idx == state.selected_index)) color = get_color("settings_sensitive" if option in sensitive_settings else "settings_default", reverse=(idx == menu_state.selected_index))
menu_pad.addstr(idx, 0, f"{display_option:<{width // 2 - 2}} {display_value}".ljust(width - 8), color) menu_pad.addstr(idx, 0, f"{display_option:<{width // 2 - 2}} {display_value}".ljust(width - 8), color)
except curses.error: except curses.error:
pass pass
if state.show_save_option: if menu_state.show_save_option:
save_position = menu_height - 2 save_position = menu_height - 2
menu_win.addstr(save_position, (width - len(save_option)) // 2, save_option, get_color("settings_save", reverse=(state.selected_index == len(state.current_menu)))) menu_win.addstr(save_position, (width - len(save_option)) // 2, save_option, get_color("settings_save", reverse=(menu_state.selected_index == len(menu_state.current_menu))))
# Draw help window with dynamically updated max_help_lines # Draw help window with dynamically updated max_help_lines
draw_help_window(start_y, start_x, menu_height, max_help_lines, transformed_path, state) draw_help_window(start_y, start_x, menu_height, max_help_lines, transformed_path, menu_state)
menu_win.refresh() menu_win.refresh()
menu_pad.refresh( menu_pad.refresh(
state.start_index[-1], 0, menu_state.start_index[-1], 0,
menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4, menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4,
menu_win.getbegyx()[0] + 3 + menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0), menu_win.getbegyx()[0] + 3 + menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0),
menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4 menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4
) )
max_index = num_items + (1 if state.show_save_option else 0) - 1 max_index = num_items + (1 if menu_state.show_save_option else 0) - 1
visible_height = menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0) visible_height = menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0)
draw_arrows(menu_win, visible_height, max_index, state) draw_arrows(menu_win, visible_height, max_index, menu_state)
return menu_win, menu_pad return menu_win, menu_pad
def draw_help_window(menu_start_y, menu_start_x, menu_height, max_help_lines, transformed_path, state): def draw_help_window(menu_start_y, menu_start_x, menu_height, max_help_lines, transformed_path, menu_state):
global help_win global help_win
if 'help_win' not in globals(): if 'help_win' not in globals():
help_win = None # Initialize if it does not exist help_win = None # Initialize if it does not exist
selected_option = list(state.current_menu.keys())[state.selected_index] if state.current_menu else None selected_option = list(menu_state.current_menu.keys())[menu_state.selected_index] if menu_state.current_menu else None
help_y = menu_start_y + menu_height help_y = menu_start_y + menu_height
help_win = update_help_window(help_win, help_text, transformed_path, selected_option, max_help_lines, width, help_y, menu_start_x) help_win = update_help_window(help_win, help_text, transformed_path, selected_option, max_help_lines, width, help_y, menu_start_x)
@@ -251,116 +251,116 @@ def get_wrapped_help_text(help_text, transformed_path, selected_option, width, m
return wrapped_help return wrapped_help
def move_highlight(old_idx, options, menu_win, menu_pad, help_win, help_text, max_help_lines, state): def move_highlight(old_idx, options, menu_win, menu_pad, help_win, help_text, max_help_lines, menu_state):
if old_idx == state.selected_index: # No-op if old_idx == menu_state.selected_index: # No-op
return return
max_index = len(options) + (1 if state.show_save_option else 0) - 1 max_index = len(options) + (1 if menu_state.show_save_option else 0) - 1
visible_height = menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0) visible_height = menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0)
# Adjust state.start_index only when moving out of visible range # Adjust menu_state.start_index only when moving out of visible range
if state.selected_index == max_index and state.show_save_option: if menu_state.selected_index == max_index and menu_state.show_save_option:
pass pass
elif state.selected_index < state.start_index[-1]: # Moving above the visible area elif menu_state.selected_index < menu_state.start_index[-1]: # Moving above the visible area
state.start_index[-1] = state.selected_index menu_state.start_index[-1] = menu_state.selected_index
elif state.selected_index >= state.start_index[-1] + visible_height: # Moving below the visible area elif menu_state.selected_index >= menu_state.start_index[-1] + visible_height: # Moving below the visible area
state.start_index[-1] = state.selected_index - visible_height menu_state.start_index[-1] = menu_state.selected_index - visible_height
pass pass
# Ensure state.start_index is within bounds # Ensure menu_state.start_index is within bounds
state.start_index[-1] = max(0, min(state.start_index[-1], max_index - visible_height + 1)) menu_state.start_index[-1] = max(0, min(menu_state.start_index[-1], max_index - visible_height + 1))
# Clear old selection # Clear old selection
if state.show_save_option and old_idx == max_index: if menu_state.show_save_option and old_idx == max_index:
menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save")) menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save"))
else: else:
menu_pad.chgat(old_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive") if options[old_idx] in sensitive_settings else get_color("settings_default")) menu_pad.chgat(old_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive") if options[old_idx] in sensitive_settings else get_color("settings_default"))
# Highlight new selection # Highlight new selection
if state.show_save_option and state.selected_index == max_index: if menu_state.show_save_option and menu_state.selected_index == max_index:
menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save", reverse=True)) menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save", reverse=True))
else: else:
menu_pad.chgat(state.selected_index, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive", reverse=True) if options[state.selected_index] in sensitive_settings else get_color("settings_default", reverse=True)) menu_pad.chgat(menu_state.selected_index, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive", reverse=True) if options[menu_state.selected_index] in sensitive_settings else get_color("settings_default", reverse=True))
menu_win.refresh() menu_win.refresh()
# Refresh pad only if scrolling is needed # Refresh pad only if scrolling is needed
menu_pad.refresh(state.start_index[-1], 0, menu_pad.refresh(menu_state.start_index[-1], 0,
menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4, menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4,
menu_win.getbegyx()[0] + 3 + visible_height, menu_win.getbegyx()[0] + 3 + visible_height,
menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4) menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4)
# Update help window # Update help window
transformed_path = transform_menu_path(state.menu_path) transformed_path = transform_menu_path(menu_state.menu_path)
selected_option = options[state.selected_index] if state.selected_index < len(options) else None selected_option = options[menu_state.selected_index] if menu_state.selected_index < len(options) else None
help_y = menu_win.getbegyx()[0] + menu_win.getmaxyx()[0] help_y = menu_win.getbegyx()[0] + menu_win.getmaxyx()[0]
help_win = update_help_window(help_win, help_text, transformed_path, selected_option, max_help_lines, width, help_y, menu_win.getbegyx()[1]) help_win = update_help_window(help_win, help_text, transformed_path, selected_option, max_help_lines, width, help_y, menu_win.getbegyx()[1])
draw_arrows(menu_win, visible_height, max_index, state) draw_arrows(menu_win, visible_height, max_index, menu_state)
def draw_arrows(win, visible_height, max_index, state): def draw_arrows(win, visible_height, max_index, menu_state):
# vh = visible_height + (1 if show_save_option else 0) # vh = visible_height + (1 if show_save_option else 0)
mi = max_index - (2 if state.show_save_option else 0) mi = max_index - (2 if menu_state.show_save_option else 0)
if visible_height < mi: if visible_height < mi:
if state.start_index[-1] > 0: if menu_state.start_index[-1] > 0:
win.addstr(3, 2, "", get_color("settings_default")) win.addstr(3, 2, "", get_color("settings_default"))
else: else:
win.addstr(3, 2, " ", get_color("settings_default")) win.addstr(3, 2, " ", get_color("settings_default"))
if mi - state.start_index[-1] >= visible_height + (0 if state.show_save_option else 1) : if mi - menu_state.start_index[-1] >= visible_height + (0 if menu_state.show_save_option else 1) :
win.addstr(visible_height + 3, 2, "", get_color("settings_default")) win.addstr(visible_height + 3, 2, "", get_color("settings_default"))
else: else:
win.addstr(visible_height + 3, 2, " ", get_color("settings_default")) win.addstr(visible_height + 3, 2, " ", get_color("settings_default"))
def settings_menu(stdscr, interface): def settings_menu(stdscr, node_state):
curses.update_lines_cols() curses.update_lines_cols()
menu = generate_menu_from_protobuf(interface) menu = generate_menu_from_protobuf(node_state)
state.current_menu = menu["Main Menu"] menu_state.current_menu = menu["Main Menu"]
state.menu_path = ["Main Menu"] menu_state.menu_path = ["Main Menu"]
modified_settings = {} modified_settings = {}
need_redraw = True need_redraw = True
state.show_save_option = False menu_state.show_save_option = False
while True: while True:
if(need_redraw): if(need_redraw):
options = list(state.current_menu.keys()) options = list(menu_state.current_menu.keys())
state.show_save_option = ( menu_state.show_save_option = (
len(state.menu_path) > 2 and ("Radio Settings" in state.menu_path or "Module Settings" in state.menu_path) len(menu_state.menu_path) > 2 and ("Radio Settings" in menu_state.menu_path or "Module Settings" in menu_state.menu_path)
) or ( ) or (
len(state.menu_path) == 2 and "User Settings" in state.menu_path len(menu_state.menu_path) == 2 and "User Settings" in menu_state.menu_path
) or ( ) or (
len(state.menu_path) == 3 and "Channels" in state.menu_path len(menu_state.menu_path) == 3 and "Channels" in menu_state.menu_path
) )
# Display the menu # Display the menu
menu_win, menu_pad = display_menu(state) menu_win, menu_pad = display_menu(menu_state)
need_redraw = False need_redraw = False
# Capture user input # Capture user input
key = menu_win.getch() key = menu_win.getch()
max_index = len(options) + (1 if state.show_save_option else 0) - 1 max_index = len(options) + (1 if menu_state.show_save_option else 0) - 1
# max_help_lines = 4 # max_help_lines = 4
if key == curses.KEY_UP: if key == curses.KEY_UP:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = max_index if state.selected_index == 0 else state.selected_index - 1 menu_state.selected_index = max_index if menu_state.selected_index == 0 else menu_state.selected_index - 1
move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, state) move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, menu_state)
elif key == curses.KEY_DOWN: elif key == curses.KEY_DOWN:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = 0 if state.selected_index == max_index else state.selected_index + 1 menu_state.selected_index = 0 if menu_state.selected_index == max_index else menu_state.selected_index + 1
move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, state) move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, menu_state)
elif key == curses.KEY_RESIZE: elif key == curses.KEY_RESIZE:
need_redraw = True need_redraw = True
@@ -372,36 +372,36 @@ def settings_menu(stdscr, interface):
menu_win.refresh() menu_win.refresh()
help_win.refresh() help_win.refresh()
elif key == ord("\t") and state.show_save_option: elif key == ord("\t") and menu_state.show_save_option:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = max_index menu_state.selected_index = max_index
move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, state) move_highlight(old_selected_index, options, menu_win, menu_pad, help_win, help_text, max_help_lines, menu_state)
elif key == curses.KEY_RIGHT or key == ord('\n'): elif key == curses.KEY_RIGHT or key == ord('\n'):
need_redraw = True need_redraw = True
state.start_index.append(0) menu_state.start_index.append(0)
menu_win.erase() menu_win.erase()
help_win.erase() help_win.erase()
# draw_help_window(menu_win.getbegyx()[0], menu_win.getbegyx()[1], menu_win.getmaxyx()[0], max_help_lines, state.current_menu, selected_index, transform_menu_path(state.menu_path)) # draw_help_window(menu_win.getbegyx()[0], menu_win.getbegyx()[1], menu_win.getmaxyx()[0], max_help_lines, menu_state.current_menu, selected_index, transform_menu_path(menu_state.menu_path))
menu_win.refresh() menu_win.refresh()
help_win.refresh() help_win.refresh()
if state.show_save_option and state.selected_index == len(options): if menu_state.show_save_option and menu_state.selected_index == len(options):
save_changes(interface, modified_settings, state) save_changes(node_state, modified_settings, menu_state)
modified_settings.clear() modified_settings.clear()
logging.info("Changes Saved") logging.info("Changes Saved")
if len(state.menu_path) > 1: if len(menu_state.menu_path) > 1:
state.menu_path.pop() menu_state.menu_path.pop()
state.current_menu = menu["Main Menu"] menu_state.current_menu = menu["Main Menu"]
for step in state.menu_path[1:]: for step in menu_state.menu_path[1:]:
state.current_menu = state.current_menu.get(step, {}) menu_state.current_menu = menu_state.current_menu.get(step, {})
state.selected_index = 0 menu_state.selected_index = 0
continue continue
selected_option = options[state.selected_index] selected_option = options[menu_state.selected_index]
if selected_option == "Exit": if selected_option == "Exit":
break break
@@ -410,27 +410,27 @@ def settings_menu(stdscr, interface):
filename = get_text_input("Enter a filename for the config file") filename = get_text_input("Enter a filename for the config file")
if not filename: if not filename:
logging.info("Export aborted: No filename provided.") logging.info("Export aborted: No filename provided.")
state.start_index.pop() menu_state.start_index.pop()
continue # Go back to the menu continue # Go back to the menu
if not filename.lower().endswith(".yaml"): if not filename.lower().endswith(".yaml"):
filename += ".yaml" filename += ".yaml"
try: try:
config_text = config_export(interface) config_text = config_export(node_state)
yaml_file_path = os.path.join(config_folder, filename) yaml_file_path = os.path.join(config_folder, filename)
if os.path.exists(yaml_file_path): if os.path.exists(yaml_file_path):
overwrite = get_list_input(f"{filename} already exists. Overwrite?", None, ["Yes", "No"]) overwrite = get_list_input(f"{filename} already exists. Overwrite?", None, ["Yes", "No"])
if overwrite == "No": if overwrite == "No":
logging.info("Export cancelled: User chose not to overwrite.") logging.info("Export cancelled: User chose not to overwrite.")
state.start_index.pop() menu_state.start_index.pop()
continue # Return to menu continue # Return to menu
os.makedirs(os.path.dirname(yaml_file_path), exist_ok=True) os.makedirs(os.path.dirname(yaml_file_path), exist_ok=True)
with open(yaml_file_path, "w", encoding="utf-8") as file: with open(yaml_file_path, "w", encoding="utf-8") as file:
file.write(config_text) file.write(config_text)
logging.info(f"Config file saved to {yaml_file_path}") logging.info(f"Config file saved to {yaml_file_path}")
dialog(stdscr, "Config File Saved:", yaml_file_path) dialog(stdscr, "Config File Saved:", yaml_file_path)
state.start_index.pop() menu_state.start_index.pop()
continue continue
except PermissionError: except PermissionError:
logging.error(f"Permission denied: Unable to write to {yaml_file_path}") logging.error(f"Permission denied: Unable to write to {yaml_file_path}")
@@ -438,7 +438,7 @@ def settings_menu(stdscr, interface):
logging.error(f"OS error while saving config: {e}") logging.error(f"OS error while saving config: {e}")
except Exception as e: except Exception as e:
logging.error(f"Unexpected error: {e}") logging.error(f"Unexpected error: {e}")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Load Config File": elif selected_option == "Load Config File":
@@ -460,73 +460,73 @@ def settings_menu(stdscr, interface):
file_path = os.path.join(config_folder, filename) file_path = os.path.join(config_folder, filename)
overwrite = get_list_input(f"Are you sure you want to load {filename}?", None, ["Yes", "No"]) overwrite = get_list_input(f"Are you sure you want to load {filename}?", None, ["Yes", "No"])
if overwrite == "Yes": if overwrite == "Yes":
config_import(interface, file_path) config_import(node_state, file_path)
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Config URL": elif selected_option == "Config URL":
current_value = interface.localNode.getURL() current_value = node_state.interface.localNode.getURL()
new_value = get_text_input(f"Config URL is currently: {current_value}") new_value = get_text_input(f"Config URL is currently: {current_value}")
if new_value is not None: if new_value is not None:
current_value = new_value current_value = new_value
overwrite = get_list_input(f"Are you sure you want to load this config?", None, ["Yes", "No"]) overwrite = get_list_input(f"Are you sure you want to load this config?", None, ["Yes", "No"])
if overwrite == "Yes": if overwrite == "Yes":
interface.localNode.setURL(new_value) node_state.interface.localNode.setURL(new_value)
logging.info(f"New Config URL sent to node") logging.info(f"New Config URL sent to node")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Reboot": elif selected_option == "Reboot":
confirmation = get_list_input("Are you sure you want to Reboot?", None, ["Yes", "No"]) confirmation = get_list_input("Are you sure you want to Reboot?", None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
interface.localNode.reboot() node_state.interface.localNode.reboot()
logging.info(f"Node Reboot Requested by menu") logging.info(f"Node Reboot Requested by menu")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Reset Node DB": elif selected_option == "Reset Node DB":
confirmation = get_list_input("Are you sure you want to Reset Node DB?", None, ["Yes", "No"]) confirmation = get_list_input("Are you sure you want to Reset Node DB?", None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
interface.localNode.resetNodeDb() node_state.interface.localNode.resetNodeDb()
logging.info(f"Node DB Reset Requested by menu") logging.info(f"Node DB Reset Requested by menu")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Shutdown": elif selected_option == "Shutdown":
confirmation = get_list_input("Are you sure you want to Shutdown?", None, ["Yes", "No"]) confirmation = get_list_input("Are you sure you want to Shutdown?", None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
interface.localNode.shutdown() node_state.interface.localNode.shutdown()
logging.info(f"Node Shutdown Requested by menu") logging.info(f"Node Shutdown Requested by menu")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "Factory Reset": elif selected_option == "Factory Reset":
confirmation = get_list_input("Are you sure you want to Factory Reset?", None, ["Yes", "No"]) confirmation = get_list_input("Are you sure you want to Factory Reset?", None, ["Yes", "No"])
if confirmation == "Yes": if confirmation == "Yes":
interface.localNode.factoryReset() node_state.interface.localNode.factoryReset()
logging.info(f"Factory Reset Requested by menu") logging.info(f"Factory Reset Requested by menu")
state.start_index.pop() menu_state.start_index.pop()
continue continue
elif selected_option == "App Settings": elif selected_option == "App Settings":
menu_win.clear() menu_win.clear()
menu_win.refresh() menu_win.refresh()
state.menu_path.append("App Settings") menu_state.menu_path.append("App Settings")
state.menu_index.append(state.selected_index) menu_state.menu_index.append(menu_state.selected_index)
json_editor(stdscr, state) # Open the App Settings menu json_editor(stdscr, menu_state) # Open the App Settings menu
state.current_menu = menu["Main Menu"] menu_state.current_menu = menu["Main Menu"]
state.menu_path = ["Main Menu"] menu_state.menu_path = ["Main Menu"]
state.start_index.pop() menu_state.start_index.pop()
state.selected_index = 4 menu_state.selected_index = 4
continue continue
# need_redraw = True # need_redraw = True
field_info = state.current_menu.get(selected_option) field_info = menu_state.current_menu.get(selected_option)
if isinstance(field_info, tuple): if isinstance(field_info, tuple):
field, current_value = field_info field, current_value = field_info
# Transform the menu path to get the full key # Transform the menu path to get the full key
transformed_path = transform_menu_path(state.menu_path) transformed_path = transform_menu_path(menu_state.menu_path)
full_key = '.'.join(transformed_path + [selected_option]) full_key = '.'.join(transformed_path + [selected_option])
# Fetch human-readable name from field_mapping # Fetch human-readable name from field_mapping
@@ -536,70 +536,70 @@ def settings_menu(stdscr, interface):
if selected_option in ['longName', 'shortName']: if selected_option in ['longName', 'shortName']:
new_value = get_text_input(f"{human_readable_name} is currently: {current_value}") new_value = get_text_input(f"{human_readable_name} is currently: {current_value}")
new_value = current_value if new_value is None else new_value new_value = current_value if new_value is None else new_value
state.current_menu[selected_option] = (field, new_value) menu_state.current_menu[selected_option] = (field, new_value)
elif selected_option == 'isLicensed': elif selected_option == 'isLicensed':
new_value = get_list_input(f"{human_readable_name} is currently: {current_value}", str(current_value), ["True", "False"]) new_value = get_list_input(f"{human_readable_name} is currently: {current_value}", str(current_value), ["True", "False"])
new_value = new_value == "True" new_value = new_value == "True"
state.current_menu[selected_option] = (field, new_value) menu_state.current_menu[selected_option] = (field, new_value)
for option, (field, value) in state.current_menu.items(): for option, (field, value) in menu_state.current_menu.items():
modified_settings[option] = value modified_settings[option] = value
state.start_index.pop() menu_state.start_index.pop()
elif selected_option in ['latitude', 'longitude', 'altitude']: elif selected_option in ['latitude', 'longitude', 'altitude']:
new_value = get_text_input(f"{human_readable_name} is currently: {current_value}") new_value = get_text_input(f"{human_readable_name} is currently: {current_value}")
new_value = current_value if new_value is None else new_value new_value = current_value if new_value is None else new_value
state.current_menu[selected_option] = (field, new_value) menu_state.current_menu[selected_option] = (field, new_value)
for option in ['latitude', 'longitude', 'altitude']: for option in ['latitude', 'longitude', 'altitude']:
if option in state.current_menu: if option in menu_state.current_menu:
modified_settings[option] = state.current_menu[option][1] modified_settings[option] = menu_state.current_menu[option][1]
state.start_index.pop() menu_state.start_index.pop()
elif selected_option == "admin_key": elif selected_option == "admin_key":
new_values = get_admin_key_input(current_value) new_values = get_admin_key_input(current_value)
new_value = current_value if new_values is None else [base64.b64decode(key) for key in new_values] new_value = current_value if new_values is None else [base64.b64decode(key) for key in new_values]
state.start_index.pop() menu_state.start_index.pop()
elif field.type == 8: # Handle boolean type elif field.type == 8: # Handle boolean type
new_value = get_list_input(human_readable_name, str(current_value), ["True", "False"]) new_value = get_list_input(human_readable_name, str(current_value), ["True", "False"])
new_value = new_value == "True" or new_value is True new_value = new_value == "True" or new_value is True
state.start_index.pop() menu_state.start_index.pop()
elif field.label == field.LABEL_REPEATED: # Handle repeated field - Not currently used elif field.label == field.LABEL_REPEATED: # Handle repeated field - Not currently used
new_value = get_repeated_input(current_value) new_value = get_repeated_input(current_value)
new_value = current_value if new_value is None else new_value.split(", ") new_value = current_value if new_value is None else new_value.split(", ")
state.start_index.pop() menu_state.start_index.pop()
elif field.enum_type: # Enum field elif field.enum_type: # Enum field
enum_options = {v.name: v.number for v in field.enum_type.values} enum_options = {v.name: v.number for v in field.enum_type.values}
new_value_name = get_list_input(human_readable_name, current_value, list(enum_options.keys())) new_value_name = get_list_input(human_readable_name, current_value, list(enum_options.keys()))
new_value = enum_options.get(new_value_name, current_value) new_value = enum_options.get(new_value_name, current_value)
state.start_index.pop() menu_state.start_index.pop()
elif field.type == 7: # Field type 7 corresponds to FIXED32 elif field.type == 7: # Field type 7 corresponds to FIXED32
new_value = get_fixed32_input(current_value) new_value = get_fixed32_input(current_value)
state.start_index.pop() menu_state.start_index.pop()
elif field.type == 13: # Field type 13 corresponds to UINT32 elif field.type == 13: # Field type 13 corresponds to UINT32
new_value = get_text_input(f"{human_readable_name} is currently: {current_value}") new_value = get_text_input(f"{human_readable_name} is currently: {current_value}")
new_value = current_value if new_value is None else int(new_value) new_value = current_value if new_value is None else int(new_value)
state.start_index.pop() menu_state.start_index.pop()
elif field.type == 2: # Field type 13 corresponds to INT64 elif field.type == 2: # Field type 13 corresponds to INT64
new_value = get_text_input(f"{human_readable_name} is currently: {current_value}") new_value = get_text_input(f"{human_readable_name} is currently: {current_value}")
new_value = current_value if new_value is None else float(new_value) new_value = current_value if new_value is None else float(new_value)
state.start_index.pop() menu_state.start_index.pop()
else: # Handle other field types else: # Handle other field types
new_value = get_text_input(f"{human_readable_name} is currently: {current_value}") new_value = get_text_input(f"{human_readable_name} is currently: {current_value}")
new_value = current_value if new_value is None else new_value new_value = current_value if new_value is None else new_value
state.start_index.pop() menu_state.start_index.pop()
for key in state.menu_path[3:]: # Skip "Main Menu" for key in menu_state.menu_path[3:]: # Skip "Main Menu"
modified_settings = modified_settings.setdefault(key, {}) modified_settings = modified_settings.setdefault(key, {})
# Add the new value to the appropriate level # Add the new value to the appropriate level
@@ -610,12 +610,12 @@ def settings_menu(stdscr, interface):
enum_value_descriptor = field.enum_type.values_by_number.get(new_value) enum_value_descriptor = field.enum_type.values_by_number.get(new_value)
new_value = enum_value_descriptor.name if enum_value_descriptor else new_value new_value = enum_value_descriptor.name if enum_value_descriptor else new_value
state.current_menu[selected_option] = (field, new_value) menu_state.current_menu[selected_option] = (field, new_value)
else: else:
state.current_menu = state.current_menu[selected_option] menu_state.current_menu = menu_state.current_menu[selected_option]
state.menu_path.append(selected_option) menu_state.menu_path.append(selected_option)
state.menu_index.append(state.selected_index) menu_state.menu_index.append(menu_state.selected_index)
state.selected_index = 0 menu_state.selected_index = 0
elif key == curses.KEY_LEFT: elif key == curses.KEY_LEFT:
@@ -625,30 +625,30 @@ def settings_menu(stdscr, interface):
help_win.erase() help_win.erase()
# max_help_lines = 4 # max_help_lines = 4
# draw_help_window(menu_win.getbegyx()[0], menu_win.getbegyx()[1], menu_win.getmaxyx()[0], max_help_lines, state.current_menu, selected_index, transform_menu_path(state.menu_path)) # draw_help_window(menu_win.getbegyx()[0], menu_win.getbegyx()[1], menu_win.getmaxyx()[0], max_help_lines, menu_state.current_menu, selected_index, transform_menu_path(menu_state.menu_path))
menu_win.refresh() menu_win.refresh()
help_win.refresh() help_win.refresh()
if len(state.menu_path) < 2: if len(menu_state.menu_path) < 2:
modified_settings.clear() modified_settings.clear()
# Navigate back to the previous menu # Navigate back to the previous menu
if len(state.menu_path) > 1: if len(menu_state.menu_path) > 1:
state.menu_path.pop() menu_state.menu_path.pop()
state.current_menu = menu["Main Menu"] menu_state.current_menu = menu["Main Menu"]
for step in state.menu_path[1:]: for step in menu_state.menu_path[1:]:
state.current_menu = state.current_menu.get(step, {}) menu_state.current_menu = menu_state.current_menu.get(step, {})
state.selected_index = state.menu_index.pop() menu_state.selected_index = menu_state.menu_index.pop()
state.start_index.pop() menu_state.start_index.pop()
elif key == 27: # Escape key elif key == 27: # Escape key
menu_win.erase() menu_win.erase()
menu_win.refresh() menu_win.refresh()
break break
def set_region(interface): def set_region(node_state):
node = interface.getNode('^local') node = node_state.interface.getNode('^local')
device_config = node.localConfig device_config = node.localConfig
lora_descriptor = device_config.lora.DESCRIPTOR lora_descriptor = device_config.lora.DESCRIPTOR

View File

@@ -5,4 +5,23 @@ class MenuState:
self.selected_index = 0 # Selected Row self.selected_index = 0 # Selected Row
self.current_menu = {} # Contents of the current menu self.current_menu = {} # Contents of the current menu
self.menu_path = [] # Menu Path self.menu_path = [] # Menu Path
self.show_save_option = False self.show_save_option = False
class NodeState:
def __init__(self):
self.interface = None
self.myNodeNum = 0
# self.lock = None
# self.display_log = False
# self.all_messages = {}
# self.channel_list = []
# self.notifications = []
# self.packet_buffer = []
# self.node_list = []
# self.selected_channel = 0
# self.selected_message = 0
# self.selected_node = 0
# self.current_window = 0

View File

@@ -20,7 +20,7 @@ def edit_color_pair(key, current_value):
return [fg_color, bg_color] return [fg_color, bg_color]
def edit_value(key, current_value, state): def edit_value(key, current_value, menu_state):
height = 10 height = 10
input_width = width - 16 # Allow space for "New Value: " input_width = width - 16 # Allow space for "New Value: "
@@ -96,17 +96,17 @@ def edit_value(key, current_value, state):
return user_input if user_input else current_value return user_input if user_input else current_value
def display_menu(state): def display_menu(menu_state):
""" """
Render the configuration menu with a Save button directly added to the window. Render the configuration menu with a Save button directly added to the window.
""" """
num_items = len(state.current_menu) + (1 if state.show_save_option else 0) num_items = len(menu_state.current_menu) + (1 if menu_state.show_save_option else 0)
# Determine menu items based on the type of current_menu # Determine menu items based on the type of current_menu
if isinstance(state.current_menu, dict): if isinstance(menu_state.current_menu, dict):
options = list(state.current_menu.keys()) options = list(menu_state.current_menu.keys())
elif isinstance(state.current_menu, list): elif isinstance(menu_state.current_menu, list):
options = [f"[{i}]" for i in range(len(state.current_menu))] options = [f"[{i}]" for i in range(len(menu_state.current_menu))]
else: else:
options = [] # Fallback in case of unexpected data types options = [] # Fallback in case of unexpected data types
@@ -130,70 +130,70 @@ def display_menu(state):
menu_pad.bkgd(get_color("background")) menu_pad.bkgd(get_color("background"))
# Display the menu path # Display the menu path
header = " > ".join(state.menu_path) header = " > ".join(menu_state.menu_path)
if len(header) > width - 4: if len(header) > width - 4:
header = header[:width - 7] + "..." header = header[:width - 7] + "..."
menu_win.addstr(1, 2, header, get_color("settings_breadcrumbs", bold=True)) menu_win.addstr(1, 2, header, get_color("settings_breadcrumbs", bold=True))
# Populate the pad with menu options # Populate the pad with menu options
for idx, key in enumerate(options): for idx, key in enumerate(options):
value = state.current_menu[key] if isinstance(state.current_menu, dict) else state.current_menu[int(key.strip("[]"))] value = menu_state.current_menu[key] if isinstance(menu_state.current_menu, dict) else menu_state.current_menu[int(key.strip("[]"))]
display_key = f"{key}"[:width // 2 - 2] display_key = f"{key}"[:width // 2 - 2]
display_value = ( display_value = (
f"{value}"[:width // 2 - 8] f"{value}"[:width // 2 - 8]
) )
color = get_color("settings_default", reverse=(idx == state.selected_index)) color = get_color("settings_default", reverse=(idx == menu_state.selected_index))
menu_pad.addstr(idx, 0, f"{display_key:<{width // 2 - 2}} {display_value}".ljust(width - 8), color) menu_pad.addstr(idx, 0, f"{display_key:<{width // 2 - 2}} {display_value}".ljust(width - 8), color)
# Add Save button to the main window # Add Save button to the main window
if state.show_save_option: if menu_state.show_save_option:
save_position = menu_height - 2 save_position = menu_height - 2
menu_win.addstr(save_position, (width - len(save_option)) // 2, save_option, get_color("settings_save", reverse=(state.selected_index == len(state.current_menu)))) menu_win.addstr(save_position, (width - len(save_option)) // 2, save_option, get_color("settings_save", reverse=(menu_state.selected_index == len(menu_state.current_menu))))
menu_win.refresh() menu_win.refresh()
menu_pad.refresh( menu_pad.refresh(
state.start_index[-1], 0, menu_state.start_index[-1], 0,
menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4, menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4,
menu_win.getbegyx()[0] + 3 + menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0), menu_win.getbegyx()[0] + 3 + menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0),
menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4 menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4
) )
max_index = num_items + (1 if state.show_save_option else 0) - 1 max_index = num_items + (1 if menu_state.show_save_option else 0) - 1
visible_height = menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0) visible_height = menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0)
draw_arrows(menu_win, visible_height, max_index, state) draw_arrows(menu_win, visible_height, max_index, menu_state)
return menu_win, menu_pad, options return menu_win, menu_pad, options
def move_highlight(old_idx, new_idx, options, menu_win, menu_pad, state): def move_highlight(old_idx, new_idx, options, menu_win, menu_pad, menu_state):
if old_idx == new_idx: # No-op if old_idx == new_idx: # No-op
return return
max_index = len(options) + (1 if state.show_save_option else 0) - 1 max_index = len(options) + (1 if menu_state.show_save_option else 0) - 1
visible_height = menu_win.getmaxyx()[0] - 5 - (2 if state.show_save_option else 0) visible_height = menu_win.getmaxyx()[0] - 5 - (2 if menu_state.show_save_option else 0)
# Adjust state.start_index only when moving out of visible range # Adjust menu_state.start_index only when moving out of visible range
if new_idx == max_index and state.show_save_option: if new_idx == max_index and menu_state.show_save_option:
pass pass
elif new_idx < state.start_index[-1]: # Moving above the visible area elif new_idx < menu_state.start_index[-1]: # Moving above the visible area
state.start_index[-1] = new_idx menu_state.start_index[-1] = new_idx
elif new_idx >= state.start_index[-1] + visible_height: # Moving below the visible area elif new_idx >= menu_state.start_index[-1] + visible_height: # Moving below the visible area
state.start_index[-1] = new_idx - visible_height menu_state.start_index[-1] = new_idx - visible_height
pass pass
# Ensure state.start_index is within bounds # Ensure menu_state.start_index is within bounds
state.start_index[-1] = max(0, min(state.start_index[-1], max_index - visible_height + 1)) menu_state.start_index[-1] = max(0, min(menu_state.start_index[-1], max_index - visible_height + 1))
# Clear old selection # Clear old selection
if state.show_save_option and old_idx == max_index: if menu_state.show_save_option and old_idx == max_index:
menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save")) menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save"))
else: else:
menu_pad.chgat(old_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive") if options[old_idx] in sensitive_settings else get_color("settings_default")) menu_pad.chgat(old_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive") if options[old_idx] in sensitive_settings else get_color("settings_default"))
# Highlight new selection # Highlight new selection
if state.show_save_option and new_idx == max_index: if menu_state.show_save_option and new_idx == max_index:
menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save", reverse=True)) menu_win.chgat(menu_win.getmaxyx()[0] - 2, (width - len(save_option)) // 2, len(save_option), get_color("settings_save", reverse=True))
else: else:
menu_pad.chgat(new_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive", reverse=True) if options[new_idx] in sensitive_settings else get_color("settings_default", reverse=True)) menu_pad.chgat(new_idx, 0, menu_pad.getmaxyx()[1], get_color("settings_sensitive", reverse=True) if options[new_idx] in sensitive_settings else get_color("settings_default", reverse=True))
@@ -201,39 +201,39 @@ def move_highlight(old_idx, new_idx, options, menu_win, menu_pad, state):
menu_win.refresh() menu_win.refresh()
# Refresh pad only if scrolling is needed # Refresh pad only if scrolling is needed
menu_pad.refresh(state.start_index[-1], 0, menu_pad.refresh(menu_state.start_index[-1], 0,
menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4, menu_win.getbegyx()[0] + 3, menu_win.getbegyx()[1] + 4,
menu_win.getbegyx()[0] + 3 + visible_height, menu_win.getbegyx()[0] + 3 + visible_height,
menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4) menu_win.getbegyx()[1] + menu_win.getmaxyx()[1] - 4)
draw_arrows(menu_win, visible_height, max_index, state) draw_arrows(menu_win, visible_height, max_index, menu_state)
def draw_arrows(win, visible_height, max_index, state): def draw_arrows(win, visible_height, max_index, menu_state):
mi = max_index - (2 if state.show_save_option else 0) mi = max_index - (2 if menu_state.show_save_option else 0)
if visible_height < mi: if visible_height < mi:
if state.start_index[-1] > 0: if menu_state.start_index[-1] > 0:
win.addstr(3, 2, "", get_color("settings_default")) win.addstr(3, 2, "", get_color("settings_default"))
else: else:
win.addstr(3, 2, " ", get_color("settings_default")) win.addstr(3, 2, " ", get_color("settings_default"))
if mi - state.start_index[-1] >= visible_height + (0 if state.show_save_option else 1) : if mi - menu_state.start_index[-1] >= visible_height + (0 if menu_state.show_save_option else 1) :
win.addstr(visible_height + 3, 2, "", get_color("settings_default")) win.addstr(visible_height + 3, 2, "", get_color("settings_default"))
else: else:
win.addstr(visible_height + 3, 2, " ", get_color("settings_default")) win.addstr(visible_height + 3, 2, " ", get_color("settings_default"))
def json_editor(stdscr, state): def json_editor(stdscr, menu_state):
state.selected_index = 0 # Track the selected option menu_state.selected_index = 0 # Track the selected option
script_dir = os.path.dirname(os.path.abspath(__file__)) script_dir = os.path.dirname(os.path.abspath(__file__))
parent_dir = os.path.abspath(os.path.join(script_dir, os.pardir)) parent_dir = os.path.abspath(os.path.join(script_dir, os.pardir))
file_path = os.path.join(parent_dir, "config.json") file_path = os.path.join(parent_dir, "config.json")
state.show_save_option = True # Always show the Save button menu_state.show_save_option = True # Always show the Save button
# Ensure the file exists # Ensure the file exists
if not os.path.exists(file_path): if not os.path.exists(file_path):
@@ -245,37 +245,37 @@ def json_editor(stdscr, state):
original_data = json.load(f) original_data = json.load(f)
data = original_data # Reference to the original data data = original_data # Reference to the original data
state.current_menu = data # Track the current level of the menu menu_state.current_menu = data # Track the current level of the menu
# Render the menu # Render the menu
menu_win, menu_pad, options = display_menu(state) menu_win, menu_pad, options = display_menu(menu_state)
need_redraw = True need_redraw = True
while True: while True:
if(need_redraw): if(need_redraw):
menu_win, menu_pad, options = display_menu(state) menu_win, menu_pad, options = display_menu(menu_state)
menu_win.refresh() menu_win.refresh()
need_redraw = False need_redraw = False
max_index = len(options) + (1 if state.show_save_option else 0) - 1 max_index = len(options) + (1 if menu_state.show_save_option else 0) - 1
key = menu_win.getch() key = menu_win.getch()
if key == curses.KEY_UP: if key == curses.KEY_UP:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = max_index if state.selected_index == 0 else state.selected_index - 1 menu_state.selected_index = max_index if menu_state.selected_index == 0 else menu_state.selected_index - 1
move_highlight(old_selected_index, state.selected_index, options, state.show_save_option, menu_win, menu_pad,state) move_highlight(old_selected_index, menu_state.selected_index, options, menu_win, menu_pad, menu_state)
elif key == curses.KEY_DOWN: elif key == curses.KEY_DOWN:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = 0 if state.selected_index == max_index else state.selected_index + 1 menu_state.selected_index = 0 if menu_state.selected_index == max_index else menu_state.selected_index + 1
move_highlight(old_selected_index, state.selected_index, options, menu_win, menu_pad, state) move_highlight(old_selected_index, menu_state.selected_index, options, menu_win, menu_pad, menu_state)
elif key == ord("\t") and state.show_save_option: elif key == ord("\t") and menu_state.show_save_option:
old_selected_index = state.selected_index old_selected_index = menu_state.selected_index
state.selected_index = max_index menu_state.selected_index = max_index
move_highlight(old_selected_index, state.selected_index, options, menu_win, menu_pad, state) move_highlight(old_selected_index, menu_state.selected_index, options, menu_win, menu_pad, menu_state)
elif key in (curses.KEY_RIGHT, 10, 13): # 10 = \n, 13 = carriage return elif key in (curses.KEY_RIGHT, 10, 13): # 10 = \n, 13 = carriage return
@@ -283,41 +283,41 @@ def json_editor(stdscr, state):
menu_win.erase() menu_win.erase()
menu_win.refresh() menu_win.refresh()
if state.selected_index < len(options): # Handle selection of a menu item if menu_state.selected_index < len(options): # Handle selection of a menu item
selected_key = options[state.selected_index] selected_key = options[menu_state.selected_index]
state.menu_path.append(str(selected_key)) menu_state.menu_path.append(str(selected_key))
state.start_index.append(0) menu_state.start_index.append(0)
state.menu_index.append(state.selected_index) menu_state.menu_index.append(menu_state.selected_index)
# Handle nested data # Handle nested data
if isinstance(state.current_menu, dict): if isinstance(menu_state.current_menu, dict):
if selected_key in state.current_menu: if selected_key in menu_state.current_menu:
selected_data = state.current_menu[selected_key] selected_data = menu_state.current_menu[selected_key]
else: else:
continue # Skip invalid key continue # Skip invalid key
elif isinstance(state.current_menu, list): elif isinstance(menu_state.current_menu, list):
selected_data = state.current_menu[int(selected_key.strip("[]"))] selected_data = menu_state.current_menu[int(selected_key.strip("[]"))]
if isinstance(selected_data, list) and len(selected_data) == 2: if isinstance(selected_data, list) and len(selected_data) == 2:
# Edit color pair # Edit color pair
new_value = edit_color_pair(selected_key, selected_data) new_value = edit_color_pair(selected_key, selected_data)
state.menu_path.pop() menu_state.menu_path.pop()
state.start_index.pop() menu_state.start_index.pop()
state.menu_index.pop() menu_state.menu_index.pop()
state.current_menu[selected_key] = new_value menu_state.current_menu[selected_key] = new_value
elif isinstance(selected_data, (dict, list)): elif isinstance(selected_data, (dict, list)):
# Navigate into nested data # Navigate into nested data
state.current_menu = selected_data menu_state.current_menu = selected_data
state.selected_index = 0 # Reset the selected index menu_state.selected_index = 0 # Reset the selected index
else: else:
# General value editing # General value editing
new_value = edit_value(selected_key, selected_data, state) new_value = edit_value(selected_key, selected_data, menu_state)
state.menu_path.pop() menu_state.menu_path.pop()
state.menu_index.pop() menu_state.menu_index.pop()
state.start_index.pop() menu_state.start_index.pop()
state.current_menu[selected_key] = new_value menu_state.current_menu[selected_key] = new_value
need_redraw = True need_redraw = True
else: else:
@@ -332,16 +332,16 @@ def json_editor(stdscr, state):
menu_win.erase() menu_win.erase()
menu_win.refresh() menu_win.refresh()
# state.selected_index = state.menu_index[-1] # menu_state.selected_index = menu_state.menu_index[-1]
# Navigate back in the menu # Navigate back in the menu
if len(state.menu_path) > 2: if len(menu_state.menu_path) > 2:
state.menu_path.pop() menu_state.menu_path.pop()
state.start_index.pop() menu_state.start_index.pop()
state.current_menu = data menu_state.current_menu = data
for path in state.menu_path[2:]: for path in menu_state.menu_path[2:]:
state.current_menu = state.current_menu[path] if isinstance(state.current_menu, dict) else state.current_menu[int(path.strip("[]"))] menu_state.current_menu = menu_state.current_menu[path] if isinstance(menu_state.current_menu, dict) else menu_state.current_menu[int(path.strip("[]"))]
else: else:
# Exit the editor # Exit the editor
@@ -360,14 +360,14 @@ def save_json(file_path, data):
def main(stdscr): def main(stdscr):
from contact.ui.ui_state import MenuState from contact.ui.ui_state import MenuState
state = MenuState() menu_state = MenuState()
if len(state.menu_path) == 0: if len(menu_state.menu_path) == 0:
state.menu_path = ["App Settings"] # Initialize if not set menu_state.menu_path = ["App Settings"] # Initialize if not set
curses.curs_set(0) curses.curs_set(0)
stdscr.keypad(True) stdscr.keypad(True)
setup_colors() setup_colors()
json_editor(stdscr, state) json_editor(stdscr, menu_state)
if __name__ == "__main__": if __name__ == "__main__":
curses.wrapper(main) curses.wrapper(main)

View File

@@ -123,24 +123,24 @@ def setPref(config, comp_name, raw_val) -> bool:
def config_import(interface, filename): def config_import(node_state, filename):
with open(filename, encoding="utf8") as file: with open(filename, encoding="utf8") as file:
configuration = yaml.safe_load(file) configuration = yaml.safe_load(file)
closeNow = True closeNow = True
interface.getNode('^local', False).beginSettingsTransaction() node_state.interface.getNode('^local', False).beginSettingsTransaction()
if "owner" in configuration: if "owner" in configuration:
logging.info(f"Setting device owner to {configuration['owner']}") logging.info(f"Setting device owner to {configuration['owner']}")
waitForAckNak = True waitForAckNak = True
interface.getNode('^local', False).setOwner(configuration["owner"]) node_state.interface.getNode('^local', False).setOwner(configuration["owner"])
if "owner_short" in configuration: if "owner_short" in configuration:
logging.info( logging.info(
f"Setting device owner short to {configuration['owner_short']}" f"Setting device owner short to {configuration['owner_short']}"
) )
waitForAckNak = True waitForAckNak = True
interface.getNode('^local', False).setOwner( node_state.interface.getNode('^local', False).setOwner(
long_name=None, short_name=configuration["owner_short"] long_name=None, short_name=configuration["owner_short"]
) )
@@ -149,23 +149,23 @@ def config_import(interface, filename):
f"Setting device owner short to {configuration['ownerShort']}" f"Setting device owner short to {configuration['ownerShort']}"
) )
waitForAckNak = True waitForAckNak = True
interface.getNode('^local', False).setOwner( node_state.interface.getNode('^local', False).setOwner(
long_name=None, short_name=configuration["ownerShort"] long_name=None, short_name=configuration["ownerShort"]
) )
if "channel_url" in configuration: if "channel_url" in configuration:
logging.info(f"Setting channel url to {configuration['channel_url']}") logging.info(f"Setting channel url to {configuration['channel_url']}")
interface.getNode('^local').setURL(configuration["channel_url"]) node_state.interface.getNode('^local').setURL(configuration["channel_url"])
if "channelUrl" in configuration: if "channelUrl" in configuration:
logging.info(f"Setting channel url to {configuration['channelUrl']}") logging.info(f"Setting channel url to {configuration['channelUrl']}")
interface.getNode('^local').setURL(configuration["channelUrl"]) node_state.interface.getNode('^local').setURL(configuration["channelUrl"])
if "location" in configuration: if "location" in configuration:
alt = 0 alt = 0
lat = 0.0 lat = 0.0
lon = 0.0 lon = 0.0
localConfig = interface.localNode.localConfig localConfig = node_state.interface.localNode.localConfig
if "alt" in configuration["location"]: if "alt" in configuration["location"]:
alt = int(configuration["location"]["alt"] or 0) alt = int(configuration["location"]["alt"] or 0)
@@ -177,43 +177,43 @@ def config_import(interface, filename):
lon = float(configuration["location"]["lon"] or 0) lon = float(configuration["location"]["lon"] or 0)
logging.info(f"Fixing longitude at {lon} degrees") logging.info(f"Fixing longitude at {lon} degrees")
logging.info("Setting device position") logging.info("Setting device position")
interface.localNode.setFixedPosition(lat, lon, alt) node_state.interface.localNode.setFixedPosition(lat, lon, alt)
if "config" in configuration: if "config" in configuration:
localConfig = interface.getNode('^local').localConfig localConfig = node_state.interface.getNode('^local').localConfig
for section in configuration["config"]: for section in configuration["config"]:
traverseConfig( traverseConfig(
section, configuration["config"][section], localConfig section, configuration["config"][section], localConfig
) )
interface.getNode('^local').writeConfig( node_state.interface.getNode('^local').writeConfig(
camel_to_snake(section) camel_to_snake(section)
) )
if "module_config" in configuration: if "module_config" in configuration:
moduleConfig = interface.getNode('^local').moduleConfig moduleConfig = node_state.interface.getNode('^local').moduleConfig
for section in configuration["module_config"]: for section in configuration["module_config"]:
traverseConfig( traverseConfig(
section, section,
configuration["module_config"][section], configuration["module_config"][section],
moduleConfig, moduleConfig,
) )
interface.getNode('^local').writeConfig( node_state.interface.getNode('^local').writeConfig(
camel_to_snake(section) camel_to_snake(section)
) )
interface.getNode('^local', False).commitSettingsTransaction() node_state.interface.getNode('^local', False).commitSettingsTransaction()
logging.info("Writing modified configuration to device") logging.info("Writing modified configuration to device")
def config_export(interface) -> str: def config_export(node_state) -> str:
"""used in --export-config""" """used in --export-config"""
configObj = {} configObj = {}
owner = interface.getLongName() owner = node_state.interface.getLongName()
owner_short = interface.getShortName() owner_short = node_state.interface.getShortName()
channel_url = interface.localNode.getURL() channel_url = node_state.interface.localNode.getURL()
myinfo = interface.getMyNodeInfo() myinfo = node_state.interface.getMyNodeInfo()
pos = myinfo.get("position") pos = myinfo.get("position")
lat = None lat = None
lon = None lon = None
@@ -238,7 +238,7 @@ def config_export(interface) -> str:
if alt: if alt:
configObj["location"]["alt"] = alt configObj["location"]["alt"] = alt
config = MessageToDict(interface.localNode.localConfig) #checkme - Used as a dictionary here and a string below config = MessageToDict(node_state.interface.localNode.localConfig) #checkme - Used as a dictionary here and a string below
if config: if config:
# Convert inner keys to correct snake/camelCase # Convert inner keys to correct snake/camelCase
prefs = {} prefs = {}
@@ -261,7 +261,7 @@ def config_export(interface) -> str:
else: else:
configObj["config"] = config configObj["config"] = config
module_config = MessageToDict(interface.localNode.moduleConfig) module_config = MessageToDict(node_state.interface.localNode.moduleConfig)
if module_config: if module_config:
# Convert inner keys to correct snake/camelCase # Convert inner keys to correct snake/camelCase
prefs = {} prefs = {}

View File

@@ -7,9 +7,9 @@ from contact.utilities.utils import decimal_to_hex
import contact.ui.default_config as config import contact.ui.default_config as config
import contact.globals as globals import contact.globals as globals
def get_table_name(channel): def get_table_name(channel, node_state):
# Construct the table name # Construct the table name
table_name = f"{str(globals.myNodeNum)}_{channel}_messages" table_name = f"{str(node_state.myNodeNum)}_{channel}_messages"
quoted_table_name = f'"{table_name}"' # Quote the table name becuase we begin with numerics and contain spaces quoted_table_name = f'"{table_name}"' # Quote the table name becuase we begin with numerics and contain spaces
return quoted_table_name return quoted_table_name
@@ -47,7 +47,7 @@ def save_message_to_db(channel, user_id, message_text):
logging.error(f"Unexpected error in save_message_to_db: {e}") logging.error(f"Unexpected error in save_message_to_db: {e}")
def update_ack_nak(channel, timestamp, message, ack): def update_ack_nak(channel, timestamp, message, ack, node_state):
try: try:
with sqlite3.connect(config.db_file_path) as db_connection: with sqlite3.connect(config.db_file_path) as db_connection:
db_cursor = db_connection.cursor() db_cursor = db_connection.cursor()
@@ -59,7 +59,7 @@ def update_ack_nak(channel, timestamp, message, ack):
message_text = ? message_text = ?
""" """
db_cursor.execute(update_query, (ack, str(globals.myNodeNum), timestamp, message)) db_cursor.execute(update_query, (ack, str(node_state.myNodeNum), timestamp, message))
db_connection.commit() db_connection.commit()
except sqlite3.Error as e: except sqlite3.Error as e:
@@ -69,14 +69,14 @@ def update_ack_nak(channel, timestamp, message, ack):
logging.error(f"Unexpected error in update_ack_nak: {e}") logging.error(f"Unexpected error in update_ack_nak: {e}")
def load_messages_from_db(): def load_messages_from_db(node_state):
"""Load messages from the database for all channels and update globals.all_messages and globals.channel_list.""" """Load messages from the database for all channels and update globals.all_messages and globals.channel_list."""
try: try:
with sqlite3.connect(config.db_file_path) as db_connection: with sqlite3.connect(config.db_file_path) as db_connection:
db_cursor = db_connection.cursor() db_cursor = db_connection.cursor()
query = "SELECT name FROM sqlite_master WHERE type='table' AND name LIKE ?" query = "SELECT name FROM sqlite_master WHERE type='table' AND name LIKE ?"
db_cursor.execute(query, (f"{str(globals.myNodeNum)}_%_messages",)) db_cursor.execute(query, (f"{str(node_state.myNodeNum)}_%_messages",))
tables = [row[0] for row in db_cursor.fetchall()] tables = [row[0] for row in db_cursor.fetchall()]
# Iterate through each table and fetch its messages # Iterate through each table and fetch its messages
@@ -123,10 +123,10 @@ def load_messages_from_db():
elif ack_type == "Nak": elif ack_type == "Nak":
ack_str = config.nak_str ack_str = config.nak_str
if user_id == str(globals.myNodeNum): if user_id == str(node_state.myNodeNum):
formatted_message = (f"{config.sent_message_prefix}{ack_str}: ", message) formatted_message = (f"{config.sent_message_prefix}{ack_str}: ", message)
else: else:
formatted_message = (f"{config.message_prefix} {get_name_from_database(int(user_id), 'short')}: ", message) formatted_message = (f"{config.message_prefix} {get_name_from_database(int(user_id), node_state,'short')}: ", message)
hourly_messages[hour].append(formatted_message) hourly_messages[hour].append(formatted_message)
@@ -142,14 +142,14 @@ def load_messages_from_db():
logging.error(f"SQLite error in load_messages_from_db: {e}") logging.error(f"SQLite error in load_messages_from_db: {e}")
def init_nodedb(): def init_nodedb(node_state):
"""Initialize the node database and update it with nodes from the interface.""" """Initialize the node database and update it with nodes from the interface."""
try: try:
if not globals.interface.nodes: if not globals.interface.nodes:
return # No nodes to initialize return # No nodes to initialize
ensure_node_table_exists() # Ensure the table exists before insertion ensure_node_table_exists(node_state) # Ensure the table exists before insertion
nodes_snapshot = list(globals.interface.nodes.values()) nodes_snapshot = list(globals.interface.nodes.values())
# Insert or update all nodes # Insert or update all nodes
@@ -172,7 +172,7 @@ def init_nodedb():
logging.error(f"Unexpected error in init_nodedb: {e}") logging.error(f"Unexpected error in init_nodedb: {e}")
def maybe_store_nodeinfo_in_db(packet): def maybe_store_nodeinfo_in_db(packet, node_state):
"""Save nodeinfo unless that record is already there, updating if necessary.""" """Save nodeinfo unless that record is already there, updating if necessary."""
try: try:
user_id = packet['from'] user_id = packet['from']
@@ -183,21 +183,21 @@ def maybe_store_nodeinfo_in_db(packet):
role = packet['decoded']['user'].get('role', 'CLIENT') role = packet['decoded']['user'].get('role', 'CLIENT')
public_key = packet['decoded']['user'].get('publicKey', '') public_key = packet['decoded']['user'].get('publicKey', '')
update_node_info_in_db(user_id, long_name, short_name, hw_model, is_licensed, role, public_key) update_node_info_in_db(node_state, user_id, long_name, short_name, hw_model, is_licensed, role, public_key)
except sqlite3.Error as e: except sqlite3.Error as e:
logging.error(f"SQLite error in maybe_store_nodeinfo_in_db: {e}") logging.error(f"SQLite error in maybe_store_nodeinfo_in_db: {e}")
except Exception as e: except Exception as e:
logging.error(f"Unexpected error in maybe_store_nodeinfo_in_db: {e}") logging.error(f"Unexpected error in maybe_store_nodeinfo_in_db: {e}")
def update_node_info_in_db(user_id, long_name=None, short_name=None, hw_model=None, is_licensed=None, role=None, public_key=None, chat_archived=None): def update_node_info_in_db(node_state, user_id, long_name=None, short_name=None, hw_model=None, is_licensed=None, role=None, public_key=None, chat_archived=None):
"""Update or insert node information into the database, preserving unchanged fields.""" """Update or insert node information into the database, preserving unchanged fields."""
try: try:
ensure_node_table_exists() # Ensure the table exists before any operation ensure_node_table_exists(node_state) # Ensure the table exists before any operation
with sqlite3.connect(config.db_file_path) as db_connection: with sqlite3.connect(config.db_file_path) as db_connection:
db_cursor = db_connection.cursor() db_cursor = db_connection.cursor()
table_name = f'"{globals.myNodeNum}_nodedb"' # Quote in case of numeric names table_name = f'"{node_state.myNodeNum}_nodedb"' # Quote in case of numeric names
table_columns = [i[1] for i in db_cursor.execute(f'PRAGMA table_info({table_name})')] table_columns = [i[1] for i in db_cursor.execute(f'PRAGMA table_info({table_name})')]
if "chat_archived" not in table_columns: if "chat_archived" not in table_columns:
@@ -249,9 +249,9 @@ def update_node_info_in_db(user_id, long_name=None, short_name=None, hw_model=No
logging.error(f"Unexpected error in update_node_info_in_db: {e}") logging.error(f"Unexpected error in update_node_info_in_db: {e}")
def ensure_node_table_exists(): def ensure_node_table_exists(node_state):
"""Ensure the node database table exists.""" """Ensure the node database table exists."""
table_name = f'"{globals.myNodeNum}_nodedb"' # Quote for safety table_name = f'"{node_state.myNodeNum}_nodedb"' # Quote for safety
schema = ''' schema = '''
user_id TEXT PRIMARY KEY, user_id TEXT PRIMARY KEY,
long_name TEXT, long_name TEXT,
@@ -279,7 +279,7 @@ def ensure_table_exists(table_name, schema):
logging.error(f"Unexpected error in ensure_table_exists({table_name}): {e}") logging.error(f"Unexpected error in ensure_table_exists({table_name}): {e}")
def get_name_from_database(user_id, type="long"): def get_name_from_database(user_id, node_state, type="long"):
""" """
Retrieve a user's name (long or short) from the node database. Retrieve a user's name (long or short) from the node database.
@@ -292,7 +292,7 @@ def get_name_from_database(user_id, type="long"):
db_cursor = db_connection.cursor() db_cursor = db_connection.cursor()
# Construct table name # Construct table name
table_name = f"{str(globals.myNodeNum)}_nodedb" table_name = f"{str(node_state.myNodeNum)}_nodedb"
nodeinfo_table = f'"{table_name}"' # Quote table name for safety nodeinfo_table = f'"{table_name}"' # Quote table name for safety
# Determine the correct column to fetch # Determine the correct column to fetch
@@ -313,11 +313,11 @@ def get_name_from_database(user_id, type="long"):
logging.error(f"Unexpected error in get_name_from_database: {e}") logging.error(f"Unexpected error in get_name_from_database: {e}")
return "Unknown" return "Unknown"
def is_chat_archived(user_id): def is_chat_archived(user_id, node_state):
try: try:
with sqlite3.connect(config.db_file_path) as db_connection: with sqlite3.connect(config.db_file_path) as db_connection:
db_cursor = db_connection.cursor() db_cursor = db_connection.cursor()
table_name = f"{str(globals.myNodeNum)}_nodedb" table_name = f"{str(node_state.myNodeNum)}_nodedb"
nodeinfo_table = f'"{table_name}"' nodeinfo_table = f'"{table_name}"'
query = f"SELECT chat_archived FROM {nodeinfo_table} WHERE user_id = ?" query = f"SELECT chat_archived FROM {nodeinfo_table} WHERE user_id = ?"
db_cursor.execute(query, (user_id,)) db_cursor.execute(query, (user_id,))

View File

@@ -4,7 +4,7 @@ import logging
import base64 import base64
import time import time
def save_changes(interface, modified_settings, state): def save_changes(node_state, modified_settings, menu_state):
""" """
Save changes to the device based on modified settings. Save changes to the device based on modified settings.
:param interface: Meshtastic interface instance :param interface: Meshtastic interface instance
@@ -16,7 +16,7 @@ def save_changes(interface, modified_settings, state):
logging.info("No changes to save. modified_settings is empty.") logging.info("No changes to save. modified_settings is empty.")
return return
node = interface.getNode('^local') node = node_state.interface.getNode('^local')
admin_key_backup = None admin_key_backup = None
if 'admin_key' in modified_settings: if 'admin_key' in modified_settings:
# Get reference to security config # Get reference to security config
@@ -52,19 +52,19 @@ def save_changes(interface, modified_settings, state):
if not modified_settings: if not modified_settings:
return return
if state.menu_path[1] == "Radio Settings" or state.menu_path[1] == "Module Settings": if menu_state.menu_path[1] == "Radio Settings" or menu_state.menu_path[1] == "Module Settings":
config_category = state.menu_path[2].lower() # for radio and module configs config_category = menu_state.menu_path[2].lower() # for radio and module configs
if {'latitude', 'longitude', 'altitude'} & modified_settings.keys(): if {'latitude', 'longitude', 'altitude'} & modified_settings.keys():
lat = float(modified_settings.get('latitude', 0.0)) lat = float(modified_settings.get('latitude', 0.0))
lon = float(modified_settings.get('longitude', 0.0)) lon = float(modified_settings.get('longitude', 0.0))
alt = int(modified_settings.get('altitude', 0)) alt = int(modified_settings.get('altitude', 0))
interface.localNode.setFixedPosition(lat, lon, alt) node_state.interface.localNode.setFixedPosition(lat, lon, alt)
logging.info(f"Updated {config_category} with Latitude: {lat} and Longitude {lon} and Altitude {alt}") logging.info(f"Updated {config_category} with Latitude: {lat} and Longitude {lon} and Altitude {alt}")
return return
elif state.menu_path[1] == "User Settings": # for user configs elif menu_state.menu_path[1] == "User Settings": # for user configs
config_category = "User Settings" config_category = "User Settings"
long_name = modified_settings.get("longName") long_name = modified_settings.get("longName")
short_name = modified_settings.get("shortName") short_name = modified_settings.get("shortName")
@@ -77,11 +77,11 @@ def save_changes(interface, modified_settings, state):
return return
elif state.menu_path[1] == "Channels": # for channel configs elif menu_state.menu_path[1] == "Channels": # for channel configs
config_category = "Channels" config_category = "Channels"
try: try:
channel = state.menu_path[-1] channel = menu_state.menu_path[-1]
channel_num = int(channel.split()[-1]) - 1 channel_num = int(channel.split()[-1]) - 1
except (IndexError, ValueError) as e: except (IndexError, ValueError) as e:
channel_num = None channel_num = None

View File

@@ -67,10 +67,9 @@ def refresh_node_list():
return True return True
return False return False
def get_nodeNum(): def get_nodeNum(node_state):
myinfo = globals.interface.getMyNodeInfo() myinfo = node_state.interface.getMyNodeInfo()
myNodeNum = myinfo['num'] node_state.myNodeNum = myinfo['num']
return myNodeNum
def decimal_to_hex(decimal_number): def decimal_to_hex(decimal_number):
return f"!{decimal_number:08x}" return f"!{decimal_number:08x}"