1 files changed, 84 insertions, 0 deletions
diff --git a/chromium/build/fuchsia/update_sdk.py b/chromium/build/fuchsia/update_sdk.py
new file mode 100755
index 00000000000..545103e7b1b
--- /dev/null
+++ b/chromium/build/fuchsia/update_sdk.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+# Copyright 2022 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.
+"""Check out the Fuchsia SDK from a given GCS path. Should be used in a
+'hooks_os' entry so that it only runs when .gclient's custom_vars includes
+'fuchsia'."""
+
+import argparse
+import logging
+import os
+import platform
+import subprocess
+import sys
+
+from common import GetHostOsFromPlatform, SDK_ROOT
+from update_images import DownloadAndUnpackFromCloudStorage, \
+                          MakeCleanDirectory
+
+
+def _GetHostArch():
+  host_arch = platform.machine()
+  # platform.machine() returns AMD64 on 64-bit Windows.
+  if host_arch in ['x86_64', 'AMD64']:
+    return 'amd64'
+  elif host_arch == 'aarch64':
+    return 'arm64'
+  raise Exception('Unsupported host architecture: %s' % host_arch)
+
+
+def _GetTarballPath(gcs_tarball_prefix: str) -> str:
+  """Get the full path to the sdk tarball on GCS"""
+  platform = GetHostOsFromPlatform()
+  arch = _GetHostArch()
+  return f'{gcs_tarball_prefix}/{platform}-{arch}/gn.tar.gz'
+
+
+def main():
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--cipd-prefix', help='CIPD base directory for the SDK.')
+  parser.add_argument('--version', help='Specifies the SDK version.')
+  parser.add_argument('--verbose',
+                      '-v',
+                      action='store_true',
+                      help='Enable debug-level logging.')
+  args = parser.parse_args()
+
+  logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO)
+
+  # Exit if there's no SDK support for this platform.
+  try:
+    host_plat = GetHostOsFromPlatform()
+  except:
+    logging.warning('Fuchsia SDK is not supported on this platform.')
+    return 0
+
+  sdk_override = os.path.join(os.path.dirname(__file__), 'sdk_override.txt')
+
+  # Download from CIPD if there is no override file.
+  if not os.path.isfile(sdk_override):
+    if not args.cipd_prefix:
+      parser.exit(1, '--cipd-prefix must be specified.')
+    if not args.version:
+      parser.exit(2, '--version must be specified.')
+    ensure_file = '%s%s-%s %s' % (args.cipd_prefix, host_plat, _GetHostArch(),
+                                  args.version)
+    subprocess.run(('cipd', 'ensure', '-ensure-file', '-', '-root', SDK_ROOT),
+                   check=True,
+                   input=ensure_file.encode('utf-8'))
+    return 0
+
+  with open(sdk_override, 'r') as f:
+    gcs_tarball_prefix = f.read()
+
+  # Always re-download the SDK.
+  logging.info('Downloading GN SDK...')
+  MakeCleanDirectory(SDK_ROOT)
+  DownloadAndUnpackFromCloudStorage(_GetTarballPath(gcs_tarball_prefix),
+                                    SDK_ROOT)
+  return 0
+
+
+if __name__ == '__main__':
+  sys.exit(main())