view hgext/narrow/narrowrepo.py @ 50256:a6e0b7d4ae9d

dirstate: write the `branch` as part of the transaction if any

Bypassing the transaction means we could get out of sync with the dirstatemap
content.

The branch is stil written right away if no transaction is around, but at least
it no longer bypass the transaction.

Actual caller of this still need to be updated.

# narrowrepo.py - repository which supports narrow revlogs, lazy loading
#
# Copyright 2017 Google, Inc.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.


from mercurial import wireprototypes

from . import narrowdirstate


def wraprepo(repo):
    """Enables narrow clone functionality on a single local repository."""

    class narrowrepository(repo.__class__):
        def _makedirstate(self):
            dirstate = super(narrowrepository, self)._makedirstate()
            return narrowdirstate.wrapdirstate(self, dirstate)

        def peer(self, path=None):
            peer = super(narrowrepository, self).peer(path=path)
            peer._caps.add(wireprototypes.NARROWCAP)
            peer._caps.add(wireprototypes.ELLIPSESCAP)
            return peer

    repo.__class__ = narrowrepository