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
20import xml.etree.ElementTree as ET
21
22from sa_info_process.merge_sa_info import JsonSAInfoMerger
23
24sys.path.append(
25    os.path.dirname(os.path.dirname(os.path.dirname(
26        os.path.abspath(__file__)))))
27from scripts.util.file_utils import write_json_file, \
28    read_json_file  # noqa: E402
29from scripts.util import build_utils  # noqa: E402
30
31
32def _get_src_sa_info(src_sa_install_info_file: str, depfiles: list):
33    src_sa_install_info = read_json_file(src_sa_install_info_file)
34    if src_sa_install_info is None:
35        raise Exception("read src_sa_install_info_file failed.")
36    install_info_file_list = []
37    for _install_info in src_sa_install_info:
38        if _install_info.get('type') == 'sa':
39            install_info_file_list.append(
40                _install_info.get('install_info_file'))
41
42    depfiles.extend(install_info_file_list)
43    json_all_sa_input_files = []
44    for _install_info_file in install_info_file_list:
45        _install_info = read_json_file(_install_info_file)
46        if _install_info is None:
47            raise Exception("read install_info_file '{}' failed.".format(
48                _install_info_file))
49        sa_info_files = _install_info.get('sa_info_files')
50        for part_sa_info_files in sa_info_files:
51            if str(part_sa_info_files).endswith(".xml"):
52                parser = ET.XMLParser()
53                tree = ET.parse(part_sa_info_files, parser)
54                root = tree.getroot()
55                process_nodes = root.findall('process')
56                process_name = process_nodes[0].text.strip()
57                raise Exception('Please use profile in json format, processname: {}'.format(process_name))
58            elif str(part_sa_info_files).endswith(".json"):
59                json_all_sa_input_files += [part_sa_info_files]
60    depfiles.extend(json_all_sa_input_files)
61    return json_all_sa_input_files
62
63
64def _sa_profile_merge(json_sa_input_files: str, no_src_subsystem_sa_zipfile: str,
65                      merge_out_dir: str, merged_zipfile: str, target_cpu: str):
66    with build_utils.temp_dir() as tmp:
67        build_utils.extract_all(no_src_subsystem_sa_zipfile, tmp)
68        for root, _, files in os.walk(tmp):
69            for sa_file in files:
70                json_sa_input_files.append(os.path.join(root, sa_file))
71
72        if not os.path.exists(merge_out_dir):
73            os.makedirs(merge_out_dir, exist_ok=True)
74
75        is_64bit_arch = target_cpu not in ["arm", "x86"]
76        # call merge tool
77        json_merge_tool = JsonSAInfoMerger()
78        json_result_file_list = json_merge_tool.merge(sorted(json_sa_input_files),
79                                            merge_out_dir)
80    build_utils.zip_dir(merged_zipfile, merge_out_dir)
81    shutil.rmtree(merge_out_dir)
82    return json_result_file_list
83
84
85def _generate_install_info(sa_result_file_list: list, sa_info_install_dest_dir: str,
86                           sa_install_info_file: str):
87    module_install_info_list = []
88    for _sa_file in sa_result_file_list:
89        _install_dest = os.path.join(sa_info_install_dest_dir,
90                                     os.path.basename(_sa_file))
91        module_install_info = {
92            'type': 'sa_info',
93            'source': _sa_file,
94            'install_enable': True,
95            'dest': [_install_dest]
96        }
97        module_install_info_list.append(module_install_info)
98    write_json_file(sa_install_info_file, module_install_info_list)
99
100
101def main():
102    parser = argparse.ArgumentParser()
103    parser.add_argument('--src-sa-install-info-file', required=True)
104    parser.add_argument('--no-src-sa-install-info-file', required=True)
105    parser.add_argument('--sa-output-dir', required=True)
106    parser.add_argument('--merged-sa-profile', required=True)
107    parser.add_argument('--sa-install-info-file', required=True)
108    parser.add_argument('--sa-info-install-dest-dir', required=True)
109    parser.add_argument('--target-cpu', default='arm64')
110    parser.add_argument('--depfile', required=True)
111    args = parser.parse_args()
112
113    depfiles = []
114    json_src_subsystem_file_list = _get_src_sa_info(args.src_sa_install_info_file,
115                                               depfiles)
116    no_src_sa_profile_zip = args.no_src_sa_install_info_file
117
118    result_file_list = _sa_profile_merge(json_src_subsystem_file_list,
119                                         no_src_sa_profile_zip,
120                                         args.sa_output_dir,
121                                         args.merged_sa_profile,
122                                         args.target_cpu)
123    _generate_install_info(result_file_list, args.sa_info_install_dest_dir,
124                           args.sa_install_info_file)
125    build_utils.write_depfile(args.depfile, args.sa_install_info_file,
126                              depfiles)
127    return 0
128
129
130if __name__ == '__main__':
131    sys.exit(main())
132