# Licensed under a 3-clause BSD style license - see LICENSE.rst
from astropy.utils.collections import HomogeneousList
from textwrap import indent
from astropy.io.votable.exceptions import vo_raise, warn_or_raise
from ...utils.xml.elements import (
    Element, ContentMixin, xmlelement, xmlattribute)
from . import voresource as vr
from .exceptions import (
    W05, W06, W19, W20, W21, W22, W23, W24, W25, W26, W27, W28, W29, W30, W31,
    E06, E08, E09, VOSIError)
__all__ = [
    "TAPCapRestriction", "TableAccess", "DataModelType", "Language", "Version",
    "LanguageFeatureList", "LanguageFeature", "OutputFormat", "UploadMethod",
    "TimeLimits", "DataLimits", "DataLimit"]
INDENT = 4 * " "
######################################################################
# ELEMENT CLASSES
[docs]
class DataModelType(ContentMixin, Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        ivo_id = kwargs.get('ivo-id', None)
        if ivo_id is None:
            warn_or_raise(W26, W26, config=config, pos=pos)
        self.ivo_id = ivo_id
    def __repr__(self):
        return '<DataModel ivo-id={}>{}</DataModel>'.format(
            self.ivo_id, self.content)
[docs]
    def describe(self):
        """
        Prints out a human readable description
        """
        print(f"Datamodel {self.content}")
        print(indent(self.ivo_id, INDENT))
        print() 
    @xmlattribute(name='ivo-id')
    def ivo_id(self):
        """The IVORN of the data model."""
        return self._ivo_id
    @ivo_id.setter
    def ivo_id(self, ivo_id):
        self._ivo_id = ivo_id 
[docs]
class UploadMethod(Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        ivo_id = kwargs.get('ivo-id')
        self.ivo_id = ivo_id
    def __repr__(self):
        return f'<UploadMethod ivo-id="{self.ivo_id}"/>'
[docs]
    def describe(self):
        """
        Prints out a human readable description
        """
        print("Upload method supported")
        print(indent(self.ivo_id, INDENT))
        print() 
    @xmlattribute(name='ivo-id')
    def ivo_id(self):
        """The IVORN of the upload model."""
        return self._ivo_id
    @ivo_id.setter
    def ivo_id(self, ivo_id):
        self._ivo_id = ivo_id 
[docs]
class TimeLimits(Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        self._default = None
        self._hard = None
    def __repr__(self):
        return '<TimeLimits default={} hard={}/>'.format(
            self.default, self.hard)
    @xmlelement(plain=True, multiple_exc=W29)
    def default(self):
        return self._default
    @default.setter
    def default(self, default):
        self._default = int(default)
    @xmlelement(plain=True, multiple_exc=W30)
    def hard(self):
        return self._hard
    @hard.setter
    def hard(self, hard):
        self._hard = int(hard) 
[docs]
class LanguageFeature(Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        self.form = None
        self.description = None
    @xmlelement(plain=True, multiple_exc=W27)
    def form(self):
        return self._form
    @form.setter
    def form(self, form):
        self._form = form
    @xmlelement(plain=True, multiple_exc=W06)
    def description(self):
        return self._description
    @description.setter
    def description(self, description):
        self._description = description
[docs]
    def parse(self, iterator, config):
        super().parse(iterator, config)
        if not self.form:
            vo_raise(E09, self._element_name, config=config, pos=self._pos) 
 
[docs]
class LanguageFeatureList(Element, HomogeneousList):
    def __init__(
        self, config=None, pos=None, _name='languageFeatures', **kwargs
    ):
        Element.__init__(self, config, pos, _name, **kwargs)
        HomogeneousList.__init__(self, LanguageFeature)
        self.type = kwargs.get('type')
        self._features = HomogeneousList(LanguageFeature)
    @xmlattribute
    def type(self):
        return self._type
    @type.setter
    def type(self, type_):
        self._type = type_
    @xmlelement(name='feature', cls=LanguageFeature)
    def features(self):
        return self 
[docs]
class Version(ContentMixin, Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        ivo_id = kwargs.get('ivo-id')
        self.ivo_id = ivo_id
    def __repr__(self):
        return '<Version ivo-id={}>{}</Version>'.format(
            self.ivo_id, self.content)
    @xmlattribute(name='ivo-id')
    def ivo_id(self):
        """The IVORN of the version."""
        return self._ivo_id
    @ivo_id.setter
    def ivo_id(self, ivo_id):
        self._ivo_id = ivo_id 
[docs]
class Language(Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        self.name = None
        self._versions = HomogeneousList(Version)
        self.description = None
        self._languagefeaturelists = HomogeneousList(LanguageFeatureList)
    def __repr__(self):
        return f'<Language>{self.name}</Language>'
[docs]
    def describe(self):
        """
        Prints out a human readable description
        """
        print(f"Language {self.name}")
        for languagefeaturelist in self.languagefeaturelists:
            print(indent(languagefeaturelist.type, INDENT))
            for feature in languagefeaturelist:
                print(indent(feature.form, 2 * INDENT))
                if feature.description:
                    print(indent(feature.description, 3 * INDENT))
                print()
            print() 
[docs]
    def get_feature_list(self, ivoid):
        """
        returns a list of features groupd with the features id ivoid.
        Parameters
        ----------
        ivoid : the ivoid of a TAPRegExt feature list.  It is compared
            case-insensitively against the service's ivoids.
        Returns
        -------
        A (possibly empty) list of `~pyvo.io.vosi.tapregext.LanguageFeature` elements
        """
        ivoid = ivoid.lower()
        for features in self.languagefeaturelists:
            if features.type.lower() == ivoid:
                return features
        return [] 
[docs]
    def get_feature(self, ivoid, form):
        """
        returns the `~pyvo.io.vosi.tapregext.LanguageFeature` with ivoid and form if present.
        We return None rather than raising an error because we expect
        the normal pattern of usage here will be "if feature is present",
        and with None-s this is simpler to write than with exceptions.
        Since it's hard to predict the form of UDFs, for those rather
        use the get_udf method.
        ivoid (regrettably) has to be compared case-insensitively;
        form is compared case-sensitively.
        Parameters
        ----------
        ivoid : str
            The IVOA identifier of the feature group the form is in
        form : str
            The form of the feature requested
        Returns
        -------
        A `~pyvo.io.vosi.tapregext.LanguageFeature` or None.
        """
        for feature in self.get_feature_list(ivoid):
            if feature.form == form:
                return feature
        return None 
[docs]
    def get_udf(self, function_name):
        """
        returns a `~pyvo.io.vosi.tapregext.LanguageFeature` corresponding to an ADQL user defined
        function on the server, on None if the UDF is not available.
        This is a bit heuristic in that it tries to parse the form, which
        is specified only so-so.
        Parameters
        ----------
        function_name : str
            A function name.  This is matched against the server's function
            names case-insensitively, as guided by ADQL's case insensitivity.
        Returns:
            A `~pyvo.io.vosi.tapregext.LanguageFeature` instance or None.
        """
        function_name = function_name.lower()
        for udf in self.get_feature_list(
                "ivo://ivoa.net/std/TAPRegExt#features-udf"):
            this_name = udf.form.split("(")[0].strip()
            if this_name.lower() == function_name:
                return udf
        return None 
    @xmlelement(plain=True, multiple_exc=W05)
    def name(self):
        return self._name
    @name.setter
    def name(self, name):
        self._name = name
    @xmlelement(name='version', cls=Version)
    def versions(self):
        return self._versions
    @xmlelement(plain=True, multiple_exc=W06)
    def description(self):
        return self._description
    @description.setter
    def description(self, description):
        self._description = description
    @xmlelement(name='languageFeatures', cls=LanguageFeatureList)
    def languagefeaturelists(self):
        return self._languagefeaturelists
[docs]
    def parse(self, iterator, config):
        super().parse(iterator, config)
        if not self.name:
            vo_raise(E06, self._element_name, config=config, pos=self._pos)
        if not self.versions:
            vo_raise(E08, self._element_name, config=config, pos=self._pos) 
 
[docs]
class DataLimit(ContentMixin, Element):
    def __init__(self, unit=None, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        self.unit = unit
    def __str__(self):
        return f"{self.unit}:{self.content}"
    @xmlattribute
    def unit(self):
        return self._unit
    @unit.setter
    def unit(self, unit):
        self._unit = unit
    @property
    def content(self):
        return self._content
    @content.setter
    def content(self, content):
        self._content = int(content)
[docs]
    def parse(self, iterator, config):
        super().parse(iterator, config)
        if self.unit not in ('byte', 'row'):
            warn_or_raise(W31, W31, config=config, pos=self._pos) 
 
[docs]
class DataLimits(Element):
    def __init__(self, config=None, pos=None, **kwargs):
        super().__init__(config=config, pos=pos, **kwargs)
        self.default = None
        self.hard = None
    def __repr__(self):
        parts = []
        if self.default is not None:
            parts.append("default={self.default}")
        if self.hard is not None:
            parts.append("hard={self.hard}")
        return '<DataLimits {}/>'.format(" ".join(parts))
    @xmlelement(cls=DataLimit, multiple_exc=W29)
    def default(self):
        return self._default
    @default.setter
    def default(self, default):
        self._default = default
    @xmlelement(cls=DataLimit, multiple_exc=W30)
    def hard(self):
        return self._hard
    @hard.setter
    def hard(self, hard):
        self._hard = hard 
[docs]
class TAPCapRestriction(vr.Capability):
    def __init__(
        self, config=None, pos=None, _name='capability', standardID=None,
        **kwargs
    ):
        if standardID != 'ivo://ivoa.net/std/TAP':
            warn_or_raise(W19, W19, config=config, pos=pos)
        super().__init__(
            config, pos, _name, standardID='ivo://ivoa.net/std/TAP', **kwargs) 
[docs]
@vr.Capability.register_xsi_type('tr:TableAccess')
class TableAccess(TAPCapRestriction):
    def __init__(self, config=None, pos=None, _name='capability', **kwargs):
        super().__init__(config, pos, _name, **kwargs)
        self._datamodels = HomogeneousList(DataModelType)
        self._languages = HomogeneousList(Language)
        self._outputformats = HomogeneousList(OutputFormat)
        self._uploadmethods = HomogeneousList(UploadMethod)
        self.retentionperiod = None
        self.executionduration = None
        self.outputlimit = None
        self.uploadlimit = None
[docs]
    def describe(self):
        """
        Prints out a human readable description
        """
        super().describe()
        for datamodel in self.datamodels:
            datamodel.describe()
        for language in self.languages:
            language.describe()
        for outputformat in self.outputformats:
            outputformat.describe()
        for uploadmethod in self.uploadmethods:
            uploadmethod.describe()
        if self.retentionperiod:
            print("Time a job is kept (in seconds)")
            print(indent(f"Default {self.retentionperiod.default}", INDENT))
            if self.retentionperiod.hard:
                print(indent(f"Maximum {self.retentionperiod.hard}", INDENT))
            print()
        if self.executionduration:
            print("Maximal run time of a job")
            print(indent(f"Default {self.executionduration.default}", INDENT))
            if self.executionduration.hard:
                print(indent(f"Maximum {self.executionduration.hard}", INDENT))
            print()
        if self.outputlimit:
            print("Maximum size of resultsets")
            print(indent("Default {} {}".format(
                self.outputlimit.default.content,
                self.outputlimit.default.unit), INDENT)
            )
            if self.outputlimit.hard:
                print(indent("Maximum {} {}".format(
                    self.outputlimit.hard.content, self.outputlimit.hard.unit), INDENT)
                )
            print()
        if self.uploadlimit and self.uploadlimit.hard:
            print("Maximal size of uploads")
            print(indent("Maximum {} {}".format(
                self.uploadlimit.hard.content, self.uploadlimit.hard.unit), INDENT))
            print() 
[docs]
    def get_adql(self):
        """
        returns the (first) ADQL language element on this service.
        ADQL support is mandatory for IVOA TAP, so in general you can
        rely on this being present.
        """
        for lang in self.languages:
            if lang.name == "ADQL":
                return lang
        raise VOSIError(
            "Invalid TAP service: Does not declare an ADQL language") 
    @xmlelement(name='dataModel', cls=DataModelType)
    def datamodels(self):
        """Identifier of IVOA-approved data model supported by the service."""
        return self._datamodels
    @xmlelement(name='language', cls=Language)
    def languages(self):
        """Languages supported by the service."""
        return self._languages
    @xmlelement(name='outputFormat', cls=OutputFormat)
    def outputformats(self):
        """Output formats supported by the service."""
        return self._outputformats
    @xmlelement(name='uploadMethod', cls=UploadMethod)
    def uploadmethods(self):
        """
        Upload methods supported by the service.
        The absence of upload methods indicates that the service does not
        support uploads at all.
        """
        return self._uploadmethods
    @xmlelement(name='retentionPeriod', cls=TimeLimits, multiple_exc=W22)
    def retentionperiod(self):
        """Limits on the time between job creation and destruction time."""
        return self._retentionperiod
    @retentionperiod.setter
    def retentionperiod(self, retentionperiod):
        self._retentionperiod = retentionperiod
    @xmlelement(name='executionDuration', cls=TimeLimits, multiple_exc=W23)
    def executionduration(self):
        """Limits on executionDuration."""
        return self._executionduration
    @executionduration.setter
    def executionduration(self, executionduration):
        self._executionduration = executionduration
    @xmlelement(name='outputLimit', cls=DataLimits, multiple_exc=W24)
    def outputlimit(self):
        """Limits on the size of data returned."""
        return self._outputlimit
    @outputlimit.setter
    def outputlimit(self, outputlimit):
        self._outputlimit = outputlimit
    @xmlelement(name='uploadLimit', cls=DataLimits, multiple_exc=W25)
    def uploadlimit(self):
        return self._uploadlimit
    @uploadlimit.setter
    def uploadlimit(self, uploadlimit, cls=DataLimits):
        self._uploadlimit = uploadlimit
[docs]
    def parse(self, iterator, config):
        super().parse(iterator, config)
        if not self.languages:
            warn_or_raise(W20, W20, config=config, pos=self._pos)
        if not self.outputformats:
            warn_or_raise(W21, W21, config=config, pos=self._pos)