|
|
# 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,
|
|
|
)
|
|
|
|