mirror of
https://github.com/SpudGunMan/meshing-around.git
synced 2026-05-07 05:44:28 +02:00
965 lines
40 KiB
Python
965 lines
40 KiB
Python
# helper functions and init for system related tasks
|
|
# K7MHI Kelly Keeton 2024
|
|
|
|
import meshtastic.serial_interface #pip install meshtastic
|
|
import meshtastic.tcp_interface
|
|
import meshtastic.ble_interface
|
|
import time
|
|
import asyncio
|
|
import contextlib # for suppressing output on watchdog
|
|
import io # for suppressing output on watchdog
|
|
from modules.log import *
|
|
|
|
# Global Variables
|
|
trap_list = ("cmd","cmd?") # default trap list
|
|
help_message = "CMD?:"
|
|
asyncLoop = asyncio.new_event_loop()
|
|
games_enabled = False
|
|
multiPingList = [{'message_from_id': 0, 'count': 0, 'type': '', 'deviceID': 0}]
|
|
lastTelemetryRequest = 0
|
|
numPacketsTx, numPacketsRx, numPacketsTxErr, numPacketsRxErr = 0, 0, 0, 0
|
|
|
|
# Ping Configuration
|
|
if ping_enabled:
|
|
# ping, pinging, ack, testing, test, pong
|
|
trap_list_ping = ("ping", "pinging", "ack", "testing", "test", "pong", "🔔", "CQCQ", "CQCQCQ")
|
|
trap_list = trap_list + trap_list_ping
|
|
help_message = help_message + "ping"
|
|
|
|
# Sitrep Configuration
|
|
if sitrep_enabled:
|
|
trap_list_sitrep = ("sitrep", "lheard")
|
|
trap_list = trap_list + trap_list_sitrep
|
|
help_message = help_message + ", sitrep"
|
|
|
|
# MOTD Configuration
|
|
if motd_enabled:
|
|
trap_list_motd = ("motd",)
|
|
trap_list = trap_list + trap_list_motd
|
|
help_message = help_message + ", motd"
|
|
|
|
# whoami Configuration
|
|
if whoami_enabled:
|
|
trap_list_whoami = ("whoami", "📍")
|
|
trap_list = trap_list + trap_list_whoami
|
|
help_message = help_message + ", whoami"
|
|
|
|
# Solar Conditions Configuration
|
|
if solar_conditions_enabled:
|
|
from modules.solarconditions import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + trap_list_solarconditions # items hfcond, solar, sun, moon
|
|
help_message = help_message + ", sun, hfcond, solar, moon"
|
|
else:
|
|
hf_band_conditions = False
|
|
|
|
# Command History Configuration
|
|
if enableCmdHistory:
|
|
trap_list = trap_list + ("history",)
|
|
#help_message = help_message + ", history"
|
|
|
|
# Location Configuration
|
|
if location_enabled:
|
|
from modules.locationdata import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + trap_list_location # items tide, whereami, wxc, wx
|
|
help_message = help_message + ", whereami, wx, wxc"
|
|
|
|
# Open-Meteo Configuration for worldwide weather
|
|
if use_meteo_wxApi:
|
|
from modules.wx_meteo import * # from the spudgunman/meshing-around repo
|
|
else:
|
|
# NOAA only features
|
|
help_message = help_message + ", wxa, tide"
|
|
|
|
# BBS Configuration
|
|
if bbs_enabled:
|
|
from modules.bbstools import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + trap_list_bbs # items bbslist, bbspost, bbsread, bbsdelete, bbshelp
|
|
help_message = help_message + ", bbslist, bbshelp"
|
|
else:
|
|
bbs_help = False
|
|
bbs_list_messages = False
|
|
|
|
|
|
# Dad Jokes Configuration
|
|
if dad_jokes_enabled:
|
|
from dadjokes import Dadjoke # pip install dadjokes
|
|
trap_list = trap_list + ("joke",)
|
|
help_message = help_message + ", joke"
|
|
|
|
# Wikipedia Search Configuration
|
|
if wikipedia_enabled:
|
|
import wikipedia # pip install wikipedia
|
|
trap_list = trap_list + ("wiki:", "wiki?",)
|
|
help_message = help_message + ", wiki:"
|
|
|
|
# LLM Configuration
|
|
if llm_enabled:
|
|
from modules.llm import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + trap_list_llm # items ask:
|
|
help_message = help_message + ", askai"
|
|
|
|
# DopeWars Configuration
|
|
if dopewars_enabled:
|
|
from modules.games.dopewar import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("dopewars",)
|
|
games_enabled = True
|
|
|
|
# Lemonade Stand Configuration
|
|
if lemonade_enabled:
|
|
from modules.games.lemonade import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("lemonstand",)
|
|
games_enabled = True
|
|
|
|
# BlackJack Configuration
|
|
if blackjack_enabled:
|
|
from modules.games.blackjack import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("blackjack",)
|
|
games_enabled = True
|
|
|
|
# Video Poker Configuration
|
|
if videoPoker_enabled:
|
|
from modules.games.videopoker import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("videopoker",)
|
|
games_enabled = True
|
|
|
|
if mastermind_enabled:
|
|
from modules.games.mmind import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("mastermind",)
|
|
games_enabled = True
|
|
|
|
if golfSim_enabled:
|
|
from modules.games.golfsim import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("golfsim",)
|
|
games_enabled = True
|
|
|
|
if uno_enabled:
|
|
from modules.games.uno import * # from the spudgunman/meshing-around repo
|
|
trap_list = trap_list + ("playuno",)
|
|
games_enabled = True
|
|
|
|
# Games Configuration
|
|
if games_enabled is True:
|
|
help_message = help_message + ", games"
|
|
trap_list = trap_list + ("games",)
|
|
gTnW_enabled = True
|
|
gamesCmdList = "Play via DM🕹️ CMD: "
|
|
if dopewars_enabled:
|
|
gamesCmdList += "dopeWars, "
|
|
if lemonade_enabled:
|
|
gamesCmdList += "lemonStand, "
|
|
if gTnW_enabled:
|
|
trap_list = trap_list + ("globalthermonuclearwar",)
|
|
if blackjack_enabled:
|
|
gamesCmdList += "blackJack, "
|
|
if videoPoker_enabled:
|
|
gamesCmdList += "videoPoker, "
|
|
if mastermind_enabled:
|
|
gamesCmdList += "masterMind, "
|
|
if golfSim_enabled:
|
|
gamesCmdList += "golfSim, "
|
|
if uno_enabled:
|
|
gamesCmdList += "playuno, "
|
|
gamesCmdList = gamesCmdList[:-2] # remove the last comma
|
|
else:
|
|
gamesCmdList = ""
|
|
|
|
# Scheduled Broadcast Configuration
|
|
if scheduler_enabled:
|
|
import schedule # pip install schedule
|
|
# Reminder Scheduler is enabled every Monday at noon send a log message
|
|
schedule.every().monday.at("12:00").do(lambda: logger.info("System: Scheduled Broadcast Reminder"))
|
|
|
|
# Sentry Configuration
|
|
if sentry_enabled:
|
|
from math import sqrt
|
|
import geopy.distance # pip install geopy
|
|
|
|
# Store and Forward Configuration
|
|
if store_forward_enabled:
|
|
trap_list = trap_list + ("messages",)
|
|
help_message = help_message + ", messages"
|
|
|
|
# Radio Monitor Configuration
|
|
if radio_detection_enabled:
|
|
from modules.radio import * # from the spudgunman/meshing-around repo
|
|
|
|
# BLE dual interface prevention
|
|
if interface1_type == 'ble' and interface2_type == 'ble':
|
|
logger.critical(f"System: BLE Interface1 and Interface2 cannot both be BLE. Exiting")
|
|
exit()
|
|
|
|
# Interface1 Configuration
|
|
try:
|
|
logger.debug(f"System: Initializing Interface1")
|
|
if interface1_type == 'serial':
|
|
interface1 = meshtastic.serial_interface.SerialInterface(port1)
|
|
elif interface1_type == 'tcp':
|
|
interface1 = meshtastic.tcp_interface.TCPInterface(hostname1)
|
|
elif interface1_type == 'ble':
|
|
interface1 = meshtastic.ble_interface.BLEInterface(mac1)
|
|
else:
|
|
logger.critical(f"System: Interface Type: {interface1_type} not supported. Validate your config against config.template Exiting")
|
|
exit()
|
|
except Exception as e:
|
|
logger.critical(f"System: script abort. Initializing Interface1 {e}")
|
|
exit()
|
|
|
|
# Interface2 Configuration
|
|
if interface2_enabled:
|
|
logger.debug(f"System: Initializing Interface2")
|
|
try:
|
|
if interface2_type == 'serial':
|
|
interface2 = meshtastic.serial_interface.SerialInterface(port2)
|
|
elif interface2_type == 'tcp':
|
|
interface2 = meshtastic.tcp_interface.TCPInterface(hostname2)
|
|
elif interface2_type == 'ble':
|
|
interface2 = meshtastic.ble_interface.BLEInterface(mac2)
|
|
else:
|
|
logger.critical(f"System: Interface Type: {interface2_type} not supported. Validate your config against config.template Exiting")
|
|
exit()
|
|
except Exception as e:
|
|
logger.critical(f"System: script abort. Initializing Interface2 {e}")
|
|
exit()
|
|
|
|
#Get the node number of the device, check if the device is connected
|
|
try:
|
|
myinfo = interface1.getMyNodeInfo()
|
|
myNodeNum1 = myinfo['num']
|
|
except Exception as e:
|
|
logger.critical(f"System: script abort. {e}")
|
|
exit()
|
|
|
|
if interface2_enabled:
|
|
try:
|
|
myinfo2 = interface2.getMyNodeInfo()
|
|
myNodeNum2 = myinfo2['num']
|
|
except Exception as e:
|
|
logger.critical(f"System: script abort. {e}")
|
|
exit()
|
|
else:
|
|
myNodeNum2 = 777
|
|
|
|
# functions below
|
|
|
|
def decimal_to_hex(decimal_number):
|
|
return f"!{decimal_number:08x}"
|
|
|
|
def get_name_from_number(number, type='long', nodeInt=1):
|
|
name = ""
|
|
if nodeInt == 1:
|
|
for node in interface1.nodes.values():
|
|
if number == node['num']:
|
|
if type == 'long':
|
|
name = node['user']['longName']
|
|
return name
|
|
elif type == 'short':
|
|
name = node['user']['shortName']
|
|
return name
|
|
else:
|
|
pass
|
|
else:
|
|
name = str(decimal_to_hex(number)) # If name not found, use the ID as string
|
|
return name
|
|
|
|
if nodeInt == 2:
|
|
for node in interface2.nodes.values():
|
|
if number == node['num']:
|
|
if type == 'long':
|
|
name = node['user']['longName']
|
|
return name
|
|
elif type == 'short':
|
|
name = node['user']['shortName']
|
|
return name
|
|
else:
|
|
pass
|
|
else:
|
|
name = str(decimal_to_hex(number)) # If name not found, use the ID as string
|
|
return name
|
|
return number
|
|
|
|
def get_num_from_short_name(short_name, nodeInt=1):
|
|
# Get the node number from the short name, converting all to lowercase for comparison (good practice?)
|
|
logger.debug(f"System: Getting Node Number from Short Name: {short_name} on Device: {nodeInt}")
|
|
if nodeInt == 1:
|
|
for node in interface1.nodes.values():
|
|
#logger.debug(f"System: Checking Node: {node['user']['shortName']} against {short_name} for number {node['num']}")
|
|
if short_name == node['user']['shortName']:
|
|
return node['num']
|
|
elif str(short_name.lower()) == node['user']['shortName'].lower():
|
|
return node['num']
|
|
else:
|
|
# try other interface
|
|
if interface2_enabled:
|
|
for node in interface2.nodes.values():
|
|
if short_name == node['user']['shortName']:
|
|
return node['num']
|
|
elif str(short_name.lower()) == node['user']['shortName'].lower():
|
|
return node['num']
|
|
if nodeInt == 2:
|
|
for node in interface2.nodes.values():
|
|
if short_name == node['user']['shortName']:
|
|
return node['num']
|
|
elif str(short_name.lower()) == node['user']['shortName'].lower():
|
|
return node['num']
|
|
else:
|
|
# try other interface
|
|
if interface2_enabled:
|
|
for node in interface1.nodes.values():
|
|
if short_name == node['user']['shortName']:
|
|
return node['num']
|
|
elif str(short_name.lower()) == node['user']['shortName'].lower():
|
|
return node['num']
|
|
return 0
|
|
|
|
def get_node_list(nodeInt=1):
|
|
# Get a list of nodes on the device
|
|
node_list = ""
|
|
node_list1 = []
|
|
node_list2 = []
|
|
short_node_list = []
|
|
last_heard = 0
|
|
if nodeInt == 1:
|
|
if interface1.nodes:
|
|
for node in interface1.nodes.values():
|
|
# ignore own
|
|
if node['num'] != myNodeNum2 and node['num'] != myNodeNum1:
|
|
node_name = get_name_from_number(node['num'], 'long', nodeInt)
|
|
snr = node.get('snr', 0)
|
|
|
|
# issue where lastHeard is not always present
|
|
last_heard = node.get('lastHeard', 0)
|
|
|
|
# make a list of nodes with last heard time and SNR
|
|
item = (node_name, last_heard, snr)
|
|
node_list1.append(item)
|
|
else:
|
|
logger.warning(f"System: No nodes found")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
if nodeInt == 2:
|
|
if interface2.nodes:
|
|
for node in interface2.nodes.values():
|
|
# ignore own
|
|
if node['num'] != myNodeNum2 and node['num'] != myNodeNum1:
|
|
node_name = get_name_from_number(node['num'], 'long', nodeInt)
|
|
snr = node.get('snr', 0)
|
|
|
|
# issue where lastHeard is not always present
|
|
last_heard = node.get('lastHeard', 0)
|
|
|
|
# make a list of nodes with last heard time and SNR
|
|
item = (node_name, last_heard, snr)
|
|
node_list2.append(item)
|
|
else:
|
|
logger.warning(f"System: No nodes found")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
try:
|
|
#print (f"Node List: {node_list1[:5]}\n")
|
|
node_list1.sort(key=lambda x: x[1] if x[1] is not None else 0, reverse=True)
|
|
#print (f"Node List: {node_list1[:5]}\n")
|
|
if interface2_enabled:
|
|
node_list2.sort(key=lambda x: x[1] if x[1] is not None else 0, reverse=True)
|
|
except Exception as e:
|
|
logger.error(f"System: Error sorting node list: {e}")
|
|
logger.debug(f"Node List1: {node_list1[:5]}\n")
|
|
if interface2_enabled:
|
|
logger.debug(f"Node List2: {node_list2[:5]}\n")
|
|
node_list = ERROR_FETCHING_DATA
|
|
|
|
try:
|
|
# make a nice list for the user
|
|
for x in node_list1[:SITREP_NODE_COUNT]:
|
|
short_node_list.append(f"{x[0]} SNR:{x[2]}")
|
|
for x in node_list2[:SITREP_NODE_COUNT]:
|
|
short_node_list.append(f"{x[0]} SNR:{x[2]}")
|
|
|
|
for x in short_node_list:
|
|
if x != "" or x != '\n':
|
|
node_list += x + "\n"
|
|
except Exception as e:
|
|
logger.error(f"System: Error creating node list: {e}")
|
|
node_list = ERROR_FETCHING_DATA
|
|
|
|
return node_list
|
|
|
|
def get_node_location(number, nodeInt=1, channel=0):
|
|
# Get the location of a node by its number from nodeDB on device
|
|
latitude = latitudeValue
|
|
longitude = longitudeValue
|
|
position = [latitudeValue,longitudeValue]
|
|
lastheard = 0
|
|
if nodeInt == 1:
|
|
if interface1.nodes:
|
|
for node in interface1.nodes.values():
|
|
if number == node['num']:
|
|
if 'position' in node:
|
|
try:
|
|
latitude = node['position']['latitude']
|
|
longitude = node['position']['longitude']
|
|
except Exception as e:
|
|
logger.warning(f"System: Error getting location data for {number}")
|
|
logger.debug(f"System: location data for {number} is {latitude},{longitude}")
|
|
position = [latitude,longitude]
|
|
return position
|
|
else:
|
|
logger.warning(f"System: No location data for {number} using default location")
|
|
# request location data
|
|
# try:
|
|
# logger.debug(f"System: Requesting location data for {number}")
|
|
# if nodeInt == 1:
|
|
# interface1.sendPosition(destinationId=number, wantResponse=False, channelIndex=channel)
|
|
# if nodeInt == 2:
|
|
# interface2.sendPosition(destinationId=number, wantResponse=False, channelIndex=channel)
|
|
# except Exception as e:
|
|
# logger.error(f"System: Error requesting location data for {number}. Error: {e}")
|
|
return position
|
|
else:
|
|
logger.warning(f"System: No nodes found")
|
|
return position
|
|
if nodeInt == 2:
|
|
if interface2.nodes:
|
|
for node in interface2.nodes.values():
|
|
if number == node['num']:
|
|
if 'position' in node:
|
|
try:
|
|
latitude = node['position']['latitude']
|
|
longitude = node['position']['longitude']
|
|
except Exception as e:
|
|
logger.warning(f"System: Error getting location data for {number}")
|
|
logger.info(f"System: location data for {number} is {latitude},{longitude}")
|
|
position = [latitude,longitude]
|
|
return position
|
|
else:
|
|
logger.warning(f"System: No location data for {number}")
|
|
return position
|
|
else:
|
|
logger.warning(f"System: No nodes found")
|
|
return position
|
|
return position
|
|
|
|
def get_closest_nodes(nodeInt=1,returnCount=3):
|
|
node_list = []
|
|
|
|
if nodeInt == 1:
|
|
if interface1.nodes:
|
|
for node in interface1.nodes.values():
|
|
if 'position' in node:
|
|
try:
|
|
nodeID = node['num']
|
|
latitude = node['position']['latitude']
|
|
longitude = node['position']['longitude']
|
|
|
|
#lastheard time in unix time
|
|
lastheard = node.get('lastHeard', 0)
|
|
#if last heard is over 24 hours ago, ignore the node
|
|
if lastheard < (time.time() - 86400):
|
|
continue
|
|
|
|
# Calculate distance to node from config.ini location
|
|
distance = round(geopy.distance.geodesic((latitudeValue, longitudeValue), (latitude, longitude)).m, 2)
|
|
|
|
if (distance < sentry_radius):
|
|
if nodeID != myNodeNum1 and myNodeNum2 and str(nodeID) not in sentryIgnoreList:
|
|
node_list.append({'id': nodeID, 'latitude': latitude, 'longitude': longitude, 'distance': distance})
|
|
|
|
except Exception as e:
|
|
pass
|
|
# else:
|
|
# # request location data
|
|
# try:
|
|
# logger.debug(f"System: Requesting location data for {node['id']}")
|
|
# interface1.sendPosition(destinationId=node['id'], wantResponse=False, channelIndex=publicChannel)
|
|
# except Exception as e:
|
|
# logger.error(f"System: Error requesting location data for {node['id']}. Error: {e}")
|
|
|
|
# sort by distance closest
|
|
#node_list.sort(key=lambda x: (x['latitude']-latitudeValue)**2 + (x['longitude']-longitudeValue)**2)
|
|
node_list.sort(key=lambda x: x['distance'])
|
|
# return the first 3 closest nodes by default
|
|
return node_list[:returnCount]
|
|
else:
|
|
logger.warning(f"System: No nodes found in closest_nodes on interface {nodeInt}")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
if nodeInt == 2:
|
|
if interface2.nodes:
|
|
for node in interface2.nodes.values():
|
|
if 'position' in node:
|
|
try:
|
|
nodeID = node['num']
|
|
latitude = node['position']['latitude']
|
|
longitude = node['position']['longitude']
|
|
|
|
#lastheard time in unix time
|
|
lastheard = node.get('lastHeard', 0)
|
|
#if last heard is over 24 hours ago, ignore the node
|
|
if lastheard < (time.time() - 86400):
|
|
continue
|
|
|
|
# Calculate distance to node from config.ini location
|
|
distance = round(geopy.distance.geodesic((latitudeValue, longitudeValue), (latitude, longitude)).m, 2)
|
|
|
|
if (distance < sentry_radius):
|
|
if nodeID != myNodeNum1 and myNodeNum2 and str(nodeID) not in sentryIgnoreList:
|
|
node_list.append({'id': nodeID, 'latitude': latitude, 'longitude': longitude, 'distance': distance})
|
|
|
|
except Exception as e:
|
|
pass
|
|
# sort by distance closest
|
|
node_list.sort(key=lambda x: x['distance'])
|
|
# return the first 3 closest nodes by default
|
|
return node_list[:returnCount]
|
|
else:
|
|
logger.warning(f"System: No nodes found in closest_nodes on interface {nodeInt}")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
def send_message(message, ch, nodeid=0, nodeInt=1):
|
|
if message == "" or message == None or len(message) == 0:
|
|
return
|
|
# if message over MESSAGE_CHUNK_SIZE characters, split it into multiple messages
|
|
if len(message) > MESSAGE_CHUNK_SIZE:
|
|
logger.debug(f"System: Splitting Message, Message Length: {len(message)}")
|
|
|
|
# split the message into MESSAGE_CHUNK_SIZE 160 character chunks
|
|
message = message.replace('\n', ' NEWLINE ') # replace newlines with NEWLINE to keep them in split chunks
|
|
|
|
split_message = message.split()
|
|
line = ''
|
|
message_list = []
|
|
|
|
for word in split_message:
|
|
if len(line + word) < MESSAGE_CHUNK_SIZE:
|
|
if 'NEWLINE' in word or '\n' in word or '\r' in word:
|
|
# chunk by newline if it exists
|
|
message_list.append(line)
|
|
line = ''
|
|
else:
|
|
line += word + ' '
|
|
else:
|
|
message_list.append(line)
|
|
line = word + ' '
|
|
|
|
message_list.append(line) # needed add contents of the last 'line' into the list
|
|
message_list = [m.replace('NEWLINE', '') for m in message_list]
|
|
|
|
for m in message_list:
|
|
if nodeid == 0:
|
|
# Send to channel
|
|
logger.info(f"Device:{nodeInt} Channel:{ch} " + CustomFormatter.red + "Sending Multi-Chunk Message: " + CustomFormatter.white + m.replace('\n', ' '))
|
|
if nodeInt == 1:
|
|
interface1.sendText(text=m, channelIndex=ch)
|
|
if nodeInt == 2:
|
|
interface2.sendText(text=m, channelIndex=ch)
|
|
else:
|
|
# Send to DM
|
|
logger.info(f"Device:{nodeInt} " + CustomFormatter.red + "Sending Multi-Chunk DM: " + CustomFormatter.white + m.replace('\n', ' ') + CustomFormatter.purple +\
|
|
" To: " + CustomFormatter.white + f"{get_name_from_number(nodeid, 'long', nodeInt)}")
|
|
if nodeInt == 1:
|
|
interface1.sendText(text=m, channelIndex=ch, destinationId=nodeid)
|
|
if nodeInt == 2:
|
|
interface2.sendText(text=m, channelIndex=ch, destinationId=nodeid)
|
|
time.sleep(splitDelay) # wait an amout of time between sending each split message
|
|
else: # message is less than MESSAGE_CHUNK_SIZE characters
|
|
if nodeid == 0:
|
|
# Send to channel
|
|
logger.info(f"Device:{nodeInt} Channel:{ch} " + CustomFormatter.red + "SendingChannel: " + CustomFormatter.white + message.replace('\n', ' '))
|
|
if nodeInt == 1:
|
|
interface1.sendText(text=message, channelIndex=ch)
|
|
if nodeInt == 2:
|
|
interface2.sendText(text=message, channelIndex=ch)
|
|
else:
|
|
# Send to DM
|
|
logger.info(f"Device:{nodeInt} " + CustomFormatter.red + "Sending DM: " + CustomFormatter.white + message.replace('\n', ' ') + CustomFormatter.purple +\
|
|
" To: " + CustomFormatter.white + f"{get_name_from_number(nodeid, 'long', nodeInt)}")
|
|
if nodeInt == 1:
|
|
interface1.sendText(text=message, channelIndex=ch, destinationId=nodeid)
|
|
if nodeInt == 2:
|
|
interface2.sendText(text=message, channelIndex=ch, destinationId=nodeid)
|
|
|
|
def tell_joke():
|
|
# tell a dad joke, does it need an explanationn :)
|
|
if dad_jokes_enabled:
|
|
dadjoke = Dadjoke()
|
|
return dadjoke.joke
|
|
else:
|
|
return ''
|
|
|
|
def get_wikipedia_summary(search_term):
|
|
wikipedia_search = wikipedia.search(search_term, results=3)
|
|
wikipedia_suggest = wikipedia.suggest(search_term)
|
|
#wikipedia_aroundme = wikipedia.geosearch(location[0], location[1], results=3)
|
|
#logger.debug(f"System: Wikipedia Nearby:{wikipedia_aroundme}")
|
|
|
|
if len(wikipedia_search) == 0:
|
|
logger.warning(f"System: No Wikipedia Results for:{search_term}")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
try:
|
|
logger.debug(f"System: Searching Wikipedia for:{search_term}, First Result:{wikipedia_search[0]}, Suggest Word:{wikipedia_suggest}")
|
|
summary = wikipedia.summary(search_term, sentences=wiki_return_limit, auto_suggest=False, redirect=True)
|
|
except wikipedia.DisambiguationError as e:
|
|
logger.warning(f"System: Disambiguation Error for:{search_term} trying {wikipedia_search[0]}")
|
|
summary = wikipedia.summary(wikipedia_search[0], sentences=wiki_return_limit, auto_suggest=True, redirect=True)
|
|
except wikipedia.PageError as e:
|
|
logger.warning(f"System: Wikipedia Page Error for:{search_term} {e} trying {wikipedia_search[0]}")
|
|
summary = wikipedia.summary(wikipedia_search[0], sentences=wiki_return_limit, auto_suggest=True, redirect=True)
|
|
except Exception as e:
|
|
logger.warning(f"System: Error with Wikipedia for:{search_term} {e}")
|
|
return ERROR_FETCHING_DATA
|
|
|
|
return summary
|
|
|
|
def getPrettyTime(seconds):
|
|
# convert unix time to minutes, hours, or days, or years for simple display
|
|
designator = "s"
|
|
if seconds > 0:
|
|
seconds = round(seconds / 60)
|
|
designator = "m"
|
|
if seconds > 60:
|
|
seconds = round(seconds / 60)
|
|
designator = "h"
|
|
if seconds > 24:
|
|
seconds = round(seconds / 24)
|
|
designator = "d"
|
|
if seconds > 365:
|
|
seconds = round(seconds / 365)
|
|
designator = "y"
|
|
return str(seconds) + designator
|
|
|
|
def messageTrap(msg):
|
|
# Check if the message contains a trap word, this is the first filter for listning to messages
|
|
# after this the message is passed to the command_handler in the bot.py which is switch case filter for applying word to function
|
|
|
|
# Split Message on assumed words spaces m for m = msg.split(" ")
|
|
# t in trap_list, built by the config and system.py not the user
|
|
message_list=msg.split(" ")
|
|
for m in message_list:
|
|
for t in trap_list:
|
|
# if word in message is in the trap list, return True
|
|
if t.lower() == m.lower():
|
|
return True
|
|
# if no trap words found, run a search for near misses like ping? or cmd?
|
|
for m in message_list:
|
|
for t in range(len(trap_list)):
|
|
if m.endswith('?') and m[:-1].lower() == trap_list[t]:
|
|
return True
|
|
return False
|
|
|
|
def getNodeFirmware(nodeID=0, nodeInt=1):
|
|
# get the firmware version of the node
|
|
# this is a workaround because .localNode.getMetadata spits out a lot of debug info which cant be suppressed
|
|
# Create a StringIO object to capture the
|
|
if nodeInt == 1:
|
|
output_capture = io.StringIO()
|
|
with contextlib.redirect_stdout(output_capture):
|
|
interface1.localNode.getMetadata()
|
|
console_output = output_capture.getvalue()
|
|
if "firmware_version" in console_output:
|
|
fwVer = console_output.split("firmware_version: ")[1].split("\n")[0]
|
|
elif nodeInt == 2:
|
|
output_capture = io.StringIO()
|
|
with contextlib.redirect_stdout(output_capture):
|
|
interface2.localNode.getMetadata()
|
|
console_output = output_capture.getvalue()
|
|
if "firmware_version" in console_output:
|
|
fwVer = console_output.split("firmware_version: ")[1].split("\n")[0]
|
|
else:
|
|
return -1
|
|
return fwVer
|
|
|
|
def getNodeTelemetry(nodeID=0, nodeInt=1):
|
|
# throttle the telemetry requests to prevent spamming the device
|
|
global lastTelemetryRequest, numPacketsTx, numPacketsRx, numPacketsTxErr, numPacketsRxErr
|
|
if time.time() - lastTelemetryRequest < 600:
|
|
return -1
|
|
lastTelemetryRequest = time.time()
|
|
# get the telemetry data for a node
|
|
dataResponse = ""
|
|
if nodeInt == 1:
|
|
chutil = round(interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("deviceMetrics", {}).get("channelUtilization", 0), 1)
|
|
airUtilTx = round(interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("deviceMetrics", {}).get("airUtilTx", 0), 1)
|
|
uptimeSeconds = interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("deviceMetrics", {}).get("uptimeSeconds", 0)
|
|
batteryLevel = interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("deviceMetrics", {}).get("batteryLevel", 0)
|
|
voltage = interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("deviceMetrics", {}).get("voltage", 0)
|
|
#numPacketsRx = interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("localStats", {}).get("numPacketsRx", 0)
|
|
#numPacketsTx = interface1.nodes.get(decimal_to_hex(myNodeNum1), {}).get("localStats", {}).get("numPacketsTx", 0)
|
|
numTotalNodes = len(interface1.nodes)
|
|
elif nodeInt == 2:
|
|
chutil = round(interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("deviceMetrics", {}).get("channelUtilization", 0), 1)
|
|
airUtilTx = round(interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("deviceMetrics", {}).get("airUtilTx", 0), 1)
|
|
uptimeSeconds = interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("deviceMetrics", {}).get("uptimeSeconds", 0)
|
|
batteryLevel = interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("deviceMetrics", {}).get("batteryLevel", 0)
|
|
voltage = interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("deviceMetrics", {}).get("voltage", 0)
|
|
#numPacketsRx = interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("localStats", {}).get("numPacketsRx", 0)
|
|
#numPacketsTx = interface2.nodes.get(decimal_to_hex(myNodeNum2), {}).get("localStats", {}).get("numPacketsTx", 0)
|
|
numTotalNodes = len(interface2.nodes)
|
|
else:
|
|
return -1
|
|
|
|
# packet telemetry
|
|
if nodeInt == 1:
|
|
dataResponse += f"Telemetry:{nodeInt} numPacketsTx:{numPacketsTx} numPacketsRx:{numPacketsRx} numPacketsTxErr:{numPacketsTxErr} numPacketsRxErr:{numPacketsRxErr}"
|
|
if nodeInt == 2:
|
|
dataResponse += f"Telemetry:{nodeInt} numPacketsTx:{numPacketsTx2} numPacketsRx:{numPacketsRx2} numPacketsTxErr:{numPacketsTxErr2} numPacketsRxErr:{numPacketsRxErr2}"
|
|
|
|
|
|
# Channel utilization and airUtilTx
|
|
dataResponse += " ChUtil%:" + str(round(chutil, 2)) + " AirTx%:" + str(round(airUtilTx, 2))
|
|
|
|
if chutil > 40:
|
|
logger.warning(f"System: High Channel Utilization {chutil}% on Device: {nodeInt}")
|
|
|
|
if airUtilTx > 25:
|
|
logger.warning(f"System: High Air Utilization {airUtilTx}% on Device: {nodeInt}")
|
|
|
|
# add packet Rx/Tx info to the response
|
|
dataResponse += f" Rx#:{numPacketsRx} Tx#:{numPacketsTx}"
|
|
|
|
# Number of nodes
|
|
dataResponse += " Nodes:" + str(numTotalNodes)
|
|
|
|
# Uptime
|
|
uptimeSeconds = getPrettyTime(uptimeSeconds)
|
|
dataResponse += " Uptime:" + str(uptimeSeconds)
|
|
|
|
# add battery info to the response
|
|
emji = "🔌" if batteryLevel == 101 else "🪫" if batteryLevel < 10 else "🔋"
|
|
dataResponse += f" Volt:{round(voltage, 1)}"
|
|
|
|
if batteryLevel < 25:
|
|
logger.warning(f"System: Low Battery Level: {batteryLevel}{emji} on Device: {nodeInt}")
|
|
elif batteryLevel < 10:
|
|
logger.critical(f"System: Critical Battery Level: {batteryLevel}{emji} on Device: {nodeInt}")
|
|
|
|
return dataResponse
|
|
|
|
def handleMultiPing(nodeID=0, deviceID=1):
|
|
global multiPingList
|
|
if len(multiPingList) > 1:
|
|
mPlCpy = multiPingList.copy()
|
|
for i in range(len(mPlCpy)):
|
|
message_id_from = mPlCpy[i]['message_from_id']
|
|
count = mPlCpy[i]['count']
|
|
type = mPlCpy[i]['type']
|
|
deviceID = mPlCpy[i]['deviceID']
|
|
|
|
if count > 1 and deviceID == 1:
|
|
count -= 1
|
|
# update count in the list
|
|
multiPingList[i]['count'] = count
|
|
|
|
send_message(f"🔂{count} {type}", publicChannel, message_id_from, 1)
|
|
if count < 2:
|
|
# remove the item from the list
|
|
for j in range(len(multiPingList)):
|
|
if multiPingList[j]['message_from_id'] == message_id_from:
|
|
multiPingList.pop(j)
|
|
break
|
|
|
|
def exit_handler():
|
|
# Close the interface and save the BBS messages
|
|
logger.debug(f"System: Closing Autoresponder")
|
|
try:
|
|
interface1.close()
|
|
logger.debug(f"System: Interface1 Closed")
|
|
if interface2_enabled:
|
|
interface2.close()
|
|
logger.debug(f"System: Interface2 Closed")
|
|
except Exception as e:
|
|
logger.error(f"System: closing: {e}")
|
|
if bbs_enabled:
|
|
save_bbsdb()
|
|
save_bbsdm()
|
|
logger.debug(f"System: BBS Messages Saved")
|
|
logger.debug(f"System: Exiting")
|
|
asyncLoop.stop()
|
|
asyncLoop.close()
|
|
exit (0)
|
|
|
|
async def BroadcastScheduler():
|
|
# handle schedule checks for the broadcast of messages
|
|
while True:
|
|
schedule.run_pending()
|
|
await asyncio.sleep(1)
|
|
|
|
async def handleSignalWatcher():
|
|
global lastHamLibAlert, antiSpam, sigWatchBroadcastCh
|
|
# monitor rigctld for signal strength and frequency
|
|
while True:
|
|
msg = await signalWatcher()
|
|
if msg != ERROR_FETCHING_DATA and msg is not None:
|
|
logger.debug(f"System: Detected Alert from Hamlib {msg}")
|
|
|
|
# check we are not spammig the channel limit messages to once per minute
|
|
if time.time() - lastHamLibAlert > 60:
|
|
lastHamLibAlert = time.time()
|
|
# if sigWatchBrodcastCh list contains multiple channels, broadcast to all
|
|
if type(sigWatchBroadcastCh) is list:
|
|
for ch in sigWatchBroadcastCh:
|
|
if antiSpam and ch != publicChannel:
|
|
send_message(msg, int(ch), 0, 1)
|
|
if interface2_enabled:
|
|
send_message(msg, int(ch), 0, 2)
|
|
else:
|
|
logger.warning(f"System: antiSpam prevented Alert from Hamlib {msg}")
|
|
else:
|
|
if antiSpam and sigWatchBroadcastCh != publicChannel:
|
|
send_message(msg, int(sigWatchBroadcastCh), 0, 1)
|
|
if interface2_enabled:
|
|
send_message(msg, int(sigWatchBroadcastCh), 0, 2)
|
|
else:
|
|
logger.warning(f"System: antiSpam prevented Alert from Hamlib {msg}")
|
|
|
|
await asyncio.sleep(1)
|
|
pass
|
|
|
|
async def retry_interface(nodeID=1):
|
|
global interface1, interface2, retry_int1, retry_int2, max_retry_count1, max_retry_count2
|
|
# retry connecting to the interface
|
|
# add a check to see if the interface is already open or trying to open
|
|
if nodeID==1:
|
|
if interface1 is not None:
|
|
retry_int1 = True
|
|
max_retry_count1 -= 1
|
|
try:
|
|
interface1.close()
|
|
except Exception as e:
|
|
logger.error(f"System: closing interface1: {e}")
|
|
if nodeID==2:
|
|
if interface2 is not None:
|
|
retry_int2 = True
|
|
max_retry_count2 -= 1
|
|
try:
|
|
interface2.close()
|
|
except Exception as e:
|
|
logger.error(f"System: closing interface2: {e}")
|
|
|
|
logger.debug(f"System: Retrying interface in 15 seconds")
|
|
if max_retry_count1 == 0:
|
|
logger.critical(f"System: Max retry count reached for interface1")
|
|
exit_handler()
|
|
if max_retry_count2 == 0:
|
|
logger.critical(f"System: Max retry count reached for interface2")
|
|
exit_handler()
|
|
# wait 15 seconds before retrying
|
|
await asyncio.sleep(15)
|
|
|
|
# retry the interface
|
|
try:
|
|
if nodeID==1 and retry_int1:
|
|
interface1 = None
|
|
logger.debug(f"System: Retrying Interface1")
|
|
if interface1_type == 'serial':
|
|
interface1 = meshtastic.serial_interface.SerialInterface(port1)
|
|
elif interface1_type == 'tcp':
|
|
interface1 = meshtastic.tcp_interface.TCPInterface(hostname1)
|
|
elif interface1_type == 'ble':
|
|
interface1 = meshtastic.ble_interface.BLEInterface(mac1)
|
|
logger.debug(f"System: Interface1 Opened!")
|
|
retry_int1 = False
|
|
except Exception as e:
|
|
logger.error(f"System: opening interface1 on: {e}")
|
|
|
|
try:
|
|
if nodeID==2 and retry_int2:
|
|
interface2 = None
|
|
logger.debug(f"System: Retrying Interface2")
|
|
if interface2_type == 'serial':
|
|
interface2 = meshtastic.serial_interface.SerialInterface(port2)
|
|
elif interface2_type == 'tcp':
|
|
interface2 = meshtastic.tcp_interface.TCPInterface(hostname2)
|
|
elif interface2_type == 'ble':
|
|
interface2 = meshtastic.ble_interface.BLEInterface(mac2)
|
|
logger.debug(f"System: Interface2 Opened!")
|
|
retry_int2 = False
|
|
except Exception as e:
|
|
logger.error(f"System: opening interface2: {e}")
|
|
|
|
|
|
handleSentinel_spotted = ""
|
|
handleSentinel_loop = 0
|
|
async def handleSentinel(deviceID=1):
|
|
global handleSentinel_spotted, handleSentinel_loop
|
|
# Locate Closest Nodes and report them to a secure channel
|
|
# async function for possibly demanding back location data
|
|
enemySpotted = ""
|
|
closest_nodes = get_closest_nodes(deviceID)
|
|
if closest_nodes != ERROR_FETCHING_DATA and closest_nodes:
|
|
if closest_nodes[0]['id'] is not None:
|
|
enemySpotted = get_name_from_number(closest_nodes[0]['id'], 'long', 1)
|
|
enemySpotted += ", " + get_name_from_number(closest_nodes[0]['id'], 'short', 1)
|
|
enemySpotted += ", " + str(closest_nodes[0]['id'])
|
|
enemySpotted += ", " + decimal_to_hex(closest_nodes[0]['id'])
|
|
enemySpotted += f" at {closest_nodes[0]['distance']}m"
|
|
|
|
if handleSentinel_loop >= sentry_holdoff and handleSentinel_spotted != enemySpotted:
|
|
logger.warning(f"System: {enemySpotted} is close to your location on Interface1")
|
|
send_message(f"Sentry{deviceID}: {enemySpotted}", secure_channel, 0, deviceID)
|
|
handleSentinel_loop = 0
|
|
handleSentinel_spotted = enemySpotted
|
|
else:
|
|
handleSentinel_loop += 1
|
|
|
|
async def watchdog():
|
|
global retry_int1, retry_int2
|
|
int1Data, int2Data = "", ""
|
|
while True:
|
|
await asyncio.sleep(20)
|
|
#print(f"MeshBot System: watchdog running\r", end="")
|
|
|
|
if interface1 is not None and not retry_int1:
|
|
# getting firmware is a heartbeat to check if the interface is still connected
|
|
try:
|
|
firmware = getNodeFirmware(0, 1)
|
|
except Exception as e:
|
|
logger.error(f"System: communicating with interface1, trying to reconnect: {e}")
|
|
retry_int1 = True
|
|
|
|
if not retry_int1:
|
|
# Locate Closest Nodes and report them to a secure channel
|
|
if sentry_enabled:
|
|
await handleSentinel(1)
|
|
|
|
# multiPing handler
|
|
handleMultiPing(0,1)
|
|
|
|
# Telemetry data
|
|
int1Data = getNodeTelemetry(0, 1)
|
|
if int1Data != -1:
|
|
logger.debug(int1Data + f" Firmware:{firmware}")
|
|
|
|
if retry_int1:
|
|
try:
|
|
await retry_interface(1)
|
|
except Exception as e:
|
|
logger.error(f"System: retrying interface1: {e}")
|
|
|
|
if interface2_enabled:
|
|
if interface2 is not None and not retry_int2:
|
|
# getting firmware is a heartbeat to check if the interface is still connected
|
|
try:
|
|
firmware2 = getNodeFirmware(0, 1)
|
|
except Exception as e:
|
|
logger.error(f"System: communicating with interface1, trying to reconnect: {e}")
|
|
retry_int2 = True
|
|
|
|
if not retry_int2:
|
|
# Locate Closest Nodes and report them to a secure channel
|
|
if sentry_enabled:
|
|
await handleSentinel(2)
|
|
|
|
# multiPing handler
|
|
handleMultiPing(0,2)
|
|
|
|
# Telemetry data
|
|
int2Data = getNodeTelemetry(0, 2)
|
|
if int1Data != -1:
|
|
logger.debug(int2Data + f" Firmware:{firmware2}")
|
|
|
|
if retry_int2:
|
|
try:
|
|
await retry_interface(2)
|
|
except Exception as e:
|
|
logger.error(f"System: retrying interface2: {e}")
|