from __future__ import absolute_import

from mercurial import (

    error,

    registrar,

    revset,

    util,

)

from . import (

    destination,

    stack,

)

try:

    mkmatcher = revset._stringmatcher

except AttributeError:

    try:

        from mercurial.utils import stringutil

        mkmatcher = stringutil.stringmatcher

    except (ImportError, AttributeError):

        mkmatcher = util.stringmatcher

revsetpredicate = registrar.revsetpredicate()

def getstringstrict(x, err):

    if x and x[0] == b'string':

        return x[1]

    raise error.ParseError(err)

@revsetpredicate(b'topic([string or set])')

def topicset(repo, subset, x):

    """All changesets with the specified topic or the topics of the given

    changesets. Without the argument, all changesets with any topic specified.

    If `string` starts with `re:` the remainder of the name is treated

    as a regular expression.

    """

    args = revset.getargs(x, 0, 1, b'topic takes one or no arguments')

    mutable = revset._notpublic(repo, revset.fullreposet(repo), ())

    if not args:

        return (subset & mutable).filter(lambda r: bool(repo[r].topic()))

    try:

        topic = getstringstrict(args[0], b'')

    except error.ParseError:

        # not a string, but another revset

        pass

    else:

        kind, pattern, matcher = mkmatcher(topic)

        if topic.startswith(b'literal:') and pattern not in repo.topics:

            raise error.RepoLookupError(b"topic '%s' does not exist" % pattern)

        def matches(r):

            topic = repo[r].topic()

            if not topic:

                return False

            return matcher(topic)

        return (subset & mutable).filter(matches)

    s = revset.getset(repo, revset.fullreposet(repo), x)

    topics = {repo[r].topic() for r in s}

    topics.discard(b'')

    def matches(r):

        if r in s:

            return True

        topic = repo[r].topic()

        if not topic:

            return False

        return topic in topics

    return (subset & mutable).filter(matches)

@revsetpredicate(b'ngtip([branch])')

def ngtipset(repo, subset, x):

    """The untopiced tip.

    Name is horrible so that people change it.

    """

    args = revset.getargs(x, 1, 1, b'ngtip takes one argument')

    # match a specific topic

    branch = revset.getstring(args[0], b'ngtip requires a string')

    if branch == b'.':

        branch = repo[b'.'].branch()

    return subset & revset.baseset(destination.ngtip(repo, branch))

@revsetpredicate(b'stack()')

def stackset(repo, subset, x):

    """All relevant changes in the current topic,

    This is roughly equivalent to 'topic(.) - obsolete' with a sorting moving

    unstable changeset after there future parent (as if evolve where already

    run).

    """

    err = b'stack takes no arguments, it works on current topic'

    revset.getargs(x, 0, 0, err)

    topic = None

    branch = None

    if repo.currenttopic:

        topic = repo.currenttopic

    else:

        branch = repo[None].branch()

    return revset.baseset(stack.stack(repo, branch=branch, topic=topic)[1:]) & subset

if util.safehasattr(revset, 'subscriptrelations'):

    def stackrel(repo, subset, x, rel, z, order):

        """This is a revset-flavored implementation of stack aliases.

        The syntax is: rev#stack[n] or rev#s[n]. Plenty of logic is borrowed

        from topic._namemap, but unlike that function, which prefers to abort

        (e.g. when stack index is too high), this returns empty set to be more

        revset-friendly.

        """

        # hg 4.9 provides a number or None, hg 5.0 provides a tuple of tokens

        if isinstance(z, tuple):

            a, b = revset.getintrange(

                z,

                b'relation subscript must be an integer or a range',

                b'relation subscript bounds must be integers',

                None, None)

        else:

            # hg <= 4.9 (431cf2c8c839+13f7a6a4f0db)

            a = b = z

        s = revset.getset(repo, revset.fullreposet(repo), x)

        if not s:

            return revset.baseset()

        def getrange(st, a, b):

            start = 1 if a is None else a

            end = len(st.revs) if b is None else b + 1

            return range(start, end)

        revs = []

        for r in s:

            topic = repo[r].topic()

            if topic:

                st = stack.stack(repo, topic=topic)

            else:

                st = stack.stack(repo, branch=repo[r].branch())

            for n in getrange(st, a, b):

                if abs(n) >= len(st.revs):

                    # also means stack base is not accessible with n < 0, which

                    # is by design

                    continue

                if n == 0 and b != 0 and a != 0:

                    # quirk: we don't want stack base unless specifically asked

                    # for it (at least one of the indices is 0)

                    continue

                rev = st.revs[n]

                if rev == -1 and n == 0:

                    continue

                if rev not in revs:

                    revs.append(rev)

        return subset & revset.baseset(revs)

    revset.subscriptrelations[b'stack'] = stackrel

    revset.subscriptrelations[b's'] = stackrel

    def topicrel(repo, subset, x, *args):

        subset &= topicset(repo, subset, x)

        # not using revset.generationssubrel directly because it was renamed

        # hg <= 5.3 (8859de3e83dc)

        generationssubrel = revset.subscriptrelations[b'generations']

        return generationssubrel(repo, subset, x, *args)

    revset.subscriptrelations[b'topic'] = topicrel

    revset.subscriptrelations[b't'] = topicrel