ext-sidedata-2.py
50 lines
| 1.5 KiB
| text/x-python
|
PythonLexer
|
r47452 | # coding: utf8 | ||
# ext-sidedata-2.py - small extension to test (differently) the sidedata logic | ||||
# | ||||
# Simulates a client for a complex sidedata exchange. | ||||
# | ||||
# Copyright 2021 Raphaël Gomès <rgomes@octobus.net> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
from __future__ import absolute_import | ||||
import hashlib | ||||
import struct | ||||
from mercurial.revlogutils import sidedata as sidedatamod | ||||
def compute_sidedata_1(repo, revlog, rev, sidedata, text=None): | ||||
sidedata = sidedata.copy() | ||||
if text is None: | ||||
text = revlog.revision(rev) | ||||
sidedata[sidedatamod.SD_TEST1] = struct.pack('>I', len(text)) | ||||
return sidedata | ||||
def compute_sidedata_2(repo, revlog, rev, sidedata, text=None): | ||||
sidedata = sidedata.copy() | ||||
if text is None: | ||||
text = revlog.revision(rev) | ||||
sha256 = hashlib.sha256(text).digest() | ||||
sidedata[sidedatamod.SD_TEST2] = struct.pack('>32s', sha256) | ||||
return sidedata | ||||
def reposetup(ui, repo): | ||||
# Sidedata keys happen to be the same as the categories, easier for testing. | ||||
for kind in (b'changelog', b'manifest', b'filelog'): | ||||
repo.register_sidedata_computer( | ||||
kind, | ||||
sidedatamod.SD_TEST1, | ||||
(sidedatamod.SD_TEST1,), | ||||
compute_sidedata_1, | ||||
) | ||||
repo.register_sidedata_computer( | ||||
kind, | ||||
sidedatamod.SD_TEST2, | ||||
(sidedatamod.SD_TEST2,), | ||||
compute_sidedata_2, | ||||
) | ||||