mirror of
https://github.com/DJ2LS/FreeDATA
synced 2024-05-14 08:04:33 +00:00
246 lines
8.2 KiB
Python
246 lines
8.2 KiB
Python
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()
|
|
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()
|