#! /usr/bin/env python3

import SoapySDR as soapy
from radio import Radio

from flask import Flask, request, jsonify
from flasgger import Swagger

app = Flask(__name__)
swag = Swagger(app)

radios = {}

@app.route('/radio/report')
def report():
    """List radio devices available to the system.
    ---
    responses:
      200:
        description: A list of radio devices.
    """

    devices = [dict(device) for device in soapy.Device.enumerate()]
    return jsonify(devices)

@app.route('/radio/<radio>/connect')
def connect(radio):
    """Connect to a radio device, by driver name or serial number.
    ---
    parameters:
      - name: radio
        description: Radio device driver name, or serial number.
        in: path
        type: string
        required: true
    responses:
      200:
        description: Successfully connected to a radio.
      400:
        description: No radio device by that name is available.
      500:
        description: Failed to connect to radio.
    """
    if radio in radios:
        return "Radio device already connected", 400

    devices = [dict(device) for device in soapy.Device.enumerate()]
    for device in devices:
        if radio in device.values():
            try:
                radios[radio] = Radio(device)
                return "", 200
            except Exception as e:
                radios.pop(radio)
                return str(e), 500


    return "Radio device not found", 400

@app.route('/radio/<radio>/disconnect')
def disconnect(radio):
    """Disconnect from a radio device.
    ---
    parameters:
      - name: radio
        description: Radio device driver name, or serial number.
        in: path
        type: string
        required: true
    responses:
      200:
        description: Successfully connected to a radio.
      400:
        description: No radio device by that name is available.
      500:
        description: An unknown error occurred.
    """
    if radio in radios:
        radios.pop(radio)
        return "", 200
    else:
        return "Radio not connected", 400


@app.route('/radio/<radio>/configure/<frequency>')
def configure(radio, frequency):
    """Tune the radio to a frequency.
    You must connect to the radio before attempting to configure it.
    The radio must be configured before streaming can be started.
    ---
    parameters:
      - name: radio
        description: Radio device driver name, or serial number.
        in: path
        type: string
        required: true
      - name: frequency
        description: Frequency (in Hz) to tune the radio to.
        in: path
        type: string
        required: true
    responses:
      200:
        description: JSON
      400:
        description: The specified radio is not connected.
    """
    if radio in radios:
        return jsonify(radios[radio].configure(frequency))
    else:
        return "Radio not connected", 400

'''
@app.route('/radio/<radio>/start')
def start_stream(radio):
    pass

@app.route('/radio/<radio>/end')
def end_stream(radio):
    pass
'''

if __name__ == '__main__':
    app.run(debug=True)