1#!/usr/bin/env python 2# -*- coding: utf-8 -*- 3# Copyright (c) 2021 Huawei Device Co., Ltd. 4# Licensed under the Apache License, Version 2.0 (the "License"); 5# you may not use this file except in compliance with the License. 6# You may obtain a copy of the License at 7# 8# http://www.apache.org/licenses/LICENSE-2.0 9# 10# Unless required by applicable law or agreed to in writing, software 11# distributed under the License is distributed on an "AS IS" BASIS, 12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13# See the License for the specific language governing permissions and 14# limitations under the License. 15 16import argparse 17import os 18import sys 19import shutil 20 21from sa_info_process.merge_sa_info import SAInfoMerger 22 23sys.path.append( 24 os.path.dirname(os.path.dirname(os.path.dirname( 25 os.path.abspath(__file__))))) 26from scripts.util.file_utils import write_json_file, \ 27 read_json_file # noqa: E402 28from scripts.util import build_utils # noqa: E402 29 30 31def _get_src_sa_info(src_sa_install_info_file, depfiles): 32 src_sa_install_info = read_json_file(src_sa_install_info_file) 33 if src_sa_install_info is None: 34 raise Exception("read src_sa_install_info_file failed.") 35 install_info_file_list = [] 36 for _install_info in src_sa_install_info: 37 if _install_info.get('type') == 'sa': 38 install_info_file_list.append( 39 _install_info.get('install_info_file')) 40 41 depfiles.extend(install_info_file_list) 42 all_sa_input_files = [] 43 for _install_info_file in install_info_file_list: 44 _install_info = read_json_file(_install_info_file) 45 if _install_info is None: 46 raise Exception("read install_info_file '{}' failed.".format( 47 _install_info_file)) 48 sa_info_files = _install_info.get('sa_info_files') 49 all_sa_input_files.extend(sa_info_files) 50 depfiles.extend(all_sa_input_files) 51 return all_sa_input_files 52 53 54def _sa_profile_merge(sa_input_files, no_src_subsystem_sa_zipfile, 55 merge_out_dir, merged_zipfile, target_cpu): 56 with build_utils.temp_dir() as tmp: 57 build_utils.extract_all(no_src_subsystem_sa_zipfile, tmp) 58 for root, _, files in os.walk(tmp): 59 for sa_file in files: 60 sa_input_files.append(os.path.join(root, sa_file)) 61 62 if not os.path.exists(merge_out_dir): 63 os.makedirs(merge_out_dir, exist_ok=True) 64 65 is_64bit_arch = target_cpu not in ["arm", "x86"] 66 # call merge tool 67 merge_tool = SAInfoMerger(is_64bit_arch) 68 result_file_list = merge_tool.merge(sorted(sa_input_files), 69 merge_out_dir) 70 build_utils.zip_dir(merged_zipfile, merge_out_dir) 71 shutil.rmtree(merge_out_dir) 72 return result_file_list 73 74 75def _generate_install_info(sa_result_file_list, sa_info_install_dest_dir, 76 sa_install_info_file): 77 module_install_info_list = [] 78 for _sa_file in sa_result_file_list: 79 _install_dest = os.path.join(sa_info_install_dest_dir, 80 os.path.basename(_sa_file)) 81 module_install_info = { 82 'type': 'sa_info', 83 'source': _sa_file, 84 'install_enable': True, 85 'dest': [_install_dest] 86 } 87 module_install_info_list.append(module_install_info) 88 write_json_file(sa_install_info_file, module_install_info_list) 89 90 91def main(): 92 parser = argparse.ArgumentParser() 93 parser.add_argument('--src-sa-install-info-file', required=True) 94 parser.add_argument('--no-src-sa-install-info-file', required=True) 95 parser.add_argument('--sa-output-dir', required=True) 96 parser.add_argument('--merged-sa-profile', required=True) 97 parser.add_argument('--sa-install-info-file', required=True) 98 parser.add_argument('--sa-info-install-dest-dir', required=True) 99 parser.add_argument('--target-cpu', default='arm64') 100 parser.add_argument('--depfile', required=True) 101 args = parser.parse_args() 102 103 depfiles = [] 104 src_subsystem_file_list = _get_src_sa_info(args.src_sa_install_info_file, 105 depfiles) 106 no_src_sa_profile_zip = args.no_src_sa_install_info_file 107 108 result_file_list = _sa_profile_merge(src_subsystem_file_list, 109 no_src_sa_profile_zip, 110 args.sa_output_dir, 111 args.merged_sa_profile, 112 args.target_cpu) 113 _generate_install_info(result_file_list, args.sa_info_install_dest_dir, 114 args.sa_install_info_file) 115 build_utils.write_depfile(args.depfile, args.sa_install_info_file, 116 depfiles) 117 return 0 118 119 120if __name__ == '__main__': 121 sys.exit(main()) 122