Mercurial > hg
annotate hgext/fastexport.py @ 48116:5ced12cfa41b
errors: raise InputError on bad revset to revrange() iff provided by the user
Most callers of `scmutil.revrange()` pass in a revset provided by the
user. If there are problems resolving that, it should result in an
`InputError` and exit code 10 (when using detailed exit
codes). However, there are also some callers that pass in revsets not
provided by the user. `InputError` is not appropriate in those
cases. This patch therefore introduces a wrapper around
`scmutil.revrange()` that simply converts the exception type. I put it
in `logcmdutil.py` since that seems to be the lowest-level module in
the (poorly defined) UI layer.
Differential Revision: https://phab.mercurial-scm.org/D11560
author | Martin von Zweigbergk <martinvonz@google.com> |
---|---|
date | Tue, 28 Sep 2021 08:47:11 -0700 |
parents | edffab2cf0ea |
children | 6000f5b25c9b |
rev | line source |
---|---|
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
1 # Copyright 2020 Joerg Sonnenberger <joerg@bec.de> |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
2 # |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
3 # This software may be used and distributed according to the terms of the |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
4 # GNU General Public License version 2 or any later version. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
5 """export repositories as git fast-import stream""" |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
6 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
7 # The format specification for fast-import streams can be found at |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
8 # https://git-scm.com/docs/git-fast-import#_input_format |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
9 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
10 from __future__ import absolute_import |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
11 import re |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
12 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
13 from mercurial.i18n import _ |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
14 from mercurial.node import hex, nullrev |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
15 from mercurial.utils import stringutil |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
16 from mercurial import ( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
17 error, |
48116
5ced12cfa41b
errors: raise InputError on bad revset to revrange() iff provided by the user
Martin von Zweigbergk <martinvonz@google.com>
parents:
44821
diff
changeset
|
18 logcmdutil, |
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
19 pycompat, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
20 registrar, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
21 scmutil, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
22 ) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
23 from .convert import convcmd |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
24 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
25 # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
26 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
27 # be specifying the version(s) of Mercurial they are tested with, or |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
28 # leave the attribute unspecified. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
29 testedwith = b"ships-with-hg-core" |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
30 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
31 cmdtable = {} |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
32 command = registrar.command(cmdtable) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
33 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
34 GIT_PERSON_PROHIBITED = re.compile(b'[<>\n"]') |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
35 GIT_EMAIL_PROHIBITED = re.compile(b"[<> \n]") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
36 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
37 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
38 def convert_to_git_user(authormap, user, rev): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
39 mapped_user = authormap.get(user, user) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
40 user_person = stringutil.person(mapped_user) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
41 user_email = stringutil.email(mapped_user) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
42 if GIT_EMAIL_PROHIBITED.match(user_email) or GIT_PERSON_PROHIBITED.match( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
43 user_person |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
44 ): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
45 raise error.Abort( |
44690
44aff45b556d
fastexport: make a diagnostics message more localizable
Joerg Sonnenberger <joerg@bec.de>
parents:
44280
diff
changeset
|
46 _(b"Unable to parse user into person and email for revision %s") |
44aff45b556d
fastexport: make a diagnostics message more localizable
Joerg Sonnenberger <joerg@bec.de>
parents:
44280
diff
changeset
|
47 % rev |
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
48 ) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
49 if user_person: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
50 return b'"' + user_person + b'" <' + user_email + b'>' |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
51 else: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
52 return b"<" + user_email + b">" |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
53 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
54 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
55 def convert_to_git_date(date): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
56 timestamp, utcoff = date |
44821
edffab2cf0ea
fastexport: adjust output to be more canonical
Joerg Sonnenberger <joerg@bec.de>
parents:
44811
diff
changeset
|
57 tzsign = b"+" if utcoff <= 0 else b"-" |
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
58 if utcoff % 60 != 0: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
59 raise error.Abort( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
60 _(b"UTC offset in %b is not an integer number of seconds") % (date,) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
61 ) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
62 utcoff = abs(utcoff) // 60 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
63 tzh = utcoff // 60 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
64 tzmin = utcoff % 60 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
65 return b"%d " % int(timestamp) + tzsign + b"%02d%02d" % (tzh, tzmin) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
66 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
67 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
68 def convert_to_git_ref(branch): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
69 # XXX filter/map depending on git restrictions |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
70 return b"refs/heads/" + branch |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
71 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
72 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
73 def write_data(buf, data, skip_newline): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
74 buf.append(b"data %d\n" % len(data)) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
75 buf.append(data) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
76 if not skip_newline or data[-1:] != b"\n": |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
77 buf.append(b"\n") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
78 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
79 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
80 def export_commit(ui, repo, rev, marks, authormap): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
81 ctx = repo[rev] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
82 revid = ctx.hex() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
83 if revid in marks: |
44811
5d309906ed0e
fastexport: downgrade message about already exported changesets to debug
Joerg Sonnenberger <joerg@bec.de>
parents:
44690
diff
changeset
|
84 ui.debug(b"warning: revision %s already exported, skipped\n" % revid) |
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
85 return |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
86 parents = [p for p in ctx.parents() if p.rev() != nullrev] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
87 for p in parents: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
88 if p.hex() not in marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
89 ui.warn( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
90 _(b"warning: parent %s of %s has not been exported, skipped\n") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
91 % (p, revid) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
92 ) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
93 return |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
94 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
95 # For all files modified by the commit, check if they have already |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
96 # been exported and otherwise dump the blob with the new mark. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
97 for fname in ctx.files(): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
98 if fname not in ctx: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
99 continue |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
100 filectx = ctx.filectx(fname) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
101 filerev = hex(filectx.filenode()) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
102 if filerev not in marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
103 mark = len(marks) + 1 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
104 marks[filerev] = mark |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
105 data = filectx.data() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
106 buf = [b"blob\n", b"mark :%d\n" % mark] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
107 write_data(buf, data, False) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
108 ui.write(*buf, keepprogressbar=True) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
109 del buf |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
110 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
111 # Assign a mark for the current revision for references by |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
112 # latter merge commits. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
113 mark = len(marks) + 1 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
114 marks[revid] = mark |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
115 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
116 ref = convert_to_git_ref(ctx.branch()) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
117 buf = [ |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
118 b"commit %s\n" % ref, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
119 b"mark :%d\n" % mark, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
120 b"committer %s %s\n" |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
121 % ( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
122 convert_to_git_user(authormap, ctx.user(), revid), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
123 convert_to_git_date(ctx.date()), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
124 ), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
125 ] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
126 write_data(buf, ctx.description(), True) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
127 if parents: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
128 buf.append(b"from :%d\n" % marks[parents[0].hex()]) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
129 if len(parents) == 2: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
130 buf.append(b"merge :%d\n" % marks[parents[1].hex()]) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
131 p0ctx = repo[parents[0]] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
132 files = ctx.manifest().diff(p0ctx.manifest()) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
133 else: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
134 files = ctx.files() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
135 filebuf = [] |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
136 for fname in files: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
137 if fname not in ctx: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
138 filebuf.append((fname, b"D %s\n" % fname)) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
139 else: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
140 filectx = ctx.filectx(fname) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
141 filerev = filectx.filenode() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
142 fileperm = b"755" if filectx.isexec() else b"644" |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
143 changed = b"M %s :%d %s\n" % (fileperm, marks[hex(filerev)], fname) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
144 filebuf.append((fname, changed)) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
145 filebuf.sort() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
146 buf.extend(changed for (fname, changed) in filebuf) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
147 del filebuf |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
148 buf.append(b"\n") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
149 ui.write(*buf, keepprogressbar=True) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
150 del buf |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
151 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
152 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
153 isrev = re.compile(b"^[0-9a-f]{40}$") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
154 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
155 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
156 @command( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
157 b"fastexport", |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
158 [ |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
159 (b"r", b"rev", [], _(b"revisions to export"), _(b"REV")), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
160 (b"i", b"import-marks", b"", _(b"old marks file to read"), _(b"FILE")), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
161 (b"e", b"export-marks", b"", _(b"new marks file to write"), _(b"FILE")), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
162 ( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
163 b"A", |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
164 b"authormap", |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
165 b"", |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
166 _(b"remap usernames using this file"), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
167 _(b"FILE"), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
168 ), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
169 ], |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
170 _(b"[OPTION]... [REV]..."), |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
171 helpcategory=command.CATEGORY_IMPORT_EXPORT, |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
172 ) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
173 def fastexport(ui, repo, *revs, **opts): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
174 """export repository as git fast-import stream |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
175 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
176 This command lets you dump a repository as a human-readable text stream. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
177 It can be piped into corresponding import routines like "git fast-import". |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
178 Incremental dumps can be created by using marks files. |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
179 """ |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
180 opts = pycompat.byteskwargs(opts) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
181 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
182 revs += tuple(opts.get(b"rev", [])) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
183 if not revs: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
184 revs = scmutil.revrange(repo, [b":"]) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
185 else: |
48116
5ced12cfa41b
errors: raise InputError on bad revset to revrange() iff provided by the user
Martin von Zweigbergk <martinvonz@google.com>
parents:
44821
diff
changeset
|
186 revs = logcmdutil.revrange(repo, revs) |
44280
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
187 if not revs: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
188 raise error.Abort(_(b"no revisions matched")) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
189 authorfile = opts.get(b"authormap") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
190 if authorfile: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
191 authormap = convcmd.readauthormap(ui, authorfile) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
192 else: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
193 authormap = {} |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
194 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
195 import_marks = opts.get(b"import_marks") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
196 marks = {} |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
197 if import_marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
198 with open(import_marks, "rb") as import_marks_file: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
199 for line in import_marks_file: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
200 line = line.strip() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
201 if not isrev.match(line) or line in marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
202 raise error.Abort(_(b"Corrupted marks file")) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
203 marks[line] = len(marks) + 1 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
204 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
205 revs.sort() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
206 with ui.makeprogress( |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
207 _(b"exporting"), unit=_(b"revisions"), total=len(revs) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
208 ) as progress: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
209 for rev in revs: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
210 export_commit(ui, repo, rev, marks, authormap) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
211 progress.increment() |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
212 |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
213 export_marks = opts.get(b"export_marks") |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
214 if export_marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
215 with open(export_marks, "wb") as export_marks_file: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
216 output_marks = [None] * len(marks) |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
217 for k, v in marks.items(): |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
218 output_marks[v - 1] = k |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
219 for k in output_marks: |
93a05cb223da
hgext: initial version of fastexport extension
Joerg Sonnenberger <joerg@bec.de>
parents:
diff
changeset
|
220 export_marks_file.write(k + b"\n") |