init
This commit is contained in:
Logan Cusano
184
app/main.py
Normal file
184
app/main.py
Normal file
@@ -0,0 +1,184 @@
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
from datetime import datetime
|
||||
from fastapi import FastAPI
|
||||
from routers.op25_controller import create_op25_router
|
||||
from internal.logger import create_logger
|
||||
from internal.op25_config_utls import scan_local_library
|
||||
import paho.mqtt.client as mqtt
|
||||
import requests
|
||||
|
||||
# Initialize logging
|
||||
LOGGER = create_logger(__name__)
|
||||
|
||||
app = FastAPI(title="Radio Edge Supervisor")
|
||||
|
||||
# Add the router
|
||||
app.include_router(create_op25_router(), prefix="/op25")
|
||||
|
||||
NODE_ID = os.getenv("NODE_ID", "standalone-node")
|
||||
MQTT_BROKER = os.getenv("MQTT_BROKER", None)
|
||||
|
||||
def handle_c2_command(topic, payload):
|
||||
"""
|
||||
Parses and routes commands received from the C2 server by calling the
|
||||
local supervisor's API.
|
||||
"""
|
||||
try:
|
||||
data = json.loads(payload)
|
||||
command_type = data.get("command")
|
||||
|
||||
LOGGER.info(f"Received C2 Command: {command_type} on {topic}")
|
||||
|
||||
# Base URL for the local supervisor API
|
||||
base_url = "http://localhost:8001/op25"
|
||||
|
||||
if command_type == "start":
|
||||
response = requests.post(f"{base_url}/start")
|
||||
response.raise_for_status()
|
||||
LOGGER.info("Successfully executed 'start' command via API.")
|
||||
|
||||
elif command_type == "stop":
|
||||
response = requests.post(f"{base_url}/stop")
|
||||
response.raise_for_status()
|
||||
LOGGER.info("Successfully executed 'stop' command via API.")
|
||||
|
||||
elif command_type == "restart":
|
||||
LOGGER.info("Executing 'restart' command...")
|
||||
stop_response = requests.post(f"{base_url}/stop")
|
||||
stop_response.raise_for_status()
|
||||
time.sleep(2) # Give it a moment for services to die
|
||||
start_response = requests.post(f"{base_url}/start")
|
||||
start_response.raise_for_status()
|
||||
LOGGER.info("Successfully executed 'restart' command via API.")
|
||||
|
||||
elif command_type in ["update", "set_active_config"]:
|
||||
config_payload = data.get("config")
|
||||
if not config_payload:
|
||||
LOGGER.error(f"Command '{command_type}' missing 'config' payload.")
|
||||
return
|
||||
|
||||
elif command_type == "update":
|
||||
LOGGER.info("Updating local configuration...")
|
||||
# Placeholder: update_local_config(data.get("config"))
|
||||
restart = data.get("restart", True)
|
||||
response = requests.post(f"{base_url}/set_active_config?restart={restart}", json=config_payload)
|
||||
response.raise_for_status()
|
||||
LOGGER.info(f"Successfully executed '{command_type}' command via API.")
|
||||
|
||||
elif command_type == "load_from_library":
|
||||
system_name = data.get("system_name")
|
||||
if not system_name:
|
||||
LOGGER.error("Command 'load_from_library' missing 'system_name' payload.")
|
||||
return
|
||||
|
||||
response = requests.post(f"{base_url}/load_from_library?system_name={system_name}")
|
||||
response.raise_for_status()
|
||||
LOGGER.info(f"Successfully executed 'load_from_library' for {system_name} via API.")
|
||||
|
||||
elif command_type == "tune":
|
||||
freq_mhz = data.get("system")
|
||||
if not freq_mhz:
|
||||
LOGGER.error("Command 'tune' missing 'frequency' payload.")
|
||||
return
|
||||
|
||||
try:
|
||||
# OP25 terminal expects frequency in Hz
|
||||
freq_hz = int(float(freq_mhz) * 1_000_000)
|
||||
# The port is hardcoded as it's the default for the OP25 terminal
|
||||
op25_terminal_url = f"http://localhost:8081/tuning?chan=0&freq={freq_hz}"
|
||||
response = requests.get(op25_terminal_url, timeout=5)
|
||||
response.raise_for_status()
|
||||
LOGGER.info(f"Successfully sent tune command to OP25 terminal for {freq_mhz} MHz.")
|
||||
except ValueError:
|
||||
LOGGER.error(f"Invalid frequency format for tune command: {freq_mhz}")
|
||||
except requests.exceptions.RequestException as e:
|
||||
LOGGER.error(f"Failed to connect to OP25 terminal for tuning: {e}")
|
||||
|
||||
else:
|
||||
LOGGER.warning(f"Unknown command type received: {command_type}")
|
||||
|
||||
except json.JSONDecodeError:
|
||||
LOGGER.error(f"Failed to decode command payload: {payload}")
|
||||
except requests.exceptions.RequestException as e:
|
||||
LOGGER.error(f"Failed to call local API for command '{data.get('command')}': {e}")
|
||||
except Exception as e:
|
||||
LOGGER.error(f"Error processing C2 command: {e}")
|
||||
|
||||
async def mqtt_phone_home():
|
||||
"""
|
||||
Maintains a persistent C2 connection using a single MQTT client.
|
||||
Handles check-ins and command subscriptions via callbacks.
|
||||
"""
|
||||
if not MQTT_BROKER:
|
||||
LOGGER.info("No MQTT_BROKER defined. Running in standalone mode.")
|
||||
return
|
||||
|
||||
# Create a single client instance
|
||||
client = mqtt.Client(client_id=NODE_ID)
|
||||
|
||||
def on_connect(client, userdata, flags, rc):
|
||||
if rc == 0:
|
||||
LOGGER.info(f"Successfully connected to MQTT Broker: {MQTT_BROKER}")
|
||||
|
||||
# 1. Subscribe to command topics for this specific node
|
||||
command_topic = f"nodes/{NODE_ID}/commands"
|
||||
client.subscribe(command_topic)
|
||||
LOGGER.info(f"Subscribed to {command_topic}")
|
||||
|
||||
# 2. Perform Initial Check-In with OP25 status
|
||||
try:
|
||||
status_response = requests.get("http://localhost:8001/op25/status")
|
||||
op25_status = status_response.json() if status_response.ok else {}
|
||||
except requests.RequestException:
|
||||
op25_status = {"is_running": False, "active_system": None}
|
||||
|
||||
checkin_data = {
|
||||
"node_id": NODE_ID,
|
||||
"status": "online",
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"version": "1.0.0",
|
||||
"is_listening": op25_status.get("is_running", False),
|
||||
"active_system": op25_status.get("active_system"),
|
||||
"available_systems": scan_local_library()
|
||||
}
|
||||
client.publish(f"nodes/{NODE_ID}/checkin", json.dumps(checkin_data), retain=True)
|
||||
else:
|
||||
LOGGER.error(f"MQTT Connection failed with return code {rc}")
|
||||
|
||||
def on_message(client, userdata, msg):
|
||||
# Handle messages arriving on subscribed topics
|
||||
handle_c2_command(msg.topic, msg.payload.decode())
|
||||
|
||||
def on_disconnect(client, userdata, rc):
|
||||
if rc != 0:
|
||||
LOGGER.warning("Unexpected MQTT disconnection. Paho will attempt to reconnect...")
|
||||
|
||||
# Set up callbacks and LWT
|
||||
client.on_connect = on_connect
|
||||
client.on_message = on_message
|
||||
client.on_disconnect = on_disconnect
|
||||
|
||||
lwt_payload = json.dumps({"node_id": NODE_ID, "status": "offline"})
|
||||
client.will_set(f"nodes/{NODE_ID}/status", lwt_payload, qos=1, retain=True)
|
||||
|
||||
try:
|
||||
# Connect and start the background loop thread
|
||||
# loop_start() handles reconnections automatically without spaming new clients
|
||||
client.connect(MQTT_BROKER, 1883, 60)
|
||||
client.loop_start()
|
||||
|
||||
# Keep the async task alive indefinitely
|
||||
while True:
|
||||
await asyncio.sleep(3600)
|
||||
|
||||
except Exception as e:
|
||||
LOGGER.error(f"Fatal error in MQTT supervisor: {e}")
|
||||
finally:
|
||||
client.loop_stop()
|
||||
|
||||
@app.on_event("startup")
|
||||
async def startup_event():
|
||||
# Start the C2 connection in the background.
|
||||
asyncio.create_task(mqtt_phone_home())
|
||||
Reference in New Issue
Block a user