3pp/bazel_bootstrap/fetch.py

#!/usr/bin/env python3
# Copyright 2022 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import json
import os
import urllib.request


def do_latest():
  print(
      urllib.request.urlopen(
          'https://github.com/bazelbuild/bazel/releases/latest').geturl().split(
              '/')[-1])


_PLATFORMS = {
    'linux-amd64': 'linux-x86_64',
    'linux-arm64': 'linux-arm64',
    'mac-amd64': 'darwin-x86_64',
    'mac-arm64': 'darwin-arm64',
    'windows-amd64': 'windows-x86_64',
    'windows-arm64': 'windows-arm64',
}

_EXTENSION = {
    'linux': '',
    'mac': '',
    'windows': '.zip',
}


def get_download_url(version, platform):
  if platform not in _PLATFORMS:
    raise ValueError('unsupported platform {}'.format(platform))

  extension = _EXTENSION[platform.split('-')[0]]

  url = ('https://github.com/bazelbuild/bazel/releases/download/{version}/'
         'bazel-{version}-{platform}{extension}'.format(
             version=version,
             platform=_PLATFORMS[platform],
             extension=extension,
         ))

  manifest = {
      'url': [url],
      'ext': extension,
  }

  print(json.dumps(manifest))


def main():
  ap = argparse.ArgumentParser()
  sub = ap.add_subparsers(dest='action', required=True)

  latest = sub.add_parser("latest")
  latest.set_defaults(func=lambda _opts: do_latest())

  download = sub.add_parser("get_url")
  download.set_defaults(func=lambda opts: get_download_url(
      os.environ['_3PP_VERSION'], os.environ['_3PP_PLATFORM']))

  opts = ap.parse_args()
  opts.func(opts)


if __name__ == '__main__':
  main()