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 sys
17import os
18import argparse
19sys.path.append(
20    os.path.dirname(os.path.dirname(os.path.dirname(
21        os.path.abspath(__file__)))))
22from scripts.util.file_utils import read_json_file, write_json_file  # noqa: E402
23from scripts.util import build_utils  # noqa: E402
24
25
26def sa_info_process(sa_profile_infos_file: str, toolchain_platform: dict):
27    sa_profile_infos = read_json_file(sa_profile_infos_file)
28    if sa_profile_infos is None:
29        raise Exception("read file '{}' failed.".format(sa_profile_infos))
30    processed_sa_infos = []
31    for _sa_info in sa_profile_infos:
32        origin_part_name = _sa_info.get('part_name')
33        if origin_part_name is None or origin_part_name == '':
34            raise Exception(
35                "get part name failed in sa info, part_name='{}'".format(
36                    origin_part_name))
37        _toolchain = _sa_info.get('toolchain')
38        platform_name = toolchain_platform.get(_toolchain)
39        if platform_name != 'phone':
40            part_name = '{}_{}'.format(origin_part_name, platform_name)
41        else:
42            part_name = origin_part_name
43        _sa_info['part_name'] = part_name
44        _sa_info['origin_part_name'] = origin_part_name
45        processed_sa_infos.append(_sa_info)
46    return processed_sa_infos
47
48
49def main():
50    parser = argparse.ArgumentParser()
51    parser.add_argument('--variants-toolchain-file', required=True)
52    parser.add_argument('--sa-profile-infos-file', required=False)
53    parser.add_argument('--output-file', required=True)
54    parser.add_argument('--depfile', required=True)
55    args = parser.parse_args()
56
57    depfiles = [args.variants_toolchain_file]
58    variants_toolchain_info = read_json_file(args.variants_toolchain_file)
59    if variants_toolchain_info is None:
60        raise Exception(
61            "read file '{}' failed.".format(args.variants_toolchain_file))
62    toolchain_platform = variants_toolchain_info.get('toolchain_platform')
63    processed_sa_infos = sa_info_process(args.sa_profile_infos_file,
64                                         toolchain_platform)
65    write_json_file(args.output_file, processed_sa_infos)
66    build_utils.write_depfile(args.depfile, args.output_file, depfiles)
67    return 0
68
69
70if __name__ == '__main__':
71    sys.exit(main())
72