1*c8dee2aaSAndroid Build Coastguard Worker#!/usr/bin/env python3 2*c8dee2aaSAndroid Build Coastguard Worker 3*c8dee2aaSAndroid Build Coastguard Worker# Copyright 2023 Google Inc. 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 Workerimport hashlib 9*c8dee2aaSAndroid Build Coastguard Workerimport json 10*c8dee2aaSAndroid Build Coastguard Workerimport os 11*c8dee2aaSAndroid Build Coastguard Workerimport platform 12*c8dee2aaSAndroid Build Coastguard Workerimport re 13*c8dee2aaSAndroid Build Coastguard Workerimport stat 14*c8dee2aaSAndroid Build Coastguard Workerimport sys 15*c8dee2aaSAndroid Build Coastguard Workerimport tempfile 16*c8dee2aaSAndroid Build Coastguard Workerimport zipfile 17*c8dee2aaSAndroid Build Coastguard Worker 18*c8dee2aaSAndroid Build Coastguard Workerfrom urllib.request import urlopen 19*c8dee2aaSAndroid Build Coastguard Worker 20*c8dee2aaSAndroid Build Coastguard WorkerTESTDATA_REL_DIR = 'third_party/externals/googlefonts_testdata' 21*c8dee2aaSAndroid Build Coastguard WorkerZIP_ARCHIVE_NAME = 'googlefonts_testdata.zip' 22*c8dee2aaSAndroid Build Coastguard WorkerVERSION_FILE = 'googlefonts_testdata.version' 23*c8dee2aaSAndroid Build Coastguard Worker 24*c8dee2aaSAndroid Build Coastguard Workerdef sha256sum(path): 25*c8dee2aaSAndroid Build Coastguard Worker try: 26*c8dee2aaSAndroid Build Coastguard Worker with open(path, 'rb') as f: 27*c8dee2aaSAndroid Build Coastguard Worker return hashlib.sha256(f.read()).hexdigest() 28*c8dee2aaSAndroid Build Coastguard Worker except OSError: 29*c8dee2aaSAndroid Build Coastguard Worker return '' 30*c8dee2aaSAndroid Build Coastguard Worker 31*c8dee2aaSAndroid Build Coastguard Worker 32*c8dee2aaSAndroid Build Coastguard Workeros.chdir(os.path.join(os.path.dirname(__file__), os.pardir)) 33*c8dee2aaSAndroid Build Coastguard Worker 34*c8dee2aaSAndroid Build Coastguard Workerwith open('DEPS', 'rb') as f: 35*c8dee2aaSAndroid Build Coastguard Worker deps = f.read().decode() 36*c8dee2aaSAndroid Build Coastguard Workerfound = re.findall(r"'googlefonts_testdata_version':\s*'(\S+)'", deps) 37*c8dee2aaSAndroid Build Coastguard Workerif len(found) != 1: 38*c8dee2aaSAndroid Build Coastguard Worker print('Unable to find desired revision in DEPS', file=sys.stderr) 39*c8dee2aaSAndroid Build Coastguard Worker exit(1) 40*c8dee2aaSAndroid Build Coastguard Workerdesired_version = found[0] 41*c8dee2aaSAndroid Build Coastguard Worker 42*c8dee2aaSAndroid Build Coastguard Worker# Determine which version (if any) we currently have. 43*c8dee2aaSAndroid Build Coastguard Workerzip_archive_path = os.path.join(TESTDATA_REL_DIR, ZIP_ARCHIVE_NAME) 44*c8dee2aaSAndroid Build Coastguard Workercurrent_sha256 = sha256sum(zip_archive_path) 45*c8dee2aaSAndroid Build Coastguard Workerarchive_version_path = os.path.join(TESTDATA_REL_DIR, VERSION_FILE) 46*c8dee2aaSAndroid Build Coastguard Worker 47*c8dee2aaSAndroid Build Coastguard Workercurrent_version = { 48*c8dee2aaSAndroid Build Coastguard Worker 'version': '', 49*c8dee2aaSAndroid Build Coastguard Worker 'sha256': '', 50*c8dee2aaSAndroid Build Coastguard Worker} 51*c8dee2aaSAndroid Build Coastguard Workertry: 52*c8dee2aaSAndroid Build Coastguard Worker with open(archive_version_path, 'r', encoding='utf8') as f: 53*c8dee2aaSAndroid Build Coastguard Worker current_version = json.load(f) 54*c8dee2aaSAndroid Build Coastguard Workerexcept OSError: 55*c8dee2aaSAndroid Build Coastguard Worker pass 56*c8dee2aaSAndroid Build Coastguard Worker 57*c8dee2aaSAndroid Build Coastguard Workerif desired_version != current_version['version']: 58*c8dee2aaSAndroid Build Coastguard Worker print('Version "%s" requested by DEPS differs from current version "%s"' % ( 59*c8dee2aaSAndroid Build Coastguard Worker desired_version, current_version['version'])) 60*c8dee2aaSAndroid Build Coastguard Workerelif current_sha256 != current_version['sha256']: 61*c8dee2aaSAndroid Build Coastguard Worker print('sha256 sum "%s" does not match last-downloaded version "%s"' % ( 62*c8dee2aaSAndroid Build Coastguard Worker current_sha256, current_version['sha256'])) 63*c8dee2aaSAndroid Build Coastguard Workerelse: 64*c8dee2aaSAndroid Build Coastguard Worker print('Already up to date.') 65*c8dee2aaSAndroid Build Coastguard Worker exit(0) 66*c8dee2aaSAndroid Build Coastguard Worker 67*c8dee2aaSAndroid Build Coastguard Workerprint('Fetching %s at %s.' % (ZIP_ARCHIVE_NAME, desired_version)) 68*c8dee2aaSAndroid Build Coastguard Worker 69*c8dee2aaSAndroid Build Coastguard Workeros.makedirs(os.path.dirname(zip_archive_path), exist_ok=True) 70*c8dee2aaSAndroid Build Coastguard Workerwith open(zip_archive_path, 'wb') as f: 71*c8dee2aaSAndroid Build Coastguard Worker url = f'https://chrome-infra-packages.appspot.com/dl/chromium/third_party/googlefonts_testdata/+/{desired_version}' 72*c8dee2aaSAndroid Build Coastguard Worker f.write(urlopen(url).read()) 73*c8dee2aaSAndroid Build Coastguard Worker 74*c8dee2aaSAndroid Build Coastguard Workerwith zipfile.ZipFile(zip_archive_path, 'r') as f: 75*c8dee2aaSAndroid Build Coastguard Worker f.extractall(TESTDATA_REL_DIR) 76*c8dee2aaSAndroid Build Coastguard Worker 77*c8dee2aaSAndroid Build Coastguard Worker# Write the downloaded version info to a file. 78*c8dee2aaSAndroid Build Coastguard Workercurrent_version['version'] = desired_version 79*c8dee2aaSAndroid Build Coastguard Workercurrent_version['sha256'] = sha256sum(zip_archive_path) 80*c8dee2aaSAndroid Build Coastguard Workerwith open(archive_version_path, 'w', encoding='utf8') as f: 81*c8dee2aaSAndroid Build Coastguard Worker json.dump(current_version, f, sort_keys=True, indent=2) 82