from flask import Flask, request, jsonify, make_response from flask_sock import Sock from flask_cors import CORS import os import serial_ports from config import CONFIG import audio import queue import server_commands import service_manager import state_manager import threading import ujson as json app = Flask(__name__) CORS(app) CORS(app, resources={r"/*": {"origins": "*"}}) sock = Sock(app) app.config['SOCK_SERVER_OPTIONS'] = {'ping_interval': 5} print(app.config) print(app.config['SOCK_SERVER_OPTIONS']) # set config file to use def set_config(): if 'FREEDATA_CONFIG' in os.environ: config_file = os.environ['FREEDATA_CONFIG'] else: config_file = 'config.ini' if os.path.exists(config_file): print(f"Using config from {config_file}") else: print(f"Config file '{config_file}' not found. Exiting.") exit(1) app.config_manager = CONFIG(config_file) set_config() # start modem app.state_queue = queue.Queue() # queue which holds latest states app.modem_events = queue.Queue() # queue which holds latest events app.modem_fft = queue.Queue() # queue which holds latest fft data app.modem_service = queue.Queue() # start / stop modem service # init state manager app.states = state_manager.STATES(app.state_queue) # start service manager service_manager.SM(app) # start modem service app.modem_service.put("start") # returns a standard API response def api_response(data): return make_response(jsonify(data), 200) ## REST API @app.route('/', methods=['GET']) def index(): return api_response({'name': 'FreeDATA API', 'description': '', 'api_version': 1, 'license': 'GPL3.0', 'documentation': 'https://wiki.freedata.app', }) # get and set config @app.route('/config', methods=['GET', 'POST']) def config(): if request.method in ['POST']: set_config = app.config_manager.write(request.json) app.modem_service.put("restart") if not set_config: response = api_response(None, 'error writing config') else: response = api_response(set_config) return response elif request.method == 'GET': return api_response(app.config_manager.read()) @app.route('/devices/audio', methods=['GET']) def get_audio_devices(): dev_in, dev_out = audio.get_audio_devices() devices = { 'in': dev_in, 'out': dev_out } return api_response(devices) @app.route('/devices/serial', methods=['GET']) def get_serial_devices(): devices = serial_ports.get_ports() return api_response(devices) @app.route('/modem/state', methods=['GET']) def get_modem_state(): return api_response(app.states.sendState()) @app.route('/modem/cqcqcq', methods=['POST', 'GET']) def post_cqcqcq(): if request.method not in ['POST']: return api_response({"info": "endpoint for triggering a CQ via POST"}) if app.states.is_modem_running: server_commands.cqcqcq(request.json) return api_response({"cmd": "cqcqcq"}) @app.route('/modem/beacon', methods=['POST']) def post_beacon(): if request.method not in ['POST']: return api_response({"info": "endpoint for controlling BEACON STATE via POST"}) if app.states.is_modem_running: server_commands.beacon(request.json) return api_response(request.json) @app.route('/modem/ping_ping', methods=['POST']) def post_ping(): if request.method not in ['POST']: return api_response({"info": "endpoint for controlling PING via POST"}) if app.states.is_modem_running: server_commands.ping_ping(request.json) return api_response(request.json) @app.route('/modem/send_test_frame', methods=['POST']) def post_send_test_frame(): if request.method not in ['POST']: return api_response({"info": "endpoint for triggering a TEST_FRAME via POST"}) if app.states.is_modem_running: server_commands.modem_send_test_frame() return api_response({"cmd": "test_frame"}) @app.route('/modem/fec_transmit', methods=['POST']) def post_send_fec_frame(): if request.method not in ['POST']: return api_response({"info": "endpoint for triggering a FEC frame via POST"}) if app.states.is_modem_running: server_commands.modem_fec_transmit(request.json) return api_response(request.json) @app.route('/modem/fec_is_writing', methods=['POST']) def post_send_fec_is_writing(): if request.method not in ['POST']: return api_response({"info": "endpoint for triggering a IS WRITING frame via POST"}) if app.states.is_modem_running: server_commands.modem_fec_is_writing(request.json) return api_response(request.json) @app.route('/modem/start', methods=['POST']) def post_modem_start(): if request.method not in ['POST']: return api_response({"info": "endpoint for STARTING modem via POST"}) print("start received...") app.modem_service.put("start") return api_response(request.json) @app.route('/modem/stop', methods=['POST']) def post_modem_stop(): if request.method not in ['POST']: return api_response({"info": "endpoint for STOPPING modem via POST"}) print("stop received...") app.modem_service.put("stop") return api_response(request.json) @app.route('/version', methods=['GET']) def get_modem_version(): return api_response({"version": 0}) # @app.route('/modem/arq_connect', methods=['POST']) # @app.route('/modem/arq_disconnect', methods=['POST']) # @app.route('/modem/send_raw', methods=['POST']) # @app.route('/modem/stop_transmission', methods=['POST']) # @app.route('/modem/listen', methods=['POST']) # not needed if we are restarting modem on changing settings # @app.route('/modem/record_audio', methods=['POST']) # @app.route('/modem/responde_to_call', methods=['POST']) # not needed if we are restarting modem on changing settings # @app.route('/modem/responde_to_cq', methods=['POST']) # not needed if we are restarting modem on changing settings # @app.route('/modem/audio_levels', methods=['POST']) # tx and rx # not needed if we are restarting modem on changing settings # @app.route('/modem/mesh_ping', methods=['POST']) # @app.route('/mesh/routing_table', methods=['GET']) # @app.route('/modem/get_rx_buffer', methods=['GET']) # @app.route('/modem/del_rx_buffer', methods=['POST']) # @app.route('/rig/status', methods=['GET']) # @app.route('/rig/mode', methods=['POST']) # @app.route('/rig/frequency', methods=['POST']) # @app.route('/rig/test_hamlib', methods=['POST']) def transmit_sock_data_worker(client_list, event_queue): while True: event = event_queue.get() clients = client_list.copy() for client in clients: try: client.send(event) except Exception: client_list.remove(client) def sock_watchdog(sock, client_list, event_queue): event_queue.put(json.dumps({"freedata-message": "hello-client"})) client_list.add(sock) while True: try: sock.receive(timeout=1) except Exception as e: print(f"client connection lost: {e}") try: client_list.remove(sock) except Exception as err: print(f"error removing client from list: {e} | {err}") break return # Event websocket @sock.route('/events') def sock_events(sock): sock_watchdog(sock, events_client_list, app.modem_events) @sock.route('/fft') def sock_fft(sock): sock_watchdog(sock, fft_client_list, app.modem_fft) @sock.route('/states') def sock_states(sock): sock_watchdog(sock, states_client_list, app.state_queue) # websocket multi client support for using with queued information. # our client set which contains all connected websocket clients events_client_list = set() fft_client_list = set() states_client_list = set() # start a worker thread for every socket endpoint events_thread = threading.Thread(target=transmit_sock_data_worker, daemon=True, args=(events_client_list, app.modem_events)) events_thread.start() states_thread = threading.Thread(target=transmit_sock_data_worker, daemon=True, args=(states_client_list, app.state_queue)) states_thread.start() fft_thread = threading.Thread(target=transmit_sock_data_worker, daemon=True, args=(fft_client_list, app.modem_fft)) fft_thread.start()