from pydantic import BaseModel
from typing import List, Optional
from enum import Enum


class BotConfig(BaseModel):
    token: str

class VoiceChannelRequest(BaseModel):
    guild_id: int
    channel_id: int

class DecodeMode(str, Enum):
    P25 = "P25"
    DMR = "DMR"
    ANALOG = "NBFM"

class TalkgroupTag(BaseModel):
    talkgroup: str
    tagDec: int

class ConfigGenerator(BaseModel):
    type: DecodeMode
    systemName: str
    channels: List[str]
    tags: Optional[List[TalkgroupTag]]
    whitelist: Optional[List[int]]

class DemodType(str, Enum):
    CQPSK = "cqpsk"
    FSK4 = "fsk4"

class FilterType(str, Enum):
    RC = "rc"
    WIDEPULSE = "widepulse"

class ChannelConfig(BaseModel):
    name: str
    trunking_sysname: Optional[str]
    enable_analog: str
    demod_type: DemodType
    filter_type: FilterType
    device: Optional[str] = "sdr"
    cqpsk_tracking: Optional[bool] = None
    frequency: Optional[float] = None
    nbfmSquelch: Optional[float] = None
    destination: Optional[str] = "udp://127.0.0.1:23456"
    tracking_threshold: Optional[int] = 120
    tracking_feedback: Optional[float] = 0.75
    excess_bw: Optional[float] = 0.2
    if_rate: Optional[int] = 24000
    plot: Optional[str] = ""
    symbol_rate: Optional[int] = 4800
    blacklist: Optional[str] = ""
    whitelist: Optional[str] = ""

class DeviceConfig(BaseModel):
    args: Optional[str] = "rtl"
    gains: Optional[str] = "lna:39"
    gain_mode: Optional[bool] = False
    name: Optional[str] = "sdr"
    offset: Optional[int] = 0
    ppm: Optional[float] = 0.0
    rate: Optional[int] = 1920000
    usable_bw_pct: Optional[float] = 0.85
    tunable: Optional[bool] = True

class TrunkingChannelConfig(BaseModel):
    sysname: str
    control_channel_list: str
    tagsFile: Optional[str] = None
    whitelist: Optional[str] = None
    nac: Optional[str] = ""
    wacn: Optional[str] = ""
    tdma_cc: Optional[bool] = False
    crypt_behavior: Optional[int] = 2

class TrunkingConfig(BaseModel):
    module: str
    chans: List[TrunkingChannelConfig]

class AudioInstanceConfig(BaseModel):
    instance_name: Optional[str] = "audio0"
    device_name: Optional[str] = "pulse"
    udp_port: Optional[int] = 23456
    audio_gain: Optional[float] = 2.5
    number_channels: Optional[int] = 1

class AudioConfig(BaseModel):
    module: Optional[str] = "sockaudio.py"
    instances: Optional[List[AudioInstanceConfig]] = [AudioInstanceConfig()]

class TerminalConfig(BaseModel):
    module: Optional[str] = "terminal.py"
    terminal_type: Optional[str] = "http:0.0.0.0:8081"
    terminal_timeout: Optional[float] = 5.0
    curses_plot_interval: Optional[float] = 0.2
    http_plot_interval: Optional[float] = 1.0
    http_plot_directory: Optional[str] = "../www/images"
    tuning_step_large: Optional[int] = 1200
    tuning_step_small: Optional[int] = 100