comparison hgext/logtoprocess.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

import os

from mercurial import (
    pycompat,
    util,
)
from mercurial.utils import (
    procutil,
)

    def tracked(self, event):
        return bool(self._scripts.get(event))

    def log(self, ui, event, msg, opts):
        script = self._scripts.get(event)
        if not script:
            return
        env = {
            b'EVENT': event,
            b'HGPID': os.getpid(),
            b'MSG1': msg[0] % msg[1:],
        }

        env.update((b'OPT_%s' % key.upper(), value)
                   for key, value in pycompat.byteskwargs(opts).items())
        fullenv = procutil.shellenviron(env)
        procutil.runbgcommand(script, fullenv, shell=True)

def uisetup(ui):

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

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

        def log(self, event, *msg, **opts):
            self._ltplogger.log(self, event, msg, opts)
            return super(logtoprocessui, self).log(event, *msg, **opts)

    # Replace the class for this instance and all clones created from it:
    ui.__class__ = logtoprocessui

40730:55b053af7196

import os

from mercurial import (
    pycompat,
)
from mercurial.utils import (
    procutil,
)

    def tracked(self, event):
        return bool(self._scripts.get(event))

    def log(self, ui, event, msg, opts):
        script = self._scripts[event]
        env = {
            b'EVENT': event,
            b'HGPID': os.getpid(),
            b'MSG1': msg[0] % msg[1:],
        }

        env.update((b'OPT_%s' % key.upper(), value)
                   for key, value in pycompat.byteskwargs(opts).items())
        fullenv = procutil.shellenviron(env)
        procutil.runbgcommand(script, fullenv, shell=True)

def uipopulate(ui):
    ui.setlogger(b'logtoprocess', processlogger(ui))