mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 00:45:18 +03:00
4e272b6fad
Summary: This gives type checker some hints. Pyre was able to find some issues, which are fixed. Reviewed By: markbt Differential Revision: D19674436 fbshipit-source-id: 44d6df6037f27e2edeff4d343a9fdec5e51420b3
168 lines
5.1 KiB
Python
168 lines
5.1 KiB
Python
# Portions Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# filelog.py - file history class for mercurial
|
|
#
|
|
# Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
|
|
#
|
|
# 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 re
|
|
import struct
|
|
from typing import Dict
|
|
|
|
from . import error, mdiff, revlog
|
|
from .pycompat import decodeutf8, encodeutf8
|
|
|
|
|
|
_mdre = re.compile(b"\1\n")
|
|
|
|
|
|
def parsemeta(text):
|
|
"""return (metadatadict, metadatasize)"""
|
|
# text can be buffer, so we can't use .startswith or .index
|
|
if bytes(text[:2]) != b"\1\n":
|
|
return None, None
|
|
s = _mdre.search(text, 2).start()
|
|
mtext = decodeutf8(text[2:s])
|
|
meta = {}
|
|
for l in mtext.splitlines():
|
|
k, v = l.split(": ", 1)
|
|
meta[k] = v
|
|
return meta, (s + 2)
|
|
|
|
|
|
def packmeta(meta, text):
|
|
# type: (Dict[str, str], bytes) -> bytes
|
|
keys = sorted(meta)
|
|
metatext = encodeutf8("".join("%s: %s\n" % (k, meta[k]) for k in keys))
|
|
return b"".join((b"\1\n", metatext, b"\1\n", text))
|
|
|
|
|
|
def _censoredtext(text):
|
|
m, offs = parsemeta(text)
|
|
return m and b"censored" in m
|
|
|
|
|
|
class filelog(revlog.revlog):
|
|
def __init__(self, opener, path):
|
|
super(filelog, self).__init__(opener, "/".join(("data", path + ".i")))
|
|
|
|
def read(self, node):
|
|
t = self.revision(node)
|
|
if not t.startswith(b"\1\n"):
|
|
return t
|
|
s = t.index(b"\1\n", 2)
|
|
return t[s + 2 :]
|
|
|
|
def add(self, text, meta, transaction, link, p1=None, p2=None):
|
|
if meta or text.startswith(b"\1\n"):
|
|
text = packmeta(meta, text)
|
|
return self.addrevision(text, transaction, link, p1, p2)
|
|
|
|
def renamed(self, node):
|
|
if self.parents(node)[0] != revlog.nullid:
|
|
return False
|
|
t = self.revision(node)
|
|
m = parsemeta(t)[0]
|
|
if m and "copy" in m:
|
|
return (m["copy"], revlog.bin(m["copyrev"]))
|
|
return False
|
|
|
|
def size(self, rev):
|
|
# type: (int) -> int
|
|
"""return the size of a given revision"""
|
|
|
|
# for revisions with renames, we have to go the slow way
|
|
node = self.node(rev)
|
|
if self.renamed(node):
|
|
return len(self.read(node))
|
|
if self.iscensored(rev):
|
|
return 0
|
|
|
|
# XXX if self.read(node).startswith("\1\n"), this returns (size+4)
|
|
return super(filelog, self).size(rev)
|
|
|
|
def cmp(self, node, text):
|
|
"""compare text with a given file revision
|
|
|
|
returns True if text is different than what is stored.
|
|
"""
|
|
|
|
t = text
|
|
if text.startswith(b"\1\n"):
|
|
t = b"\1\n\1\n" + text
|
|
|
|
samehashes = not super(filelog, self).cmp(node, t)
|
|
if samehashes:
|
|
return False
|
|
|
|
# censored files compare against the empty file
|
|
if self.iscensored(self.rev(node)):
|
|
return text != b""
|
|
|
|
# renaming a file produces a different hash, even if the data
|
|
# remains unchanged. Check if it's the case (slow):
|
|
if self.renamed(node):
|
|
t2 = self.read(node)
|
|
return t2 != text
|
|
|
|
return True
|
|
|
|
def checkhash(self, text, node, p1=None, p2=None, rev=None):
|
|
try:
|
|
super(filelog, self).checkhash(text, node, p1=p1, p2=p2, rev=rev)
|
|
except error.RevlogError:
|
|
if _censoredtext(text):
|
|
raise error.CensoredNodeError(self.indexfile, node, text)
|
|
raise
|
|
|
|
def iscensored(self, rev):
|
|
"""Check if a file revision is censored."""
|
|
return self.flags(rev) & revlog.REVIDX_ISCENSORED
|
|
|
|
def _peek_iscensored(self, baserev, delta, flush):
|
|
"""Quickly check if a delta produces a censored revision."""
|
|
# Fragile heuristic: unless new file meta keys are added alphabetically
|
|
# preceding "censored", all censored revisions are prefixed by
|
|
# "\1\ncensored:". A delta producing such a censored revision must be a
|
|
# full-replacement delta, so we inspect the first and only patch in the
|
|
# delta for this prefix.
|
|
hlen = struct.calcsize(">lll")
|
|
if len(delta) <= hlen:
|
|
return False
|
|
|
|
oldlen = self.rawsize(baserev)
|
|
newlen = len(delta) - hlen
|
|
if delta[:hlen] != mdiff.replacediffheader(oldlen, newlen):
|
|
return False
|
|
|
|
add = "\1\ncensored:"
|
|
addlen = len(add)
|
|
return newlen >= addlen and delta[hlen : hlen + addlen] == add
|
|
|
|
|
|
class fileslog(object):
|
|
"""Top level object representing all the file storage.
|
|
|
|
Eventually filelog content access should go through this, but for now it's
|
|
just used to handle remotefilelog writes.
|
|
"""
|
|
|
|
def __init__(self, repo):
|
|
self.ui = repo.ui
|
|
self.repo = repo
|
|
|
|
def commitpending(self):
|
|
"""Used in alternative filelog implementations to commit pending
|
|
additions."""
|
|
|
|
def abortpending(self):
|
|
"""Used in alternative filelog implementations to throw out pending
|
|
additions."""
|