comparison hgext/fsmonitor/__init__.py @ 33386:5a9b4370bb3b

fsmonitor: centralize setup procedures for dirstate

33385:8bb4c69c4000

        hashignore = _hashignore(wctx.repo().dirstate._ignore)
        notefiles = (status.modified + status.added + status.removed +
                     status.deleted + status.unknown)
        wctx.repo()._fsmonitorstate.set(clock, hashignore, notefiles)

def makedirstate(cls):
    class fsmonitordirstate(cls):
        def _fsmonitorinit(self, fsmonitorstate, watchmanclient):
            # _fsmonitordisable is used in paranoid mode
            self._fsmonitordisable = False
            self._fsmonitorstate = fsmonitorstate
            self._watchmanclient = watchmanclient

        def invalidate(self, *args, **kwargs):
            self._fsmonitorstate.invalidate()
            return super(fsmonitordirstate, self).invalidate(*args, **kwargs)

    return fsmonitordirstate

def wrapdirstate(orig, self):
    ds = orig(self)
    # only override the dirstate when Watchman is available for the repo
    if util.safehasattr(self, '_fsmonitorstate'):
        ds.__class__ = makedirstate(ds.__class__)
        ds._fsmonitorinit(self._fsmonitorstate, self._watchmanclient)
    return ds

def extsetup(ui):
    extensions.wrapfilecache(
        localrepo.localrepository, 'dirstate', wrapdirstate)

        repo._watchmanclient = client

        # at this point since fsmonitorstate wasn't present, repo.dirstate is
        # not a fsmonitordirstate
        dirstate = repo.dirstate
        dirstate.__class__ = makedirstate(dirstate.__class__)
        dirstate._fsmonitorinit(fsmonitorstate, client)
        # invalidate property cache, but keep filecache which contains the
        # wrapped dirstate object
        del repo.unfiltered().__dict__['dirstate']
        assert dirstate is repo._filecache['dirstate'].obj

33386:5a9b4370bb3b

        hashignore = _hashignore(wctx.repo().dirstate._ignore)
        notefiles = (status.modified + status.added + status.removed +
                     status.deleted + status.unknown)
        wctx.repo()._fsmonitorstate.set(clock, hashignore, notefiles)

def makedirstate(repo, dirstate):
    class fsmonitordirstate(dirstate.__class__):
        def _fsmonitorinit(self, fsmonitorstate, watchmanclient):
            # _fsmonitordisable is used in paranoid mode
            self._fsmonitordisable = False
            self._fsmonitorstate = fsmonitorstate
            self._watchmanclient = watchmanclient

        def invalidate(self, *args, **kwargs):
            self._fsmonitorstate.invalidate()
            return super(fsmonitordirstate, self).invalidate(*args, **kwargs)

    dirstate.__class__ = fsmonitordirstate
    dirstate._fsmonitorinit(repo._fsmonitorstate, repo._watchmanclient)

def wrapdirstate(orig, self):
    ds = orig(self)
    # only override the dirstate when Watchman is available for the repo
    if util.safehasattr(self, '_fsmonitorstate'):
        makedirstate(self, ds)
    return ds

def extsetup(ui):
    extensions.wrapfilecache(
        localrepo.localrepository, 'dirstate', wrapdirstate)

        repo._watchmanclient = client

        # at this point since fsmonitorstate wasn't present, repo.dirstate is
        # not a fsmonitordirstate
        dirstate = repo.dirstate
        makedirstate(repo, dirstate)

        # invalidate property cache, but keep filecache which contains the
        # wrapped dirstate object
        del repo.unfiltered().__dict__['dirstate']
        assert dirstate is repo._filecache['dirstate'].obj