1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3# Copyright (c) 2024 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 json
18import os
19import time
20import stat
21import utils
22
23
24def _get_args():
25    parser = argparse.ArgumentParser(add_help=True)
26    parser.add_argument(
27        "-p", "--input_path",
28        default=r"./", type=str,
29        help="Path of source code",
30    )
31    parser.add_argument(
32        "-rp", "--root_path",
33        default=r"./", type=str,
34        help="Path of root",
35    )
36    parser.add_argument(
37        "-t", "--test",
38        default=1, type=int,
39        help="whether the target contains test type. default 0 , choices: 0 or 1 2",
40    )
41    args = parser.parse_args()
42    return args
43
44
45def _judge_type(element, deps_list: list):
46    if isinstance(element, dict):
47        for k, v in element.items():
48            _judge_type(v, deps_list)
49    elif isinstance(element, list):
50        for v in element:
51            _judge_type(v, deps_list)
52    elif isinstance(element, str):
53        deps_list.append(element)
54
55
56def _inner_kits_name(inner_kits_list, deps_list):
57    if inner_kits_list:
58        for k in inner_kits_list:
59            deps_list.append(k['name'])
60
61
62def _output_build_gn(deps_list, output_path, _test_check):
63    file_name = os.path.join(output_path, 'BUILD.gn')
64    if os.path.exists(file_name):
65        os.remove(file_name)
66    flags = os.O_WRONLY | os.O_CREAT
67    modes = stat.S_IWUSR | stat.S_IRUSR
68    with os.fdopen(os.open(file_name, flags, modes), 'w') as f:
69        f.write('import("//build/ohos_var.gni")\n')
70        f.write('\n')
71        f.write('group("default") {\n')
72        if _test_check:
73            f.write('    testonly = true\n')
74        f.write('    deps = [\n')
75        for i in deps_list:
76            f.write(f"        \"{i}\",\n")
77        f.write('    ]\n')
78        f.write('}\n')
79
80
81def _get_bundle_path(source_code_path):
82    bundle_paths = dict()
83    for root, dirs, files in os.walk(source_code_path):
84        for file in files:
85            if file.endswith("bundle.json"):
86                bundle_paths.update({os.path.join(root, file): root})
87    return bundle_paths
88
89
90def _get_src_part_name(src_bundle_paths):
91    _name = ''
92    _path = ''
93    _bundle_path = ''
94    for src_bundle_path, v_path in src_bundle_paths.items():
95        src_bundle_json = utils.get_json(src_bundle_path)
96        part_name = ''
97        try:
98            part_name = src_bundle_json['component']['name']
99        except KeyError:
100            print(f'--get bundle json component name error--')
101        if part_name.endswith('_lite'):
102            pass
103        else:
104            _name = part_name
105            _bundle_path = src_bundle_path
106            _path = v_path
107    return _bundle_path, _path
108
109
110def _target_handle(ele, build_data, deps_list, _test_check):
111    if ele not in ['inner_kits', 'test', 'inner_api']:
112        _judge_type(build_data[ele], deps_list)
113    elif ele in ['inner_kits', 'inner_api']:
114        inner_kits_list = build_data[ele]
115        _inner_kits_name(inner_kits_list, deps_list)
116
117    if _test_check == 1 and ele == 'test':
118        inner_kits_list = build_data[ele]
119        for k in inner_kits_list:
120            deps_list.append(k)
121
122
123def main():
124    args = _get_args()
125    source_code_path = args.input_path
126    _test_check = args.test
127    deps_list = list()
128    bundle_paths = _get_bundle_path(source_code_path)
129    _bundle_path, dir_path = _get_src_part_name(bundle_paths)
130    bundle_json = utils.get_json(_bundle_path)
131    build_data = dict()
132    try:
133        build_data = bundle_json["component"]["build"]
134    except KeyError:
135        print(f'--get bundle json component build dict error--')
136    if _test_check != 2:
137        for ele in build_data.keys():
138            _target_handle(ele, build_data, deps_list, _test_check)
139    elif _test_check == 2:
140        inner_kits_list = build_data.get('test')
141        deps_list = []
142        if inner_kits_list:
143            for k in inner_kits_list:
144                deps_list.append(k)
145        else:
146            deps_list = []
147
148    output_path = os.path.join(args.root_path, 'out')
149    _output_build_gn(deps_list, output_path, _test_check)
150
151
152if __name__ == '__main__':
153    main()
154