#!/usr/bin/env python
# -*- coding: utf-8 -*-

# See COPYING file for copyrights details.

from itertools import repeat, islice, chain

## URI parsing functions

def split_as_dict(s, sep, labels):
    return dict(zip(labels, islice(chain(s.split(sep), repeat("")), len(labels))))

def parse_tcp(loc):
    return split_as_dict(loc, ":", ["host", "port"])

def parse_sslpsk(loc):
    locals().update(**split_as_dict(loc, "#", ["hostport", "ID"]))
    return dict(**parse_tcp(hostport), ID=ID) # type: ignore

def parse_serial(loc):
    return split_as_dict(loc, "@", ["device", "baudrate"])

def parse_usb(loc):
    return split_as_dict(loc, ":", ["VID", "PID", "serialnumber"])

## URI building functions

def build_tcp(fields):
    if fields['port']:
        return "{host}:{port}".format(**fields)
    return fields['host']

def build_sslpsk(fields):
    return "{hostport}#{ID}".format(hostport=build_tcp(fields), **fields)

def build_serial(fields):
    if fields['baudrate']:
        return "{device}@{baudrate}".format(**fields)
    return fields['device']

def build_usb(fields):
    if fields['serialnumber']:
        return "{VID}:{PID}:{serialnumber}".format(**fields)
    if fields['PID']:
        return "{VID}:{PID}".format(**fields)
    return fields['VID']
    
## Dialog fields definition

model_tcp = [('host', _("Host:")),
             ('port', _("Port:"))]

model_serial = [('device', _("Device:")),
                ('baudrate', _("Baud rate:"))]

model_usb = [('VID', _("Vendor ID:")),
             ('PID', _("Product ID:")),
             ('serialnumber', _("Serial number:"))]


## Schemes description

schemes_desc = [
#   ( scheme name ,  data model , use ID,   parser   ,    builder  )
    ("LOCAL",       [],           False, lambda x: {}, lambda x: ""),
    ("ERPC",        model_tcp,    False, parse_tcp,    build_tcp   ),
    ("ERPCS",       model_tcp,    True,  parse_sslpsk, build_sslpsk),
    ("ERPC-SERIAL", model_serial, False, parse_serial, build_serial),
    ("ERPC-USB",    model_usb,    False, parse_usb,    build_usb   )]

per_scheme_model = {sch: desc for sch, *desc in schemes_desc}