# __init__.py - topic extension

#

# This software may be used and distributed according to the terms of the

# GNU General Public License version 2 or any later version.

"""support for topic branches

Topic branches are lightweight branches which disappear when changes are

finalized (move to the public phase).

Compared to bookmark, topic is reference carried by each changesets of the

series instead of just the single head revision.  Topic are quite similar to

the way named branch work, except they eventually fade away when the changeset

becomes part of the immutable history. Changeset can belong to both a topic and

a named branch, but as long as it is mutable, its topic identity will prevail.

As a result, default destination for 'update', 'merge', etc...  will take topic

into account. When a topic is active these operations will only consider other

changesets on that topic (and, in some occurrence, bare changeset on same

branch).  When no topic is active, changeset with topic will be ignored and

only bare one on the same branch will be taken in account.

There is currently two commands to be used with that extension: 'topics' and

'stack'.

The 'hg topics' command is used to set the current topic, change and list

existing one. 'hg topics --verbose' will list various information related to

each topic.

The 'stack' will show you information about the stack of commit belonging to

your current topic.

Topic is offering you aliases reference to changeset in your current topic

stack as 't#'. For example, 't1' refers to the root of your stack, 't2' to the

second commits, etc. The 'hg stack' command show these number.

Push behavior will change a bit with topic. When pushing to a publishing

repository the changesets will turn public and the topic data on them will fade

away. The logic regarding pushing new heads will behave has before, ignore any

topic related data. When pushing to a non-publishing repository (supporting

topic), the head checking will be done taking topic data into account.

Push will complain about multiple heads on a branch if you push multiple heads

with no topic information on them (or multiple public heads). But pushing a new

topic will not requires any specific flag. However, pushing multiple heads on a

topic will be met with the usual warning.

The 'evolve' extension takes 'topic' into account. 'hg evolve --all'

will evolve all changesets in the active topic. In addition, by default. 'hg

next' and 'hg prev' will stick to the current topic.

Be aware that this extension is still an experiment, commands and other features

are likely to be change/adjusted/dropped over time as we refine the concept.

topic-mode

==========

The topic extension can be configured to ensure the user do not forget to add

a topic when committing a new topic::

    [experimental]

    # behavior when commit is made without an active topic

    topic-mode = ignore # do nothing special (default)

    topic-mode = warning # print a warning

    topic-mode = enforce # abort the commit (except for merge)

    topic-mode = enforce-all # abort the commit (even for merge)

    topic-mode = random # use a randomized generated topic (except for merge)

    topic-mode = random-all # use a randomized generated topic (even for merge)

Single head enforcing

=====================

The extensions come with an option to enforce that there is only one heads for

each name in the repository at any time.

::

    [experimental]

    enforce-single-head = yes

Publishing behavior

===================

Topic vanish when changeset move to the public phases. Moving to the public

phase usually happens on push, but it is possible to update that behavior. The

server needs to have specific config for this.

* everything pushed become public (the default)::

    [phase]

    publish = yes

* nothing push turned public::

    [phase]

    publish = no

* topic branches are not published, changeset without topic are::

    [phase]

    publish = no

    [experimental]

    topic.publish-bare-branch = yes

In addition, the topic extension adds a ``--publish`` flag on :hg:`push`. When

used, the pushed revisions are published if the push succeeds. It also applies

to common revisions selected by the push.

"""

from __future__ import absolute_import

import functools

import re

import time

import weakref

from mercurial.i18n import _

from mercurial import (

    bookmarks,

    changelog,

    cmdutil,

    commands,

    context,

    error,

    extensions,

    hg,

    localrepo,

    lock as lockmod,

    merge,

    namespaces,

    node,

    obsolete,

    patch,

    phases,

    registrar,

    scmutil,

    templatefilters,

    templatekw,

    util,

)

from . import (

    compat,

    constants,

    destination,

    discovery,

    evolvebits,

    flow,

    randomname,

    revset as topicrevset,

    stack,

    topicmap,

)

if util.safehasattr(registrar, 'command'):

    commandfunc = registrar.command

else: # compat with hg < 4.3

    commandfunc = cmdutil.command

cmdtable = {}

command = commandfunc(cmdtable)

colortable = {'topic.active': 'green',

              'topic.list.troubledcount': 'red',

              'topic.list.headcount.multiple': 'yellow',

              'topic.list.behindcount': 'cyan',

              'topic.list.behinderror': 'red',

              'topic.stack.index': 'yellow',

              'topic.stack.index.base': 'none dim',

              'topic.stack.desc.base': 'none dim',

              'topic.stack.shortnode.base': 'none dim',

              'topic.stack.state.base': 'dim',

              'topic.stack.state.clean': 'green',

              'topic.stack.index.current': 'cyan',       # random pick

              'topic.stack.state.current': 'cyan bold',  # random pick

              'topic.stack.desc.current': 'cyan',        # random pick

              'topic.stack.shortnode.current': 'cyan',   # random pick

              'topic.stack.state.unstable': 'red',

              'topic.stack.summary.behindcount': 'cyan',

              'topic.stack.summary.behinderror': 'red',

              'topic.stack.summary.headcount.multiple': 'yellow',

              # default color to help log output and thg

              # (first pick I could think off, update as needed

              'log.topic': 'green_background',

              'topic.active': 'green',

             }

testedwith = '4.3.3 4.4.2 4.5.2 4.6.2 4.7'

minimumhgversion = '4.3'

buglink = 'https://bz.mercurial-scm.org/'

if util.safehasattr(registrar, 'configitem'):

    from mercurial import configitems

    configtable = {}

    configitem = registrar.configitem(configtable)

    configitem('experimental', 'enforce-topic',

               default=False,

    )

    configitem('experimental', 'enforce-single-head',

               default=False,

    )

    configitem('experimental', 'topic-mode',

               default=None,

    )

    configitem('experimental', 'topic.publish-bare-branch',

               default=False,

    )

    configitem('_internal', 'keep-topic',

               default=False,

    )

    configitem('experimental', 'topic-mode.server',

               default=configitems.dynamicdefault,

    )

    def extsetup(ui):

        # register config that strictly belong to other code (thg, core, etc)

        #

        # To ensure all config items we used are registered, we register them if

        # nobody else did so far.

        from mercurial import configitems

        extraitem = functools.partial(configitems._register, ui._knownconfig)

        if ('experimental' not in ui._knownconfig

                or not ui._knownconfig['experimental'].get('thg.displaynames')):

            extraitem('experimental', 'thg.displaynames',

                      default=None,

            )

        if ('devel' not in ui._knownconfig

                or not ui._knownconfig['devel'].get('random')):

            extraitem('devel', 'randomseed',

                      default=None,

            )

def _contexttopic(self, force=False):

    if not (force or self.mutable()):

        return ''

    return self.extra().get(constants.extrakey, '')

context.basectx.topic = _contexttopic

def _contexttopicidx(self):

    topic = self.topic()

    if not topic:

        # XXX we might want to include t0 here,

        # however t0 is related to  'currenttopic' which has no place here.

        return None

    revlist = stack.stack(self._repo, topic=topic)

    try:

        return revlist.index(self.rev())

    except ValueError:

        if self.obsolete():

            succ = evolvebits._singlesuccessor(self._repo, self)

            if succ not in revlist:

                return None

            return revlist.index(succ)

    except IndexError:

        # Lets move to the last ctx of the current topic

        return None

context.basectx.topicidx = _contexttopicidx

topicrev = re.compile(r'^t\d+$')

branchrev = re.compile(r'^b\d+$')

def _namemap(repo, name):

    revs = None

    if topicrev.match(name):

        idx = int(name[1:])

        ttype = 'topic'

        tname = topic = repo.currenttopic

        if not tname:

            raise error.Abort(_('cannot resolve "%s": no active topic') % name)

        revs = list(stack.stack(repo, topic=topic))

    elif branchrev.match(name):

        ttype = 'branch'

        idx = int(name[1:])

        tname = branch = repo[None].branch()

        revs = list(stack.stack(repo, branch=branch))

    if revs is not None:

        try:

            r = revs[idx]

        except IndexError:

            msg = _('cannot resolve "%s": %s "%s" has only %d changesets')

            raise error.Abort(msg % (name, ttype, tname, len(revs) - 1))

        # b0 or t0 can be None

        if r == -1 and idx == 0:

            msg = _('the %s "%s" has no %s')

            raise error.Abort(msg % (ttype, tname, name))

        return [repo[r].node()]

    if name not in repo.topics:

        return []

    node = repo.changelog.node

    return [node(rev) for rev in repo.revs('topic(%s)', name)]

def _nodemap(repo, node):

    ctx = repo[node]

    t = ctx.topic()

    if t and ctx.phase() > phases.public:

        return [t]

    return []

def uisetup(ui):

    destination.modsetup(ui)

    discovery.modsetup(ui)

    topicmap.modsetup(ui)

    setupimportexport(ui)

    extensions.afterloaded('rebase', _fixrebase)

    flow.installpushflag(ui)

    entry = extensions.wrapcommand(commands.table, 'commit', commitwrap)

    entry[1].append(('t', 'topic', '',

                     _("use specified topic"), _('TOPIC')))

    entry = extensions.wrapcommand(commands.table, 'push', pushoutgoingwrap)

    entry[1].append(('t', 'topic', '',

                     _("topic to push"), _('TOPIC')))

    entry = extensions.wrapcommand(commands.table, 'outgoing',

                                   pushoutgoingwrap)

    entry[1].append(('t', 'topic', '',

                     _("topic to push"), _('TOPIC')))

    extensions.wrapfunction(cmdutil, 'buildcommittext', committextwrap)

    extensions.wrapfunction(merge, 'update', mergeupdatewrap)

    # We need to check whether t0 or b0 is passed to override the default update

    # behaviour of changing topic and I can't find a better way

    # to do that as scmutil.revsingle returns the rev number and hence we can't

    # plug into logic for this into mergemod.update().

    extensions.wrapcommand(commands.table, 'update', checkt0)

    try:

        evolve = extensions.find('evolve')

        extensions.wrapfunction(evolve.rewriteutil, "presplitupdate",

                                presplitupdatetopic)

    except (KeyError, AttributeError):

        pass

    cmdutil.summaryhooks.add('topic', summaryhook)

    templatekw.keywords['topic'] = topickw

    # Wrap workingctx extra to return the topic name

    extensions.wrapfunction(context.workingctx, '__init__', wrapinit)

    # Wrap changelog.add to drop empty topic

    extensions.wrapfunction(changelog.changelog, 'add', wrapadd)

def reposetup(ui, repo):

    if not isinstance(repo, localrepo.localrepository):

        return # this can be a peer in the ssh case (puzzling)

    repo = repo.unfiltered()

    if repo.ui.config('experimental', 'thg.displaynames') is None:

        repo.ui.setconfig('experimental', 'thg.displaynames', 'topics',

                          source='topic-extension')

    class topicrepo(repo.__class__):

        def _restrictcapabilities(self, caps):

            caps = super(topicrepo, self)._restrictcapabilities(caps)

            caps.add('topics')

            return caps

        def commit(self, *args, **kwargs):

            backup = self.ui.backupconfig('ui', 'allowemptycommit')

            try:

                if self.currenttopic != self['.'].topic():

                    # bypass the core "nothing changed" logic

                    self.ui.setconfig('ui', 'allowemptycommit', True)

                return super(topicrepo, self).commit(*args, **kwargs)

            finally:

                self.ui.restoreconfig(backup)

        def commitctx(self, ctx, error=None):

            topicfilter = topicmap.topicfilter(self.filtername)

            if topicfilter != self.filtername:

                other = self.filtered(topicmap.topicfilter(self.filtername))

                other.commitctx(ctx, error=error)

            if isinstance(ctx, context.workingcommitctx):

                current = self.currenttopic

                if current:

                    ctx.extra()[constants.extrakey] = current

            if (isinstance(ctx, context.memctx) and

                ctx.extra().get('amend_source') and

                ctx.topic() and

                not self.currenttopic):

                # we are amending and need to remove a topic

                del ctx.extra()[constants.extrakey]

            return super(topicrepo, self).commitctx(ctx, error=error)

        @property