1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
# Copyright (c) 2010 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Compresses input arguments to a zip archive after running text files
through version/branding token replacement.
"""
import optparse
import os
import shutil
import sys
import zipfile
sys.path.append(
os.path.join(os.path.dirname(sys.argv[0]), '../../chrome/tools/build'))
import version
# Special token to separate the regular files needed in the XPI file from
# the debug-only files. The regular files are listed first, then the separator,
# followed by the debug only files.
_DEBUG_ONLY_FILE_SEPARATOR = '##'
# Usage string for option parser.
_USAGE='Usage: %prog [options] <file-list> [## <debug-only-file-list>]'
# Help description text for option parser.
_DESCRIPTION="""Creates a zip file containing the list of files
specified on the command line. Also expands constants from VERSION
and BRANDING files.
The files before the ## token are always added to the zip file. The files
after the ## token are added only if the configuration specified is Debug.
If relative paths are used to name the files, they are assumed to be relative
to the current working directory. Any directory structure will be preserved
unless the -p command line option is used to place the files in a specific
folder of the zip file.
The --version and --branding flags must be used to specify the paths
to VERSION and BRANDING files, respectively. Add any binary files
(and/or text files you know won't need expanding) to the
--ignore_extensions list.
"""
def ParseCommandLine():
"""Constructs and configures an option parser for this script and parses
the command line arguments.
Returns:
The parsed options.
"""
parser = optparse.OptionParser(usage=_USAGE, description=_DESCRIPTION)
parser.add_option('-i', '--input', dest='input',
help='An archive file to append to.')
parser.add_option('-o', '--output', dest='output',
help='Output file for archive.')
parser.add_option('-p', '--path', dest='path',
help='Path within the archive to write the files to.')
parser.add_option('-c', '--config', dest='config',
help='Configuration being built, either Debug or Release.')
parser.add_option('-v', '--version', dest='version_file',
help='Path to VERSION file.')
parser.add_option('-b', '--branding', dest='branding_file',
help='Path to BRANDING file.')
parser.add_option('-x', '--ignore_extensions', dest='ignore_extensions',
default='',
help='Comma-separated list of extensions to not expand.')
(options, args) = parser.parse_args()
if not options.output:
parser.error('You must provide an output file.')
if not options.config:
parser.error('You must provide configuration.')
if not options.version_file or not options.branding_file:
parser.error('You must provide --version and --branding.')
return (options, args)
def Main():
"""Zip input arguments to Zip output archive."""
(options, args) = ParseCommandLine()
# Test that all the input files exist before we blow the output archive away
# only to fail part way.
for path in args:
if path == _DEBUG_ONLY_FILE_SEPARATOR:
continue
if not os.path.exists(path):
raise Exception('Input file does not exist:', path)
if os.path.isdir(path):
raise Exception('Input file is a path:', path)
ignore_extensions = options.ignore_extensions.split(',')
# Set up variable substitution mapping.
variable_files = [options.version_file, options.branding_file]
subst_values = version.fetch_values(variable_files)
# Open the output file.
if options.input:
shutil.copyfile(options.input, options.output)
mode = 'a'
else:
mode = 'w'
output = zipfile.ZipFile(options.output, mode, zipfile.ZIP_DEFLATED)
# Add the files to the output archive.
for path in args:
if path == _DEBUG_ONLY_FILE_SEPARATOR:
if options.config != 'Debug':
break
else:
dest_path = path
if options.path:
dest_path = os.path.join(options.path, os.path.basename(path))
extension = os.path.splitext(path)[1]
if extension:
extension = extension[1:]
if extension in ignore_extensions:
output.write(path, dest_path)
else:
contents = version.subst_file(path, subst_values)
output.writestr(dest_path, contents)
output.close()
return 0
if __name__ == '__main__':
sys.exit(Main())
|
