comparison mercurial/revlogutils/rewrite.py @ 47471:aab064416f0c

censor: rename `rl` to `revlog` in the main function

Now that the bulk of the code moved to smaller function we are less restricted
on line length and we can use more explicite naming.

Differential Revision: https://phab.mercurial-scm.org/D10902

47470:d6afe1478a2a

    rl.clearcaches()
    rl._loadindex()


def v2_censor(rl, tr, censornode, tombstone=b''):
    """censors a revision in a "version 2" revlog"""
    # General principle
    #
    # We create new revlog files (index/data/sidedata) to copy the content of
    # the existing data without the censored data.

    # final destination.
    #
    # All data before the censored data can be blindly copied. The rest needs
    # to be copied as we go and the associated index entry needs adjustement.

    assert rl._format_version != REVLOGV0, rl._format_version
    assert rl._format_version != REVLOGV1, rl._format_version

    old_index = rl.index
    docket = rl._docket

    censor_rev = rl.rev(censornode)
    tombstone = storageutil.packmeta({b'censored': tombstone}, b'')

    censored_entry = rl.index[censor_rev]
    index_cutoff = rl.index.entry_size * censor_rev
    data_cutoff = censored_entry[ENTRY_DATA_OFFSET] >> 16
    sidedata_cutoff = rl.sidedata_cut_off(censor_rev)

    with pycompat.unnamedtempfile(mode=b"w+b") as tmp_storage:
        # rev → (new_base, data_start, data_end, compression_mode)
        rewritten_entries = _precompute_rewritten_delta(
            rl,
            old_index,
            {censor_rev},
            tmp_storage,
        )

        all_files = _setup_new_files(
            rl,
            index_cutoff,
            data_cutoff,
            sidedata_cutoff,
        )

                new_sidedata_file,
            ) = open_files

            # writing the censored revision
            _rewrite_censor(
                rl,
                old_index,
                open_files,
                censor_rev,
                tombstone,
            )

            # Writing all subsequent revisions
            for rev in range(censor_rev + 1, len(old_index)):
                _rewrite_simple(
                    rl,
                    old_index,
                    open_files,
                    rev,
                    rewritten_entries,
                    tmp_storage,

47471:aab064416f0c

    rl.clearcaches()
    rl._loadindex()


def v2_censor(revlog, tr, censornode, tombstone=b''):
    """censors a revision in a "version 2" revlog"""
    # General principle
    #
    # We create new revlog files (index/data/sidedata) to copy the content of
    # the existing data without the censored data.

    # final destination.
    #
    # All data before the censored data can be blindly copied. The rest needs
    # to be copied as we go and the associated index entry needs adjustement.

    assert revlog._format_version != REVLOGV0, revlog._format_version
    assert revlog._format_version != REVLOGV1, revlog._format_version

    old_index = revlog.index
    docket = revlog._docket

    censor_rev = revlog.rev(censornode)
    tombstone = storageutil.packmeta({b'censored': tombstone}, b'')

    censored_entry = revlog.index[censor_rev]
    index_cutoff = revlog.index.entry_size * censor_rev
    data_cutoff = censored_entry[ENTRY_DATA_OFFSET] >> 16
    sidedata_cutoff = revlog.sidedata_cut_off(censor_rev)

    with pycompat.unnamedtempfile(mode=b"w+b") as tmp_storage:
        # rev → (new_base, data_start, data_end, compression_mode)
        rewritten_entries = _precompute_rewritten_delta(
            revlog,
            old_index,
            {censor_rev},
            tmp_storage,
        )

        all_files = _setup_new_files(
            revlog,
            index_cutoff,
            data_cutoff,
            sidedata_cutoff,
        )

                new_sidedata_file,
            ) = open_files

            # writing the censored revision
            _rewrite_censor(
                revlog,
                old_index,
                open_files,
                censor_rev,
                tombstone,
            )

            # Writing all subsequent revisions
            for rev in range(censor_rev + 1, len(old_index)):
                _rewrite_simple(
                    revlog,
                    old_index,
                    open_files,
                    rev,
                    rewritten_entries,
                    tmp_storage,