grpc/tools/buildgen/plugins/transitive_dependencies.py

# Copyright 2015, Google Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#     * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following disclaimer
# in the documentation and/or other materials provided with the
# distribution.
#     * Neither the name of Google Inc. nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

"""Buildgen transitive dependencies

This takes the list of libs, node_modules, and targets from our
yaml dictionary, and adds to each the transitive closure
of the list of dependencies.

"""

def get_lib(libs, name):
  try:
    return next(lib for lib in libs if lib['name']==name)
  except StopIteration:
    return None

def transitive_deps(lib, libs):
  if lib is not None and 'deps' in lib:
    # Recursively call transitive_deps on each dependency, and take the union
    return set.union(set(lib['deps']),
                     *[set(transitive_deps(get_lib(libs, dep), libs))
                       for dep in lib['deps']])
  else:
    return set()

def mako_plugin(dictionary):
  """The exported plugin code for transitive_dependencies.

  Each item in libs, node_modules, and targets can have a deps list.
  We add a transitive_deps property to each with the transitive closure
  of those dependency lists.
  """
  libs = dictionary.get('libs')
  node_modules = dictionary.get('node_modules')
  targets = dictionary.get('targets')

  for target_list in (libs, targets, node_modules):
    for target in target_list:
      target['transitive_deps'] = transitive_deps(target, libs)

  python_dependencies = dictionary.get('python_dependencies')
  python_dependencies['transitive_deps'] = (
      transitive_deps(python_dependencies, libs))