License generation script for build_aar.py.

tools_webrtc/generate_licenses.py

Index: tools_webrtc/generate_licenses.py
diff --git a/tools_webrtc/ios/generate_licenses.py b/tools_webrtc/generate_licenses.py
similarity index 65%
rename from tools_webrtc/ios/generate_licenses.py
rename to tools_webrtc/generate_licenses.py
index b91314783f30407920966c549107e103f3cba337..2f2115d79d8c39ace28f4b420c17d9c6dd2ed741 100755
--- a/tools_webrtc/ios/generate_licenses.py
+++ b/tools_webrtc/generate_licenses.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python
+#!/usr/bin/env python
 
 #  Copyright 2016 The WebRTC project authors. All Rights Reserved.
 #
@@ -8,7 +8,7 @@
 #  in the file PATENTS.  All contributing project authors may
 #  be found in the AUTHORS file in the root of the source tree.
 
-"""Generates license HTML for a prebuilt version of WebRTC for iOS."""
+"""Generates license markdown for a prebuilt version of WebRTC."""
 
 import sys
 
@@ -16,53 +16,58 @@ import argparse
 import cgi
 import os
 import re
-import textwrap
 import subprocess
 
 
 LIB_TO_LICENSES_DICT = {
+    'android_tools': ['third_party/android_tools/LICENSE'],
     'boringssl': ['third_party/boringssl/src/LICENSE'],
     'expat': ['third_party/expat/files/COPYING'],
+    'ijar': ['third_party/ijar/LICENSE'],
     'jsoncpp': ['third_party/jsoncpp/LICENSE'],
+    'libc++': ['buildtools/third_party/libc++/trunk/LICENSE.TXT'],
+    'libc++abi': ['buildtools/third_party/libc++abi/trunk/LICENSE.TXT'],
+    'libevent': ['base/third_party/libevent/LICENSE'],
+    'libjpeg_turbo': ['third_party/libjpeg_turbo/LICENSE.md'],
+    'libsrtp': ['third_party/libsrtp/LICENSE'],
+    'libvpx': ['third_party/libvpx/source/libvpx/LICENSE'],
+    'libyuv': ['third_party/libyuv/LICENSE'],
+    'openmax_dl': ['third_party/openmax_dl/LICENSE'],
     'opus': ['third_party/opus/src/COPYING'],
+    'ow2_asm': ['third_party/ow2_asm/LICENSE'],

[kjellander_webrtc, 2017/08/31 09:45:21]

AFAIK this is only used by tests, so it shouldn't really be included since this
is only building production code, right?

See https://cs.chromium.org/search/?q=ow2_asm+f:%5C.gn&type=cs

[sakal, 2017/08/31 14:26:58]

I modified build_aar.py to not include test targets and now it is no longer
required.

     'protobuf': ['third_party/protobuf/LICENSE'],
-    'libsrtp': ['third_party/libsrtp/LICENSE'],
     'usrsctp': ['third_party/usrsctp/LICENSE'],
     'webrtc': ['webrtc/LICENSE', 'webrtc/LICENSE_THIRD_PARTY'],
-    'libvpx': ['third_party/libvpx/source/libvpx/LICENSE'],
-    'libyuv': ['third_party/libyuv/LICENSE'],
 }
 
 SCRIPT_DIR = os.path.dirname(os.path.realpath(sys.argv[0]))
 CHECKOUT_ROOT = os.path.abspath(os.path.join(SCRIPT_DIR, os.pardir, os.pardir))
 WEBRTC_ROOT = os.path.join(CHECKOUT_ROOT, 'webrtc')
+THIRD_PARTY_DEP_REGEX = r'^\s*//.*third_party/([^:/]+).*$'
 
 
 def GetThirdPartyLibraries(buildfile_dir, target_name):
-  def ExtractLibName(string_list):
-    # Sample input:
-    # ["   //third_party/usrsctp:usrsctp", "    //webrtc:webrtc_common"]
-    # Sample output:
-    # ["usrsctp"]
-    return re.sub(r'\(.*\)', '', string_list).strip().split(
-        os.path.sep)[-1].split(':')[0]
   output = subprocess.check_output(
     ["gn", "desc", buildfile_dir, target_name, '--all']) .split(os.linesep)
-  return [ExtractLibName(x) for x in output if re.search(r'third_party', x)]
+  matches = [re.match(THIRD_PARTY_DEP_REGEX, line) for line in output]
+  libraries = [match.group(1) for match in matches if match] # Filter out Nones
+  return libraries
 
 
 class LicenseBuilder(object):
 
-  def __init__(self, buildfile_dirs, target_name):
+  def __init__(self, buildfile_dirs, targets):
     self.buildfile_dirs = buildfile_dirs
-    self.target_name = target_name
+    self.targets = targets
 
   def GenerateLicenseText(self, output_dir):
     # Get a list of third_party libs from gn. For fat libraries we must consider
     # all architectures, hence the multiple buildfile directories.
     # The `sum` function flattens the 2d list.
-    third_party_libs = sum([GetThirdPartyLibraries(buildfile, self.target_name)
-                            for buildfile in self.buildfile_dirs], [])
+    third_party_libs = sum((
+        sum((GetThirdPartyLibraries(buildfile, target)
+            for buildfile in self.buildfile_dirs), [])
+                for target in self.targets), [])
     assert len(third_party_libs) > 0
 
     # Generate amalgamated list of libraries. Will exit with error if a
@@ -82,49 +87,33 @@ class LicenseBuilder(object):
     license_libs = sorted(license_libs)
     license_libs.insert(0, 'webrtc')
 
-    # Generate HTML.
-    output_license_file = open(os.path.join(output_dir, 'LICENSE.html'), 'w+')
-    output_license_file.write('<!DOCTYPE html>\n')
-    output_license_file.write('<html>\n<head>\n')
-    output_license_file.write('<meta charset="UTF-8">\n')
-    output_license_file.write('<title>Licenses</title>\n')
-    style_tag = textwrap.dedent('''\
-    <style>
-      body { margin: 0; font-family: sans-serif; }
-      pre { background-color: #eeeeee; padding: 1em; white-space: pre-wrap; }
-      p { margin: 1em; white-space: nowrap; }
-    </style>
-    ''')
-    output_license_file.write(style_tag)
-    output_license_file.write('</head>\n')
-
+    # Generate markdown.
+    output_license_file = open(os.path.join(output_dir, 'LICENSE.md'), 'w+')
     for license_lib in license_libs:
-      output_license_file.write('<p>%s<br/></p>\n' % license_lib)
-      output_license_file.write('<pre>\n')
+      output_license_file.write('# %s\n' % license_lib)
+      output_license_file.write('```\n')
       for path in LIB_TO_LICENSES_DICT[license_lib]:
         license_path = os.path.join(CHECKOUT_ROOT, path)
         with open(license_path, 'r') as license_file:
           license_text = cgi.escape(license_file.read(), quote=True)
           output_license_file.write(license_text)
           output_license_file.write('\n')
-      output_license_file.write('</pre>\n')
+      output_license_file.write('```\n\n')
 
-    output_license_file.write('</body>\n')
-    output_license_file.write('</html>')
     output_license_file.close()
     return 0
 
 
 def main():
-  parser = argparse.ArgumentParser(description='Generate WebRTC LICENSE.html')
-  parser.add_argument('target_name',
+  parser = argparse.ArgumentParser(description='Generate WebRTC LICENSE.md')
+  parser.add_argument('--target', required=True, action='append', default=[],
                       help='Name of the GN target to generate a license for')
   parser.add_argument('output_dir',
-                      help='Directory to output LICENSE.html to.')
+                      help='Directory to output LICENSE.md to.')
   parser.add_argument('buildfile_dirs', nargs="+",
                       help='Directories containing gn generated ninja files')
   args = parser.parse_args()
-  builder = LicenseBuilder(args.buildfile_dirs, args.target_name)
+  builder = LicenseBuilder(args.buildfile_dirs, args.target)
   sys.exit(builder.GenerateLicenseText(args.output_dir))