The cr-android EWS should actually be building Chromium for Android
[WebKit-https.git] / Tools / Scripts / webkitpy / tool / multicommandtool.py
1 # Copyright (c) 2009 Google Inc. All rights reserved.
2 # Copyright (c) 2009 Apple Inc. All rights reserved.
3 #
4 # Redistribution and use in source and binary forms, with or without
5 # modification, are permitted provided that the following conditions are
6 # met:
7
8 #     * Redistributions of source code must retain the above copyright
9 # notice, this list of conditions and the following disclaimer.
10 #     * Redistributions in binary form must reproduce the above
11 # copyright notice, this list of conditions and the following disclaimer
12 # in the documentation and/or other materials provided with the
13 # distribution.
14 #     * Neither the name of Google Inc. nor the names of its
15 # contributors may be used to endorse or promote products derived from
16 # this software without specific prior written permission.
17
18 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 #
30 # MultiCommandTool provides a framework for writing svn-like/git-like tools
31 # which are called with the following format:
32 # tool-name [global options] command-name [command options]
33
34 import sys
35
36 from optparse import OptionParser, IndentedHelpFormatter, SUPPRESS_USAGE, make_option
37
38 from webkitpy.tool.grammar import pluralize
39 from webkitpy.common.system.deprecated_logging import log
40
41
42 class TryAgain(Exception):
43     pass
44
45
46 class Command(object):
47     name = None
48     show_in_main_help = False
49     def __init__(self, help_text, argument_names=None, options=None, long_help=None, requires_local_commits=False):
50         self.help_text = help_text
51         self.long_help = long_help
52         self.argument_names = argument_names
53         self.required_arguments = self._parse_required_arguments(argument_names)
54         self.options = options
55         self.requires_local_commits = requires_local_commits
56         self._tool = None
57         # option_parser can be overriden by the tool using set_option_parser
58         # This default parser will be used for standalone_help printing.
59         self.option_parser = HelpPrintingOptionParser(usage=SUPPRESS_USAGE, add_help_option=False, option_list=self.options)
60
61     def _exit(self, code):
62         sys.exit(code)
63
64     # This design is slightly awkward, but we need the
65     # the tool to be able to create and modify the option_parser
66     # before it knows what Command to run.
67     def set_option_parser(self, option_parser):
68         self.option_parser = option_parser
69         self._add_options_to_parser()
70
71     def _add_options_to_parser(self):
72         options = self.options or []
73         for option in options:
74             self.option_parser.add_option(option)
75
76     # The tool calls bind_to_tool on each Command after adding it to its list.
77     def bind_to_tool(self, tool):
78         # Command instances can only be bound to one tool at a time.
79         if self._tool and tool != self._tool:
80             raise Exception("Command already bound to tool!")
81         self._tool = tool
82
83     @staticmethod
84     def _parse_required_arguments(argument_names):
85         required_args = []
86         if not argument_names:
87             return required_args
88         split_args = argument_names.split(" ")
89         for argument in split_args:
90             if argument[0] == '[':
91                 # For now our parser is rather dumb.  Do some minimal validation that
92                 # we haven't confused it.
93                 if argument[-1] != ']':
94                     raise Exception("Failure to parse argument string %s.  Argument %s is missing ending ]" % (argument_names, argument))
95             else:
96                 required_args.append(argument)
97         return required_args
98
99     def name_with_arguments(self):
100         usage_string = self.name
101         if self.options:
102             usage_string += " [options]"
103         if self.argument_names:
104             usage_string += " " + self.argument_names
105         return usage_string
106
107     def parse_args(self, args):
108         return self.option_parser.parse_args(args)
109
110     def check_arguments_and_execute(self, options, args, tool=None):
111         if len(args) < len(self.required_arguments):
112             log("%s required, %s provided.  Provided: %s  Required: %s\nSee '%s help %s' for usage." % (
113                 pluralize("argument", len(self.required_arguments)),
114                 pluralize("argument", len(args)),
115                 "'%s'" % " ".join(args),
116                 " ".join(self.required_arguments),
117                 tool.name(),
118                 self.name))
119             return 1
120         return self.execute(options, args, tool) or 0
121
122     def standalone_help(self):
123         help_text = self.name_with_arguments().ljust(len(self.name_with_arguments()) + 3) + self.help_text + "\n\n"
124         if self.long_help:
125             help_text += "%s\n\n" % self.long_help
126         help_text += self.option_parser.format_option_help(IndentedHelpFormatter())
127         return help_text
128
129     def execute(self, options, args, tool):
130         raise NotImplementedError, "subclasses must implement"
131
132     # main() exists so that Commands can be turned into stand-alone scripts.
133     # Other parts of the code will likely require modification to work stand-alone.
134     def main(self, args=sys.argv):
135         (options, args) = self.parse_args(args)
136         # Some commands might require a dummy tool
137         return self.check_arguments_and_execute(options, args)
138
139
140 # FIXME: This should just be rolled into Command.  help_text and argument_names do not need to be instance variables.
141 class AbstractDeclarativeCommand(Command):
142     help_text = None
143     argument_names = None
144     long_help = None
145     def __init__(self, options=None, **kwargs):
146         Command.__init__(self, self.help_text, self.argument_names, options=options, long_help=self.long_help, **kwargs)
147
148
149 class HelpPrintingOptionParser(OptionParser):
150     def __init__(self, epilog_method=None, *args, **kwargs):
151         self.epilog_method = epilog_method
152         OptionParser.__init__(self, *args, **kwargs)
153
154     def error(self, msg):
155         self.print_usage(sys.stderr)
156         error_message = "%s: error: %s\n" % (self.get_prog_name(), msg)
157         # This method is overriden to add this one line to the output:
158         error_message += "\nType \"%s --help\" to see usage.\n" % self.get_prog_name()
159         self.exit(1, error_message)
160
161     # We override format_epilog to avoid the default formatting which would paragraph-wrap the epilog
162     # and also to allow us to compute the epilog lazily instead of in the constructor (allowing it to be context sensitive).
163     def format_epilog(self, epilog):
164         if self.epilog_method:
165             return "\n%s\n" % self.epilog_method()
166         return ""
167
168
169 class HelpCommand(AbstractDeclarativeCommand):
170     name = "help"
171     help_text = "Display information about this program or its subcommands"
172     argument_names = "[COMMAND]"
173
174     def __init__(self):
175         options = [
176             make_option("-a", "--all-commands", action="store_true", dest="show_all_commands", help="Print all available commands"),
177         ]
178         AbstractDeclarativeCommand.__init__(self, options)
179         self.show_all_commands = False # A hack used to pass --all-commands to _help_epilog even though it's called by the OptionParser.
180
181     def _help_epilog(self):
182         # Only show commands which are relevant to this checkout's SCM system.  Might this be confusing to some users?
183         if self.show_all_commands:
184             epilog = "All %prog commands:\n"
185             relevant_commands = self._tool.commands[:]
186         else:
187             epilog = "Common %prog commands:\n"
188             relevant_commands = filter(self._tool.should_show_in_main_help, self._tool.commands)
189         longest_name_length = max(map(lambda command: len(command.name), relevant_commands))
190         relevant_commands.sort(lambda a, b: cmp(a.name, b.name))
191         command_help_texts = map(lambda command: "   %s   %s\n" % (command.name.ljust(longest_name_length), command.help_text), relevant_commands)
192         epilog += "%s\n" % "".join(command_help_texts)
193         epilog += "See '%prog help --all-commands' to list all commands.\n"
194         epilog += "See '%prog help COMMAND' for more information on a specific command.\n"
195         return epilog.replace("%prog", self._tool.name()) # Use of %prog here mimics OptionParser.expand_prog_name().
196
197     # FIXME: This is a hack so that we don't show --all-commands as a global option:
198     def _remove_help_options(self):
199         for option in self.options:
200             self.option_parser.remove_option(option.get_opt_string())
201
202     def execute(self, options, args, tool):
203         if args:
204             command = self._tool.command_by_name(args[0])
205             if command:
206                 print command.standalone_help()
207                 return 0
208
209         self.show_all_commands = options.show_all_commands
210         self._remove_help_options()
211         self.option_parser.print_help()
212         return 0
213
214
215 class MultiCommandTool(object):
216     global_options = None
217
218     def __init__(self, name=None, commands=None):
219         self._name = name or OptionParser(prog=name).get_prog_name() # OptionParser has nice logic for fetching the name.
220         # Allow the unit tests to disable command auto-discovery.
221         self.commands = commands or [cls() for cls in self._find_all_commands() if cls.name]
222         self.help_command = self.command_by_name(HelpCommand.name)
223         # Require a help command, even if the manual test list doesn't include one.
224         if not self.help_command:
225             self.help_command = HelpCommand()
226             self.commands.append(self.help_command)
227         for command in self.commands:
228             command.bind_to_tool(self)
229
230     @classmethod
231     def _add_all_subclasses(cls, class_to_crawl, seen_classes):
232         for subclass in class_to_crawl.__subclasses__():
233             if subclass not in seen_classes:
234                 seen_classes.add(subclass)
235                 cls._add_all_subclasses(subclass, seen_classes)
236
237     @classmethod
238     def _find_all_commands(cls):
239         commands = set()
240         cls._add_all_subclasses(Command, commands)
241         return sorted(commands)
242
243     def name(self):
244         return self._name
245
246     def _create_option_parser(self):
247         usage = "Usage: %prog [options] COMMAND [ARGS]"
248         return HelpPrintingOptionParser(epilog_method=self.help_command._help_epilog, prog=self.name(), usage=usage)
249
250     @staticmethod
251     def _split_command_name_from_args(args):
252         # Assume the first argument which doesn't start with "-" is the command name.
253         command_index = 0
254         for arg in args:
255             if arg[0] != "-":
256                 break
257             command_index += 1
258         else:
259             return (None, args[:])
260
261         command = args[command_index]
262         return (command, args[:command_index] + args[command_index + 1:])
263
264     def command_by_name(self, command_name):
265         for command in self.commands:
266             if command_name == command.name:
267                 return command
268         return None
269
270     def path(self):
271         raise NotImplementedError, "subclasses must implement"
272
273     def command_completed(self):
274         pass
275
276     def should_show_in_main_help(self, command):
277         return command.show_in_main_help
278
279     def should_execute_command(self, command):
280         return True
281
282     def _add_global_options(self, option_parser):
283         global_options = self.global_options or []
284         for option in global_options:
285             option_parser.add_option(option)
286
287     def handle_global_options(self, options):
288         pass
289
290     def main(self, argv=sys.argv):
291         (command_name, args) = self._split_command_name_from_args(argv[1:])
292
293         option_parser = self._create_option_parser()
294         self._add_global_options(option_parser)
295
296         command = self.command_by_name(command_name) or self.help_command
297         if not command:
298             option_parser.error("%s is not a recognized command" % command_name)
299
300         command.set_option_parser(option_parser)
301         (options, args) = command.parse_args(args)
302         self.handle_global_options(options)
303
304         (should_execute, failure_reason) = self.should_execute_command(command)
305         if not should_execute:
306             log(failure_reason)
307             return 0 # FIXME: Should this really be 0?
308
309         while True:
310             try:
311                 result = command.check_arguments_and_execute(options, args, self)
312                 break
313             except TryAgain, e:
314                 pass
315
316         self.command_completed()
317         return result