from owrx.source.connector import ConnectorSource, ConnectorDeviceDescription
from owrx.command import Argument, Flag, Option
from owrx.form.input import Input, DropdownInput, DropdownEnum, CheckboxInput
from owrx.form.input.device import RemoteInput
from owrx.form.input.validator import Range
from typing import List


class RundsSource(ConnectorSource):
    def getCommandMapper(self):
        return (
            super()
            .getCommandMapper()
            .setBase("runds_connector")
            .setMappings(
                {
                    "long": Flag("-l"),
                    "remote": Argument(),
                    "protocol": Option("-m"),
                }
            )
        )


class ProtocolOptions(DropdownEnum):
    PROTOCOL_EB200 = ("eb200", "EB200 protocol")
    PROTOCOL_AMMOS = ("ammos", "Ammos protocol")

    def __new__(cls, *args, **kwargs):
        value, description = args
        obj = object.__new__(cls)
        obj._value_ = value
        obj.description = description
        return obj

    def __str__(self):
        return self.description


class RundsDeviceDescription(ConnectorDeviceDescription):
    def getName(self):
        return "R&S device using EB200 or Ammos protocol"

    def supportsPpm(self):
        # currently not implemented in the connector
        return False

    def getInputs(self) -> List[Input]:
        return super().getInputs() + [
            RemoteInput(),
            DropdownInput("protocol", "Protocol", ProtocolOptions),
            CheckboxInput("long", "Use 32-bit sample size (LONG)"),
        ]

    def getDeviceMandatoryKeys(self):
        return super().getDeviceMandatoryKeys() + ["remote"]

    def getDeviceOptionalKeys(self):
        return super().getDeviceOptionalKeys() + ["protocol", "long"]

    def getSampleRateRanges(self) -> List[Range]:
        # can't be very specific here due to the wide range of devices, so this is more of a sanity check.
        return [Range(0, 20000000)]