1#!/usr/local/bin/python
2# $DragonFly: src/tools/tools/splitpatch/splitpatch.py,v 1.2 2005/05/21 10:47:43 corecode Exp $
3
4"""Split a patch file into one patch for each file."""
5
6__copyright__ = """
7Copyright (c) 2004 Joerg Sonnenberger.  All rights reserved.
8
9Redistribution and use in source and binary forms, with or without
10modification, are permitted provided that the following conditions are
11met:
12
13  * Redistributions of source code must retain the above copyright
14    notice, this list of conditions and the following disclaimer.
15
16  * Redistributions in binary form must reproduce the above copyright
17    notice, this list of conditions and the following disclaimer in the
18    documentation and/or other materials provided with the distribution.
19
20  * Neither the name of the authors nor the names of there
21    contributors may be used to endorse or promote products derived from
22    this software without specific prior written permission.
23
24THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
25IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
26TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
27PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR
28CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
29EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
30PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
31PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
32LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
33NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
34SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35"""
36
37
38import os.path
39
40def directory_save(filename, patch, suffix = None, root = None, forceful = False):
41	"""
42	Saves patch into filename.suffix relative to root.
43	"""
44	if suffix is None:
45		suffix = ".patch"
46	if root is None:
47		root = ""
48	output_name  = os.path.join(root, "%s%s" % (filename.replace('/',','), suffix))
49
50	if os.path.exists(output_name) and not forceful:
51		raise IOError, 'file exists'
52	f = open(output_name,"w")
53	f.write(patch)
54	f.close()
55
56def splitpatch(source, output = directory_save, quiet = False, tag = False):
57	"""
58	Split the patch in source into independent pieces
59	and call output on the result with the guessed filename
60	and the patch itself.
61
62	source has to provide an iterator like file objects.
63	"""
64	diff_line = { " ": True, "+": True, "-": True, "@": True, "!": True, '*': True }
65	buf = []
66	if tag:
67		buf.append('$''DragonFly$\n')
68	filename = None
69	for line in source:
70		if not filename:
71			if line.startswith('***'):
72				# context diff
73				filename = line.split()[1]
74			elif line.startswith('+++'):
75				# unified diff
76				filename = line.split()[1]
77
78			if filename and not quiet:
79				print "Found patch for %s" % filename
80
81			buf.append(line)
82		elif diff_line.get(line[0]):
83			# valid line for a patch
84			buf.append(line)
85		else:
86			# invalid line for a patch, write out current buffer
87			output(filename, "".join(buf))
88
89			filename = None
90			buf = []
91			if tag:
92				buf.append('$''DragonFly$\n')
93
94	if filename:
95		output(filename, "".join(buf))
96
97def main():
98	from optparse import OptionParser
99	import sys
100	parser = OptionParser("usage: %prog [-q] [-f] [-s suffix] [input]")
101	parser.add_option("-q", "--quiet", action="store_true", dest="quiet", help="do not print names of the subpatches")
102	parser.add_option("-f", "--force", action="store_true", dest="force", help="overwrite existing patches")
103	parser.add_option("-s", "--suffix", type="string", dest="suffix", help="use SUFFIX instead of .patch for the created patches")
104	parser.add_option("-d", "--directory", type="string", dest="directory", help="create patches in DIRECTORY")
105	parser.add_option("-t", "--tag", action="store_true", dest="tag", help="add DragonFly CVS tag")
106	(options, args) = parser.parse_args()
107	if len(args) > 1:
108		parser.error("incorrect number of arguments")
109	if args:
110		source = open(args[0])
111	else:
112		source = sys.stdin
113	splitpatch(source, lambda filename, patch: directory_save(filename, patch, forceful = options.force,
114				suffix = options.suffix, root = options.directory), quiet = options.quiet, tag = options.tag)
115
116if __name__ == '__main__':
117	main()
118