from mercurial import branchmap

from . import topicmap

def _headssummary(orig, repo, remote, outgoing):

    publishing = ('phases' not in remote.listkeys('namespaces')

                  or bool(remote.listkeys('phases').get('publishing', False)))

    if publishing:

        return orig(repo, remote, outgoing)

    oldgetitem = repo.__getitem__

    oldrepo = repo.__class__

    oldbranchcache = branchmap.branchcache

    oldfilename = branchmap._filename

    try:

        class repocls(repo.__class__):

            def __getitem__(self, key):

                ctx = super(repocls, self).__getitem__(key)

                oldbranch = ctx.branch

                def branch():

                    branch = oldbranch()

                    topic = ctx.topic()

                    if topic:

                        branch = "%s:%s" % (branch, topic)

                    return branch

                ctx.branch = branch

                return ctx

        repo.__class__ = repocls

        branchmap.branchcache = topicmap.topiccache

        branchmap._filename = topicmap._filename

        summary = orig(repo, remote, outgoing)

        for key, value in summary.iteritems():

            if ':' in key: # This is a topic

                if value[0] is None and value[1]:

                    summary[key] = ([value[1].pop(0)], ) + value[1:]

        return summary

    finally:

        repo.__class__ = oldrepo

        branchmap.branchcache = oldbranchcache

        branchmap._filename = oldfilename

def wireprotobranchmap(orig, repo, proto):

    oldrepo = repo.__class__

    print repo

    try:

        class repocls(repo.__class__):

            def branchmap(self):

                usetopic = not self.publishing()

                return super(repocls, self).branchmap(topic=usetopic)

        repo.__class__ = repocls

        return orig(repo, proto)

    finally:

        repo.__class__ = oldrepo