comparison mercurial/exchange.py @ 22239:0688010ee38f

push: move bookmark discovery with other discovery steps

The discovery of necessary bookmark updates is now done within the "discovery
phase". This opens the door to the inclusion of bookmarks in a unified bundle2
push.

22238:c894fdff56d1

        self.outdatedphases = None
        # phases changes that must be pushed if changeset push fails
        self.fallbackoutdatedphases = None
        # outgoing obsmarkers
        self.outobsmarkers = set()

    @util.propertycache
    def futureheads(self):
        """future remote heads if the changeset push succeeds"""
        return self.outgoing.missingheads

@pushdiscovery('obsmarker')
def _pushdiscoveryobsmarkers(pushop):
    pushop.outobsmarkers = pushop.repo.obsstore

def _pushcheckoutgoing(pushop):
    outgoing = pushop.outgoing
    unfi = pushop.repo.unfiltered()
    if not outgoing.missing:
        # nothing to push

def _pushbookmark(pushop):
    """Update bookmark position on remote"""
    if pushop.ret == 0:
        return
    ui = pushop.ui
    repo = pushop.repo.unfiltered()
    remote = pushop.remote
    ui.debug("checking for updated bookmarks\n")
    ancestors = ()
    if pushop.revs:
        revnums = map(repo.changelog.rev, pushop.revs)
        ancestors = repo.changelog.ancestors(revnums, inclusive=True)
    remotebookmark = remote.listkeys('bookmarks')
    comp = bookmarks.compare(repo, repo._bookmarks, remotebookmark, srchex=hex)
    (addsrc, adddst, advsrc, advdst, diverge, differ, invalid) = comp
    for b, scid, dcid in advsrc:
        if ancestors and repo[scid].rev() not in ancestors:
            continue
        if remote.pushkey('bookmarks', b, dcid, scid):
            ui.status(_("updating bookmark %s\n") % b)
        else:
            ui.warn(_('updating bookmark %s failed!\n') % b)

class pulloperation(object):

22239:0688010ee38f

        self.outdatedphases = None
        # phases changes that must be pushed if changeset push fails
        self.fallbackoutdatedphases = None
        # outgoing obsmarkers
        self.outobsmarkers = set()
        # outgoing bookmarks
        self.outbookmarks = []

    @util.propertycache
    def futureheads(self):
        """future remote heads if the changeset push succeeds"""
        return self.outgoing.missingheads

@pushdiscovery('obsmarker')
def _pushdiscoveryobsmarkers(pushop):
    pushop.outobsmarkers = pushop.repo.obsstore

@pushdiscovery('bookmarks')
def _pushdiscoverybookmarks(pushop):
    ui = pushop.ui
    repo = pushop.repo.unfiltered()
    remote = pushop.remote
    ui.debug("checking for updated bookmarks\n")
    ancestors = ()
    if pushop.revs:
        revnums = map(repo.changelog.rev, pushop.revs)
        ancestors = repo.changelog.ancestors(revnums, inclusive=True)
    remotebookmark = remote.listkeys('bookmarks')

    comp = bookmarks.compare(repo, repo._bookmarks, remotebookmark, srchex=hex)
    (addsrc, adddst, advsrc, advdst, diverge, differ, invalid) = comp
    for b, scid, dcid in advsrc:
        if not ancestors or repo[scid].rev() in ancestors:
            pushop.outbookmarks.append((b, dcid, scid))

def _pushcheckoutgoing(pushop):
    outgoing = pushop.outgoing
    unfi = pushop.repo.unfiltered()
    if not outgoing.missing:
        # nothing to push

def _pushbookmark(pushop):
    """Update bookmark position on remote"""
    if pushop.ret == 0:
        return
    ui = pushop.ui
    remote = pushop.remote
    for b, old, new in pushop.outbookmarks:
        if remote.pushkey('bookmarks', b, old, new):
            ui.status(_("updating bookmark %s\n") % b)
        else:
            ui.warn(_('updating bookmark %s failed!\n') % b)

class pulloperation(object):