hg
566 lines
| 17.2 KiB
| text/plain
|
TextLexer
|
r0 | #!/usr/bin/env python | ||
# | ||||
# mercurial - a minimal scalable distributed SCM | ||||
|
r193 | # v0.5b "katje" | ||
|
r0 | # | ||
# Copyright 2005 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. | ||||
|
r46 | # the psyco compiler makes commits a bit faster | ||
# and makes changegroup merge about 20 times slower! | ||||
# try: | ||||
# import psyco | ||||
# psyco.full() | ||||
# except: | ||||
# pass | ||||
|
r0 | |||
|
r23 | import sys, os, time | ||
|
r209 | from mercurial import hg, mdiff, fancyopts, ui, commands | ||
|
r0 | |||
|
r18 | def help(): | ||
|
r175 | ui.status("""\ | ||
|
r18 | commands: | ||
|
r150 | add [files...] add the given files in the next commit | ||
addremove add all new files, delete all missing files | ||||
annotate [files...] show changeset number per file line | ||||
|
r18 | branch <path> create a branch of <path> in this directory | ||
checkout [changeset] checkout the latest or given changeset | ||||
commit commit all changes to the repository | ||||
|
r150 | diff [files...] diff working directory (or selected files) | ||
|
r18 | dump <file> [rev] dump the latest or given revision of a file | ||
dumpmanifest [rev] dump the latest or given revision of the manifest | ||||
|
r196 | export <rev> dump the changeset header and diffs for a revision | ||
|
r150 | history show changeset history | ||
init create a new repository in this directory | ||||
log <file> show revision history of a single file | ||||
merge <path> merge changes from <path> into local repository | ||||
|
r163 | recover rollback an interrupted transaction | ||
|
r150 | remove [files...] remove the given files in the next commit | ||
|
r158 | serve export the repository via HTTP | ||
|
r150 | status show new, missing, and changed files in working dir | ||
|
r67 | tags show current changeset tags | ||
|
r163 | undo undo the last transaction | ||
|
r175 | """) | ||
|
r18 | |||
|
r33 | def filterfiles(list, files): | ||
l = [ x for x in list if x in files ] | ||||
|
r31 | |||
|
r33 | for f in files: | ||
if f[-1] != os.sep: f += os.sep | ||||
l += [ x for x in list if x.startswith(f) ] | ||||
return l | ||||
|
r31 | |||
|
r34 | def diff(files = None, node1 = None, node2 = None): | ||
|
r64 | def date(c): | ||
return time.asctime(time.gmtime(float(c[2].split(' ')[0]))) | ||||
|
r34 | |||
if node2: | ||||
change = repo.changelog.read(node2) | ||||
mmap2 = repo.manifest.read(change[0]) | ||||
(c, a, d) = repo.diffrevs(node1, node2) | ||||
def read(f): return repo.file(f).read(mmap2[f]) | ||||
|
r64 | date2 = date(change) | ||
|
r34 | else: | ||
|
r64 | date2 = time.asctime() | ||
|
r34 | if not node1: | ||
node1 = repo.current | ||||
(c, a, d) = repo.diffdir(repo.root, node1) | ||||
|
r129 | a = [] # ignore unknown files in repo, by popular request | ||
|
r64 | def read(f): return file(os.path.join(repo.root, f)).read() | ||
|
r34 | |||
change = repo.changelog.read(node1) | ||||
mmap = repo.manifest.read(change[0]) | ||||
|
r64 | date1 = date(change) | ||
|
r34 | |||
if files: | ||||
|
r129 | c, a, d = map(lambda x: filterfiles(x, files), (c, a, d)) | ||
|
r34 | |||
for f in c: | ||||
|
r130 | to = "" | ||
if mmap.has_key(f): | ||||
to = repo.file(f).read(mmap[f]) | ||||
|
r34 | tn = read(f) | ||
|
r64 | sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f)) | ||
|
r129 | for f in a: | ||
to = "" | ||||
tn = read(f) | ||||
sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f)) | ||||
|
r34 | for f in d: | ||
to = repo.file(f).read(mmap[f]) | ||||
tn = "" | ||||
|
r64 | sys.stdout.write(mdiff.unidiff(to, date1, tn, date2, f)) | ||
|
r34 | |||
|
r209 | |||
try: | ||||
sys.exit(commands.dispatch(sys.argv[1:])) | ||||
except commands.UnknownCommand: | ||||
# fall through | ||||
pass | ||||
|
r0 | options = {} | ||
opts = [('v', 'verbose', None, 'verbose'), | ||||
|
r83 | ('d', 'debug', None, 'debug'), | ||
|
r107 | ('q', 'quiet', None, 'quiet'), | ||
('y', 'noninteractive', None, 'run non-interactively'), | ||||
] | ||||
|
r0 | |||
args = fancyopts.fancyopts(sys.argv[1:], opts, options, | ||||
'hg [options] <command> [command options] [files]') | ||||
try: | ||||
cmd = args[0] | ||||
args = args[1:] | ||||
except: | ||||
|
r206 | cmd = "help" | ||
|
r0 | |||
|
r207 | ui = ui.ui(options["verbose"], options["debug"], options["quiet"], | ||
|
r107 | not options["noninteractive"]) | ||
|
r0 | |||
if cmd == "init": | ||||
repo = hg.repository(ui, ".", create=1) | ||||
sys.exit(0) | ||||
elif cmd == "branch" or cmd == "clone": | ||||
os.system("cp -al %s/.hg .hg" % args[0]) | ||||
sys.exit(0) | ||||
|
r18 | elif cmd == "help": | ||
help() | ||||
sys.exit(0) | ||||
|
r0 | else: | ||
|
r18 | try: | ||
repo = hg.repository(ui=ui) | ||||
|
r74 | except IOError: | ||
|
r83 | ui.warn("Unable to open repository\n") | ||
|
r18 | sys.exit(0) | ||
|
r0 | |||
|
r103 | relpath = None | ||
if os.getcwd() != repo.root: | ||||
relpath = os.getcwd()[len(repo.root) + 1: ] | ||||
|
r0 | if cmd == "checkout" or cmd == "co": | ||
node = repo.changelog.tip() | ||||
|
r39 | if args: | ||
|
r67 | node = repo.lookup(args[0]) | ||
|
r0 | repo.checkout(node) | ||
elif cmd == "add": | ||||
repo.add(args) | ||||
elif cmd == "remove" or cmd == "rm" or cmd == "del" or cmd == "delete": | ||||
repo.remove(args) | ||||
elif cmd == "commit" or cmd == "checkin" or cmd == "ci": | ||||
if 1: | ||||
if len(args) > 0: | ||||
|
r31 | repo.commit(repo.current, args) | ||
|
r0 | else: | ||
|
r31 | repo.commit(repo.current) | ||
|
r203 | elif cmd == "rawcommit": | ||
"raw commit interface" | ||||
rc = {} | ||||
opts = [('p', 'parent', [], 'parent'), | ||||
('d', 'date', "", 'data'), | ||||
('u', 'user', "", 'user'), | ||||
('F', 'files', "", 'file list'), | ||||
('t', 'text', "", 'commit text'), | ||||
('l', 'logfile', "", 'commit text file') | ||||
] | ||||
args = fancyopts.fancyopts(args, opts, rc, | ||||
"hg rawcommit [options] files") | ||||
text = rc['text'] | ||||
if not text and rc['logfile']: | ||||
try: text = open(rc['logfile']).read() | ||||
except IOError: pass | ||||
if not text and not rc['logfile']: | ||||
print "missing commit text" | ||||
sys.exit(0) | ||||
if rc['files']: | ||||
files = open(rc['files']).read().splitlines() | ||||
else: | ||||
files = args | ||||
repo.rawcommit(files, text, rc['user'], rc['date'], *rc['parent']) | ||||
|
r0 | elif cmd == "import" or cmd == "patch": | ||
|
r69 | try: | ||
import psyco | ||||
psyco.full() | ||||
except: | ||||
pass | ||||
|
r0 | ioptions = {} | ||
opts = [('p', 'strip', 1, 'path strip'), | ||||
|
r50 | ('b', 'base', "", 'base path'), | ||
('q', 'quiet', "", 'silence diff') | ||||
] | ||||
|
r0 | |||
args = fancyopts.fancyopts(args, opts, ioptions, | ||||
'hg import [options] <patch names>') | ||||
d = ioptions["base"] | ||||
strip = ioptions["strip"] | ||||
|
r50 | quiet = ioptions["quiet"] and "> /dev/null" or "" | ||
|
r0 | |||
for patch in args: | ||||
ui.status("applying %s\n" % patch) | ||||
|
r49 | pf = os.path.join(d, patch) | ||
|
r62 | text = "" | ||
for l in file(pf): | ||||
|
r69 | if l[:4] == "--- ": break | ||
|
r62 | text += l | ||
|
r0 | f = os.popen("lsdiff --strip %d %s" % (strip, pf)) | ||
|
r54 | files = filter(None, map(lambda x: x.rstrip(), f.read().splitlines())) | ||
|
r0 | f.close() | ||
|
r69 | |||
if files: | ||||
if os.system("patch -p%d < %s %s" % (strip, pf, quiet)): | ||||
raise "patch failed!" | ||||
|
r49 | repo.commit(repo.current, files, text) | ||
|
r0 | |||
elif cmd == "status": | ||||
|
r33 | (c, a, d) = repo.diffdir(repo.root, repo.current) | ||
|
r103 | if relpath: | ||
(c, a, d) = map(lambda x: filterfiles(x, [ relpath ]), (c, a, d)) | ||||
|
r0 | for f in c: print "C", f | ||
for f in a: print "?", f | ||||
for f in d: print "R", f | ||||
elif cmd == "diff": | ||||
|
r34 | revs = [] | ||
|
r31 | |||
|
r33 | if args: | ||
|
r38 | doptions = {} | ||
|
r33 | opts = [('r', 'revision', [], 'revision')] | ||
args = fancyopts.fancyopts(args, opts, doptions, | ||||
'hg diff [options] [files]') | ||||
|
r67 | revs = map(lambda x: repo.lookup(x), doptions['revision']) | ||
|
r33 | |||
if len(revs) > 2: | ||||
|
r83 | self.ui.warn("too many revisions to diff\n") | ||
|
r33 | sys.exit(1) | ||
|
r64 | |||
|
r103 | if relpath: | ||
|
r64 | if not args: args = [ relpath ] | ||
else: args = [ os.path.join(relpath, x) for x in args ] | ||||
diff(args, *revs) | ||||
|
r33 | |||
|
r34 | elif cmd == "export": | ||
|
r67 | node = repo.lookup(args[0]) | ||
|
r68 | prev, other = repo.changelog.parents(node) | ||
change = repo.changelog.read(node) | ||||
print "# HG changeset patch" | ||||
print "# User %s" % change[1] | ||||
print "# Node ID %s" % hg.hex(node) | ||||
print "# Parent %s" % hg.hex(prev) | ||||
if other != hg.nullid: | ||||
print "# Parent %s" % hg.hex(other) | ||||
print change[4] | ||||
|
r34 | diff(None, prev, node) | ||
|
r0 | |||
|
r46 | elif cmd == "debugchangegroup": | ||
|
r67 | newer = repo.newer(map(repo.lookup, args)) | ||
|
r62 | for chunk in repo.changegroup(newer): | ||
sys.stdout.write(chunk) | ||||
|
r46 | |||
elif cmd == "debugaddchangegroup": | ||||
data = sys.stdin.read() | ||||
repo.addchangegroup(data) | ||||
|
r0 | elif cmd == "addremove": | ||
|
r29 | (c, a, d) = repo.diffdir(repo.root, repo.current) | ||
|
r0 | repo.add(a) | ||
repo.remove(d) | ||||
elif cmd == "history": | ||||
for i in range(repo.changelog.count()): | ||||
n = repo.changelog.node(i) | ||||
changes = repo.changelog.read(n) | ||||
(p1, p2) = repo.changelog.parents(n) | ||||
(h, h1, h2) = map(hg.hex, (n, p1, p2)) | ||||
(i1, i2) = map(repo.changelog.rev, (p1, p2)) | ||||
print "rev: %4d:%s" % (i, h) | ||||
print "parents: %4d:%s" % (i1, h1) | ||||
if i2: print " %4d:%s" % (i2, h2) | ||||
print "manifest: %4d:%s" % (repo.manifest.rev(changes[0]), | ||||
hg.hex(changes[0])) | ||||
print "user:", changes[1] | ||||
|
r21 | print "date:", time.asctime( | ||
time.localtime(float(changes[2].split(' ')[0]))) | ||||
|
r123 | if ui.verbose: print "files:", " ".join(changes[3]) | ||
|
r0 | print "description:" | ||
print changes[4] | ||||
|
r89 | elif cmd == "tip": | ||
n = repo.changelog.tip() | ||||
t = repo.changelog.rev(n) | ||||
ui.status("%d:%s\n" % (t, hg.hex(n))) | ||||
|
r0 | elif cmd == "log": | ||
|
r103 | |||
if len(args) == 1: | ||||
if relpath: | ||||
args[0] = os.path.join(relpath, args[0]) | ||||
|
r0 | r = repo.file(args[0]) | ||
for i in range(r.count()): | ||||
n = r.node(i) | ||||
(p1, p2) = r.parents(n) | ||||
(h, h1, h2) = map(hg.hex, (n, p1, p2)) | ||||
(i1, i2) = map(r.rev, (p1, p2)) | ||||
cr = r.linkrev(n) | ||||
cn = hg.hex(repo.changelog.node(cr)) | ||||
print "rev: %4d:%s" % (i, h) | ||||
print "changeset: %4d:%s" % (cr, cn) | ||||
print "parents: %4d:%s" % (i1, h1) | ||||
if i2: print " %4d:%s" % (i2, h2) | ||||
|
r103 | changes = repo.changelog.read(repo.changelog.node(cr)) | ||
|
r105 | print "user: %s" % changes[1] | ||
print "date: %s" % time.asctime( | ||||
|
r106 | time.localtime(float(changes[2].split(' ')[0]))) | ||
|
r103 | print "description:" | ||
print changes[4] | ||||
elif len(args) > 1: | ||||
print "too many args" | ||||
|
r0 | else: | ||
print "missing filename" | ||||
elif cmd == "dump": | ||||
if args: | ||||
r = repo.file(args[0]) | ||||
n = r.tip() | ||||
|
r39 | if len(args) > 1: n = r.lookup(args[1]) | ||
|
r0 | sys.stdout.write(r.read(n)) | ||
else: | ||||
print "missing filename" | ||||
elif cmd == "dumpmanifest": | ||||
n = repo.manifest.tip() | ||||
if len(args) > 0: | ||||
|
r39 | n = repo.manifest.lookup(args[0]) | ||
|
r0 | m = repo.manifest.read(n) | ||
files = m.keys() | ||||
files.sort() | ||||
for f in files: | ||||
print hg.hex(m[f]), f | ||||
|
r24 | elif cmd == "debugindex": | ||
|
r91 | if ".hg" not in args[0]: | ||
args[0] = ".hg/data/" + repo.file(args[0]).encodepath(args[0]) + "i" | ||||
|
r24 | r = hg.revlog(open, args[0], "") | ||
|
r70 | print " rev offset length base linkrev"+\ | ||
|
r24 | " p1 p2 nodeid" | ||
for i in range(r.count()): | ||||
e = r.index[i] | ||||
|
r70 | print "% 6d % 9d % 7d % 6d % 7d %s.. %s.. %s.." % ( | ||
|
r24 | i, e[0], e[1], e[2], e[3], | ||
hg.hex(e[4][:5]), hg.hex(e[5][:5]), hg.hex(e[6][:5])) | ||||
|
r92 | elif cmd == "debugindexdot": | ||
if ".hg" not in args[0]: | ||||
args[0] = ".hg/data/" + repo.file(args[0]).encodepath(args[0]) + "i" | ||||
r = hg.revlog(open, args[0], "") | ||||
print "digraph G {" | ||||
for i in range(r.count()): | ||||
e = r.index[i] | ||||
print "\t%d -> %d" % (r.rev(e[4]), i) | ||||
if e[5] != hg.nullid: | ||||
print "\t%d -> %d" % (r.rev(e[5]), i) | ||||
print "}" | ||||
|
r0 | elif cmd == "merge": | ||
|
r174 | (c, a, d) = repo.diffdir(repo.root, repo.current) | ||
if c: | ||||
ui.warn("aborting (outstanding changes in working directory)\n") | ||||
sys.exit(1) | ||||
|
r0 | if args: | ||
|
r175 | paths = {} | ||
try: | ||||
pf = os.path.join(os.environ["HOME"], ".hgpaths") | ||||
for l in file(pf): | ||||
name, path = l.split() | ||||
paths[name] = path | ||||
except: | ||||
pass | ||||
if args[0] in paths: args[0] = paths[args[0]] | ||||
|
r0 | other = hg.repository(ui, args[0]) | ||
|
r62 | cg = repo.getchangegroup(other) | ||
repo.addchangegroup(cg) | ||||
else: | ||||
print "missing source repository" | ||||
|
r67 | elif cmd == "tags": | ||
repo.lookup(0) # prime the cache | ||||
i = repo.tags.items() | ||||
i.sort() | ||||
for k, n in i: | ||||
try: | ||||
r = repo.changelog.rev(n) | ||||
except KeyError: | ||||
r = "?" | ||||
print "%-30s %5d:%s" % (k, repo.changelog.rev(n), hg.hex(n)) | ||||
|
r162 | elif cmd == "recover": | ||
repo.recover() | ||||
|
r0 | elif cmd == "verify": | ||
filelinkrevs = {} | ||||
filenodes = {} | ||||
manifestchangeset = {} | ||||
changesets = revisions = files = 0 | ||||
|
r87 | errors = 0 | ||
|
r0 | |||
|
r87 | ui.status("checking changesets\n") | ||
|
r0 | for i in range(repo.changelog.count()): | ||
changesets += 1 | ||||
n = repo.changelog.node(i) | ||||
|
r87 | for p in repo.changelog.parents(n): | ||
if p not in repo.changelog.nodemap: | ||||
ui.warn("changeset %s has unknown parent %s\n" % | ||||
(hg.short(n), hg.short(p))) | ||||
errors += 1 | ||||
try: | ||||
changes = repo.changelog.read(n) | ||||
|
r145 | except Exception, inst: | ||
|
r87 | ui.warn("unpacking changeset %s: %s\n" % (short(n), inst)) | ||
errors += 1 | ||||
|
r0 | manifestchangeset[changes[0]] = n | ||
for f in changes[3]: | ||||
revisions += 1 | ||||
filelinkrevs.setdefault(f, []).append(i) | ||||
|
r87 | ui.status("checking manifests\n") | ||
|
r0 | for i in range(repo.manifest.count()): | ||
n = repo.manifest.node(i) | ||||
|
r87 | for p in repo.manifest.parents(n): | ||
if p not in repo.manifest.nodemap: | ||||
ui.warn("manifest %s has unknown parent %s\n" % | ||||
(hg.short(n), hg.short(p))) | ||||
errors += 1 | ||||
|
r0 | ca = repo.changelog.node(repo.manifest.linkrev(n)) | ||
cc = manifestchangeset[n] | ||||
if ca != cc: | ||||
|
r87 | ui.warn("manifest %s points to %s, not %s\n" % | ||
(hg.hex(n), hg.hex(ca), hg.hex(cc))) | ||||
errors += 1 | ||||
try: | ||||
|
r121 | delta = mdiff.patchtext(repo.manifest.delta(n)) | ||
except KeyboardInterrupt: | ||||
print "aborted" | ||||
sys.exit(0) | ||||
|
r93 | except Exception, inst: | ||
|
r87 | ui.warn("unpacking manifest %s: %s\n" % (hg.short(n), inst)) | ||
errors += 1 | ||||
|
r142 | |||
|
r121 | ff = [ l.split('\0') for l in delta.splitlines() ] | ||
for f, fn in ff: | ||||
filenodes.setdefault(f, {})[hg.bin(fn)] = 1 | ||||
|
r0 | |||
|
r87 | ui.status("crosschecking files in changesets and manifests\n") | ||
|
r0 | for f in filenodes: | ||
if f not in filelinkrevs: | ||||
|
r87 | ui.warn("file %s in manifest but not in changesets\n" % f) | ||
errors += 1 | ||||
|
r0 | |||
for f in filelinkrevs: | ||||
if f not in filenodes: | ||||
|
r121 | ui.warn("file %s in changeset but not in manifest\n" % f) | ||
|
r87 | errors += 1 | ||
|
r0 | |||
|
r87 | ui.status("checking files\n") | ||
|
r121 | ff = filenodes.keys() | ||
ff.sort() | ||||
for f in ff: | ||||
if f == "/dev/null": continue | ||||
|
r0 | files += 1 | ||
fl = repo.file(f) | ||||
|
r87 | nodes = { hg.nullid: 1 } | ||
|
r0 | for i in range(fl.count()): | ||
n = fl.node(i) | ||||
|
r17 | |||
|
r0 | if n not in filenodes[f]: | ||
|
r121 | ui.warn("%s: %d:%s not in manifests\n" % (f, i, hg.short(n))) | ||
print len(filenodes[f].keys()), fl.count(), f | ||||
|
r87 | errors += 1 | ||
|
r17 | else: | ||
del filenodes[f][n] | ||||
flr = fl.linkrev(n) | ||||
if flr not in filelinkrevs[f]: | ||||
|
r87 | ui.warn("%s:%s points to unexpected changeset rev %d\n" | ||
% (f, hg.short(n), fl.linkrev(n))) | ||||
errors += 1 | ||||
|
r17 | else: | ||
filelinkrevs[f].remove(flr) | ||||
# verify contents | ||||
|
r87 | try: | ||
t = fl.read(n) | ||||
|
r145 | except Exception, inst: | ||
ui.warn("unpacking file %s %s: %s\n" % (f, hg.short(n), inst)) | ||||
|
r87 | errors += 1 | ||
|
r17 | # verify parents | ||
|
r0 | (p1, p2) = fl.parents(n) | ||
if p1 not in nodes: | ||||
|
r87 | ui.warn("file %s:%s unknown parent 1 %s" % | ||
(f, hg.short(n), hg.short(p1))) | ||||
errors += 1 | ||||
|
r0 | if p2 not in nodes: | ||
|
r87 | ui.warn("file %s:%s unknown parent 2 %s" % | ||
(f, hg.short(n), hg.short(p1))) | ||||
errors += 1 | ||||
|
r17 | nodes[n] = 1 | ||
|
r0 | |||
|
r17 | # cross-check | ||
for flr in filelinkrevs[f]: | ||||
|
r87 | ui.warn("changeset rev %d not in %s\n" % (flr, f)) | ||
errors += 1 | ||||
|
r17 | |||
for node in filenodes[f]: | ||||
|
r87 | ui.warn("node %s in manifests not in %s\n" % (hg.hex(n), f)) | ||
errors += 1 | ||||
|
r0 | |||
|
r87 | ui.status("%d files, %d changesets, %d total revisions\n" % | ||
(files, changesets, revisions)) | ||||
if errors: | ||||
|
r93 | ui.warn("%d integrity errors encountered!\n" % errors) | ||
|
r87 | sys.exit(1) | ||
|
r158 | |||
elif cmd == "serve": | ||||
from mercurial import hgweb | ||||
soptions = {} | ||||
opts = [('p', 'port', 8000, 'listen port'), | ||||
('a', 'address', '', 'interface address'), | ||||
('n', 'name', os.getcwd(), 'repository name'), | ||||
('t', 'templates', "", 'template map') | ||||
] | ||||
args = fancyopts.fancyopts(args, opts, soptions, | ||||
'hg serve [options]') | ||||
hgweb.server(repo.root, soptions["name"], soptions["templates"], | ||||
soptions["address"], soptions["port"]) | ||||
|
r0 | |||
else: | ||||
|
r175 | if cmd: ui.warn("unknown command\n\n") | ||
|
r18 | help() | ||
|
r0 | sys.exit(1) | ||