#!/usr/bin/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.

"""This script runs "pub get" on all directories within the tree that have
pubspec.yaml files.

See https://www.dartlang.org/tools/pub/get-started.html for information about
the pub tool."""

import argparse
import os
import subprocess
import sys

def pub_get(dart_sdk_path, target_directory, upgrade):
  cmd = [
    os.path.join(dart_sdk_path, "bin/pub")
  ]
  if upgrade:
    cmd.extend(["upgrade"])
  else:
    cmd.extend(["get"])

  # Cache the downloaded pubs inside the repo to avoid the chance of multiple
  # simultaneous builds in different repos stomping on each other.
  env = os.environ.copy()
  env["PUB_CACHE"] = os.path.join(os.getcwd(), "dart-pub-cache")
  try:
      subprocess.check_output(cmd, shell=False,
                              stderr=subprocess.STDOUT,
                              cwd=target_directory,
                              env=env)
  except subprocess.CalledProcessError as e:
    print('Error running pub get in %s' % target_directory)
    print(e.output)
    raise e


def main(repository_root, dart_sdk_path, dirs_to_ignore, upgrade):
  os.chdir(repository_root)

  # Relativize dart_sdk_path to repository_root.
  dart_sdk_path_from_root = os.path.join(repository_root,
      os.path.relpath(dart_sdk_path, repository_root))

  cmd = ["git", "ls-files", "*/pubspec.yaml"]
  pubspec_yaml_files = subprocess.check_output(cmd,
                                               shell=False,
                                               stderr=subprocess.STDOUT)

  for f in pubspec_yaml_files.split():
    ignore = reduce(lambda x, y: x or f.startswith(y), dirs_to_ignore, False)
    if ignore:
      continue
    pub_get(dart_sdk_path_from_root, os.path.dirname(f), upgrade)


if __name__ == '__main__':
  parser = argparse.ArgumentParser(
      description="Run 'pub get' on all directories with checked-in "
                  "pubspec.yaml files")
  parser.add_argument("--repository-root",
                      metavar="<repository-root>",
                      type=str,
                      required=True,
                      help="Path to the root of the Git repository, "
                           "specified as a relative path from this directory.")
  parser.add_argument("--dart-sdk-directory",
                      metavar="<dart-sdk-directory>",
                      type=str,
                      required=True,
                      help="Path to the directory containing the Dart SDK, "
                           "specified as a relative path from this directory.")
  parser.add_argument("--dirs-to-ignore",
                      metavar="<dir>",
                      nargs="+",
                      default=[],
                      type=str,
                      help="Optional list of directories to ignore, specified "
                           "relative to the root of the repo. 'pub get' will "
                           "not be run for any subdirectories of these "
                           "directories.")
  parser.add_argument("--upgrade",
                      action="store_true",
                      default=False,
                      help="Upgrade pub package dependencies")
  args = parser.parse_args()
  _current_path = os.path.dirname(os.path.realpath(__file__))
  _repository_root = os.path.join(_current_path, args.repository_root)
  _dart_sdk_path = os.path.join(_current_path, args.dart_sdk_directory)
  sys.exit(
      main(_repository_root, _dart_sdk_path, args.dirs_to_ignore, args.upgrade))