cmdutil.py
1319 lines
| 46.0 KiB
| text/x-python
|
PythonLexer
/ mercurial / cmdutil.py
Vadim Gelfer
|
r2957 | # cmdutil.py - help for command processing in mercurial | ||
Vadim Gelfer
|
r2874 | # | ||
Thomas Arendsen Hein
|
r4635 | # Copyright 2005-2007 Matt Mackall <mpm@selenic.com> | ||
Vadim Gelfer
|
r2874 | # | ||
Martin Geisler
|
r8225 | # This software may be used and distributed according to the terms of the | ||
Matt Mackall
|
r10263 | # GNU General Public License version 2 or any later version. | ||
Vadim Gelfer
|
r2874 | |||
Joel Rosdahl
|
r6211 | from node import hex, nullid, nullrev, short | ||
Matt Mackall
|
r3891 | from i18n import _ | ||
Sune Foldager
|
r14269 | import os, sys, errno, re, tempfile | ||
Matt Mackall
|
r15214 | import util, scmutil, templater, patch, error, templatekw, revlog, copies | ||
Martin Geisler
|
r12085 | import match as matchmod | ||
Matt Mackall
|
r14319 | import subrepo | ||
Vadim Gelfer
|
r2874 | |||
Brendan Cully
|
r10401 | def parsealiases(cmd): | ||
return cmd.lstrip("^").split("|") | ||||
Matt Mackall
|
r7213 | def findpossible(cmd, table, strict=False): | ||
Matt Mackall
|
r4549 | """ | ||
Return cmd -> (aliases, command table entry) | ||||
for each matching command. | ||||
Return debug commands (or their aliases) only if no normal command matches. | ||||
""" | ||||
choice = {} | ||||
debugchoice = {} | ||||
Matt Mackall
|
r15600 | |||
if cmd in table: | ||||
# short-circuit exact matches, "log" alias beats "^log|history" | ||||
keys = [cmd] | ||||
else: | ||||
keys = table.keys() | ||||
for e in keys: | ||||
Brendan Cully
|
r10401 | aliases = parsealiases(e) | ||
Matt Mackall
|
r4549 | found = None | ||
if cmd in aliases: | ||||
found = cmd | ||||
Matt Mackall
|
r7213 | elif not strict: | ||
Matt Mackall
|
r4549 | for a in aliases: | ||
if a.startswith(cmd): | ||||
found = a | ||||
break | ||||
if found is not None: | ||||
if aliases[0].startswith("debug") or found.startswith("debug"): | ||||
Matt Mackall
|
r5178 | debugchoice[found] = (aliases, table[e]) | ||
Matt Mackall
|
r4549 | else: | ||
Matt Mackall
|
r5178 | choice[found] = (aliases, table[e]) | ||
Matt Mackall
|
r4549 | |||
if not choice and debugchoice: | ||||
choice = debugchoice | ||||
return choice | ||||
Matt Mackall
|
r7213 | def findcmd(cmd, table, strict=True): | ||
Matt Mackall
|
r4549 | """Return (aliases, command table entry) for command string.""" | ||
Matt Mackall
|
r7213 | choice = findpossible(cmd, table, strict) | ||
Matt Mackall
|
r4549 | |||
Christian Ebert
|
r5915 | if cmd in choice: | ||
Matt Mackall
|
r4549 | return choice[cmd] | ||
if len(choice) > 1: | ||||
clist = choice.keys() | ||||
clist.sort() | ||||
Matt Mackall
|
r7643 | raise error.AmbiguousCommand(cmd, clist) | ||
Matt Mackall
|
r4549 | |||
if choice: | ||||
return choice.values()[0] | ||||
Matt Mackall
|
r7643 | raise error.UnknownCommand(cmd) | ||
Matt Mackall
|
r4549 | |||
Brendan Cully
|
r10402 | def findrepo(p): | ||
while not os.path.isdir(os.path.join(p, ".hg")): | ||||
oldp, p = p, os.path.dirname(p) | ||||
if p == oldp: | ||||
return None | ||||
return p | ||||
Matt Mackall
|
r14289 | def bailifchanged(repo): | ||
Matt Mackall
|
r13878 | if repo.dirstate.p2() != nullid: | ||
Matt Mackall
|
r5716 | raise util.Abort(_('outstanding uncommitted merge')) | ||
Matt Mackall
|
r4549 | modified, added, removed, deleted = repo.status()[:4] | ||
if modified or added or removed or deleted: | ||||
raise util.Abort(_("outstanding uncommitted changes")) | ||||
Eric Roshan Eisner
|
r15231 | ctx = repo[None] | ||
for s in ctx.substate: | ||||
if ctx.sub(s).dirty(): | ||||
raise util.Abort(_("uncommitted changes in subrepo %s") % s) | ||||
Matt Mackall
|
r4549 | |||
Idan Kamara
|
r14635 | def logmessage(ui, opts): | ||
Matt Mackall
|
r4549 | """ get the log message according to -m and -l option """ | ||
Alexander Solovyov
|
r7667 | message = opts.get('message') | ||
logfile = opts.get('logfile') | ||||
Matt Mackall
|
r4549 | |||
if message and logfile: | ||||
raise util.Abort(_('options --message and --logfile are mutually ' | ||||
'exclusive')) | ||||
if not message and logfile: | ||||
try: | ||||
if logfile == '-': | ||||
Idan Kamara
|
r14635 | message = ui.fin.read() | ||
Matt Mackall
|
r4549 | else: | ||
Patrick Mezard
|
r14249 | message = '\n'.join(util.readfile(logfile).splitlines()) | ||
Matt Mackall
|
r4549 | except IOError, inst: | ||
raise util.Abort(_("can't read commit message '%s': %s") % | ||||
(logfile, inst.strerror)) | ||||
return message | ||||
Thomas Arendsen Hein
|
r6190 | def loglimit(opts): | ||
"""get the log limit according to option -l/--limit""" | ||||
limit = opts.get('limit') | ||||
if limit: | ||||
try: | ||||
limit = int(limit) | ||||
except ValueError: | ||||
raise util.Abort(_('limit must be a positive integer')) | ||||
Matt Mackall
|
r10282 | if limit <= 0: | ||
raise util.Abort(_('limit must be positive')) | ||||
Thomas Arendsen Hein
|
r6190 | else: | ||
Nicolas Dumazet
|
r10111 | limit = None | ||
Thomas Arendsen Hein
|
r6190 | return limit | ||
Andrzej Bieniek
|
r14986 | def makefilename(repo, pat, node, desc=None, | ||
Vadim Gelfer
|
r2874 | total=None, seqno=None, revwidth=None, pathname=None): | ||
node_expander = { | ||||
'H': lambda: hex(node), | ||||
'R': lambda: str(repo.changelog.rev(node)), | ||||
'h': lambda: short(node), | ||||
Andrzej Bieniek
|
r14986 | 'm': lambda: re.sub('[^\w]', '_', str(desc)) | ||
Vadim Gelfer
|
r2874 | } | ||
expander = { | ||||
'%': lambda: '%', | ||||
'b': lambda: os.path.basename(repo.root), | ||||
} | ||||
try: | ||||
if node: | ||||
expander.update(node_expander) | ||||
Alexis S. L. Carvalho
|
r4836 | if node: | ||
Vadim Gelfer
|
r2874 | expander['r'] = (lambda: | ||
Alexis S. L. Carvalho
|
r4836 | str(repo.changelog.rev(node)).zfill(revwidth or 0)) | ||
Vadim Gelfer
|
r2874 | if total is not None: | ||
expander['N'] = lambda: str(total) | ||||
if seqno is not None: | ||||
expander['n'] = lambda: str(seqno) | ||||
if total is not None and seqno is not None: | ||||
Thomas Arendsen Hein
|
r3673 | expander['n'] = lambda: str(seqno).zfill(len(str(total))) | ||
Vadim Gelfer
|
r2874 | if pathname is not None: | ||
expander['s'] = lambda: os.path.basename(pathname) | ||||
expander['d'] = lambda: os.path.dirname(pathname) or '.' | ||||
expander['p'] = lambda: pathname | ||||
newname = [] | ||||
patlen = len(pat) | ||||
i = 0 | ||||
while i < patlen: | ||||
c = pat[i] | ||||
if c == '%': | ||||
i += 1 | ||||
c = pat[i] | ||||
c = expander[c]() | ||||
newname.append(c) | ||||
i += 1 | ||||
return ''.join(newname) | ||||
except KeyError, inst: | ||||
timeless
|
r8761 | raise util.Abort(_("invalid format spec '%%%s' in output filename") % | ||
Thomas Arendsen Hein
|
r3072 | inst.args[0]) | ||
Vadim Gelfer
|
r2874 | |||
Andrzej Bieniek
|
r14986 | def makefileobj(repo, pat, node=None, desc=None, total=None, | ||
Matt Mackall
|
r14291 | seqno=None, revwidth=None, mode='wb', pathname=None): | ||
Ronny Pfannschmidt
|
r7319 | |||
Adrian Buehlmann
|
r13769 | writable = mode not in ('r', 'rb') | ||
Ronny Pfannschmidt
|
r7319 | |||
Vadim Gelfer
|
r2874 | if not pat or pat == '-': | ||
Idan Kamara
|
r14637 | fp = writable and repo.ui.fout or repo.ui.fin | ||
Augie Fackler
|
r14948 | if util.safehasattr(fp, 'fileno'): | ||
Idan Kamara
|
r14638 | return os.fdopen(os.dup(fp.fileno()), mode) | ||
else: | ||||
# if this fp can't be duped properly, return | ||||
# a dummy object that can be closed | ||||
class wrappedfileobj(object): | ||||
noop = lambda x: None | ||||
def __init__(self, f): | ||||
self.f = f | ||||
def __getattr__(self, attr): | ||||
if attr == 'close': | ||||
return self.noop | ||||
else: | ||||
return getattr(self.f, attr) | ||||
return wrappedfileobj(fp) | ||||
Augie Fackler
|
r14948 | if util.safehasattr(pat, 'write') and writable: | ||
Vadim Gelfer
|
r2874 | return pat | ||
Augie Fackler
|
r14948 | if util.safehasattr(pat, 'read') and 'r' in mode: | ||
Vadim Gelfer
|
r2874 | return pat | ||
Andrzej Bieniek
|
r14986 | return open(makefilename(repo, pat, node, desc, total, seqno, revwidth, | ||
Vadim Gelfer
|
r2874 | pathname), | ||
mode) | ||||
Vadim Gelfer
|
r2882 | |||
Sune Foldager
|
r14323 | def openrevlog(repo, cmd, file_, opts): | ||
"""opens the changelog, manifest, a filelog or a given revlog""" | ||||
cl = opts['changelog'] | ||||
mf = opts['manifest'] | ||||
msg = None | ||||
if cl and mf: | ||||
msg = _('cannot specify --changelog and --manifest at the same time') | ||||
elif cl or mf: | ||||
if file_: | ||||
msg = _('cannot specify filename with --changelog or --manifest') | ||||
elif not repo: | ||||
msg = _('cannot specify --changelog or --manifest ' | ||||
'without a repository') | ||||
if msg: | ||||
raise util.Abort(msg) | ||||
r = None | ||||
if repo: | ||||
if cl: | ||||
r = repo.changelog | ||||
elif mf: | ||||
r = repo.manifest | ||||
elif file_: | ||||
filelog = repo.file(file_) | ||||
if len(filelog): | ||||
r = filelog | ||||
if not r: | ||||
if not file_: | ||||
raise error.CommandError(cmd, _('invalid arguments')) | ||||
if not os.path.isfile(file_): | ||||
raise util.Abort(_("revlog '%s' not found") % file_) | ||||
r = revlog.revlog(scmutil.opener(os.getcwd(), audit=False), | ||||
file_[:-2] + ".i") | ||||
return r | ||||
Matt Mackall
|
r5610 | def copy(ui, repo, pats, opts, rename=False): | ||
Matt Mackall
|
r5589 | # called with the repo lock held | ||
# | ||||
# hgsep => pathname that uses "/" to separate directories | ||||
# ossep => pathname that uses os.sep to separate directories | ||||
cwd = repo.getcwd() | ||||
targets = {} | ||||
Matt Mackall
|
r5607 | after = opts.get("after") | ||
dryrun = opts.get("dry_run") | ||||
Dirkjan Ochtman
|
r11303 | wctx = repo[None] | ||
Matt Mackall
|
r5589 | |||
Matt Mackall
|
r5605 | def walkpat(pat): | ||
srcs = [] | ||||
Peter Arrenbrecht
|
r11223 | badstates = after and '?' or '?r' | ||
Matt Mackall
|
r14671 | m = scmutil.match(repo[None], [pat], opts, globbed=True) | ||
Matt Mackall
|
r6586 | for abs in repo.walk(m): | ||
Matt Mackall
|
r5605 | state = repo.dirstate[abs] | ||
Matt Mackall
|
r6584 | rel = m.rel(abs) | ||
exact = m.exact(abs) | ||||
Peter Arrenbrecht
|
r11223 | if state in badstates: | ||
Matt Mackall
|
r5605 | if exact and state == '?': | ||
ui.warn(_('%s: not copying - file is not managed\n') % rel) | ||||
if exact and state == 'r': | ||||
ui.warn(_('%s: not copying - file has been marked for' | ||||
' remove\n') % rel) | ||||
continue | ||||
# abs: hgsep | ||||
# rel: ossep | ||||
srcs.append((abs, rel, exact)) | ||||
return srcs | ||||
Matt Mackall
|
r5589 | |||
# abssrc: hgsep | ||||
# relsrc: ossep | ||||
# otarget: ossep | ||||
Matt Mackall
|
r5605 | def copyfile(abssrc, relsrc, otarget, exact): | ||
Adrian Buehlmann
|
r13971 | abstarget = scmutil.canonpath(repo.root, cwd, otarget) | ||
Matt Mackall
|
r5589 | reltarget = repo.pathto(abstarget, cwd) | ||
Matt Mackall
|
r5607 | target = repo.wjoin(abstarget) | ||
Matt Mackall
|
r5589 | src = repo.wjoin(abssrc) | ||
Matt Mackall
|
r5608 | state = repo.dirstate[abstarget] | ||
Matt Mackall
|
r5607 | |||
Adrian Buehlmann
|
r13962 | scmutil.checkportable(ui, abstarget) | ||
Adrian Buehlmann
|
r13945 | |||
Matt Mackall
|
r5607 | # check for collisions | ||
prevsrc = targets.get(abstarget) | ||||
Matt Mackall
|
r5589 | if prevsrc is not None: | ||
ui.warn(_('%s: not overwriting - %s collides with %s\n') % | ||||
(reltarget, repo.pathto(abssrc, cwd), | ||||
repo.pathto(prevsrc, cwd))) | ||||
return | ||||
Matt Mackall
|
r5607 | |||
# check for overwrites | ||||
Patrick Mezard
|
r12342 | exists = os.path.lexists(target) | ||
Martin Geisler
|
r8117 | if not after and exists or after and state in 'mn': | ||
Matt Mackall
|
r5589 | if not opts['force']: | ||
ui.warn(_('%s: not overwriting - file exists\n') % | ||||
reltarget) | ||||
return | ||||
Matt Mackall
|
r5607 | |||
if after: | ||||
Matt Mackall
|
r5608 | if not exists: | ||
Steve Losh
|
r11152 | if rename: | ||
ui.warn(_('%s: not recording move - %s does not exist\n') % | ||||
(relsrc, reltarget)) | ||||
else: | ||||
ui.warn(_('%s: not recording copy - %s does not exist\n') % | ||||
(relsrc, reltarget)) | ||||
Matt Mackall
|
r5589 | return | ||
Matt Mackall
|
r5608 | elif not dryrun: | ||
Matt Mackall
|
r5589 | try: | ||
Matt Mackall
|
r5608 | if exists: | ||
os.unlink(target) | ||||
targetdir = os.path.dirname(target) or '.' | ||||
if not os.path.isdir(targetdir): | ||||
os.makedirs(targetdir) | ||||
util.copyfile(src, target) | ||||
Adrian Buehlmann
|
r14518 | srcexists = True | ||
Matt Mackall
|
r5589 | except IOError, inst: | ||
if inst.errno == errno.ENOENT: | ||||
ui.warn(_('%s: deleted in working copy\n') % relsrc) | ||||
Adrian Buehlmann
|
r14518 | srcexists = False | ||
Matt Mackall
|
r5589 | else: | ||
ui.warn(_('%s: cannot copy - %s\n') % | ||||
(relsrc, inst.strerror)) | ||||
Matt Mackall
|
r5606 | return True # report a failure | ||
Matt Mackall
|
r5607 | |||
Matt Mackall
|
r5589 | if ui.verbose or not exact: | ||
Martin Geisler
|
r7894 | if rename: | ||
ui.status(_('moving %s to %s\n') % (relsrc, reltarget)) | ||||
else: | ||||
ui.status(_('copying %s to %s\n') % (relsrc, reltarget)) | ||||
Matt Mackall
|
r5608 | |||
Matt Mackall
|
r5589 | targets[abstarget] = abssrc | ||
Matt Mackall
|
r5607 | |||
# fix up dirstate | ||||
Matt Mackall
|
r14321 | scmutil.dirstatecopy(ui, repo, wctx, abssrc, abstarget, | ||
dryrun=dryrun, cwd=cwd) | ||||
Matt Mackall
|
r5610 | if rename and not dryrun: | ||
Adrian Buehlmann
|
r14518 | if not after and srcexists: | ||
util.unlinkpath(repo.wjoin(abssrc)) | ||||
wctx.forget([abssrc]) | ||||
Matt Mackall
|
r5589 | |||
# pat: ossep | ||||
# dest ossep | ||||
# srcs: list of (hgsep, hgsep, ossep, bool) | ||||
# return: function that takes hgsep and returns ossep | ||||
def targetpathfn(pat, dest, srcs): | ||||
if os.path.isdir(pat): | ||||
Adrian Buehlmann
|
r13971 | abspfx = scmutil.canonpath(repo.root, cwd, pat) | ||
Matt Mackall
|
r5589 | abspfx = util.localpath(abspfx) | ||
if destdirexists: | ||||
striplen = len(os.path.split(abspfx)[0]) | ||||
else: | ||||
striplen = len(abspfx) | ||||
if striplen: | ||||
striplen += len(os.sep) | ||||
res = lambda p: os.path.join(dest, util.localpath(p)[striplen:]) | ||||
elif destdirexists: | ||||
res = lambda p: os.path.join(dest, | ||||
os.path.basename(util.localpath(p))) | ||||
else: | ||||
res = lambda p: dest | ||||
return res | ||||
# pat: ossep | ||||
# dest ossep | ||||
# srcs: list of (hgsep, hgsep, ossep, bool) | ||||
# return: function that takes hgsep and returns ossep | ||||
def targetpathafterfn(pat, dest, srcs): | ||||
Martin Geisler
|
r12085 | if matchmod.patkind(pat): | ||
Matt Mackall
|
r5589 | # a mercurial pattern | ||
res = lambda p: os.path.join(dest, | ||||
os.path.basename(util.localpath(p))) | ||||
else: | ||||
Adrian Buehlmann
|
r13971 | abspfx = scmutil.canonpath(repo.root, cwd, pat) | ||
Matt Mackall
|
r5589 | if len(abspfx) < len(srcs[0][0]): | ||
# A directory. Either the target path contains the last | ||||
# component of the source path or it does not. | ||||
def evalpath(striplen): | ||||
score = 0 | ||||
for s in srcs: | ||||
t = os.path.join(dest, util.localpath(s[0])[striplen:]) | ||||
Patrick Mezard
|
r12357 | if os.path.lexists(t): | ||
Matt Mackall
|
r5589 | score += 1 | ||
return score | ||||
abspfx = util.localpath(abspfx) | ||||
striplen = len(abspfx) | ||||
if striplen: | ||||
striplen += len(os.sep) | ||||
if os.path.isdir(os.path.join(dest, os.path.split(abspfx)[1])): | ||||
score = evalpath(striplen) | ||||
striplen1 = len(os.path.split(abspfx)[0]) | ||||
if striplen1: | ||||
striplen1 += len(os.sep) | ||||
if evalpath(striplen1) > score: | ||||
striplen = striplen1 | ||||
res = lambda p: os.path.join(dest, | ||||
util.localpath(p)[striplen:]) | ||||
else: | ||||
# a file | ||||
if destdirexists: | ||||
res = lambda p: os.path.join(dest, | ||||
os.path.basename(util.localpath(p))) | ||||
else: | ||||
res = lambda p: dest | ||||
return res | ||||
Matt Mackall
|
r14321 | pats = scmutil.expandpats(pats) | ||
Matt Mackall
|
r5589 | if not pats: | ||
raise util.Abort(_('no source or destination specified')) | ||||
if len(pats) == 1: | ||||
raise util.Abort(_('no destination specified')) | ||||
dest = pats.pop() | ||||
Alexis S. L. Carvalho
|
r6258 | destdirexists = os.path.isdir(dest) and not os.path.islink(dest) | ||
Matt Mackall
|
r5589 | if not destdirexists: | ||
Martin Geisler
|
r12085 | if len(pats) > 1 or matchmod.patkind(pats[0]): | ||
Matt Mackall
|
r5589 | raise util.Abort(_('with multiple sources, destination must be an ' | ||
'existing directory')) | ||||
Shun-ichi GOTO
|
r5843 | if util.endswithsep(dest): | ||
Matt Mackall
|
r5589 | raise util.Abort(_('destination %s is not a directory') % dest) | ||
Matt Mackall
|
r5607 | |||
tfn = targetpathfn | ||||
if after: | ||||
Matt Mackall
|
r5589 | tfn = targetpathafterfn | ||
copylist = [] | ||||
for pat in pats: | ||||
Matt Mackall
|
r5605 | srcs = walkpat(pat) | ||
Matt Mackall
|
r5589 | if not srcs: | ||
continue | ||||
copylist.append((tfn(pat, dest, srcs), srcs)) | ||||
if not copylist: | ||||
raise util.Abort(_('no files to copy')) | ||||
Matt Mackall
|
r5606 | errors = 0 | ||
Matt Mackall
|
r5589 | for targetpath, srcs in copylist: | ||
Matt Mackall
|
r5605 | for abssrc, relsrc, exact in srcs: | ||
Matt Mackall
|
r5606 | if copyfile(abssrc, relsrc, targetpath(abssrc), exact): | ||
errors += 1 | ||||
Matt Mackall
|
r5589 | |||
if errors: | ||||
ui.warn(_('(consider using --after)\n')) | ||||
Matt Mackall
|
r5609 | |||
Matt Mackall
|
r11177 | return errors != 0 | ||
Matt Mackall
|
r5589 | |||
Nicolas Dumazet
|
r9513 | def service(opts, parentfn=None, initfn=None, runfn=None, logfile=None, | ||
Nicolas Dumazet
|
r10012 | runargs=None, appendpid=False): | ||
Bryan O'Sullivan
|
r4380 | '''Run a command as a service.''' | ||
if opts['daemon'] and not opts['daemon_pipefds']: | ||||
Patrick Mezard
|
r10238 | # Signal child process startup with file removal | ||
lockfd, lockpath = tempfile.mkstemp(prefix='hg-service-') | ||||
Matt Mackall
|
r10282 | os.close(lockfd) | ||
Patrick Mezard
|
r10238 | try: | ||
if not runargs: | ||||
Patrick Mezard
|
r10239 | runargs = util.hgcmd() + sys.argv[1:] | ||
Patrick Mezard
|
r10238 | runargs.append('--daemon-pipefds=%s' % lockpath) | ||
# Don't pass --cwd to the child process, because we've already | ||||
# changed directory. | ||||
Matt Mackall
|
r10282 | for i in xrange(1, len(runargs)): | ||
Patrick Mezard
|
r10238 | if runargs[i].startswith('--cwd='): | ||
del runargs[i] | ||||
break | ||||
elif runargs[i].startswith('--cwd'): | ||||
Matt Mackall
|
r10282 | del runargs[i:i + 2] | ||
Patrick Mezard
|
r10238 | break | ||
Patrick Mezard
|
r10344 | def condfn(): | ||
return not os.path.exists(lockpath) | ||||
pid = util.rundetached(runargs, condfn) | ||||
if pid < 0: | ||||
raise util.Abort(_('child process failed to start')) | ||||
Patrick Mezard
|
r10238 | finally: | ||
try: | ||||
os.unlink(lockpath) | ||||
except OSError, e: | ||||
if e.errno != errno.ENOENT: | ||||
raise | ||||
Bryan O'Sullivan
|
r4380 | if parentfn: | ||
return parentfn(pid) | ||||
else: | ||||
Nicolas Dumazet
|
r9896 | return | ||
Bryan O'Sullivan
|
r4380 | |||
if initfn: | ||||
initfn() | ||||
if opts['pid_file']: | ||||
Nicolas Dumazet
|
r10012 | mode = appendpid and 'a' or 'w' | ||
fp = open(opts['pid_file'], mode) | ||||
Bryan O'Sullivan
|
r4380 | fp.write(str(os.getpid()) + '\n') | ||
fp.close() | ||||
if opts['daemon_pipefds']: | ||||
Patrick Mezard
|
r10238 | lockpath = opts['daemon_pipefds'] | ||
Bryan O'Sullivan
|
r4380 | try: | ||
os.setsid() | ||||
except AttributeError: | ||||
pass | ||||
Patrick Mezard
|
r10238 | os.unlink(lockpath) | ||
Patrick Mezard
|
r10240 | util.hidewindow() | ||
Bryan O'Sullivan
|
r4380 | sys.stdout.flush() | ||
sys.stderr.flush() | ||||
Nicolas Dumazet
|
r8789 | |||
nullfd = os.open(util.nulldev, os.O_RDWR) | ||||
logfilefd = nullfd | ||||
if logfile: | ||||
logfilefd = os.open(logfile, os.O_RDWR | os.O_CREAT | os.O_APPEND) | ||||
os.dup2(nullfd, 0) | ||||
os.dup2(logfilefd, 1) | ||||
os.dup2(logfilefd, 2) | ||||
if nullfd not in (0, 1, 2): | ||||
os.close(nullfd) | ||||
if logfile and logfilefd not in (0, 1, 2): | ||||
os.close(logfilefd) | ||||
Bryan O'Sullivan
|
r4380 | |||
if runfn: | ||||
return runfn() | ||||
Benoit Boissinot
|
r10611 | def export(repo, revs, template='hg-%h.patch', fp=None, switch_parent=False, | ||
opts=None): | ||||
'''export changesets as hg patches.''' | ||||
total = len(revs) | ||||
revwidth = max([len(str(rev)) for rev in revs]) | ||||
def single(rev, seqno, fp): | ||||
ctx = repo[rev] | ||||
node = ctx.node() | ||||
parents = [p.node() for p in ctx.parents() if p] | ||||
branch = ctx.branch() | ||||
if switch_parent: | ||||
parents.reverse() | ||||
prev = (parents and parents[0]) or nullid | ||||
Dan Villiom Podlaski Christiansen
|
r13400 | shouldclose = False | ||
Benoit Boissinot
|
r10611 | if not fp: | ||
Andrzej Bieniek
|
r14986 | desc_lines = ctx.description().rstrip().split('\n') | ||
desc = desc_lines[0] #Commit always has a first line. | ||||
fp = makefileobj(repo, template, node, desc=desc, total=total, | ||||
seqno=seqno, revwidth=revwidth, mode='ab') | ||||
Waqas Hussain
|
r13467 | if fp != template: | ||
shouldclose = True | ||||
Augie Fackler
|
r14948 | if fp != sys.stdout and util.safehasattr(fp, 'name'): | ||
Benoit Boissinot
|
r10611 | repo.ui.note("%s\n" % fp.name) | ||
fp.write("# HG changeset patch\n") | ||||
fp.write("# User %s\n" % ctx.user()) | ||||
fp.write("# Date %d %d\n" % ctx.date()) | ||||
Martin Geisler
|
r11821 | if branch and branch != 'default': | ||
Benoit Boissinot
|
r10611 | fp.write("# Branch %s\n" % branch) | ||
fp.write("# Node ID %s\n" % hex(node)) | ||||
fp.write("# Parent %s\n" % hex(prev)) | ||||
if len(parents) > 1: | ||||
fp.write("# Parent %s\n" % hex(parents[1])) | ||||
fp.write(ctx.description().rstrip()) | ||||
fp.write("\n\n") | ||||
for chunk in patch.diff(repo, prev, node, opts=opts): | ||||
fp.write(chunk) | ||||
Dan Villiom Podlaski Christiansen
|
r13400 | if shouldclose: | ||
fp.close() | ||||
Dan Villiom Podlaski Christiansen
|
r13081 | |||
Benoit Boissinot
|
r10611 | for seqno, rev in enumerate(revs): | ||
single(rev, seqno + 1, fp) | ||||
Yuya Nishihara
|
r11050 | def diffordiffstat(ui, repo, diffopts, node1, node2, match, | ||
Martin Geisler
|
r12167 | changes=None, stat=False, fp=None, prefix='', | ||
listsubrepos=False): | ||||
Yuya Nishihara
|
r11050 | '''show diff or diffstat.''' | ||
if fp is None: | ||||
write = ui.write | ||||
else: | ||||
def write(s, **kw): | ||||
fp.write(s) | ||||
if stat: | ||||
Alecs King
|
r11950 | diffopts = diffopts.copy(context=0) | ||
Yuya Nishihara
|
r11050 | width = 80 | ||
if not ui.plain(): | ||||
Augie Fackler
|
r12689 | width = ui.termwidth() | ||
Martin Geisler
|
r12167 | chunks = patch.diff(repo, node1, node2, match, changes, diffopts, | ||
prefix=prefix) | ||||
Yuya Nishihara
|
r11050 | for chunk, label in patch.diffstatui(util.iterlines(chunks), | ||
width=width, | ||||
git=diffopts.git): | ||||
write(chunk, label=label) | ||||
else: | ||||
for chunk, label in patch.diffui(repo, node1, node2, match, | ||||
Martin Geisler
|
r12167 | changes, diffopts, prefix=prefix): | ||
Yuya Nishihara
|
r11050 | write(chunk, label=label) | ||
Martin Geisler
|
r12167 | if listsubrepos: | ||
ctx1 = repo[node1] | ||||
Martin Geisler
|
r12175 | ctx2 = repo[node2] | ||
Martin Geisler
|
r12176 | for subpath, sub in subrepo.itersubrepos(ctx1, ctx2): | ||
Alistair Bell
|
r15698 | tempnode2 = node2 | ||
Renato Cunha
|
r15634 | try: | ||
if node2 is not None: | ||||
Alistair Bell
|
r15698 | tempnode2 = ctx2.substate[subpath][1] | ||
Renato Cunha
|
r15634 | except KeyError: | ||
# A subrepo that existed in node1 was deleted between node1 and | ||||
# node2 (inclusive). Thus, ctx2's substate won't contain that | ||||
# subpath. The best we can do is to ignore it. | ||||
Alistair Bell
|
r15698 | tempnode2 = None | ||
Martin Geisler
|
r12167 | submatch = matchmod.narrowmatcher(subpath, match) | ||
Alistair Bell
|
r15698 | sub.diff(diffopts, tempnode2, submatch, changes=changes, | ||
Martin Geisler
|
r12167 | stat=stat, fp=fp, prefix=prefix) | ||
Matt Mackall
|
r3643 | class changeset_printer(object): | ||
'''show changeset information when templating not requested.''' | ||||
Jim Correia
|
r7762 | def __init__(self, ui, repo, patch, diffopts, buffered): | ||
Matt Mackall
|
r3643 | self.ui = ui | ||
self.repo = repo | ||||
Matt Mackall
|
r3645 | self.buffered = buffered | ||
self.patch = patch | ||||
Jim Correia
|
r7762 | self.diffopts = diffopts | ||
Matt Mackall
|
r3738 | self.header = {} | ||
self.hunk = {} | ||||
self.lastheader = None | ||||
Robert Bachmann
|
r10152 | self.footer = None | ||
Matt Mackall
|
r3645 | |||
def flush(self, rev): | ||||
Matt Mackall
|
r3738 | if rev in self.header: | ||
h = self.header[rev] | ||||
if h != self.lastheader: | ||||
self.lastheader = h | ||||
self.ui.write(h) | ||||
del self.header[rev] | ||||
if rev in self.hunk: | ||||
self.ui.write(self.hunk[rev]) | ||||
del self.hunk[rev] | ||||
return 1 | ||||
return 0 | ||||
Matt Mackall
|
r3643 | |||
Robert Bachmann
|
r10152 | def close(self): | ||
if self.footer: | ||||
self.ui.write(self.footer) | ||||
Mads Kiilerich
|
r11488 | def show(self, ctx, copies=None, matchfn=None, **props): | ||
Matt Mackall
|
r3738 | if self.buffered: | ||
self.ui.pushbuffer() | ||||
Mads Kiilerich
|
r11488 | self._show(ctx, copies, matchfn, props) | ||
Brodie Rao
|
r10819 | self.hunk[ctx.rev()] = self.ui.popbuffer(labeled=True) | ||
Matt Mackall
|
r3738 | else: | ||
Mads Kiilerich
|
r11488 | self._show(ctx, copies, matchfn, props) | ||
Matt Mackall
|
r3738 | |||
Mads Kiilerich
|
r11488 | def _show(self, ctx, copies, matchfn, props): | ||
Matt Mackall
|
r3643 | '''show a single changeset or file revision''' | ||
Dirkjan Ochtman
|
r7369 | changenode = ctx.node() | ||
rev = ctx.rev() | ||||
Matt Mackall
|
r3643 | |||
if self.ui.quiet: | ||||
Brodie Rao
|
r10819 | self.ui.write("%d:%s\n" % (rev, short(changenode)), | ||
label='log.node') | ||||
Matt Mackall
|
r3643 | return | ||
Dirkjan Ochtman
|
r7369 | log = self.repo.changelog | ||
Greg Ward
|
r9547 | date = util.datestr(ctx.date()) | ||
Matt Mackall
|
r3643 | |||
hexfunc = self.ui.debugflag and hex or short | ||||
Thomas Arendsen Hein
|
r4825 | parents = [(p, hexfunc(log.node(p))) | ||
for p in self._meaningful_parentrevs(log, rev)] | ||||
Matt Mackall
|
r3643 | |||
Brodie Rao
|
r10819 | self.ui.write(_("changeset: %d:%s\n") % (rev, hexfunc(changenode)), | ||
label='log.changeset') | ||||
Matt Mackall
|
r3643 | |||
Adrian Buehlmann
|
r9637 | branch = ctx.branch() | ||
Alexis S. L. Carvalho
|
r4176 | # don't show the default branch name | ||
if branch != 'default': | ||||
Brodie Rao
|
r10819 | self.ui.write(_("branch: %s\n") % branch, | ||
label='log.branch') | ||||
David Soria Parra
|
r13386 | for bookmark in self.repo.nodebookmarks(changenode): | ||
self.ui.write(_("bookmark: %s\n") % bookmark, | ||||
label='log.bookmark') | ||||
Matt Mackall
|
r3643 | for tag in self.repo.nodetags(changenode): | ||
Brodie Rao
|
r10819 | self.ui.write(_("tag: %s\n") % tag, | ||
label='log.tag') | ||||
Pierre-Yves David
|
r15907 | if self.ui.debugflag and ctx.phase(): | ||
self.ui.write(_("phase: %s\n") % _(ctx.phasestr()), | ||||
label='log.phase') | ||||
Matt Mackall
|
r3643 | for parent in parents: | ||
Brodie Rao
|
r10819 | self.ui.write(_("parent: %d:%s\n") % parent, | ||
label='log.parent') | ||||
Matt Mackall
|
r3643 | |||
if self.ui.debugflag: | ||||
Greg Ward
|
r9547 | mnode = ctx.manifestnode() | ||
Matt Mackall
|
r3643 | self.ui.write(_("manifest: %d:%s\n") % | ||
Brodie Rao
|
r10819 | (self.repo.manifest.rev(mnode), hex(mnode)), | ||
label='ui.debug log.manifest') | ||||
self.ui.write(_("user: %s\n") % ctx.user(), | ||||
label='log.user') | ||||
self.ui.write(_("date: %s\n") % date, | ||||
label='log.date') | ||||
Matt Mackall
|
r3643 | |||
if self.ui.debugflag: | ||||
files = self.repo.status(log.parents(changenode)[0], changenode)[:3] | ||||
for key, value in zip([_("files:"), _("files+:"), _("files-:")], | ||||
files): | ||||
if value: | ||||
Brodie Rao
|
r10819 | self.ui.write("%-12s %s\n" % (key, " ".join(value)), | ||
label='ui.debug log.files') | ||||
Greg Ward
|
r9547 | elif ctx.files() and self.ui.verbose: | ||
Brodie Rao
|
r10819 | self.ui.write(_("files: %s\n") % " ".join(ctx.files()), | ||
label='ui.note log.files') | ||||
Matt Mackall
|
r3643 | if copies and self.ui.verbose: | ||
copies = ['%s (%s)' % c for c in copies] | ||||
Brodie Rao
|
r10819 | self.ui.write(_("copies: %s\n") % ' '.join(copies), | ||
label='ui.note log.copies') | ||||
Matt Mackall
|
r3643 | |||
Adrian Buehlmann
|
r9637 | extra = ctx.extra() | ||
Matt Mackall
|
r3643 | if extra and self.ui.debugflag: | ||
Matt Mackall
|
r8209 | for key, value in sorted(extra.items()): | ||
Matt Mackall
|
r3643 | self.ui.write(_("extra: %s=%s\n") | ||
Brodie Rao
|
r10819 | % (key, value.encode('string_escape')), | ||
label='ui.debug log.extra') | ||||
Matt Mackall
|
r3643 | |||
Greg Ward
|
r9547 | description = ctx.description().strip() | ||
Matt Mackall
|
r3643 | if description: | ||
if self.ui.verbose: | ||||
Brodie Rao
|
r10819 | self.ui.write(_("description:\n"), | ||
label='ui.note log.description') | ||||
self.ui.write(description, | ||||
label='ui.note log.description') | ||||
Matt Mackall
|
r3643 | self.ui.write("\n\n") | ||
else: | ||||
self.ui.write(_("summary: %s\n") % | ||||
Brodie Rao
|
r10819 | description.splitlines()[0], | ||
label='log.summary') | ||||
Matt Mackall
|
r3643 | self.ui.write("\n") | ||
Mads Kiilerich
|
r11488 | self.showpatch(changenode, matchfn) | ||
Matt Mackall
|
r3645 | |||
Mads Kiilerich
|
r11488 | def showpatch(self, node, matchfn): | ||
if not matchfn: | ||||
matchfn = self.patch | ||||
if matchfn: | ||||
Yuya Nishihara
|
r11061 | stat = self.diffopts.get('stat') | ||
Alecs King
|
r11950 | diff = self.diffopts.get('patch') | ||
Yuya Nishihara
|
r11061 | diffopts = patch.diffopts(self.ui, self.diffopts) | ||
Matt Mackall
|
r3645 | prev = self.repo.changelog.parents(node)[0] | ||
Alecs King
|
r11950 | if stat: | ||
diffordiffstat(self.ui, self.repo, diffopts, prev, node, | ||||
match=matchfn, stat=True) | ||||
if diff: | ||||
if stat: | ||||
self.ui.write("\n") | ||||
diffordiffstat(self.ui, self.repo, diffopts, prev, node, | ||||
match=matchfn, stat=False) | ||||
Matt Mackall
|
r3645 | self.ui.write("\n") | ||
Thomas Arendsen Hein
|
r4825 | def _meaningful_parentrevs(self, log, rev): | ||
"""Return list of meaningful (or all if debug) parentrevs for rev. | ||||
For merges (two non-nullrev revisions) both parents are meaningful. | ||||
Otherwise the first parent revision is considered meaningful if it | ||||
is not the preceding revision. | ||||
""" | ||||
parents = log.parentrevs(rev) | ||||
if not self.ui.debugflag and parents[1] == nullrev: | ||||
if parents[0] >= rev - 1: | ||||
parents = [] | ||||
else: | ||||
parents = [parents[0]] | ||||
return parents | ||||
Matt Mackall
|
r3645 | class changeset_templater(changeset_printer): | ||
Matt Mackall
|
r3643 | '''format changeset information.''' | ||
Jim Correia
|
r7762 | def __init__(self, ui, repo, patch, diffopts, mapfile, buffered): | ||
changeset_printer.__init__(self, ui, repo, patch, diffopts, buffered) | ||||
Dirkjan Ochtman
|
r8360 | formatnode = ui.debugflag and (lambda x: x) or (lambda x: x[:12]) | ||
Patrick Mezard
|
r10061 | defaulttempl = { | ||
'parent': '{rev}:{node|formatnode} ', | ||||
'manifest': '{rev}:{node|formatnode}', | ||||
'file_copy': '{name} ({source})', | ||||
'extra': '{key}={value|stringescape}' | ||||
} | ||||
# filecopy is preserved for compatibility reasons | ||||
defaulttempl['filecopy'] = defaulttempl['file_copy'] | ||||
Dirkjan Ochtman
|
r8360 | self.t = templater.templater(mapfile, {'formatnode': formatnode}, | ||
Patrick Mezard
|
r10061 | cache=defaulttempl) | ||
Patrick Mezard
|
r10057 | self.cache = {} | ||
Matt Mackall
|
r3643 | |||
def use_template(self, t): | ||||
'''set template string to use''' | ||||
self.t.cache['changeset'] = t | ||||
Alexander Solovyov
|
r7878 | def _meaningful_parentrevs(self, ctx): | ||
"""Return list of meaningful (or all if debug) parentrevs for rev. | ||||
""" | ||||
parents = ctx.parents() | ||||
if len(parents) > 1: | ||||
return parents | ||||
if self.ui.debugflag: | ||||
return [parents[0], self.repo['null']] | ||||
if parents[0].rev() >= ctx.rev() - 1: | ||||
return [] | ||||
return parents | ||||
Mads Kiilerich
|
r11488 | def _show(self, ctx, copies, matchfn, props): | ||
Matt Mackall
|
r3643 | '''show a single changeset or file revision''' | ||
Patrick Mezard
|
r10053 | showlist = templatekw.showlist | ||
Matt Mackall
|
r3643 | |||
Patrick Mezard
|
r10058 | # showparents() behaviour depends on ui trace level which | ||
# causes unexpected behaviours at templating level and makes | ||||
# it harder to extract it in a standalone function. Its | ||||
# behaviour cannot be changed so leave it here for now. | ||||
Matt Mackall
|
r3643 | def showparents(**args): | ||
Patrick Mezard
|
r10260 | ctx = args['ctx'] | ||
Alexander Solovyov
|
r7878 | parents = [[('rev', p.rev()), ('node', p.hex())] | ||
for p in self._meaningful_parentrevs(ctx)] | ||||
Matt Mackall
|
r3643 | return showlist('parent', parents, **args) | ||
props = props.copy() | ||||
Patrick Mezard
|
r10054 | props.update(templatekw.keywords) | ||
Patrick Mezard
|
r10058 | props['parents'] = showparents | ||
Patrick Mezard
|
r10053 | props['templ'] = self.t | ||
Patrick Mezard
|
r10054 | props['ctx'] = ctx | ||
Patrick Mezard
|
r10055 | props['repo'] = self.repo | ||
Patrick Mezard
|
r10058 | props['revcache'] = {'copies': copies} | ||
Patrick Mezard
|
r10057 | props['cache'] = self.cache | ||
Matt Mackall
|
r3643 | |||
Dirkjan Ochtman
|
r8013 | # find correct templates for current mode | ||
tmplmodes = [ | ||||
(True, None), | ||||
(self.ui.verbose, 'verbose'), | ||||
(self.ui.quiet, 'quiet'), | ||||
(self.ui.debugflag, 'debug'), | ||||
] | ||||
Robert Bachmann
|
r10152 | types = {'header': '', 'footer':'', 'changeset': 'changeset'} | ||
Dirkjan Ochtman
|
r8013 | for mode, postfix in tmplmodes: | ||
for type in types: | ||||
cur = postfix and ('%s_%s' % (type, postfix)) or type | ||||
if mode and cur in self.t: | ||||
types[type] = cur | ||||
Matt Mackall
|
r3643 | try: | ||
Dirkjan Ochtman
|
r8013 | |||
# write header | ||||
if types['header']: | ||||
h = templater.stringify(self.t(types['header'], **props)) | ||||
Matt Mackall
|
r3645 | if self.buffered: | ||
Alexander Solovyov
|
r7878 | self.header[ctx.rev()] = h | ||
Matt Mackall
|
r3645 | else: | ||
Simon Howkins
|
r11465 | if self.lastheader != h: | ||
self.lastheader = h | ||||
Simon Howkins
|
r11441 | self.ui.write(h) | ||
Dirkjan Ochtman
|
r8013 | |||
# write changeset metadata, then patch if requested | ||||
key = types['changeset'] | ||||
Matt Mackall
|
r3645 | self.ui.write(templater.stringify(self.t(key, **props))) | ||
Mads Kiilerich
|
r11488 | self.showpatch(ctx.node(), matchfn) | ||
Dirkjan Ochtman
|
r8013 | |||
Robert Bachmann
|
r10160 | if types['footer']: | ||
Robert Bachmann
|
r10152 | if not self.footer: | ||
self.footer = templater.stringify(self.t(types['footer'], | ||||
**props)) | ||||
Matt Mackall
|
r3643 | except KeyError, inst: | ||
Dirkjan Ochtman
|
r8013 | msg = _("%s: no key named '%s'") | ||
raise util.Abort(msg % (self.t.mapfile, inst.args[0])) | ||||
Matt Mackall
|
r3643 | except SyntaxError, inst: | ||
Martin Geisler
|
r10829 | raise util.Abort('%s: %s' % (self.t.mapfile, inst.args[0])) | ||
Matt Mackall
|
r3643 | |||
Mads Kiilerich
|
r11488 | def show_changeset(ui, repo, opts, buffered=False): | ||
Matt Mackall
|
r3643 | """show one changeset using template or regular display. | ||
Display format will be the first non-empty hit of: | ||||
1. option 'template' | ||||
2. option 'style' | ||||
3. [ui] setting 'logtemplate' | ||||
4. [ui] setting 'style' | ||||
If all of these values are either the unset or the empty string, | ||||
regular display via changeset_printer() is done. | ||||
""" | ||||
# options | ||||
Matt Mackall
|
r3837 | patch = False | ||
Yuya Nishihara
|
r11061 | if opts.get('patch') or opts.get('stat'): | ||
Matt Mackall
|
r14322 | patch = scmutil.matchall(repo) | ||
Matt Mackall
|
r3837 | |||
Matt Mackall
|
r3643 | tmpl = opts.get('template') | ||
Dirkjan Ochtman
|
r7967 | style = None | ||
Matt Mackall
|
r3643 | if tmpl: | ||
tmpl = templater.parsestring(tmpl, quoted=False) | ||||
else: | ||||
Dirkjan Ochtman
|
r7967 | style = opts.get('style') | ||
# ui settings | ||||
if not (tmpl or style): | ||||
tmpl = ui.config('ui', 'logtemplate') | ||||
if tmpl: | ||||
tmpl = templater.parsestring(tmpl) | ||||
else: | ||||
Patrick Mezard
|
r10249 | style = util.expandpath(ui.config('ui', 'style', '')) | ||
Dirkjan Ochtman
|
r7967 | |||
if not (tmpl or style): | ||||
return changeset_printer(ui, repo, patch, opts, buffered) | ||||
Matt Mackall
|
r3643 | |||
Dirkjan Ochtman
|
r7967 | mapfile = None | ||
if style and not tmpl: | ||||
mapfile = style | ||||
if not os.path.split(mapfile)[0]: | ||||
mapname = (templater.templatepath('map-cmdline.' + mapfile) | ||||
or templater.templatepath(mapfile)) | ||||
Matt Mackall
|
r10282 | if mapname: | ||
mapfile = mapname | ||||
Dirkjan Ochtman
|
r7967 | |||
try: | ||||
t = changeset_templater(ui, repo, patch, opts, mapfile, buffered) | ||||
except SyntaxError, inst: | ||||
raise util.Abort(inst.args[0]) | ||||
Matt Mackall
|
r10282 | if tmpl: | ||
t.use_template(tmpl) | ||||
Dirkjan Ochtman
|
r7967 | return t | ||
Matt Mackall
|
r3643 | |||
Matt Mackall
|
r3814 | def finddate(ui, repo, date): | ||
"""Find the tipmost changeset that matches the given date spec""" | ||||
Dirkjan Ochtman
|
r9667 | |||
mark.williamson@cl.cam.ac.uk
|
r5836 | df = util.matchdate(date) | ||
Matt Mackall
|
r14322 | m = scmutil.matchall(repo) | ||
Matt Mackall
|
r3814 | results = {} | ||
Matt Mackall
|
r9662 | |||
def prep(ctx, fns): | ||||
d = ctx.date() | ||||
if df(d[0]): | ||||
Dirkjan Ochtman
|
r9668 | results[ctx.rev()] = d | ||
Matt Mackall
|
r9662 | |||
Dirkjan Ochtman
|
r9667 | for ctx in walkchangerevs(repo, m, {'rev': None}, prep): | ||
Matt Mackall
|
r9662 | rev = ctx.rev() | ||
if rev in results: | ||||
ui.status(_("Found revision %s from %s\n") % | ||||
(rev, util.datestr(results[rev]))) | ||||
return str(rev) | ||||
Matt Mackall
|
r3814 | |||
raise util.Abort(_("revision matching date not found")) | ||||
Matt Mackall
|
r9665 | def walkchangerevs(repo, match, opts, prepare): | ||
timeless
|
r7807 | '''Iterate over files and the revs in which they changed. | ||
Matt Mackall
|
r3650 | |||
Callers most commonly need to iterate backwards over the history | ||||
timeless
|
r7807 | in which they are interested. Doing so has awful (quadratic-looking) | ||
Matt Mackall
|
r3650 | performance, so we use iterators in a "windowed" way. | ||
We walk a window of revisions in the desired order. Within the | ||||
window, we first walk forwards to gather data, then in the desired | ||||
order (usually backwards) to display it. | ||||
Matt Mackall
|
r9662 | This function returns an iterator yielding contexts. Before | ||
yielding each context, the iterator will first call the prepare | ||||
function on each context in the window in forward order.''' | ||||
Matt Mackall
|
r3650 | |||
def increasing_windows(start, end, windowsize=8, sizelimit=512): | ||||
if start < end: | ||||
while start < end: | ||||
Matt Mackall
|
r10282 | yield start, min(windowsize, end - start) | ||
Matt Mackall
|
r3650 | start += windowsize | ||
if windowsize < sizelimit: | ||||
windowsize *= 2 | ||||
else: | ||||
while start > end: | ||||
Matt Mackall
|
r10282 | yield start, min(windowsize, start - end - 1) | ||
Matt Mackall
|
r3650 | start -= windowsize | ||
if windowsize < sizelimit: | ||||
windowsize *= 2 | ||||
follow = opts.get('follow') or opts.get('follow_first') | ||||
Matt Mackall
|
r6750 | if not len(repo): | ||
Matt Mackall
|
r9652 | return [] | ||
Matt Mackall
|
r3650 | |||
if follow: | ||||
Matt Mackall
|
r6747 | defrange = '%s:0' % repo['.'].rev() | ||
Matt Mackall
|
r3650 | else: | ||
Alexis S. L. Carvalho
|
r6145 | defrange = '-1:0' | ||
Matt Mackall
|
r14319 | revs = scmutil.revrange(repo, opts['rev'] or [defrange]) | ||
Matt Mackall
|
r11281 | if not revs: | ||
return [] | ||||
Martin Geisler
|
r8152 | wanted = set() | ||
Matt Mackall
|
r9652 | slowpath = match.anypats() or (match.files() and opts.get('removed')) | ||
Matt Mackall
|
r3650 | fncache = {} | ||
Matt Mackall
|
r16108 | change = repo.changectx | ||
Matt Mackall
|
r3650 | |||
Nicolas Dumazet
|
r11632 | # First step is to fill wanted, the set of revisions that we want to yield. | ||
# When it does not induce extra cost, we also fill fncache for revisions in | ||||
# wanted: a cache of filenames that were changed (ctx.files()) and that | ||||
# match the file filtering conditions. | ||||
Matt Mackall
|
r9652 | if not slowpath and not match.files(): | ||
Matt Mackall
|
r3650 | # No files, no patterns. Display all revs. | ||
Martin Geisler
|
r8152 | wanted = set(revs) | ||
Matt Mackall
|
r3650 | copies = [] | ||
Matt Mackall
|
r9665 | |||
Matt Mackall
|
r3650 | if not slowpath: | ||
Nicolas Dumazet
|
r11632 | # We only have to read through the filelog to find wanted revisions | ||
Nicolas Dumazet
|
r11607 | minrev, maxrev = min(revs), max(revs) | ||
Nicolas Dumazet
|
r11606 | def filerevgen(filelog, last): | ||
Nicolas Dumazet
|
r11899 | """ | ||
Only files, no patterns. Check the history of each file. | ||||
Examines filelog entries within minrev, maxrev linkrev range | ||||
Returns an iterator yielding (linkrev, parentlinkrevs, copied) | ||||
tuples in backwards order | ||||
""" | ||||
Matt Mackall
|
r6750 | cl_count = len(repo) | ||
Nicolas Dumazet
|
r11608 | revs = [] | ||
Martin Geisler
|
r11634 | for j in xrange(0, last + 1): | ||
Nicolas Dumazet
|
r11608 | linkrev = filelog.linkrev(j) | ||
if linkrev < minrev: | ||||
continue | ||||
# only yield rev for which we have the changelog, it can | ||||
# happen while doing "hg log" during a pull or commit | ||||
Nicolas Dumazet
|
r12971 | if linkrev >= cl_count: | ||
Nicolas Dumazet
|
r11608 | break | ||
Nicolas Dumazet
|
r11899 | |||
parentlinkrevs = [] | ||||
for p in filelog.parentrevs(j): | ||||
if p != nullrev: | ||||
parentlinkrevs.append(filelog.linkrev(p)) | ||||
Nicolas Dumazet
|
r11608 | n = filelog.node(j) | ||
Nicolas Dumazet
|
r11899 | revs.append((linkrev, parentlinkrevs, | ||
Nicolas Dumazet
|
r11608 | follow and filelog.renamed(n))) | ||
Nicolas Dumazet
|
r11901 | return reversed(revs) | ||
Matt Mackall
|
r3650 | def iterfiles(): | ||
Matt Mackall
|
r9652 | for filename in match.files(): | ||
Matt Mackall
|
r3650 | yield filename, None | ||
for filename_node in copies: | ||||
yield filename_node | ||||
for file_, node in iterfiles(): | ||||
filelog = repo.file(file_) | ||||
Matt Mackall
|
r6750 | if not len(filelog): | ||
Patrick Mezard
|
r6536 | if node is None: | ||
# A zero count may be a directory or deleted file, so | ||||
# try to find matching entries on the slow path. | ||||
Brendan Cully
|
r7404 | if follow: | ||
Matt Mackall
|
r10282 | raise util.Abort( | ||
_('cannot follow nonexistent file: "%s"') % file_) | ||||
Patrick Mezard
|
r6536 | slowpath = True | ||
break | ||||
else: | ||||
continue | ||||
Nicolas Dumazet
|
r11606 | |||
if node is None: | ||||
last = len(filelog) - 1 | ||||
else: | ||||
last = filelog.rev(node) | ||||
Nicolas Dumazet
|
r11899 | |||
# keep track of all ancestors of the file | ||||
ancestors = set([filelog.linkrev(last)]) | ||||
# iterate from latest to oldest revision | ||||
for rev, flparentlinkrevs, copied in filerevgen(filelog, last): | ||||
Nicolas Dumazet
|
r12972 | if not follow: | ||
if rev > maxrev: | ||||
continue | ||||
else: | ||||
# Note that last might not be the first interesting | ||||
# rev to us: | ||||
# if the file has been changed after maxrev, we'll | ||||
# have linkrev(last) > maxrev, and we still need | ||||
# to explore the file graph | ||||
if rev not in ancestors: | ||||
continue | ||||
# XXX insert 1327 fix here | ||||
if flparentlinkrevs: | ||||
ancestors.update(flparentlinkrevs) | ||||
Nicolas Dumazet
|
r11899 | |||
Nicolas Dumazet
|
r11901 | fncache.setdefault(rev, []).append(file_) | ||
Nicolas Dumazet
|
r11607 | wanted.add(rev) | ||
if copied: | ||||
copies.append(copied) | ||||
Matt Mackall
|
r3650 | if slowpath: | ||
Nicolas Dumazet
|
r11632 | # We have to read the changelog to match filenames against | ||
# changed files | ||||
Matt Mackall
|
r3650 | if follow: | ||
raise util.Abort(_('can only follow copies/renames for explicit ' | ||||
timeless
|
r8761 | 'filenames')) | ||
Matt Mackall
|
r3650 | |||
# The slow path checks files modified in every changeset. | ||||
Nicolas Dumazet
|
r11631 | for i in sorted(revs): | ||
Nicolas Dumazet
|
r11609 | ctx = change(i) | ||
Matt Mackall
|
r9652 | matches = filter(match, ctx.files()) | ||
Matt Mackall
|
r3650 | if matches: | ||
Nicolas Dumazet
|
r11609 | fncache[i] = matches | ||
wanted.add(i) | ||||
Matt Mackall
|
r3650 | |||
Benoit Boissinot
|
r8778 | class followfilter(object): | ||
Matt Mackall
|
r3650 | def __init__(self, onlyfirst=False): | ||
self.startrev = nullrev | ||||
Benoit Boissinot
|
r10024 | self.roots = set() | ||
Matt Mackall
|
r3650 | self.onlyfirst = onlyfirst | ||
def match(self, rev): | ||||
def realparents(rev): | ||||
if self.onlyfirst: | ||||
return repo.changelog.parentrevs(rev)[0:1] | ||||
else: | ||||
return filter(lambda x: x != nullrev, | ||||
repo.changelog.parentrevs(rev)) | ||||
if self.startrev == nullrev: | ||||
self.startrev = rev | ||||
return True | ||||
if rev > self.startrev: | ||||
# forward: all descendants | ||||
if not self.roots: | ||||
Benoit Boissinot
|
r10024 | self.roots.add(self.startrev) | ||
Matt Mackall
|
r3650 | for parent in realparents(rev): | ||
if parent in self.roots: | ||||
Benoit Boissinot
|
r10024 | self.roots.add(rev) | ||
Matt Mackall
|
r3650 | return True | ||
else: | ||||
# backwards: all parents | ||||
if not self.roots: | ||||
Benoit Boissinot
|
r10024 | self.roots.update(realparents(self.startrev)) | ||
Matt Mackall
|
r3650 | if rev in self.roots: | ||
self.roots.remove(rev) | ||||
Benoit Boissinot
|
r10024 | self.roots.update(realparents(rev)) | ||
Matt Mackall
|
r3650 | return True | ||
return False | ||||
# it might be worthwhile to do this in the iterator if the rev range | ||||
# is descending and the prune args are all within that range | ||||
for rev in opts.get('prune', ()): | ||||
rev = repo.changelog.rev(repo.lookup(rev)) | ||||
ff = followfilter() | ||||
stop = min(revs[0], revs[-1]) | ||||
Matt Mackall
|
r10282 | for x in xrange(rev, stop - 1, -1): | ||
Martin Geisler
|
r8152 | if ff.match(x): | ||
wanted.discard(x) | ||||
Matt Mackall
|
r3650 | |||
Nicolas Dumazet
|
r11632 | # Now that wanted is correctly initialized, we can iterate over the | ||
# revision range, yielding only revisions in wanted. | ||||
Matt Mackall
|
r3650 | def iterate(): | ||
Matt Mackall
|
r9652 | if follow and not match.files(): | ||
Matt Mackall
|
r3650 | ff = followfilter(onlyfirst=opts.get('follow_first')) | ||
def want(rev): | ||||
Martin Geisler
|
r8119 | return ff.match(rev) and rev in wanted | ||
Matt Mackall
|
r3650 | else: | ||
def want(rev): | ||||
return rev in wanted | ||||
for i, window in increasing_windows(0, len(revs)): | ||||
Matt Mackall
|
r10282 | nrevs = [rev for rev in revs[i:i + window] if want(rev)] | ||
Matt Mackall
|
r8209 | for rev in sorted(nrevs): | ||
Matt Mackall
|
r3650 | fns = fncache.get(rev) | ||
Matt Mackall
|
r9654 | ctx = change(rev) | ||
Matt Mackall
|
r3650 | if not fns: | ||
def fns_generator(): | ||||
Matt Mackall
|
r9654 | for f in ctx.files(): | ||
Matt Mackall
|
r9652 | if match(f): | ||
Matt Mackall
|
r3650 | yield f | ||
fns = fns_generator() | ||||
Matt Mackall
|
r9662 | prepare(ctx, fns) | ||
Matt Mackall
|
r3650 | for rev in nrevs: | ||
Matt Mackall
|
r9662 | yield change(rev) | ||
Matt Mackall
|
r9652 | return iterate() | ||
Bryan O'Sullivan
|
r5034 | |||
David M. Carr
|
r15911 | def add(ui, repo, match, dryrun, listsubrepos, prefix, explicitonly): | ||
Martin Geisler
|
r12270 | join = lambda f: os.path.join(prefix, f) | ||
Martin Geisler
|
r12269 | bad = [] | ||
oldbad = match.bad | ||||
match.bad = lambda x, y: bad.append(x) or oldbad(x, y) | ||||
names = [] | ||||
Martin Geisler
|
r12270 | wctx = repo[None] | ||
Adrian Buehlmann
|
r14138 | cca = None | ||
abort, warn = scmutil.checkportabilityalert(ui) | ||||
if abort or warn: | ||||
cca = scmutil.casecollisionauditor(ui, abort, wctx) | ||||
Martin Geisler
|
r12269 | for f in repo.walk(match): | ||
exact = match.exact(f) | ||||
David M. Carr
|
r15911 | if exact or not explicitonly and f not in repo.dirstate: | ||
Adrian Buehlmann
|
r14138 | if cca: | ||
cca(f) | ||||
Martin Geisler
|
r12269 | names.append(f) | ||
if ui.verbose or not exact: | ||||
Martin Geisler
|
r12270 | ui.status(_('adding %s\n') % match.rel(join(f))) | ||
David M. Carr
|
r15410 | for subpath in wctx.substate: | ||
sub = wctx.sub(subpath) | ||||
try: | ||||
submatch = matchmod.narrowmatcher(subpath, match) | ||||
if listsubrepos: | ||||
David M. Carr
|
r15911 | bad.extend(sub.add(ui, submatch, dryrun, listsubrepos, prefix, | ||
False)) | ||||
David M. Carr
|
r15410 | else: | ||
David M. Carr
|
r15911 | bad.extend(sub.add(ui, submatch, dryrun, listsubrepos, prefix, | ||
True)) | ||||
David M. Carr
|
r15410 | except error.LookupError: | ||
ui.status(_("skipping missing subrepository: %s\n") | ||||
% join(subpath)) | ||||
Martin Geisler
|
r12270 | |||
Martin Geisler
|
r12269 | if not dryrun: | ||
Martin Geisler
|
r12270 | rejected = wctx.add(names, prefix) | ||
Martin Geisler
|
r12269 | bad.extend(f for f in rejected if f in match.files()) | ||
return bad | ||||
David M. Carr
|
r15912 | def forget(ui, repo, match, prefix, explicitonly): | ||
join = lambda f: os.path.join(prefix, f) | ||||
bad = [] | ||||
oldbad = match.bad | ||||
match.bad = lambda x, y: bad.append(x) or oldbad(x, y) | ||||
wctx = repo[None] | ||||
forgot = [] | ||||
s = repo.status(match=match, clean=True) | ||||
forget = sorted(s[0] + s[1] + s[3] + s[6]) | ||||
if explicitonly: | ||||
forget = [f for f in forget if match.exact(f)] | ||||
for subpath in wctx.substate: | ||||
sub = wctx.sub(subpath) | ||||
try: | ||||
submatch = matchmod.narrowmatcher(subpath, match) | ||||
subbad, subforgot = sub.forget(ui, submatch, prefix) | ||||
bad.extend([subpath + '/' + f for f in subbad]) | ||||
forgot.extend([subpath + '/' + f for f in subforgot]) | ||||
except error.LookupError: | ||||
ui.status(_("skipping missing subrepository: %s\n") | ||||
% join(subpath)) | ||||
FUJIWARA Katsunori
|
r16070 | if not explicitonly: | ||
for f in match.files(): | ||||
David M. Carr
|
r15912 | if f not in repo.dirstate and not os.path.isdir(match.rel(join(f))): | ||
if f not in forgot: | ||||
if os.path.exists(match.rel(join(f))): | ||||
ui.warn(_('not removing %s: ' | ||||
'file is already untracked\n') | ||||
% match.rel(join(f))) | ||||
bad.append(f) | ||||
for f in forget: | ||||
if ui.verbose or not match.exact(f): | ||||
ui.status(_('removing %s\n') % match.rel(join(f))) | ||||
rejected = wctx.forget(forget, prefix) | ||||
bad.extend(f for f in rejected if f in match.files()) | ||||
forgot.extend(forget) | ||||
return bad, forgot | ||||
Matt Mackall
|
r15777 | def duplicatecopies(repo, rev, p1): | ||
Matt Mackall
|
r15214 | "Reproduce copies found in the source revision in the dirstate for grafts" | ||
Matt Mackall
|
r15777 | for dst, src in copies.pathcopies(repo[p1], repo[rev]).iteritems(): | ||
repo.dirstate.copy(src, dst) | ||||
Matt Mackall
|
r15214 | |||
Bryan O'Sullivan
|
r5034 | def commit(ui, repo, commitfunc, pats, opts): | ||
'''commit the specified files or all outstanding changes''' | ||||
Thomas Arendsen Hein
|
r6139 | date = opts.get('date') | ||
if date: | ||||
opts['date'] = util.parsedate(date) | ||||
Idan Kamara
|
r14635 | message = logmessage(ui, opts) | ||
Bryan O'Sullivan
|
r5034 | |||
Kirill Smelkov
|
r5829 | # extract addremove carefully -- this function can be called from a command | ||
# that doesn't support addremove | ||||
if opts.get('addremove'): | ||||
Matt Mackall
|
r14321 | scmutil.addremove(repo, pats, opts) | ||
Kirill Smelkov
|
r5829 | |||
Matt Mackall
|
r14671 | return commitfunc(ui, repo, message, | ||
scmutil.match(repo[None], pats, opts), opts) | ||||
Matt Mackall
|
r8407 | |||
Matt Mackall
|
r8994 | def commiteditor(repo, ctx, subs): | ||
Matt Mackall
|
r8407 | if ctx.description(): | ||
return ctx.description() | ||||
Matt Mackall
|
r8994 | return commitforceeditor(repo, ctx, subs) | ||
Matt Mackall
|
r8407 | |||
Matt Mackall
|
r8994 | def commitforceeditor(repo, ctx, subs): | ||
Matt Mackall
|
r8407 | edittext = [] | ||
Matt Mackall
|
r8707 | modified, added, removed = ctx.modified(), ctx.added(), ctx.removed() | ||
Matt Mackall
|
r8407 | if ctx.description(): | ||
edittext.append(ctx.description()) | ||||
edittext.append("") | ||||
edittext.append("") # Empty line between message and comments. | ||||
edittext.append(_("HG: Enter commit message." | ||||
" Lines beginning with 'HG:' are removed.")) | ||||
Martin Geisler
|
r8535 | edittext.append(_("HG: Leave message empty to abort commit.")) | ||
Matt Mackall
|
r8407 | edittext.append("HG: --") | ||
edittext.append(_("HG: user: %s") % ctx.user()) | ||||
if ctx.p2(): | ||||
edittext.append(_("HG: branch merge")) | ||||
if ctx.branch(): | ||||
Matt Mackall
|
r13047 | edittext.append(_("HG: branch '%s'") % ctx.branch()) | ||
Matt Mackall
|
r8994 | edittext.extend([_("HG: subrepo %s") % s for s in subs]) | ||
Matt Mackall
|
r8407 | edittext.extend([_("HG: added %s") % f for f in added]) | ||
Matt Mackall
|
r8707 | edittext.extend([_("HG: changed %s") % f for f in modified]) | ||
Matt Mackall
|
r8407 | edittext.extend([_("HG: removed %s") % f for f in removed]) | ||
Matt Mackall
|
r8707 | if not added and not modified and not removed: | ||
Matt Mackall
|
r8407 | edittext.append(_("HG: no files changed")) | ||
edittext.append("") | ||||
# run editor in the repository root | ||||
olddir = os.getcwd() | ||||
os.chdir(repo.root) | ||||
text = repo.ui.edit("\n".join(edittext), ctx.user()) | ||||
Matt Mackall
|
r12900 | text = re.sub("(?m)^HG:.*(\n|$)", "", text) | ||
Matt Mackall
|
r8407 | os.chdir(olddir) | ||
if not text.strip(): | ||||
raise util.Abort(_("empty commit message")) | ||||
return text | ||||
Adrian Buehlmann
|
r14297 | |||
def command(table): | ||||
'''returns a function object bound to table which can be used as | ||||
a decorator for populating table as a command table''' | ||||
def cmd(name, options, synopsis=None): | ||||
def decorator(func): | ||||
if synopsis: | ||||
Matt Mackall
|
r14442 | table[name] = func, options[:], synopsis | ||
Adrian Buehlmann
|
r14297 | else: | ||
Matt Mackall
|
r14442 | table[name] = func, options[:] | ||
Adrian Buehlmann
|
r14297 | return func | ||
return decorator | ||||
return cmd | ||||