|
|
# context.py - changeset and file context objects for mercurial
|
|
|
#
|
|
|
# Copyright 2006 Matt Mackall <mpm@selenic.com>
|
|
|
#
|
|
|
# This software may be used and distributed according to the terms
|
|
|
# of the GNU General Public License, incorporated herein by reference.
|
|
|
|
|
|
from node import *
|
|
|
from i18n import gettext as _
|
|
|
import ancestor, bdiff, repo, revlog, util, os
|
|
|
|
|
|
class changectx(object):
|
|
|
"""A changecontext object makes access to data related to a particular
|
|
|
changeset convenient."""
|
|
|
def __init__(self, repo, changeid=None):
|
|
|
"""changeid is a revision number, node, or tag"""
|
|
|
self._repo = repo
|
|
|
|
|
|
if not changeid and changeid != 0:
|
|
|
p1, p2 = self._repo.dirstate.parents()
|
|
|
self._rev = self._repo.changelog.rev(p1)
|
|
|
if self._rev == -1:
|
|
|
changeid = 'tip'
|
|
|
else:
|
|
|
self._node = p1
|
|
|
return
|
|
|
|
|
|
self._node = self._repo.lookup(changeid)
|
|
|
self._rev = self._repo.changelog.rev(self._node)
|
|
|
|
|
|
def __str__(self):
|
|
|
return short(self.node())
|
|
|
|
|
|
def __repr__(self):
|
|
|
return "<changectx %s>" % str(self)
|
|
|
|
|
|
def __eq__(self, other):
|
|
|
try:
|
|
|
return self._rev == other._rev
|
|
|
except AttributeError:
|
|
|
return False
|
|
|
|
|
|
def __nonzero__(self):
|
|
|
return self._rev != nullrev
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
if name == '_changeset':
|
|
|
self._changeset = self._repo.changelog.read(self.node())
|
|
|
return self._changeset
|
|
|
elif name == '_manifest':
|
|
|
self._manifest = self._repo.manifest.read(self._changeset[0])
|
|
|
return self._manifest
|
|
|
elif name == '_manifestdelta':
|
|
|
md = self._repo.manifest.readdelta(self._changeset[0])
|
|
|
self._manifestdelta = md
|
|
|
return self._manifestdelta
|
|
|
else:
|
|
|
raise AttributeError, name
|
|
|
|
|
|
def changeset(self): return self._changeset
|
|
|
def manifest(self): return self._manifest
|
|
|
|
|
|
def rev(self): return self._rev
|
|
|
def node(self): return self._node
|
|
|
def user(self): return self._changeset[1]
|
|
|
def date(self): return self._changeset[2]
|
|
|
def files(self): return self._changeset[3]
|
|
|
def description(self): return self._changeset[4]
|
|
|
def branch(self): return self._changeset[5].get("branch", "")
|
|
|
|
|
|
def parents(self):
|
|
|
"""return contexts for each parent changeset"""
|
|
|
p = self._repo.changelog.parents(self._node)
|
|
|
return [changectx(self._repo, x) for x in p]
|
|
|
|
|
|
def children(self):
|
|
|
"""return contexts for each child changeset"""
|
|
|
c = self._repo.changelog.children(self._node)
|
|
|
return [changectx(self._repo, x) for x in c]
|
|
|
|
|
|
def filenode(self, path):
|
|
|
if '_manifest' in self.__dict__:
|
|
|
try:
|
|
|
return self._manifest[path]
|
|
|
except KeyError:
|
|
|
raise repo.LookupError(_("'%s' not found in manifest") % path)
|
|
|
if '_manifestdelta' in self.__dict__ or path in self.files():
|
|
|
if path in self._manifestdelta:
|
|
|
return self._manifestdelta[path]
|
|
|
node, flag = self._repo.manifest.find(self._changeset[0], path)
|
|
|
if not node:
|
|
|
raise repo.LookupError(_("'%s' not found in manifest") % path)
|
|
|
|
|
|
return node
|
|
|
|
|
|
def filectx(self, path, fileid=None):
|
|
|
"""get a file context from this changeset"""
|
|
|
if fileid is None:
|
|
|
fileid = self.filenode(path)
|
|
|
return filectx(self._repo, path, fileid=fileid, changectx=self)
|
|
|
|
|
|
def filectxs(self):
|
|
|
"""generate a file context for each file in this changeset's
|
|
|
manifest"""
|
|
|
mf = self.manifest()
|
|
|
m = mf.keys()
|
|
|
m.sort()
|
|
|
for f in m:
|
|
|
yield self.filectx(f, fileid=mf[f])
|
|
|
|
|
|
def ancestor(self, c2):
|
|
|
"""
|
|
|
return the ancestor context of self and c2
|
|
|
"""
|
|
|
n = self._repo.changelog.ancestor(self._node, c2._node)
|
|
|
return changectx(self._repo, n)
|
|
|
|
|
|
class filectx(object):
|
|
|
"""A filecontext object makes access to data related to a particular
|
|
|
filerevision convenient."""
|
|
|
def __init__(self, repo, path, changeid=None, fileid=None,
|
|
|
filelog=None, changectx=None):
|
|
|
"""changeid can be a changeset revision, node, or tag.
|
|
|
fileid can be a file revision or node."""
|
|
|
self._repo = repo
|
|
|
self._path = path
|
|
|
|
|
|
assert changeid is not None or fileid is not None
|
|
|
|
|
|
if filelog:
|
|
|
self._filelog = filelog
|
|
|
if changectx:
|
|
|
self._changectx = changectx
|
|
|
self._changeid = changectx.node()
|
|
|
|
|
|
if fileid is None:
|
|
|
self._changeid = changeid
|
|
|
else:
|
|
|
self._fileid = fileid
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
if name == '_changectx':
|
|
|
self._changectx = changectx(self._repo, self._changeid)
|
|
|
return self._changectx
|
|
|
elif name == '_filelog':
|
|
|
self._filelog = self._repo.file(self._path)
|
|
|
return self._filelog
|
|
|
elif name == '_changeid':
|
|
|
self._changeid = self._filelog.linkrev(self._filenode)
|
|
|
return self._changeid
|
|
|
elif name == '_filenode':
|
|
|
try:
|
|
|
if '_fileid' in self.__dict__:
|
|
|
self._filenode = self._filelog.lookup(self._fileid)
|
|
|
else:
|
|
|
self._filenode = self._changectx.filenode(self._path)
|
|
|
except revlog.RevlogError, inst:
|
|
|
raise repo.LookupError(str(inst))
|
|
|
return self._filenode
|
|
|
elif name == '_filerev':
|
|
|
self._filerev = self._filelog.rev(self._filenode)
|
|
|
return self._filerev
|
|
|
else:
|
|
|
raise AttributeError, name
|
|
|
|
|
|
def __nonzero__(self):
|
|
|
try:
|
|
|
n = self._filenode
|
|
|
return True
|
|
|
except repo.LookupError:
|
|
|
# file is missing
|
|
|
return False
|
|
|
|
|
|
def __str__(self):
|
|
|
return "%s@%s" % (self.path(), short(self.node()))
|
|
|
|
|
|
def __repr__(self):
|
|
|
return "<filectx %s>" % str(self)
|
|
|
|
|
|
def __eq__(self, other):
|
|
|
try:
|
|
|
return (self._path == other._path
|
|
|
and self._changeid == other._changeid)
|
|
|
except AttributeError:
|
|
|
return False
|
|
|
|
|
|
def filectx(self, fileid):
|
|
|
'''opens an arbitrary revision of the file without
|
|
|
opening a new filelog'''
|
|
|
return filectx(self._repo, self._path, fileid=fileid,
|
|
|
filelog=self._filelog)
|
|
|
|
|
|
def filerev(self): return self._filerev
|
|
|
def filenode(self): return self._filenode
|
|
|
def filelog(self): return self._filelog
|
|
|
|
|
|
def rev(self):
|
|
|
if '_changectx' in self.__dict__:
|
|
|
return self._changectx.rev()
|
|
|
return self._filelog.linkrev(self._filenode)
|
|
|
|
|
|
def node(self): return self._changectx.node()
|
|
|
def user(self): return self._changectx.user()
|
|
|
def date(self): return self._changectx.date()
|
|
|
def files(self): return self._changectx.files()
|
|
|
def description(self): return self._changectx.description()
|
|
|
def branch(self): return self._changectx.branch()
|
|
|
def manifest(self): return self._changectx.manifest()
|
|
|
def changectx(self): return self._changectx
|
|
|
|
|
|
def data(self): return self._filelog.read(self._filenode)
|
|
|
def renamed(self): return self._filelog.renamed(self._filenode)
|
|
|
def path(self): return self._path
|
|
|
def size(self): return self._filelog.size(self._filerev)
|
|
|
|
|
|
def cmp(self, text): return self._filelog.cmp(self._filenode, text)
|
|
|
|
|
|
def parents(self):
|
|
|
p = self._path
|
|
|
fl = self._filelog
|
|
|
pl = [(p, n, fl) for n in self._filelog.parents(self._filenode)]
|
|
|
|
|
|
r = self.renamed()
|
|
|
if r:
|
|
|
pl[0] = (r[0], r[1], None)
|
|
|
|
|
|
return [filectx(self._repo, p, fileid=n, filelog=l)
|
|
|
for p,n,l in pl if n != nullid]
|
|
|
|
|
|
def children(self):
|
|
|
# hard for renames
|
|
|
c = self._filelog.children(self._filenode)
|
|
|
return [filectx(self._repo, self._path, fileid=x,
|
|
|
filelog=self._filelog) for x in c]
|
|
|
|
|
|
def annotate(self, follow=False):
|
|
|
'''returns a list of tuples of (ctx, line) for each line
|
|
|
in the file, where ctx is the filectx of the node where
|
|
|
that line was last changed'''
|
|
|
|
|
|
def decorate(text, rev):
|
|
|
return ([rev] * len(text.splitlines()), text)
|
|
|
|
|
|
def pair(parent, child):
|
|
|
for a1, a2, b1, b2 in bdiff.blocks(parent[1], child[1]):
|
|
|
child[0][b1:b2] = parent[0][a1:a2]
|
|
|
return child
|
|
|
|
|
|
getlog = util.cachefunc(lambda x: self._repo.file(x))
|
|
|
def getctx(path, fileid):
|
|
|
log = path == self._path and self._filelog or getlog(path)
|
|
|
return filectx(self._repo, path, fileid=fileid, filelog=log)
|
|
|
getctx = util.cachefunc(getctx)
|
|
|
|
|
|
def parents(f):
|
|
|
# we want to reuse filectx objects as much as possible
|
|
|
p = f._path
|
|
|
if f._filerev is None: # working dir
|
|
|
pl = [(n.path(), n.filerev()) for n in f.parents()]
|
|
|
else:
|
|
|
pl = [(p, n) for n in f._filelog.parentrevs(f._filerev)]
|
|
|
|
|
|
if follow:
|
|
|
r = f.renamed()
|
|
|
if r:
|
|
|
pl[0] = (r[0], getlog(r[0]).rev(r[1]))
|
|
|
|
|
|
return [getctx(p, n) for p, n in pl if n != nullrev]
|
|
|
|
|
|
# use linkrev to find the first changeset where self appeared
|
|
|
if self.rev() != self._filelog.linkrev(self._filenode):
|
|
|
base = self.filectx(self.filerev())
|
|
|
else:
|
|
|
base = self
|
|
|
|
|
|
# find all ancestors
|
|
|
needed = {base: 1}
|
|
|
visit = [base]
|
|
|
files = [base._path]
|
|
|
while visit:
|
|
|
f = visit.pop(0)
|
|
|
for p in parents(f):
|
|
|
if p not in needed:
|
|
|
needed[p] = 1
|
|
|
visit.append(p)
|
|
|
if p._path not in files:
|
|
|
files.append(p._path)
|
|
|
else:
|
|
|
# count how many times we'll use this
|
|
|
needed[p] += 1
|
|
|
|
|
|
# sort by revision (per file) which is a topological order
|
|
|
visit = []
|
|
|
files.reverse()
|
|
|
for f in files:
|
|
|
fn = [(n._filerev, n) for n in needed.keys() if n._path == f]
|
|
|
fn.sort()
|
|
|
visit.extend(fn)
|
|
|
hist = {}
|
|
|
|
|
|
for r, f in visit:
|
|
|
curr = decorate(f.data(), f)
|
|
|
for p in parents(f):
|
|
|
if p != nullid:
|
|
|
curr = pair(hist[p], curr)
|
|
|
# trim the history of unneeded revs
|
|
|
needed[p] -= 1
|
|
|
if not needed[p]:
|
|
|
del hist[p]
|
|
|
hist[f] = curr
|
|
|
|
|
|
return zip(hist[f][0], hist[f][1].splitlines(1))
|
|
|
|
|
|
def ancestor(self, fc2):
|
|
|
"""
|
|
|
find the common ancestor file context, if any, of self, and fc2
|
|
|
"""
|
|
|
|
|
|
acache = {}
|
|
|
|
|
|
# prime the ancestor cache for the working directory
|
|
|
for c in (self, fc2):
|
|
|
if c._filerev == None:
|
|
|
pl = [(n.path(), n.filenode()) for n in c.parents()]
|
|
|
acache[(c._path, None)] = pl
|
|
|
|
|
|
flcache = {self._path:self._filelog, fc2._path:fc2._filelog}
|
|
|
def parents(vertex):
|
|
|
if vertex in acache:
|
|
|
return acache[vertex]
|
|
|
f, n = vertex
|
|
|
if f not in flcache:
|
|
|
flcache[f] = self._repo.file(f)
|
|
|
fl = flcache[f]
|
|
|
pl = [(f, p) for p in fl.parents(n) if p != nullid]
|
|
|
re = fl.renamed(n)
|
|
|
if re:
|
|
|
pl.append(re)
|
|
|
acache[vertex] = pl
|
|
|
return pl
|
|
|
|
|
|
a, b = (self._path, self._filenode), (fc2._path, fc2._filenode)
|
|
|
v = ancestor.ancestor(a, b, parents)
|
|
|
if v:
|
|
|
f, n = v
|
|
|
return filectx(self._repo, f, fileid=n, filelog=flcache[f])
|
|
|
|
|
|
return None
|
|
|
|
|
|
class workingctx(changectx):
|
|
|
"""A workingctx object makes access to data related to
|
|
|
the current working directory convenient."""
|
|
|
def __init__(self, repo):
|
|
|
self._repo = repo
|
|
|
self._rev = None
|
|
|
self._node = None
|
|
|
|
|
|
def __str__(self):
|
|
|
return str(self._parents[0]) + "+"
|
|
|
|
|
|
def __nonzero__(self):
|
|
|
return True
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
if name == '_parents':
|
|
|
self._parents = self._repo.parents()
|
|
|
return self._parents
|
|
|
if name == '_status':
|
|
|
self._status = self._repo.status()
|
|
|
return self._status
|
|
|
if name == '_manifest':
|
|
|
self._buildmanifest()
|
|
|
return self._manifest
|
|
|
else:
|
|
|
raise AttributeError, name
|
|
|
|
|
|
def _buildmanifest(self):
|
|
|
"""generate a manifest corresponding to the working directory"""
|
|
|
|
|
|
man = self._parents[0].manifest().copy()
|
|
|
copied = self._repo.dirstate.copies()
|
|
|
modified, added, removed, deleted, unknown = self._status[:5]
|
|
|
for i, l in (("a", added), ("m", modified), ("u", unknown)):
|
|
|
for f in l:
|
|
|
man[f] = man.get(copied.get(f, f), nullid) + i
|
|
|
try:
|
|
|
man.set(f, util.is_exec(self._repo.wjoin(f), man.execf(f)))
|
|
|
except OSError:
|
|
|
pass
|
|
|
|
|
|
for f in deleted + removed:
|
|
|
if f in man:
|
|
|
del man[f]
|
|
|
|
|
|
self._manifest = man
|
|
|
|
|
|
def manifest(self): return self._manifest
|
|
|
|
|
|
def user(self): return self._repo.ui.username()
|
|
|
def date(self): return util.makedate()
|
|
|
def description(self): return ""
|
|
|
def files(self):
|
|
|
f = self.modified() + self.added() + self.removed()
|
|
|
f.sort()
|
|
|
return f
|
|
|
|
|
|
def modified(self): return self._status[0]
|
|
|
def added(self): return self._status[1]
|
|
|
def removed(self): return self._status[2]
|
|
|
def deleted(self): return self._status[3]
|
|
|
def unknown(self): return self._status[4]
|
|
|
def clean(self): return self._status[5]
|
|
|
def branch(self):
|
|
|
try:
|
|
|
return self._repo.opener("branch").read().strip()
|
|
|
except IOError:
|
|
|
return ""
|
|
|
|
|
|
def parents(self):
|
|
|
"""return contexts for each parent changeset"""
|
|
|
return self._parents
|
|
|
|
|
|
def children(self):
|
|
|
return []
|
|
|
|
|
|
def filectx(self, path):
|
|
|
"""get a file context from the working directory"""
|
|
|
return workingfilectx(self._repo, path, workingctx=self)
|
|
|
|
|
|
def ancestor(self, c2):
|
|
|
"""return the ancestor context of self and c2"""
|
|
|
return self._parents[0].ancestor(c2) # punt on two parents for now
|
|
|
|
|
|
class workingfilectx(filectx):
|
|
|
"""A workingfilectx object makes access to data related to a particular
|
|
|
file in the working directory convenient."""
|
|
|
def __init__(self, repo, path, filelog=None, workingctx=None):
|
|
|
"""changeid can be a changeset revision, node, or tag.
|
|
|
fileid can be a file revision or node."""
|
|
|
self._repo = repo
|
|
|
self._path = path
|
|
|
self._changeid = None
|
|
|
self._filerev = self._filenode = None
|
|
|
|
|
|
if filelog:
|
|
|
self._filelog = filelog
|
|
|
if workingctx:
|
|
|
self._changectx = workingctx
|
|
|
|
|
|
def __getattr__(self, name):
|
|
|
if name == '_changectx':
|
|
|
self._changectx = workingctx(repo)
|
|
|
return self._changectx
|
|
|
elif name == '_repopath':
|
|
|
self._repopath = (self._repo.dirstate.copied(self._path)
|
|
|
or self._path)
|
|
|
return self._repopath
|
|
|
elif name == '_filelog':
|
|
|
self._filelog = self._repo.file(self._repopath)
|
|
|
return self._filelog
|
|
|
else:
|
|
|
raise AttributeError, name
|
|
|
|
|
|
def __nonzero__(self):
|
|
|
return True
|
|
|
|
|
|
def __str__(self):
|
|
|
return "%s@%s" % (self.path(), self._changectx)
|
|
|
|
|
|
def filectx(self, fileid):
|
|
|
'''opens an arbitrary revision of the file without
|
|
|
opening a new filelog'''
|
|
|
return filectx(self._repo, self._repopath, fileid=fileid,
|
|
|
filelog=self._filelog)
|
|
|
|
|
|
def rev(self):
|
|
|
if '_changectx' in self.__dict__:
|
|
|
return self._changectx.rev()
|
|
|
return self._filelog.linkrev(self._filenode)
|
|
|
|
|
|
def data(self): return self._repo.wread(self._path)
|
|
|
def renamed(self):
|
|
|
rp = self._repopath
|
|
|
if rp == self._path:
|
|
|
return None
|
|
|
return rp, self._workingctx._parents._manifest.get(rp, nullid)
|
|
|
|
|
|
def parents(self):
|
|
|
'''return parent filectxs, following copies if necessary'''
|
|
|
p = self._path
|
|
|
rp = self._repopath
|
|
|
pcl = self._changectx._parents
|
|
|
fl = self._filelog
|
|
|
pl = [(rp, pcl[0]._manifest.get(rp, nullid), fl)]
|
|
|
if len(pcl) > 1:
|
|
|
if rp != p:
|
|
|
fl = None
|
|
|
pl.append((p, pcl[1]._manifest.get(p, nullid), fl))
|
|
|
|
|
|
return [filectx(self._repo, p, fileid=n, filelog=l)
|
|
|
for p,n,l in pl if n != nullid]
|
|
|
|
|
|
def children(self):
|
|
|
return []
|
|
|
|
|
|
def size(self): return os.stat(self._repo.wjoin(self._path)).st_size
|
|
|
|
|
|
def cmp(self, text): return self._repo.wread(self._path) == text
|
|
|
|
