1*c8dee2aaSAndroid Build Coastguard Worker#!/usr/bin/env python3 2*c8dee2aaSAndroid Build Coastguard Worker# 3*c8dee2aaSAndroid Build Coastguard Worker# Copyright 2024 Google LLC 4*c8dee2aaSAndroid Build Coastguard Worker# 5*c8dee2aaSAndroid Build Coastguard Worker# Use of this source code is governed by a BSD-style license that can be 6*c8dee2aaSAndroid Build Coastguard Worker# found in the LICENSE file. 7*c8dee2aaSAndroid Build Coastguard Worker 8*c8dee2aaSAndroid Build Coastguard Worker 9*c8dee2aaSAndroid Build Coastguard Worker"""Create the asset.""" 10*c8dee2aaSAndroid Build Coastguard Worker 11*c8dee2aaSAndroid Build Coastguard Worker 12*c8dee2aaSAndroid Build Coastguard Workerimport argparse 13*c8dee2aaSAndroid Build Coastguard Workerimport os 14*c8dee2aaSAndroid Build Coastguard Workerimport subprocess 15*c8dee2aaSAndroid Build Coastguard Workerimport tempfile 16*c8dee2aaSAndroid Build Coastguard Worker 17*c8dee2aaSAndroid Build Coastguard WorkerURL = 'https://github.com/instrumenta/kubeval/releases/download/v0.16.1/kubeval-linux-amd64.tar.gz' 18*c8dee2aaSAndroid Build Coastguard WorkerSHA256 = '2d6f9bda1423b93787fa05d9e8dfce2fc1190fefbcd9d0936b9635f3f78ba790' 19*c8dee2aaSAndroid Build Coastguard Worker 20*c8dee2aaSAndroid Build Coastguard WorkerBINARY = URL.split('/')[-1] 21*c8dee2aaSAndroid Build Coastguard Worker 22*c8dee2aaSAndroid Build Coastguard Worker 23*c8dee2aaSAndroid Build Coastguard Workerdef create_asset(target_dir): 24*c8dee2aaSAndroid Build Coastguard Worker """Create the asset.""" 25*c8dee2aaSAndroid Build Coastguard Worker tmp = tempfile.mkdtemp() 26*c8dee2aaSAndroid Build Coastguard Worker target_file = os.path.join(tmp, 'kubeval.tar.gz') 27*c8dee2aaSAndroid Build Coastguard Worker subprocess.call(['wget', '--quiet', '--output-document', target_file, URL]) 28*c8dee2aaSAndroid Build Coastguard Worker output = subprocess.check_output(['sha256sum', target_file], encoding='utf-8') 29*c8dee2aaSAndroid Build Coastguard Worker actual_hash = output.split(' ')[0] 30*c8dee2aaSAndroid Build Coastguard Worker if actual_hash != SHA256: 31*c8dee2aaSAndroid Build Coastguard Worker raise Exception('SHA256 does not match (%s != %s)' % (actual_hash, SHA256)) 32*c8dee2aaSAndroid Build Coastguard Worker subprocess.check_call( 33*c8dee2aaSAndroid Build Coastguard Worker ['tar', 'xf', os.path.join(tmp, 'kubeval.tar.gz'), 'kubeval'], 34*c8dee2aaSAndroid Build Coastguard Worker cwd=target_dir) 35*c8dee2aaSAndroid Build Coastguard Worker subprocess.check_call(['rm', '-rf', tmp]) 36*c8dee2aaSAndroid Build Coastguard Worker 37*c8dee2aaSAndroid Build Coastguard Worker 38*c8dee2aaSAndroid Build Coastguard Workerdef main(): 39*c8dee2aaSAndroid Build Coastguard Worker parser = argparse.ArgumentParser() 40*c8dee2aaSAndroid Build Coastguard Worker parser.add_argument('--target_dir', '-t', required=True) 41*c8dee2aaSAndroid Build Coastguard Worker args = parser.parse_args() 42*c8dee2aaSAndroid Build Coastguard Worker create_asset(args.target_dir) 43*c8dee2aaSAndroid Build Coastguard Worker 44*c8dee2aaSAndroid Build Coastguard Worker 45*c8dee2aaSAndroid Build Coastguard Workerif __name__ == '__main__': 46*c8dee2aaSAndroid Build Coastguard Worker main() 47*c8dee2aaSAndroid Build Coastguard Worker 48