Implement mb - a meta-build wrapper for bots to use in the GYP->GN migration.

tools/mb/mb.py

Index: tools/mb/mb.py
diff --git a/tools/mb/mb.py b/tools/mb/mb.py
new file mode 100755
index 0000000000000000000000000000000000000000..2c25d76aa4c8d82ce57e20235b85b103fbaac9fc
--- /dev/null
+++ b/tools/mb/mb.py
@@ -0,0 +1,453 @@
+#!/usr/bin/env python
+# Copyright 2015 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.
+
+"""MB - the Meta-Build wrapper around GYP and GN
+
+MB is a wrapper script for GYP and GN that can be used to generate build files
+for sets of canned configurations and analyze them.
+"""
+
+from __future__ import print_function
+
+import argparse
+import ast
+import json
+import os
+import pipes
+import shlex
+import shutil
+import sys
+import subprocess
+import tempfile
+
+
+def main(args):
+  mb = MetaBuildWrapper()
+  mb.ParseArgs(args)
+  return mb.args.func()
+
+
+class MetaBuildWrapper(object):
+  def __init__(self):
+    p = os.path
+    d = os.path.dirname
+    self.chromium_src_dir = p.normpath(d(d(d(p.abspath(__file__)))))
+    self.default_config = p.join(self.chromium_src_dir, 'tools', 'mb',
+                                 'mb_config.pyl')
+    self.configs = {}
+    self.masters = {}
+    self.mixins = {}
+
+  def ParseArgs(self, argv):
+    def AddCommonOptions(subp):
+      subp.add_argument('-b', '--builder',
+                        help='builder name to look up config from')
+      subp.add_argument('-m', '--master',
+                        help='master name to look up config from'),
+      subp.add_argument('-c', '--config',
+                        help='configuration to analyze')
+      subp.add_argument('-f', '--config-file', metavar='PATH',
+                        default=self.default_config,
+                        help='path to config file '
+                            '(default is //tools/mb/mb_config.pyl)')
+      subp.add_argument('-g', '--goma-dir', default=self.ExpandUser('~/goma'),
+                        help='path to goma directory (default is %(default)s).')
+      subp.add_argument('-n', '--dryrun', action='store_true',
+                        help='Do a dry run (i.e., do nothing, just print '
+                             'the commands that will run)')
+      subp.add_argument('-q', '--quiet', action='store_true',
+                        help='Do not print anything, just return an exit '
+                             'code.')
+      subp.add_argument('-v', '--verbose', action='count',
+                        help='verbose logging (may specify multiple times).')
+
+    parser = argparse.ArgumentParser(prog='mb')
+    subps = parser.add_subparsers()
+
+    subp = subps.add_parser('analyze',
+                            help='analyze whether changes to a set of files '
+                                 'will cause a set of binaries to be rebuilt.')
+    AddCommonOptions(subp)
+    subp.add_argument('input-path', nargs=1,
+                      help='path to a file containing the input arguments '
+                           'as a JSON object.')
+    subp.add_argument('output-path', nargs=1,
+                      help='path to a file containing the output arguments '
+                           'as a JSON object.')
+    subp.set_defaults(func=self.CmdAnalyze)
+
+    subp = subps.add_parser('gen',
+                            help='generate a new set of build files')
+    AddCommonOptions(subp)
+    subp.add_argument('path', type=str, nargs=1,
+                      help='path to generate build into')
+    subp.set_defaults(func=self.CmdGen)
+
+    subp = subps.add_parser('lookup',
+                            help='look up the command for a given config or '
+                                 'builder')
+    AddCommonOptions(subp)
+    subp.set_defaults(func=self.CmdLookup)
+
+    subp = subps.add_parser('validate',
+                            help='validate the config file')
+    AddCommonOptions(subp)
+    subp.set_defaults(func=self.CmdValidate)
+
+    subp = subps.add_parser('help',
+                            help='Get help on a subcommand.')
+    subp.add_argument(nargs='?', action='store', dest='subcommand',
+                      help='The command to get help for.')
+    subp.set_defaults(func=self.CmdHelp)
+
+    self.args = parser.parse_args(argv)
+
+  def CmdAnalyze(self):
+    vals = self.GetConfig()
+    if vals['type'] == 'gn':
+      self.RunGNAnalyze(vals)
+    elif vals['type'] in ('gyp', 'gyp_one_config'):
+      self.RunGypAnalyze(vals)
+    else:
+      raise MBErr('Unknown meta-build type "%s"' % vals['type'])
+
+  def CmdGen(self):
+    vals = self.GetConfig()
+    if vals['type'] == 'gn':
+      self.RunGNGen(self.args.path[0], vals['gn_args'])
+    elif vals['type'] == 'gyp':
+      self.RunGYPGen(self.args.path[0], vals['gyp_defines'])
+    else:
+      raise MBErr('Unknown meta-build type "%s"' % vals['type'])
+    return 0
+
+  def CmdLookup(self):
+    vals = self.GetConfig()
+    if vals['type'] == 'gn':
+      cmd = self.GNCmd('<path>', vals['gn_args'])
+    elif vals['type'] == 'gyp':
+      cmd = self.GYPCmd('<path>', vals['gyp_defines'], vals['gyp_config'])
+    else:
+      raise MBErr('Unknown meta-build type "%s"' % vals['type'])
+
+    self.PrintCmd(cmd)
+    return 0
+
+  def CmdHelp(self):
+    if self.args.subcommand:
+      self.ParseArgs([self.args.subcommand, '--help'])
+    else:
+      self.ParseArgs(['--help'])
+
+  def CmdValidate(self):
+    errs = []
+
+    # Read the file to make sure it parses.
+    self.ReadConfigFile()
+
+    # Figure out the whole list of configs and ensure that no config is
+    # listed in more than one category.
+    all_configs = {}
+    for config in self.common_dev_configs:
+      all_configs[config] = 'common_dev_configs'
+    for config in self.private_configs:
+      if config in all_configs:
+        errs.append('config "%s" listed in "private_configs" also '
+                    'listed in "%s"' % (config, all_configs['config']))
+      else:
+        all_configs[config] = 'private_configs'
+    for config in self.unsupported_configs:
+      if config in all_configs:
+        errs.append('config "%s" listed in "unsupported_configs" also '
+                    'listed in "%s"' % (config, all_configs['config']))
+      else:
+        all_configs[config] = 'unsupported_configs'
+
+    for master in self.masters:
+      for builder in self.masters[master]:
+        config = self.masters[master][builder]
+        if config in all_configs and all_configs[config] not in self.masters:
+          errs.append('Config "%s" used by a bot is also listed in "%s".' %
+                      (config, all_configs[config]))
+        else:
+          all_configs[config] = master
+
+    # Check that every referenced config actually exists.
+    for config, loc in all_configs.items():
+      if not config in self.configs:
+        errs.append('Unknown config "%s" referenced from "%s".' %
+                    (config, loc))
+
+    # Check that every actual config is actually referenced.
+    for config in self.configs:
+      if not config in all_configs:
+        errs.append('Unused config "%s".' % config)
+
+    # Figure out the whole list of mixins, and check that every mixin
+    # listed by a config or another mixin actually exists.
+    referenced_mixins = set()
+    for config, mixins in self.configs.items():
+      for mixin in mixins:
+        if not mixin in self.mixins:
+          errs.append('Unknown mixin "%s" referenced by config "%s".' %
+                      (mixin, config))
+        referenced_mixins.add(mixin)
+
+    for mixin in self.mixins:
+      for sub_mixin in self.mixins[mixin].get('mixins', []):
+        if not sub_mixin in self.mixins:
+          errs.append('Unknown mixin "%s" referenced by mixin "%s".' %
+                      (sub_mixin, mixin))
+        referenced_mixins.add(sub_mixin)
+
+    # Check that every mixin defined is actually referenced somewhere.
+    for mixin in self.mixins:
+      if not mixin in referenced_mixins:
+        errs.append('Unreferenced mixin "%s".' % mixin)
+
+    if errs:
+      raise MBErr('mb config file %s has problems:\n  ' + '\n  '.join(errs))
+
+    if not self.args.quiet:
+      self.Print('mb config file %s looks ok.' % self.args.config_file)
+    return 0
+
+  def GetConfig(self):
+    self.ReadConfigFile()
+    config = self.ConfigFromArgs()
+    if not config in self.configs:
+      raise MBErr('Config "%s" not found in %s' %
+                  (config, self.args.config_file))
+
+    return self.FlattenConfig(config)
+
+  def ReadConfigFile(self):
+    if not self.Exists(self.args.config_file):
+      raise MBErr('config file not found at %s' % self.args.config_file)
+
+    try:
+      contents = ast.literal_eval(self.ReadFile(self.args.config_file))
+    except SyntaxError as e:
+      raise MBErr('Failed to parse config file "%s": %s' %
+                 (self.args.config_file, e))
+
+    self.common_dev_configs = contents['common_dev_configs']
+    self.configs = contents['configs']
+    self.masters = contents['masters']
+    self.mixins = contents['mixins']
+    self.private_configs = contents['private_configs']
+    self.unsupported_configs = contents['unsupported_configs']
+
+  def ConfigFromArgs(self):
+    if self.args.config:
+      if self.args.master or self.args.builder:
+        raise MBErr('Can not specific both -c/--config and -m/--master or '
+                    '-b/--builder')
+
+      return self.args.config
+
+    if not self.args.master or not self.args.builder:
+      raise MBErr('Must specify either -c/--config or '
+                  '(-m/--master and -b/--builder)')
+
+    if not self.args.master in self.masters:
+      raise MBErr('Master name "%s" not found in "%s"' %
+                  (self.args.master, self.args.config_file))
+
+    if not self.args.builder in self.masters[self.args.master]:
+      raise MBErr('Builder name "%s"  not found under masters[%s] in "%s"' %
+                  (self.args.builder, self.args.master, self.args.config_file))
+
+    return self.masters[self.args.master][self.args.builder]
+
+  def FlattenConfig(self, config):
+    mixins = self.configs[config]
+    vals = {
+      'type': None,
+      'gn_args': [],
+      'gyp_config': [],
+      'gyp_defines': [],
+    }
+
+    visited = []
+    self.FlattenMixins(mixins, vals, visited)
+    return vals
+
+  def FlattenMixins(self, mixins, vals, visited):
+    for m in mixins:
+      if m not in self.mixins:
+        raise MBErr('Unknown mixin "%s"' % m)
+      if m in visited:
+        raise MBErr('Cycle in build configs for "%s": %s' % (config, visited))
+
+      visited.append(m)
+
+      mixin_vals = self.mixins[m]
+      if 'type' in mixin_vals:
+        vals['type'] = mixin_vals['type']
+      if 'gn_args' in mixin_vals:
+        if vals['gn_args']:
+          vals['gn_args'] += ' ' + mixin_vals['gn_args']
+        else:
+          vals['gn_args'] = mixin_vals['gn_args']
+      if 'gyp_config' in mixin_vals:
+        vals['gyp_config'] = mixin_vals['gyp_config']
+      if 'gyp_defines' in mixin_vals:
+        if vals['gyp_defines']:
+          vals['gyp_defines'] += ' ' + mixin_vals['gyp_defines']
+        else:
+          vals['gyp_defines'] = mixin_vals['gyp_defines']
+      if 'mixins' in mixin_vals:
+        self.FlattenMixins(mixin_vals['mixins'], vals, visited)
+    return vals
+
+  def RunGNGen(self, path, gn_args):
+    cmd = self.GNCmd(path, gn_args)
+    ret, _, _ = self.Run(cmd)
+    return ret
+
+  def GNCmd(self, path, gn_args):
+    # TODO(dpranke): Find gn explicitly in the path ...
+    cmd = ['gn', 'gen', path]
+    gn_args = gn_args.replace("$(goma_dir)", gn_args)
+    if gn_args:
+      cmd.append('--args=%s' % ' '.join(gn_args))
+    return cmd
+
+  def RunGYPGen(self, path, gyp_defines):
+    output_dir, gyp_config = self.ParseGYPConfigPath(path)
+    if gyp_config != vals['gyp_config']:
+      raise MBErr('The last component of the path (%s) must match the '
+                  'GYP configuration specified in the config (%s), and '
+                  'it does not.' % (gyp_config, vals['gyp_config']))
+    cmd = self.GYPCmd(output_dir, gyp_defines, config=gyp_config)
+    ret, _, _ = self.Run(cmd)
+    return ret
+
+  def ParseGYPOutputPath(self, path):
+    assert(path.startswith('//'))
+    return path[2:]
+
+  def ParseGYPConfigPath(self, path):
+    assert(path.startswith('//'))
+    output_dir, _, config = path[2:].rpartition('/')
+    self.CheckGYPConfigIsSupported(config, path)
+    return output_dir, config
+
+  def CheckGYPConfigIsSupported(self, config, path):
+    if config not in ('Debug', 'Release'):
+      if (sys.platform in ('win32', 'cygwin') and
+          config not in ('Debug_x64', 'Release_x64')):
+        raise MBErr('Unknown or unsupported config type "%s" in "%s"' %
+                    config, path)
+
+  def GYPCmd(self, output_dir, gyp_defines, config):
+    gyp_defines = gyp_defines.replace("$(goma_dir)", self.args.goma_dir)
+    cmd = [
+        sys.executable,
+        os.path.join('build', 'gyp_chromium'),
+        '-G',
+        'output_dir=' + output_dir,
+        '-G',
+        'config=' + config,
+    ]
+    for d in shlex.split(gyp_defines):
+      cmd += ['-D', d]
+    return cmd
+
+  def RunGNAnalyze(self, vals):
+    # TODO: Skip this if a valid gn build already exists, which it should on
+    # the bots.
+    if self.args.dryrun:
+      tmpdir = '//out/$tmpdir'
+    else:
+      tmpdir = tempfile.mkdtemp(prefix='analyze', suffix='tmp',
+                                dir=os.path.join(self.chromium_src_dir, 'out'))
+
+    files = self.args.paths
+    try:
+      idx = files.index('-')
+      targets = files[idx + 1:]
+      files = files[:idx]
+    except ValueError:
+      targets = []
+
+    self.RunGN(tmpdir, vals['gn_args'])
+
+    cmd = ['gn', 'refs', tmpdir] + files + ['--type=executable',
+           '--all', '--as=output']
+    needed_targets = []
+    ret, out, _ = self.Run(cmd)
+
+    # TODO: handle failures from 'gn refs'
+
+    if self.args.dryrun or self.args.verbose:
+      self.Print('rm -fr', tmpdir)
+    if not self.args.dryrun:
+      shutil.rmtree(tmpdir, ignore_errors=True)
+
+    rpath = os.path.relpath(tmpdir, self.chromium_src_dir) + os.sep
+    needed_targets = [t.replace(rpath, '') for t in out.splitlines()]
+    if targets:
+      needed_targets = [nt for nt in needed_targets if nt in targets]
+
+    for nt in needed_targets:
+      self.Print(nt)
+
+  def PrintCmd(self, cmd):
+    if cmd[0] == sys.executable:
+      cmd = ['python'] + cmd[1:]
+    self.Print(*[pipes.quote(c) for c in cmd])
+
+  def Print(self, *args, **kwargs):
+    # This function largely exists so it can be overridden for testing.
+    print(*args, **kwargs)
+
+  def Run(self, cmd):
+    # This function largely exists so it can be overridden for testing.
+    if self.args.dryrun or self.args.verbose:
+      self.PrintCmd(cmd)
+    if self.args.dryrun:
+      return 0, '', ''
+    if self.args.verbose:
+      if out:
+        self.Print(out)
+      if err:
+        self.Print(err, file=sys.stderr)
+    return self.Call(cmd)
+
+  def Call(self, cmd):
+    p = subprocess.Popen(cmd, shell=False, cwd=self.chromium_src_dir,
+                         stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    out, err = p.communicate()
+    return p.returncode, out, err
+
+  def ExpandUser(self, path):
+    # This function largely exists so it can be overridden for testing.
+    return os.path.expanduser(path)
+
+  def Exists(self, path):
+    # This function largely exists so it can be overridden for testing.
+    return os.path.exists(path)
+
+  def ReadFile(self, path):
+    # This function largely exists so it can be overriden for testing.
+    with open(path) as fp:
+      return fp.read()
+
+
+class MBErr(Exception):
+  pass
+
+
+if __name__ == '__main__':
+  try:
+    sys.exit(main(sys.argv[1:]))
+  except Err as e:
+    print(e)
+    sys.exit(1)
+  except KeyboardInterrupt:
+    print("interrupted, exiting", stream=sys.stderr)
+    sys.exit(130)