# extension to emulate invoking 'dirstate.write()' at the time

# specified by '[fakedirstatewritetime] fakenow', only when

# 'dirstate.write()' is invoked via functions below:

#

#   - 'workingctx._poststatusfixup()' (= 'repo.status()')

#   - 'committablectx.markcommitted()'

from mercurial import (

    context,

    dirstatemap as dirstatemapmod,

    extensions,

    policy,

    registrar,

)

from mercurial.dirstateutils import timestamp

from mercurial.utils import dateutil

try:

    from mercurial import rustext

    rustext.__name__  # force actual import (see hgdemandimport)

except ImportError:

    rustext = None

configtable = {}

configitem = registrar.configitem(configtable)

configitem(

    b'fakedirstatewritetime',

    b'fakenow',

    default=None,

)

parsers = policy.importmod('parsers')

has_rust_dirstate = policy.importrust('dirstate') is not None

def pack_dirstate(orig, dmap, copymap, pl):

    return orig(dmap, copymap, pl)

def fakewrite(ui, func):

    # fake "now" of 'pack_dirstate' only if it is invoked while 'func'

    fakenow = ui.config(b'fakedirstatewritetime', b'fakenow')

    if not fakenow:

        # Execute original one, if fakenow isn't configured. This is

        # useful to prevent subrepos from executing replaced one,

        # because replacing 'parsers.pack_dirstate' is also effective

        # in subrepos.

        return func()

    # parsing 'fakenow' in YYYYmmddHHMM format makes comparison between

    # 'fakenow' value and 'touch -t YYYYmmddHHMM' argument easy

    fakenow = dateutil.parsedate(fakenow, [b'%Y%m%d%H%M'])[0]

    fakenow = timestamp.timestamp((fakenow, 0, False))

    if has_rust_dirstate:

        # The Rust implementation does not use public parse/pack dirstate

        # to prevent conversion round-trips

        orig_dirstatemap_write = dirstatemapmod.dirstatemap.write

        wrapper = lambda self, tr, st: orig_dirstatemap_write(self, tr, st)

        dirstatemapmod.dirstatemap.write = wrapper

    orig_get_fs_now = timestamp.get_fs_now

    wrapper = lambda *args: pack_dirstate(orig_pack_dirstate, *args)

    orig_module = parsers

    orig_pack_dirstate = parsers.pack_dirstate

    orig_module.pack_dirstate = wrapper

    timestamp.get_fs_now = (

        lambda *args: fakenow

    )  # XXX useless for this purpose now

    try:

        return func()

    finally:

        orig_module.pack_dirstate = orig_pack_dirstate

        timestamp.get_fs_now = orig_get_fs_now

        if has_rust_dirstate:

            dirstatemapmod.dirstatemap.write = orig_dirstatemap_write

def _poststatusfixup(orig, workingctx, status, fixup):

    ui = workingctx.repo().ui

    return fakewrite(ui, lambda: orig(workingctx, status, fixup))

def markcommitted(orig, committablectx, node):

    ui = committablectx.repo().ui

    return fakewrite(ui, lambda: orig(committablectx, node))

def extsetup(ui):

    extensions.wrapfunction(

        context.workingctx, '_poststatusfixup', _poststatusfixup

    )

    extensions.wrapfunction(context.workingctx, 'markcommitted', markcommitted)