1cb93a386Sopenharmony_ci#!/usr/bin/env python
2cb93a386Sopenharmony_ci#
3cb93a386Sopenharmony_ci# Copyright 2019 Google Inc.
4cb93a386Sopenharmony_ci#
5cb93a386Sopenharmony_ci# Use of this source code is governed by a BSD-style license that can be
6cb93a386Sopenharmony_ci# found in the LICENSE file.
7cb93a386Sopenharmony_ci
8cb93a386Sopenharmony_ci
9cb93a386Sopenharmony_ci"""Create the asset."""
10cb93a386Sopenharmony_ci
11cb93a386Sopenharmony_ci
12cb93a386Sopenharmony_ciimport argparse
13cb93a386Sopenharmony_ciimport os
14cb93a386Sopenharmony_ciimport subprocess
15cb93a386Sopenharmony_ciimport sys
16cb93a386Sopenharmony_ci
17cb93a386Sopenharmony_ciFILE_DIR = os.path.dirname(os.path.abspath(__file__))
18cb93a386Sopenharmony_ciINFRA_BOTS_DIR = os.path.realpath(os.path.join(FILE_DIR, os.pardir, os.pardir))
19cb93a386Sopenharmony_cisys.path.insert(0, INFRA_BOTS_DIR)
20cb93a386Sopenharmony_ciimport utils
21cb93a386Sopenharmony_ci
22cb93a386Sopenharmony_ci
23cb93a386Sopenharmony_ci# Remember to also update the go asset when this is updated.
24cb93a386Sopenharmony_ciGO_URL = "https://golang.org/dl/go1.16.3.windows-amd64.zip"
25cb93a386Sopenharmony_ci
26cb93a386Sopenharmony_ci
27cb93a386Sopenharmony_cidef create_asset(target_dir):
28cb93a386Sopenharmony_ci  """Create the asset."""
29cb93a386Sopenharmony_ci  with utils.tmp_dir():
30cb93a386Sopenharmony_ci    cwd = os.getcwd()
31cb93a386Sopenharmony_ci    zipfile = os.path.join(cwd, 'go.zip')
32cb93a386Sopenharmony_ci    subprocess.check_call(["wget", '-O', zipfile, GO_URL])
33cb93a386Sopenharmony_ci    subprocess.check_call(["unzip", zipfile, "-d", target_dir])
34cb93a386Sopenharmony_ci
35cb93a386Sopenharmony_ci
36cb93a386Sopenharmony_cidef main():
37cb93a386Sopenharmony_ci  parser = argparse.ArgumentParser()
38cb93a386Sopenharmony_ci  parser.add_argument('--target_dir', '-t', required=True)
39cb93a386Sopenharmony_ci  args = parser.parse_args()
40cb93a386Sopenharmony_ci  create_asset(args.target_dir)
41cb93a386Sopenharmony_ci
42cb93a386Sopenharmony_ci
43cb93a386Sopenharmony_ciif __name__ == '__main__':
44cb93a386Sopenharmony_ci  main()
45