Commit | Line | Data |
---|---|---|
0d2cd1e4 CM |
1 | __copyright__ = """ |
2 | Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com> | |
3 | ||
4 | This program is free software; you can redistribute it and/or modify | |
5 | it under the terms of the GNU General Public License version 2 as | |
6 | published by the Free Software Foundation. | |
7 | ||
8 | This program is distributed in the hope that it will be useful, | |
9 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
10 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
11 | GNU General Public License for more details. | |
12 | ||
13 | You should have received a copy of the GNU General Public License | |
14 | along with this program; if not, write to the Free Software | |
15 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
16 | """ | |
17 | ||
9417ece4 | 18 | import sys, os, re |
0d2cd1e4 CM |
19 | from optparse import OptionParser, make_option |
20 | ||
21 | from stgit.commands.common import * | |
22 | from stgit.utils import * | |
23 | from stgit import stack, git | |
24 | ||
25 | ||
26 | help = 'import a GNU diff file as a new patch' | |
b8a0986f | 27 | usage = """%prog [options] [<file>] |
0d2cd1e4 | 28 | |
b8a0986f CM |
29 | Create a new patch and apply the given GNU diff file (or the standard |
30 | input). By default, the file name is used as the patch name but this | |
388f63b6 | 31 | can be overridden with the '--name' option. The patch can either be a |
b8a0986f CM |
32 | normal file with the description at the top or it can have standard |
33 | mail format, the Subject, From and Date headers being used for | |
34 | generating the patch information. | |
0d2cd1e4 | 35 | |
b8a0986f | 36 | The patch description has to be separated from the data with a '---' |
99e73103 | 37 | line.""" |
0d2cd1e4 CM |
38 | |
39 | options = [make_option('-m', '--mail', | |
40 | help = 'import the patch from a standard e-mail file', | |
41 | action = 'store_true'), | |
42 | make_option('-n', '--name', | |
43 | help = 'use NAME as the patch name'), | |
b0cdad5e CM |
44 | make_option('-t', '--strip', |
45 | help = 'strip numbering and extension from patch name', | |
46 | action = 'store_true'), | |
9417ece4 CM |
47 | make_option('-s', '--series', |
48 | help = 'import a series of patches', | |
49 | action = 'store_true'), | |
50 | make_option('-i', '--ignore', | |
51 | help = 'ignore the applied patches in the series', | |
52 | action = 'store_true'), | |
034db15c CM |
53 | make_option('--replace', |
54 | help = 'replace the unapplied patches in the series', | |
55 | action = 'store_true'), | |
b21bc8d1 | 56 | make_option('-b', '--base', |
35344f86 | 57 | help = 'use BASE instead of HEAD for file importing'), |
33e580e0 CM |
58 | make_option('-e', '--edit', |
59 | help = 'invoke an editor for the patch description', | |
60 | action = 'store_true'), | |
9417ece4 | 61 | make_option('-p', '--showpatch', |
6ad48e48 PBG |
62 | help = 'show the patch content in the editor buffer', |
63 | action = 'store_true'), | |
0d2cd1e4 CM |
64 | make_option('-a', '--author', metavar = '"NAME <EMAIL>"', |
65 | help = 'use "NAME <EMAIL>" as the author details'), | |
66 | make_option('--authname', | |
67 | help = 'use AUTHNAME as the author name'), | |
68 | make_option('--authemail', | |
69 | help = 'use AUTHEMAIL as the author e-mail'), | |
70 | make_option('--authdate', | |
71 | help = 'use AUTHDATE as the author date'), | |
72 | make_option('--commname', | |
73 | help = 'use COMMNAME as the committer name'), | |
74 | make_option('--commemail', | |
75 | help = 'use COMMEMAIL as the committer e-mail')] | |
76 | ||
77 | ||
d4c43e19 PBG |
78 | def __end_descr(line): |
79 | return re.match('---\s*$', line) or re.match('diff -', line) or \ | |
80 | re.match('Index: ', line) | |
99e73103 | 81 | |
b0cdad5e | 82 | def __strip_patch_name(name): |
bcb6d890 CM |
83 | stripped = re.sub('^[0-9]+-(.*)$', '\g<1>', name) |
84 | stripped = re.sub('^(.*)\.(diff|patch)$', '\g<1>', stripped) | |
85 | ||
86 | return stripped | |
b0cdad5e | 87 | |
99e73103 CM |
88 | def __parse_description(descr): |
89 | """Parse the patch description and return the new description and | |
90 | author information (if any). | |
91 | """ | |
92 | subject = body = '' | |
0543bc5f | 93 | authname = authemail = authdate = None |
99e73103 | 94 | |
0543bc5f | 95 | descr_lines = [line.rstrip() for line in descr.split('\n')] |
99e73103 CM |
96 | if not descr_lines: |
97 | raise CmdException, "Empty patch description" | |
98 | ||
0543bc5f | 99 | lasthdr = 0 |
99e73103 CM |
100 | end = len(descr_lines) |
101 | ||
0543bc5f | 102 | # Parse the patch header |
61dabd0e | 103 | for pos in range(0, end): |
0543bc5f TM |
104 | if not descr_lines[pos]: |
105 | continue | |
106 | # check for a "From|Author:" line | |
107 | if re.match('\s*(?:from|author):\s+', descr_lines[pos], re.I): | |
108 | auth = re.findall('^.*?:\s+(.*)$', descr_lines[pos])[0] | |
109 | authname, authemail = name_email(auth) | |
110 | lasthdr = pos + 1 | |
111 | continue | |
112 | # check for a "Date:" line | |
113 | if re.match('\s*date:\s+', descr_lines[pos], re.I): | |
114 | authdate = re.findall('^.*?:\s+(.*)$', descr_lines[pos])[0] | |
115 | lasthdr = pos + 1 | |
116 | continue | |
117 | if subject: | |
118 | break | |
119 | # get the subject | |
120 | subject = descr_lines[pos] | |
121 | lasthdr = pos + 1 | |
99e73103 CM |
122 | |
123 | # get the body | |
0543bc5f TM |
124 | if lasthdr < end: |
125 | body = reduce(lambda x, y: x + '\n' + y, descr_lines[lasthdr:], '') | |
99e73103 | 126 | |
0543bc5f | 127 | return (subject + body, authname, authemail, authdate) |
99e73103 | 128 | |
0d2cd1e4 CM |
129 | def __parse_mail(filename = None): |
130 | """Parse the input file in a mail format and return (description, | |
131 | authname, authemail, authdate) | |
132 | """ | |
133 | if filename: | |
134 | f = file(filename) | |
135 | else: | |
136 | f = sys.stdin | |
137 | ||
138 | descr = authname = authemail = authdate = None | |
139 | ||
140 | # parse the headers | |
6fe6b1bd CM |
141 | while True: |
142 | line = f.readline() | |
143 | if not line: | |
144 | break | |
0d2cd1e4 CM |
145 | line = line.strip() |
146 | if re.match('from:\s+', line, re.I): | |
147 | auth = re.findall('^.*?:\s+(.*)$', line)[0] | |
148 | authname, authemail = name_email(auth) | |
149 | elif re.match('date:\s+', line, re.I): | |
150 | authdate = re.findall('^.*?:\s+(.*)$', line)[0] | |
151 | elif re.match('subject:\s+', line, re.I): | |
152 | descr = re.findall('^.*?:\s+(.*)$', line)[0] | |
153 | elif line == '': | |
154 | # end of headers | |
155 | break | |
156 | ||
186e6b6b | 157 | # remove the '[*PATCH*]' expression in the subject |
0d2cd1e4 | 158 | if descr: |
7c02f338 CM |
159 | descr = re.findall('^(\[[^\s]*[Pp][Aa][Tt][Cc][Hh].*?\])?\s*(.*)$', |
160 | descr)[0][1] | |
0d2cd1e4 CM |
161 | descr += '\n\n' |
162 | else: | |
163 | raise CmdException, 'Subject: line not found' | |
164 | ||
165 | # the rest of the patch description | |
6fe6b1bd CM |
166 | while True: |
167 | line = f.readline() | |
168 | if not line: | |
169 | break | |
d4c43e19 | 170 | if __end_descr(line): |
0d2cd1e4 CM |
171 | break |
172 | else: | |
173 | descr += line | |
174 | descr.rstrip() | |
175 | ||
176 | if filename: | |
177 | f.close() | |
178 | ||
99e73103 | 179 | # parse the description for author information |
0543bc5f | 180 | descr, descr_authname, descr_authemail, descr_authdate = __parse_description(descr) |
99e73103 CM |
181 | if descr_authname: |
182 | authname = descr_authname | |
183 | if descr_authemail: | |
184 | authemail = descr_authemail | |
0543bc5f TM |
185 | if descr_authdate: |
186 | authdate = descr_authdate | |
99e73103 | 187 | |
0d2cd1e4 CM |
188 | return (descr, authname, authemail, authdate) |
189 | ||
190 | def __parse_patch(filename = None): | |
191 | """Parse the input file and return (description, authname, | |
192 | authemail, authdate) | |
193 | """ | |
194 | if filename: | |
195 | f = file(filename) | |
196 | else: | |
197 | f = sys.stdin | |
198 | ||
0d2cd1e4 | 199 | descr = '' |
6fe6b1bd CM |
200 | while True: |
201 | line = f.readline() | |
202 | if not line: | |
203 | break | |
204 | ||
d4c43e19 | 205 | if __end_descr(line): |
0d2cd1e4 CM |
206 | break |
207 | else: | |
208 | descr += line | |
209 | descr.rstrip() | |
210 | ||
0d2cd1e4 CM |
211 | if filename: |
212 | f.close() | |
213 | ||
0543bc5f | 214 | descr, authname, authemail, authdate = __parse_description(descr) |
99e73103 CM |
215 | |
216 | # we don't yet have an agreed place for the creation date. | |
217 | # Just return None | |
0543bc5f | 218 | return (descr, authname, authemail, authdate) |
0d2cd1e4 | 219 | |
9417ece4 CM |
220 | def __import_patch(patch, filename, options): |
221 | """Import a patch from a file or standard input | |
0d2cd1e4 | 222 | """ |
0d2cd1e4 CM |
223 | # the defaults |
224 | message = author_name = author_email = author_date = committer_name = \ | |
225 | committer_email = None | |
226 | ||
227 | if options.author: | |
228 | options.authname, options.authemail = name_email(options.author) | |
229 | ||
230 | if options.mail: | |
231 | message, author_name, author_email, author_date = \ | |
232 | __parse_mail(filename) | |
233 | else: | |
234 | message, author_name, author_email, author_date = \ | |
235 | __parse_patch(filename) | |
236 | ||
fff9bce5 CM |
237 | if not patch: |
238 | patch = make_patch_name(message) | |
239 | if not patch: | |
240 | raise CmdException, 'Unknown patch name' | |
241 | ||
95742cfc PBG |
242 | # refresh_patch() will invoke the editor in this case, with correct |
243 | # patch content | |
9d15ccd8 | 244 | if not message: |
95742cfc | 245 | can_edit = False |
9d15ccd8 | 246 | |
0d2cd1e4 CM |
247 | # override the automatically parsed settings |
248 | if options.authname: | |
249 | author_name = options.authname | |
250 | if options.authemail: | |
251 | author_email = options.authemail | |
252 | if options.authdate: | |
253 | author_date = options.authdate | |
254 | if options.commname: | |
255 | committer_name = options.commname | |
256 | if options.commemail: | |
257 | committer_email = options.commemail | |
258 | ||
034db15c CM |
259 | if options.replace and patch in crt_series.get_unapplied(): |
260 | crt_series.delete_patch(patch) | |
261 | ||
95742cfc | 262 | crt_series.new_patch(patch, message = message, can_edit = False, |
0d2cd1e4 CM |
263 | author_name = author_name, |
264 | author_email = author_email, | |
265 | author_date = author_date, | |
266 | committer_name = committer_name, | |
267 | committer_email = committer_email) | |
268 | ||
9417ece4 | 269 | print 'Importing patch "%s"...' % patch, |
0d2cd1e4 CM |
270 | sys.stdout.flush() |
271 | ||
35344f86 | 272 | if options.base: |
be3e6bd9 | 273 | git.apply_patch(filename, git_id(options.base)) |
35344f86 CM |
274 | else: |
275 | git.apply_patch(filename) | |
276 | ||
6ad48e48 PBG |
277 | crt_series.refresh_patch(edit = options.edit, |
278 | show_patch = options.showpatch) | |
0d2cd1e4 CM |
279 | |
280 | print 'done' | |
9417ece4 CM |
281 | |
282 | def __import_series(filename, options): | |
283 | """Import a series of patches | |
284 | """ | |
285 | applied = crt_series.get_applied() | |
286 | ||
287 | if filename: | |
288 | f = file(filename) | |
289 | patchdir = os.path.dirname(filename) | |
290 | else: | |
291 | f = sys.stdin | |
292 | patchdir = '' | |
293 | ||
294 | for line in f: | |
295 | patch = re.sub('#.*$', '', line).strip() | |
296 | if not patch: | |
297 | continue | |
bcb6d890 CM |
298 | patchfile = os.path.join(patchdir, patch) |
299 | ||
b0cdad5e CM |
300 | if options.strip: |
301 | patch = __strip_patch_name(patch) | |
9417ece4 CM |
302 | if options.ignore and patch in applied: |
303 | print 'Ignoring already applied patch "%s"' % patch | |
304 | continue | |
305 | ||
9417ece4 CM |
306 | __import_patch(patch, patchfile, options) |
307 | ||
308 | def func(parser, options, args): | |
309 | """Import a GNU diff file as a new patch | |
310 | """ | |
311 | if len(args) > 1: | |
312 | parser.error('incorrect number of arguments') | |
313 | ||
314 | check_local_changes() | |
315 | check_conflicts() | |
316 | check_head_top_equal() | |
317 | ||
318 | if len(args) == 1: | |
319 | filename = args[0] | |
320 | else: | |
321 | filename = None | |
322 | ||
323 | if options.series: | |
324 | __import_series(filename, options) | |
325 | else: | |
326 | if options.name: | |
327 | patch = options.name | |
328 | elif filename: | |
329 | patch = os.path.basename(filename) | |
330 | else: | |
fff9bce5 | 331 | patch = '' |
b0cdad5e CM |
332 | if options.strip: |
333 | patch = __strip_patch_name(patch) | |
9417ece4 CM |
334 | |
335 | __import_patch(patch, filename, options) | |
336 | ||
0d2cd1e4 | 337 | print_crt_patch() |