Start including sdk_ext sources in sky dart-pkg

mojo/public/tools/dart_pkg.py

 #!/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.
 
 """Utility for dart_pkg and dart_pkg_app rules"""
 
 import argparse
 import errno
@@ skipping 87 matching lines @@
         dirs[:] = filter(wrapped_filter, dirs)
 
 
 def copy_or_link(from_root, to_root, filter_func=None):
     if USE_LINKS:
         link(from_root, to_root)
     else:
         copy(from_root, to_root, filter_func)
 
 
+def list_files(from_root, filter_func=None):
+    file_list = []
+    for root, dirs, files in os.walk(from_root):
+        # filter_func expects paths not names, so wrap it to make them absolute.
+        wrapped_filter = None
+        if filter_func:
+            wrapped_filter = lambda name: filter_func(os.path.join(root, name))
+        for name in filter(wrapped_filter, files):
+            path = os.path.join(root, name)
+            file_list.append(path)
+        dirs[:] = filter(wrapped_filter, dirs)
+    return file_list
+
+
 def remove_broken_symlink(path):
     try:
         link_path = os.readlink(path)
     except OSError as e:
         # Path was not a symlink.
         if e.errno == errno.EINVAL:
             pass
     else:
         if not os.path.exists(link_path):
             os.unlink(path)
@@ skipping 43 matching lines @@
                         required=True)
     parser.add_argument('--package-sources',
                         metavar='package_sources',
                         help='Package sources',
                         nargs='+')
     parser.add_argument('--mojom-sources',
                         metavar='mojom_sources',
                         help='.mojom and .mojom.dart sources',
                         nargs='*',
                         default=[])
+    parser.add_argument('--sdk-ext-directories',
+                        metavar='sdk_ext_directories',
+                        help='Directory containing .dart sources',
+                        nargs='*',
+                        default=[])
     args = parser.parse_args()
 
     # We must have a pubspec.yaml.
     assert has_pubspec_yaml(args.package_sources)
 
     # Copy or symlink package sources into pkg directory.
     target_dir = os.path.join(args.pkg_directory, args.package_name)
     common_source_prefix = os.path.commonprefix(args.package_sources)
     for source in args.package_sources:
         relative_source = os.path.relpath(source, common_source_prefix)
         target = os.path.join(target_dir, relative_source)
         copy_or_link(source, target)
 
+    # Copy sdk-ext sources into pkg directory
+    sdk_ext_dir = os.path.join(target_dir, 'sdk_ext')
+    for directory in args.sdk_ext_directories:
+        sdk_ext_sources = list_files(directory, dart_filter)
+        common_prefix = os.path.commonprefix(sdk_ext_sources)
+        for source in sdk_ext_sources:
+            relative_source = os.path.relpath(source, common_prefix)
+            target = os.path.join(sdk_ext_dir, relative_source)
+            copy_or_link(source, target)
+
     lib_path = os.path.join(target_dir, "lib")
     lib_mojom_path = os.path.join(lib_path, "mojom")
 
     # Copy mojom sources
     for mojom_source_path in args.mojom_sources:
         path = mojom_path(mojom_source_path)
         copy(mojom_source_path, os.path.join(lib_mojom_path, path))
 
     # Copy generated mojom.dart files.
     generated_mojom_lib_path = os.path.join(args.gen_directory, "mojom/lib")
@@ skipping 10 matching lines @@
     # Remove any broken symlinks in target_dir and package root.
     remove_broken_symlinks(target_dir)
     remove_broken_symlinks(args.package_root)
 
     # Write stamp file.
     with open(args.stamp_file, 'w'):
         pass
 
 if __name__ == '__main__':
     sys.exit(main())