"""
Tests for some application-specific db invariants
"""
import itertools

from sqlalchemy import func, or_, and_, literal, union_all, select, true
from sqlalchemy import orm

from clld.scripts.util import parsed_args
from clld.db.meta import DBSession
from clld.db.models import Config, Language, LanguageIdentifier, Identifier, ValueSet

from glottolog3.models import Languoid, LanguoidLevel, TreeClosureTable, Ref


class CheckMeta(type):

    __instances = []

    def __init__(self, name, bases, dct):
        super(CheckMeta, self).__init__(name, bases, dct)
        if 'invalid_query' in dct:
            self.__instances.append(self)

    def __iter__(self):
        return iter(self.__instances)


class Check(metaclass=CheckMeta):

    detail = True

    def __init__(self):
        self.query = self.invalid_query(DBSession)

    def validate(self):
        self.invalid_count = self.query.count()
        print(self)
        if self.invalid_count:  # pragma: no cover
            if self.detail:
                self.invalid = self.query.all()
                self.display()
            return False
        else:
            self.invalid = ()
            return True

    def invalid_query(self, session, **kw):
        raise NotImplementedError()  # pragma: no cover

    def display(self, number=25):  # pragma: no cover
        ids = (i.id for i in itertools.islice(self.invalid, number))
        cont = ', ...' if number < self.invalid_count else ''
        print('    %s%s' % (', '.join(ids), cont))

    def __str__(self):
        if self.invalid_count:  # pragma: no cover
            msg = '%d invalid\n    (violating %s)' % (self.invalid_count, self.__doc__)
        else:
            msg = 'OK'
        return '%s: %s' % (self.__class__.__name__, msg)


class FamiliesDistinct(Check):
    """Each family node has a unique set of member languages."""

    def invalid_query(self, session, exclude=u'Unclassified'):
        member = orm.aliased(Languoid, flat=True)
        extent = func.array(
            session.query(member.pk)
            .filter_by(active=True, level=LanguoidLevel.language)
            .join(TreeClosureTable, TreeClosureTable.child_pk == member.pk)
            .filter_by(parent_pk=Languoid.pk)
            .order_by(member.pk).as_scalar())
        cte = session.query(Languoid.id, extent.label('extent'))\
            .filter_by(active=True, level=LanguoidLevel.family)\
            .filter(~Languoid.name.startswith(exclude)).cte()
        dup = orm.aliased(cte)
        return session.query(cte.c.id)\
            .filter(session.query(dup).filter(
                dup.c.id != cte.c.id, dup.c.extent == cte.c.extent).exists())\
            .order_by(cte.c.extent, cte.c.id)


class DialectFather(Check):
    """Father of a dialect is a language or dialect."""

    def invalid_query(self, session, **kw):
        return session.query(Languoid)\
            .filter_by(active=True, level=LanguoidLevel.dialect)\
            .order_by(Languoid.id)\
            .join(Languoid.father, aliased=True)\
            .filter(Languoid.level.notin_(
                [LanguoidLevel.language, LanguoidLevel.dialect]))


class FamilyChildren(Check):
    """Family has at least one subfamily or language."""

    def invalid_query(self, session, **kw):
        return session.query(Languoid)\
            .filter_by(active=True, level=LanguoidLevel.family)\
            .filter(~Languoid.children.any(and_(
                Languoid.active == true(),
                Languoid.level.in_([LanguoidLevel.family, LanguoidLevel.language]))))\
            .order_by(Languoid.id)


class FatherFamily(Check):
    """Languoids have correct top-level family."""

    def invalid_query(self, session, **kw):
        cte = session.query(Languoid.pk, Languoid.father_pk)\
            .cte(recursive=True)
        parent = orm.aliased(Languoid)
        cte = cte.union_all(session.query(cte.c.pk, parent.father_pk)
            .join(parent, cte.c.father_pk == parent.pk)
            .filter(parent.father_pk != None))  # noqa
        family = orm.aliased(Languoid)
        return session.query(Languoid).join(cte, Languoid.pk == cte.c.pk)\
            .outerjoin(family, and_(
                cte.c.father_pk == family.pk, family.father_pk == None))\
            .filter(Languoid.family_pk != family.pk)\
            .order_by(Languoid.id)  # noqa


class TreeClosure(Check):
    """Treeclosuretable is correct."""

    detail = False

    def invalid_query(self, session, **kw):
        cte = session.query(
            Languoid.pk, Languoid.pk.label('father_pk'), literal(0).label('depth'))\
            .cte(recursive=True)
        parent = orm.aliased(Languoid)
        cte = cte.union_all(session.query(cte.c.pk, parent.father_pk, cte.c.depth + 1)
            .join(parent, cte.c.father_pk == parent.pk)
            .filter(parent.father_pk != None))  # noqa
        tree1 = session.query(
            TreeClosureTable.child_pk,
            TreeClosureTable.parent_pk,
            TreeClosureTable.depth)
        tree2 = session.query(cte.c.pk, cte.c.father_pk, cte.c.depth)
        diff = union_all(tree1.except_all(tree2), tree2.except_all(tree1))
        return session.query(diff.alias())


class ChildCounts(Check):
    """Languoids have correct child family/language/dialect counts."""

    def invalid_query(self, session, **kw):
        cte = session.query(Languoid.pk, Languoid.father_pk, Languoid.level)\
            .filter(Languoid.father_pk != None).cte(recursive=True)  # noqa
        parent = orm.aliased(Languoid)
        cte = cte.union_all(session.query(cte.c.pk, parent.father_pk, cte.c.level)\
            .join(parent, cte.c.father_pk == parent.pk)\
            .filter(parent.father_pk != None))  # noqa
        return session.query(Languoid)\
            .outerjoin(cte, Languoid.pk == cte.c.father_pk)\
            .group_by(Language.pk, Languoid.pk)\
            .having(or_(
                func.coalesce(Languoid.child_family_count, -1) !=
                    func.count(func.nullif(cte.c.level != LanguoidLevel.family, True)),
                func.coalesce(Languoid.child_language_count, -1) !=
                    func.count(func.nullif(cte.c.level != LanguoidLevel.language, True)),
                func.coalesce(Languoid.child_dialect_count, -1) !=
                    func.count(func.nullif(cte.c.level != LanguoidLevel.dialect, True))))\
            .order_by((Languoid.id))


class FamilyLanguages(Check):
    """Family has at least two languages."""

    def invalid_query(self, session, exclude=None):
        exclude = ['Unclassified', 'Unattested'] if exclude is None else exclude
        child = orm.aliased(Languoid, flat=True)
        return session.query(Languoid)\
            .filter_by(active=True, level=LanguoidLevel.family)\
            .filter(Languoid.family.has(Languoid.name.notin_(exclude)))\
            .join(TreeClosureTable, TreeClosureTable.parent_pk == Languoid.pk)\
            .outerjoin(child, and_(
                TreeClosureTable.child_pk == child.pk,
                TreeClosureTable.depth > 0,
                child.level == LanguoidLevel.language))\
            .group_by(Language.pk, Languoid.pk)\
            .having(func.count(child.pk) < 2)\
            .order_by(Languoid.id)


class BookkeepingNoChildren(Check):
    """Bookkeeping languoids lack children."""

    def invalid_query(self, session, **kw):
        return session.query(Languoid)\
            .filter(Languoid.category == 'Bookkeeping')\
            .filter(Languoid.children.any())\
            .order_by(Languoid.id)


class IsolateInactive(Check):
    """Inactive languoids lack parent and children."""

    def invalid_query(self, session, **kw):
        return session.query(Languoid)\
            .filter_by(active=False).filter(or_(
                Languoid.father_pk != None,
                Languoid.children.any()))\
            .order_by(Languoid.id)  # noqa


class UniqueIsoCode(Check):
    """Active languoids do not share iso639-3 identifiers."""

    @staticmethod
    def _ident_query(session, type=u'iso639-3'):
        lang = orm.aliased(Languoid)
        ident = orm.aliased(Identifier)
        query = session.query(lang).filter_by(active=True)\
            .join(LanguageIdentifier, LanguageIdentifier.language_pk == lang.pk)\
            .join(ident, and_(
                LanguageIdentifier.identifier_pk == ident.pk, ident.type == type))
        return lang, ident, query

    def invalid_query(self, session, **kw):
        lang, ident, query = self._ident_query(session)
        other, other_ident, other_query = self._ident_query(session)
        return query.filter(other_query.filter(
            other.pk != lang.pk, ident.name == other_ident.name).exists())\
            .order_by(lang.id)


class CleanName(Check):
    """Glottolog names lack problematic characters."""

    def invalid_query(self, session, type='name',
                      description=Languoid.GLOTTOLOG_NAME, **kw):
        return session.query(Languoid)\
            .filter(Languoid.identifiers.any(or_(
                Identifier.name.op('~')(r'^\s|\s$'),
                Identifier.name.op('~')(r'[`_*:\xa4\xab\xb6\xbc]'),
            ), type=type, description=description))\
            .order_by(Languoid.id)


class UniqueName(Check):
    """Among active languages Glottolog names are unique."""

    @staticmethod
    def _ident_query(session, type='name',
                     description=Languoid.GLOTTOLOG_NAME):
        lang = orm.aliased(Languoid)
        ident = orm.aliased(Identifier)
        query = session.query(lang).filter_by(
            active=True, level=LanguoidLevel.language)\
            .join(LanguageIdentifier, LanguageIdentifier.language_pk == lang.pk)\
            .join(ident, and_(
                LanguageIdentifier.identifier_pk == ident.pk,
                ident.type == type, ident.description == description))
        return lang, ident, query

    def invalid_query(self, session, **kw):
        lang, ident, query = self._ident_query(session)
        other, other_ident, other_query = self._ident_query(session)
        return query.filter(other_query.filter(
            other.pk != lang.pk, ident.name == other_ident.name).exists())\
            .order_by(lang.id)


class RefRedirects(Check):
    """Redirects of reference ids target an unredirected id."""

    def invalid_query(self, session, **kw):
        return session.query(
            func.regexp_replace(Config.key, u'\D', '', u'g').label('id'),
            func.nullif(Config.value, u'__gone__').label('target'))\
            .filter(Config.key.like(u'__Source_%%__'))\
            .filter(
                session.query(orm.aliased(Config))
                .filter_by(key=func.format(u'__Source_%s__', Config.value)).exists())\
            .order_by('id', 'target')


class MarkupRefLinks(Check):
    """Classification description source links are valid."""

    def invalid_query(self, session, **kw):
        vs_rid = select([
            ValueSet.pk,
            func.unnest(func.regexp_matches(
                ValueSet.description, '\*\*(\d+)\*\*', 'g')).label('ref_id')]).alias()
        return session.query(ValueSet)\
            .filter(ValueSet.pk.in_(
                session.query(vs_rid.c.pk)
                .filter(~session.query(Ref).filter_by(id=vs_rid.c.ref_id).exists())))\
            .order_by(ValueSet.id)


class RefPages(Check):
    """References do not have zero/negative page count."""

    def invalid_query(self, session, **kw):
        return session.query(Ref)\
            .filter(Ref.pages_int < 1)\
            .order_by(Ref.pk)


def main(args):
    for cls in Check:
        if cls.__name__ != 'Check':
            check = cls()
            check.validate()


if __name__ == '__main__':  # pragma: no cover
    main(parsed_args())
