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

# See COPYING file for copyrights details.

from __future__ import absolute_import

from itertools import repeat, islice, chain
import wx

from connectors.SchemeEditor import SchemeEditor


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

secure_model = model + [('ID',_("ID:"))]

models = [("LOCAL", []), ("PYRO",model), ("PYROS",secure_model)]

Schemes = list(zip(*models)[0])

ModelsDict = dict(models)

class PYRO_dialog(SchemeEditor):
    def __init__(self, scheme, *args, **kwargs):
        self.model = ModelsDict[scheme]
        SchemeEditor.__init__(self, scheme, *args, **kwargs)

    def SetLoc(self, loc):
        hostport, ID = list(islice(chain(loc.split("#"), repeat("")),2))
        host, port = list(islice(chain(hostport.split(":"), repeat("")),2))
        self.SetFields(locals())

    def GetLoc(self):
        if self.model:
            fields = self.GetFields()
            template = "{host}"
            if fields['port']:
                template += ":{port}" 

            return template.format(**fields)
        return ''