from plivo.utils.validators import *
from plivo.xml import (PlivoXMLElement,
                       map_type,
                       BreakElement,
                       )


class LangElement(PlivoXMLElement):
    _name = 'lang'
    _nestable = [
        "break",
        "emphasis",
        "lang",
        "p",
        "phoneme",
        "prosody",
        "s",
        "say-as",
        "sub",
        "w"
    ]

    @property
    def xmllang(self):
        return self.__xmllang

    @xmllang.setter
    def xmllang(self, value):
        self.__xmllang = six.text_type(
            value) if value is not None else None

    @validate_args(
        value=[of_type(six.text_type)],
    )
    def set_xmllang(self, value):
        self.xmllang = value
        return self

    def __init__(
            self,
            content=None,
            xmllang=None,
    ):

        super(LangElement, self).__init__()
        self.content = content
        self.xmllang = xmllang

    def to_dict(self):
        d = {
            '{http://www.w3.org/XML/1998/namespace}lang': self.xmllang,
        }

        return {
            k: six.text_type(map_type(v))
            for k, v in d.items() if v is not None
            }

    def add_break(
            self,
            strength=None,
            time=None
    ):
        self.add(
            BreakElement(
                strength=strength,
                time=time,
            ))
        return self

    def add_emphasis(
            self,
            content,
            level=None,
    ):
        from .emphasisElement import EmphasisElement

        self.add(
            EmphasisElement(
                content=content,
                level=level,
            ))
        return self

    def add_lang(
            self,
            content,
            xmllang=None,
    ):
        self.add(
            LangElement(
                content=content,
                xmllang=xmllang,
            ))
        return self

    def add_p(
            self,
            content,
    ):
        from .pElement import PElement

        self.add(
            PElement(
                content=content,
            ))
        return self

    def add_phoneme(
            self,
            content,
            alphabet=None,
            ph=None,
    ):
        from .phonemeElement import PhonemeElement

        self.add(
            PhonemeElement(
                content=content,
                alphabet=alphabet,
                ph=ph,
            ))
        return self

    def add_prosody(
            self,
            content,
            volume=None,
            rate=None,
            pitch=None,
    ):
        from .prosodyElement import ProsodyElement

        self.add(
            ProsodyElement(
                content=content,
                volume=volume,
                rate=rate,
                pitch=pitch,
            ))
        return self

    def add_s(
            self,
            content,
    ):
        from .sElement import SElement

        self.add(
            SElement(
                content=content,
            ))
        return self

    def add_say_as(
            self,
            content,
            interpret_as=None,
            format=None,
    ):
        from .sayAsElement import SayAsElement

        self.add(
            SayAsElement(
                content=content,
                interpret_as=interpret_as,
                format=format,
            ))
        return self

    def add_sub(
            self,
            content,
            alias=None,
    ):
        from .subElement import SubElement

        self.add(
            SubElement(
                content=content,
                alias=alias,
            ))
        return self

    def add_w(
            self,
            content,
            role=None,
    ):
        from .wElement import WElement

        self.add(
            WElement(
                content=content,
                role=role,
            ))
        return self