# copyright 2010-2011 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

"""Specific views for data sources and related entities (eg CWSource,

CWSourceHostConfig, CWSourceSchemaConfig).

"""

from cubicweb.selectors import is_instance, score_entity, match_user_groups

_pvs = uicfg.primaryview_section

_pvs.tag_object_of(('*', 'cw_for_source', 'CWSource'), 'hidden')

MAPPED_SOURCE_TYPES = set( ('pyrorql', 'datafeed') )

                  & match_user_groups('managers')

                  & score_entity(lambda x:x.type in MAPPED_SOURCE_TYPES))

        self.w('<h3>%s</h3>' % _('Entity and relation supported by this source'))

        rset = self._cw.execute(

            'Any X, SCH, XO ORDERBY ET WHERE X options XO, X cw_for_source S, S eid %(s)s, '

            'X cw_schema SCH, SCH is ET', {'s': entity.eid})

        self.wview('table', rset, 'noresult')

        # self.w('<h3>%s</h3>' % _('Relations that should not be crossed'))

        # self.w('<p>%s</p>' % _(

        #     'By default, when a relation is not supported by a source, it is '

        #     'supposed that a local relation may point to an entity from the '

        #     'external source. Relations listed here won\'t have this '

        #     '"crossing" behaviour.'))

        # self.wview('list', entity.related('cw_dont_cross'), 'noresult')

        # self.w('<h3>%s</h3>' % _('Relations that can be crossed'))

        # self.w('<p>%s</p>' % _(

        #     'By default, when a relation is supported by a source, it is '

        #     'supposed that a local relation can\'t point to an entity from the '

        #     'external source. Relations listed here may have this '

        #     '"crossing" behaviour anyway.'))

        # self.wview('list', entity.related('cw_may_cross'), 'noresult')

        checker = MAPPING_CHECKERS.get(entity.type, MappingChecker)(entity)

        checker.check()

        if (checker.errors or checker.warnings or checker.infos):

                errors = zip(repeat(_('error')), checker.errors)

                warnings = zip(repeat(_('warning')), checker.warnings)

                infos = zip(repeat(_('warning')), checker.infos)

class MappingChecker(object):

    def __init__(self, cwsource):

        self.cwsource = cwsource

        self.errors = []

        self.warnings = []

        self.infos = []

        self.schema = cwsource._cw.vreg.schema

    def init(self):

        # supported entity types

        self.sentities = set()

        # supported relations

        self.srelations = {}

        # avoid duplicated messages

        self.seen = set()

        # first get mapping as dict/sets

        for schemacfg in self.cwsource.reverse_cw_for_source:

            self.init_schemacfg(schemacfg)

    def init_schemacfg(self, schemacfg):

        cwerschema = schemacfg.schema

        if cwerschema.__regid__ == 'CWEType':

            self.sentities.add(cwerschema.name)

        elif cwerschema.__regid__ == 'CWRType':

            assert not cwerschema.name in self.srelations

            self.srelations[cwerschema.name] = None

        else: # CWAttribute/CWRelation

            self.srelations.setdefault(cwerschema.rtype.name, []).append(

                (cwerschema.stype.name, cwerschema.otype.name) )

    def check(self):

        self.init()

        error = self.errors.append

        warning = self.warnings.append

        info = self.infos.append

        for etype in self.sentities:

            eschema = self.schema[etype]

            for rschema, ttypes, role in eschema.relation_definitions():

                if rschema in META_RTYPES:

                    continue

                ttypes = [ttype for ttype in ttypes if ttype in self.sentities]

                if not rschema in self.srelations:

                    for ttype in ttypes:

                        rdef = rschema.role_rdef(etype, ttype, role)

                        self.seen.add(rdef)

                        if rdef.role_cardinality(role) in '1+':

                            error(_('relation %(type)s with %(etype)s as %(role)s '

                                    'and target type %(target)s is mandatory but '

                                    'not supported') %

                                  {'rtype': rschema, 'etype': etype, 'role': role,

                                   'target': ttype})

                        elif ttype in self.sentities:

                            warning(_('%s could be supported') % rdef)

                elif not ttypes:

                    warning(_('relation %(rtype)s with %(etype)s as %(role)s is '

                              'supported but no target type supported') %

                            {'rtype': rschema, 'role': role, 'etype': etype})

        for rtype in self.srelations:

            rschema = self.schema[rtype]

            for subj, obj in rschema.rdefs:

                if subj in self.sentities and obj in self.sentities:

                    break

                error(_('relation %s is supported but none if its definitions '

                        'matches supported entities') % rtype)

        self.custom_check()

    def custom_check(self):

        pass

class PyroRQLMappingChecker(MappingChecker):

    """pyrorql source mapping checker"""

    def init(self):

        self.dontcross = set()

        self.maycross = set()

        super(PyroRQLMappingChecker, self).init()

    def init_schemacfg(self, schemacfg):

        options = schemacfg.options or ()

        if 'dontcross' in options:

            self.dontcross.add(schemacfg.schema.name)

        else:

            super(PyroRQLMappingChecker, self).init_schemacfg(schemacfg)

            if 'maycross' in options:

                self.maycross.add(schemacfg.schema.name)

    def custom_check(self):

        error = self.errors.append

        info = self.infos.append

        for etype in self.sentities:

            eschema = self.schema[etype]

            for rschema, ttypes, role in eschema.relation_definitions():

                if rschema in META_RTYPES:

                    continue

                if not rschema in self.srelations:

                    if rschema not in self.dontcross:

                        if role == 'subject' and rschema.inlined:

                            error(_('inlined relation %(rtype)s of %(etype)s '

                                    'should be supported') %

                                  {'rtype': rschema, 'etype': etype})

                        elif (rschema not in self.seen and rschema not in self.maycross):

                            info(_('you may want to specify something for %s') %

                                 rschema)

                            self.seen.add(rschema)

                elif rschema in self.maycross and rschema.inlined:

MAPPING_CHECKERS = {

    'pyrorql': PyroRQLMappingChecker,

    }