1*333d2b36SAndroid Build Coastguard Worker# Copyright (C) 2024 The Android Open Source Project 2*333d2b36SAndroid Build Coastguard Worker# 3*333d2b36SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License"); 4*333d2b36SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License. 5*333d2b36SAndroid Build Coastguard Worker# You may obtain a copy of the License at 6*333d2b36SAndroid Build Coastguard Worker# 7*333d2b36SAndroid Build Coastguard Worker# http://www.apache.org/licenses/LICENSE-2.0 8*333d2b36SAndroid Build Coastguard Worker# 9*333d2b36SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software 10*333d2b36SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS, 11*333d2b36SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12*333d2b36SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and 13*333d2b36SAndroid Build Coastguard Worker# limitations under the License. 14*333d2b36SAndroid Build Coastguard Worker 15*333d2b36SAndroid Build Coastguard Workerimport json 16*333d2b36SAndroid Build Coastguard Workerimport os 17*333d2b36SAndroid Build Coastguard Workerimport pathlib 18*333d2b36SAndroid Build Coastguard Workerimport sys 19*333d2b36SAndroid Build Coastguard Worker 20*333d2b36SAndroid Build Coastguard Worker 21*333d2b36SAndroid Build Coastguard Workerdef OpenModuleInfoFile(): 22*333d2b36SAndroid Build Coastguard Worker product_out = os.getenv("ANDROID_PRODUCT_OUT") 23*333d2b36SAndroid Build Coastguard Worker if not product_out: 24*333d2b36SAndroid Build Coastguard Worker if os.getenv("QUIET_VERIFYMODINFO") != "true": 25*333d2b36SAndroid Build Coastguard Worker sys.stderr.write("No ANDROID_PRODUCT_OUT. Try running 'lunch' first.\n") 26*333d2b36SAndroid Build Coastguard Worker sys.exit(1) 27*333d2b36SAndroid Build Coastguard Worker try: 28*333d2b36SAndroid Build Coastguard Worker return open(pathlib.Path(product_out) / "module-info.json") 29*333d2b36SAndroid Build Coastguard Worker except (FileNotFoundError, PermissionError): 30*333d2b36SAndroid Build Coastguard Worker if os.getenv("QUIET_VERIFYMODINFO") != "true": 31*333d2b36SAndroid Build Coastguard Worker sys.stderr.write("Could not find module-info.json. Please run 'refreshmod' first.\n") 32*333d2b36SAndroid Build Coastguard Worker sys.exit(1) 33*333d2b36SAndroid Build Coastguard Worker 34*333d2b36SAndroid Build Coastguard Worker 35*333d2b36SAndroid Build Coastguard Workerdef ReadModuleInfo(): 36*333d2b36SAndroid Build Coastguard Worker with OpenModuleInfoFile() as f: 37*333d2b36SAndroid Build Coastguard Worker return json.load(f) 38*333d2b36SAndroid Build Coastguard Worker 39*333d2b36SAndroid Build Coastguard Workerdef GetModule(modules, module_name): 40*333d2b36SAndroid Build Coastguard Worker if module_name not in modules: 41*333d2b36SAndroid Build Coastguard Worker sys.stderr.write(f"Could not find module '{module_name}' (try 'refreshmod' if there have been build changes?)\n") 42*333d2b36SAndroid Build Coastguard Worker sys.exit(1) 43*333d2b36SAndroid Build Coastguard Worker return modules[module_name] 44*333d2b36SAndroid Build Coastguard Worker 45