FreeDATA/modem/server.py

126 lines
3.8 KiB
Python
Raw Normal View History

2023-11-06 12:49:37 +00:00
from flask import Flask, request, jsonify
from flask_sock import Sock
2023-11-08 09:54:08 +00:00
from flask_cors import CORS
2023-11-06 14:36:11 +00:00
import os
import serial_ports
2023-11-06 12:49:37 +00:00
from config import CONFIG
import audio
import data_handler
import modem
import queue
2023-11-09 09:37:45 +00:00
import server_commands
2023-11-06 12:49:37 +00:00
app = Flask(__name__)
2023-11-08 09:54:08 +00:00
CORS(app)
CORS(app, resources={r"/*": {"origins": "*"}})
2023-11-06 12:49:37 +00:00
sock = Sock(app)
2023-11-09 09:37:45 +00:00
print(sock)
2023-11-06 12:49:37 +00:00
2023-11-06 14:36:11 +00:00
# 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'
2023-11-06 12:49:37 +00:00
2023-11-06 14:36:11 +00:00
if os.path.exists(config_file):
print("Using config from %s" % config_file)
else:
print("Config file '%s' not found. Exiting." % config_file)
exit(1)
2023-11-06 12:49:37 +00:00
2023-11-06 14:36:11 +00:00
app.config_manager = CONFIG(config_file)
# returns a standard API response
2023-11-06 12:49:37 +00:00
def api_response(data, status = 'ok'):
response = {
'status': status,
'data': data
}
return jsonify(response)
2023-11-06 14:36:11 +00:00
set_config()
# start modem
app.modem_events = queue.Queue()
app.modem = modem.RF(app.config_manager.config, app.modem_events)
data_handler.DATA(app.config_manager.config, app.modem_events)
2023-11-06 12:49:37 +00:00
## REST API
@app.route('/', methods=['GET'])
def index():
2023-11-06 14:36:11 +00:00
return api_response({'name': 'FreeDATA API',
2023-11-06 12:49:37 +00:00
'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 == 'POST':
2023-11-06 19:52:33 +00:00
set_config = app.config_manager.write(request.json)
2023-11-06 12:49:37 +00:00
if not set_config:
response = api_response(None, 'error writing config')
else:
response = api_response(set_config)
return response
elif request.method == 'GET':
2023-11-06 14:36:11 +00:00
return api_response(app.config_manager.read())
2023-11-06 12:49:37 +00:00
@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)
2023-11-08 18:21:36 +00:00
# @app.route('/modem/listen', methods=['POST'])
# @app.route('/modem/record_audio', methods=['POST'])
# @app.route('/modem/responde_to_call', methods=['POST'])
# @app.route('/modem/responde_to_cq', methods=['POST'])
# @app.route('/modem/audio_levels', methods=['POST']) # tx and rx
# @app.route('/modem/send_test_frame', methods=['POST'])
# @app.route('/modem/fec_transmit', methods=['POST'])
# @app.route('/modem/fec_is_writing', methods=['POST'])
2023-11-09 09:37:45 +00:00
@app.route('/modem/cqcqcq', methods=['POST', 'GET'])
def post_cqcqcq():
if request.method in ['POST']:
server_commands.cqcqcq()
return api_response({"cmd": "cqcqcq"})
else:
return api_response({"info": "endpoint for triggering a CQ via POST"})
2023-11-08 18:21:36 +00:00
# @app.route('/modem/beacon', methods=['POST']) # on/off
# @app.route('/modem/mesh_ping', methods=['POST'])
# @app.route('/modem/ping_ping', methods=['POST'])
# @app.route('/modem/arc_connect', methods=['POST'])
# @app.route('/modem/arc_disconnect', methods=['POST'])
# @app.route('/modem/send_raw', methods=['POST'])
# @app.route('/modem/stop_transmission', 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('/modem/start', methods=['POST'])
# @app.route('/modem/stop', 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'])
2023-11-06 12:49:37 +00:00
# Event websocket
@sock.route('/events')
def echo(sock):
ev = app.modem_events.get()
2023-11-09 09:37:45 +00:00
print(ev)
ev = {'test': 'ok'}
sock.send(ev)