209
|
1 import os, re
|
|
2 from mercurial import fancyopts, ui, hg
|
|
3
|
|
4 class UnknownCommand(Exception): pass
|
|
5
|
|
6 def relpath(repo, args):
|
|
7 if os.getcwd() != repo.root:
|
|
8 p = os.getcwd()[len(repo.root) + 1: ]
|
|
9 return [ os.path.join(p, x) for x in args ]
|
|
10 return args
|
|
11
|
|
12 def help(ui, args):
|
|
13 ui.status("""\
|
|
14 hg commands:
|
|
15
|
|
16 add [files...] add the given files in the next commit
|
|
17 addremove add all new files, delete all missing files
|
|
18 annotate [files...] show changeset number per file line
|
|
19 branch <path> create a branch of <path> in this directory
|
|
20 checkout [changeset] checkout the latest or given changeset
|
|
21 commit commit all changes to the repository
|
|
22 diff [files...] diff working directory (or selected files)
|
|
23 dump <file> [rev] dump the latest or given revision of a file
|
|
24 dumpmanifest [rev] dump the latest or given revision of the manifest
|
|
25 export <rev> dump the changeset header and diffs for a revision
|
|
26 history show changeset history
|
|
27 init create a new repository in this directory
|
|
28 log <file> show revision history of a single file
|
|
29 merge <path> merge changes from <path> into local repository
|
|
30 recover rollback an interrupted transaction
|
|
31 remove [files...] remove the given files in the next commit
|
|
32 serve export the repository via HTTP
|
|
33 status show new, missing, and changed files in working dir
|
|
34 tags show current changeset tags
|
|
35 undo undo the last transaction
|
|
36 """)
|
|
37
|
|
38 def init(ui, args):
|
|
39 """create a repository"""
|
|
40 hg.repository(ui, ".", create=1)
|
|
41
|
|
42 def checkout(u, repo, args):
|
|
43 node = repo.changelog.tip()
|
|
44 if args:
|
|
45 node = repo.lookup(args[0])
|
|
46 repo.checkout(node)
|
|
47
|
|
48 def annotate(u, repo, args, **ops):
|
|
49 if not args:
|
|
50 return
|
|
51
|
|
52 def getnode(rev):
|
|
53 return hg.short(repo.changelog.node(rev))
|
|
54
|
|
55 def getname(rev):
|
|
56 try:
|
|
57 return bcache[rev]
|
|
58 except KeyError:
|
|
59 cl = repo.changelog.read(repo.changelog.node(rev))
|
|
60 name = cl[1]
|
|
61 f = name.find('@')
|
|
62 if f >= 0:
|
|
63 name = name[:f]
|
|
64 bcache[rev] = name
|
|
65 return name
|
|
66
|
|
67 bcache = {}
|
|
68 opmap = [['user', getname], ['number', str], ['changeset', getnode]]
|
|
69 if not ops['user'] and not ops['changeset']:
|
|
70 ops['number'] = 1
|
|
71
|
|
72 args = relpath(repo, args)
|
|
73 node = repo.current
|
|
74 if ops['revision']:
|
|
75 node = repo.changelog.lookup(ops['revision'])
|
|
76 change = repo.changelog.read(node)
|
|
77 mmap = repo.manifest.read(change[0])
|
|
78 maxuserlen = 0
|
|
79 maxchangelen = 0
|
|
80 for f in args:
|
|
81 lines = repo.file(f).annotate(mmap[f])
|
|
82 pieces = []
|
|
83
|
|
84 for o, f in opmap:
|
|
85 if ops[o]:
|
|
86 l = [ f(n) for n,t in lines ]
|
|
87 m = max(map(len, l))
|
|
88 pieces.append([ "%*s" % (m, x) for x in l])
|
|
89
|
|
90 for p,l in zip(zip(*pieces), lines):
|
|
91 u.write(" ".join(p) + ": " + l[1])
|
|
92
|
|
93 table = {
|
|
94 "init": (init, [], 'hg init'),
|
|
95 "help": (help, [], 'hg init'),
|
|
96 "checkout|co": (checkout, [], 'hg init'),
|
|
97 "ann|annotate": (annotate,
|
|
98 [('r', 'revision', '', 'revision'),
|
|
99 ('u', 'user', None, 'show user'),
|
|
100 ('n', 'number', None, 'show revision number'),
|
|
101 ('c', 'changeset', None, 'show changeset')],
|
|
102 'hg annotate [-u] [-c] [-n] [-r id] [files]'),
|
|
103 }
|
|
104
|
|
105 norepo = "init branch help"
|
|
106
|
|
107 def dispatch(args):
|
|
108 options = {}
|
|
109 opts = [('v', 'verbose', None, 'verbose'),
|
|
110 ('d', 'debug', None, 'debug'),
|
|
111 ('q', 'quiet', None, 'quiet'),
|
|
112 ('y', 'noninteractive', None, 'run non-interactively'),
|
|
113 ]
|
|
114
|
|
115 args = fancyopts.fancyopts(args, opts, options,
|
|
116 'hg [options] <command> [options] [files]')
|
|
117
|
|
118 if not args:
|
|
119 cmd = "help"
|
|
120 else:
|
|
121 cmd, args = args[0], args[1:]
|
|
122
|
|
123 u = ui.ui(options["verbose"], options["debug"], options["quiet"],
|
|
124 not options["noninteractive"])
|
|
125
|
|
126 i = None
|
|
127 for e in table.keys():
|
|
128 if re.match(e + "$", cmd):
|
|
129 i = table[e]
|
|
130
|
|
131 # deal with this internally later
|
|
132 if not i: raise UnknownCommand(cmd)
|
|
133
|
|
134 cmdoptions = {}
|
|
135 args = fancyopts.fancyopts(args, i[1], cmdoptions, i[2])
|
|
136
|
|
137 if cmd not in norepo.split():
|
|
138 repo = hg.repository(ui = u)
|
|
139 d = lambda: i[0](u, repo, args, **cmdoptions)
|
|
140 else:
|
|
141 d = lambda: i[0](u, args, **cmdoptions)
|
|
142
|
|
143 try:
|
|
144 d()
|
|
145 except KeyboardInterrupt:
|
|
146 u.warn("interrupted!\n")
|