comparison hgext/blackbox.py @ 40730:55b053af7196

ui: manage logger instances and event filtering by core ui

The setup code in blackbox needs more tweaks since it has lots of black
magics. I'll fix them by follow-up patches.

To be clear, the goal of this series is to provide a proper way for command
server to install its own logger. I need it to debug in-memory repository
cache.

40729:c93d046d4300

from mercurial import (
    encoding,
    pycompat,
    registrar,
    ui as uimod,
    util,
)
from mercurial.utils import (
    dateutil,
    procutil,
)

    def tracked(self, event):
        return b'*' in self._trackedevents or event in self._trackedevents

    def log(self, ui, event, msg, opts):
        global _lastlogger
        if not self.tracked(event):
            return

        if self._bbvfs:
            _lastlogger = self
        elif _lastlogger and _lastlogger._bbvfs:
            # certain logger instances exist outside the context of
            # a repo, so just default to the last blackbox logger that

    def setrepo(self, repo):
        self._repo = repo

def wrapui(ui):
    class blackboxui(ui.__class__):
        def __init__(self, src=None):
            super(blackboxui, self).__init__(src)
            if src and r'_bblogger' in src.__dict__:
                self._bblogger = src._bblogger

        # trick to initialize logger after configuration is loaded, which
        # can be replaced later with blackboxlogger(ui) in uisetup(), where
        # both user and repo configurations should be available.
        @util.propertycache
        def _bblogger(self):
            return blackboxlogger(self)

        def debug(self, *msg, **opts):
            super(blackboxui, self).debug(*msg, **opts)
            if self.debugflag:
                self.log('debug', '%s', ''.join(msg))

        def log(self, event, *msg, **opts):
            super(blackboxui, self).log(event, *msg, **opts)
            self._bblogger.log(self, event, msg, opts)

    ui.__class__ = blackboxui
    uimod.ui = blackboxui

def uisetup(ui):
    wrapui(ui)

def reposetup(ui, repo):
    # During 'hg pull' a httppeer repo is created to represent the remote repo.
    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.
    if not repo.local():
        return

    logger = getattr(ui, '_bblogger', None)
    if logger:
        logger.setrepo(repo)

        # Set _lastlogger even if ui.log is not called. This gives blackbox a
        # fallback place to log.

40730:55b053af7196

from mercurial import (
    encoding,
    pycompat,
    registrar,
    ui as uimod,
)
from mercurial.utils import (
    dateutil,
    procutil,
)

    def tracked(self, event):
        return b'*' in self._trackedevents or event in self._trackedevents

    def log(self, ui, event, msg, opts):
        global _lastlogger
        if self._bbvfs:
            _lastlogger = self
        elif _lastlogger and _lastlogger._bbvfs:
            # certain logger instances exist outside the context of
            # a repo, so just default to the last blackbox logger that

    def setrepo(self, repo):
        self._repo = repo

def wrapui(ui):
    class blackboxui(ui.__class__):
        def debug(self, *msg, **opts):
            super(blackboxui, self).debug(*msg, **opts)
            if self.debugflag:
                self.log('debug', '%s', ''.join(msg))

    ui.__class__ = blackboxui
    uimod.ui = blackboxui

def uisetup(ui):
    wrapui(ui)

def uipopulate(ui):
    ui.setlogger(b'blackbox', blackboxlogger(ui))

def reposetup(ui, repo):
    # During 'hg pull' a httppeer repo is created to represent the remote repo.
    # It doesn't have a .hg directory to put a blackbox in, so we don't do
    # the blackbox setup for it.
    if not repo.local():
        return

    # Since blackbox.log is stored in the repo directory, the logger should be
    # instantiated per repository.
    logger = blackboxlogger(ui)
    ui.setlogger(b'blackbox', logger)
    if logger:
        logger.setrepo(repo)

        # Set _lastlogger even if ui.log is not called. This gives blackbox a
        # fallback place to log.