1b8a72a62Sopenharmony_ci#!/usr/bin/env python
2b8a72a62Sopenharmony_ci# -*- coding: utf-8 -*-
3b8a72a62Sopenharmony_ci# Copyright (c) 2022-2024 Huawei Device Co., Ltd.
4b8a72a62Sopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License");
5b8a72a62Sopenharmony_ci# you may not use this file except in compliance with the License.
6b8a72a62Sopenharmony_ci# You may obtain a copy of the License at
7b8a72a62Sopenharmony_ci#
8b8a72a62Sopenharmony_ci#     http://www.apache.org/licenses/LICENSE-2.0
9b8a72a62Sopenharmony_ci#
10b8a72a62Sopenharmony_ci# Unless required by applicable law or agreed to in writing, software
11b8a72a62Sopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS,
12b8a72a62Sopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13b8a72a62Sopenharmony_ci# See the License for the specific language governing permissions and
14b8a72a62Sopenharmony_ci# limitations under the License.
15b8a72a62Sopenharmony_ci"""
16b8a72a62Sopenharmony_ciArchives a set of files.
17b8a72a62Sopenharmony_ci"""
18b8a72a62Sopenharmony_ci
19b8a72a62Sopenharmony_ci
20b8a72a62Sopenharmony_ciimport os
21b8a72a62Sopenharmony_ciimport sys
22b8a72a62Sopenharmony_ciimport argparse
23b8a72a62Sopenharmony_ciimport subprocess
24b8a72a62Sopenharmony_ciimport re
25b8a72a62Sopenharmony_ci
26b8a72a62Sopenharmony_ci
27b8a72a62Sopenharmony_cidef run_cmd(title, cmd, time_out):
28b8a72a62Sopenharmony_ci    child_process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
29b8a72a62Sopenharmony_ci    cmd_out, cmd_err = child_process.communicate(timeout=time_out)
30b8a72a62Sopenharmony_ci    if child_process.returncode != 0:
31b8a72a62Sopenharmony_ci        print(cmd_out.decode('utf-8'))
32b8a72a62Sopenharmony_ci        print(cmd_err.decode('utf-8'))
33b8a72a62Sopenharmony_ci        raise Exception('compile module: ' + title + ' failed!')
34b8a72a62Sopenharmony_ci
35b8a72a62Sopenharmony_ci
36b8a72a62Sopenharmony_cidef get_compile_str(src_path, java_sources):
37b8a72a62Sopenharmony_ci    compile_java = ''
38b8a72a62Sopenharmony_ci    for src in java_sources:
39b8a72a62Sopenharmony_ci        if len(compile_java) == 0:
40b8a72a62Sopenharmony_ci            compile_java = '{}'.format(os.path.join(src_path, src))
41b8a72a62Sopenharmony_ci        else:
42b8a72a62Sopenharmony_ci            compile_java = '{} {}'.format(compile_java, os.path.join(src_path, src))
43b8a72a62Sopenharmony_ci    return compile_java
44b8a72a62Sopenharmony_ci
45b8a72a62Sopenharmony_ci
46b8a72a62Sopenharmony_cidef compile_haptobin_tool(root_path, src_path, jar_output, out_path,
47b8a72a62Sopenharmony_ci                          toolchain, fastjson_jar):
48b8a72a62Sopenharmony_ci    java_sources = [
49b8a72a62Sopenharmony_ci        'BinaryTool.java',
50b8a72a62Sopenharmony_ci        'BundleException.java',
51b8a72a62Sopenharmony_ci        'ConvertHapToBin.java',
52b8a72a62Sopenharmony_ci        'FileUtils.java',
53b8a72a62Sopenharmony_ci        'Log.java',
54b8a72a62Sopenharmony_ci        'PackFormatter.java',
55b8a72a62Sopenharmony_ci        'Utility.java'
56b8a72a62Sopenharmony_ci    ]
57b8a72a62Sopenharmony_ci    compile_java = get_compile_str(src_path, java_sources)
58b8a72a62Sopenharmony_ci
59b8a72a62Sopenharmony_ci    shell_path = os.path.join(root_path, "haptobin.sh")
60b8a72a62Sopenharmony_ci    command = ['bash', shell_path, root_path, jar_output, out_path,
61b8a72a62Sopenharmony_ci               toolchain, compile_java, fastjson_jar]
62b8a72a62Sopenharmony_ci    print('command: ', command)
63b8a72a62Sopenharmony_ci    run_cmd('haptobin_tool', command, 5000)
64b8a72a62Sopenharmony_ci    return
65b8a72a62Sopenharmony_ci
66b8a72a62Sopenharmony_ci
67b8a72a62Sopenharmony_cidef compile_unpacking_tool(root_path, src_path, jar_output, out_path, big_version,
68b8a72a62Sopenharmony_ci                           toolchain, fastjson_jar):
69b8a72a62Sopenharmony_ci    java_sources = [
70b8a72a62Sopenharmony_ci        'AbilityFormInfo.java', 'AbilityInfo.java', 'ApiVersion.java',
71b8a72a62Sopenharmony_ci        'AppInfo.java', 'AppModel.java', 'APPQFResult.java',
72b8a72a62Sopenharmony_ci        'BundleException.java', 'CommandParser.java', 'CommonEvent.java',
73b8a72a62Sopenharmony_ci        'CountryCode.java', 'CustomizeData.java', 'DefinePermission.java',
74b8a72a62Sopenharmony_ci        'DefPermissionGroup.java', 'DefPermission.java', 'DependencyItem.java',
75b8a72a62Sopenharmony_ci        'DeviceConfig.java', 'DistroFilter.java', 'Distro.java',
76b8a72a62Sopenharmony_ci        'ExtensionAbilityInfo.java', 'FileUtils.java', 'FormInfo.java',
77b8a72a62Sopenharmony_ci        'HapInfo.java', 'HapZipInfo.java', 'HQFInfo.java',
78b8a72a62Sopenharmony_ci        'IntentInfo.java', 'JsInfo.java', 'JsonUtil.java',
79b8a72a62Sopenharmony_ci        'Log.java', 'MetaDataInfo.java', 'MetaData.java',
80b8a72a62Sopenharmony_ci        'ModuleAbilityInfo.java', 'ModuleAdaption.java', 'ModuleAppInfo.java',
81b8a72a62Sopenharmony_ci        'ModuleAtomicService.java', 'ModuleDeviceType.java', 'ModuleInfo.java',
82b8a72a62Sopenharmony_ci        'ModuleMetadataInfo.java', 'ModuleProfileInfo.java', 'ModuleResult.java',
83b8a72a62Sopenharmony_ci        'ModuleShortcut.java', 'PackFormatter.java', 'PackInfo.java',
84b8a72a62Sopenharmony_ci        'PreloadItem.java', 'ProfileInfo.java', 'ReqPermission.java',
85b8a72a62Sopenharmony_ci        'ResourceIndexResult.java', 'ResourcesParser.java', 'ScreenDensity.java',
86b8a72a62Sopenharmony_ci        'ScreenShape.java', 'ScreenWindow.java', 'Shortcut.java',
87b8a72a62Sopenharmony_ci        'ShowHelp.java', 'SkillInfo.java', 'UncompressEntrance.java',
88b8a72a62Sopenharmony_ci        'Uncompress.java', 'UncompressResult.java', 'UncompressVerify.java',
89b8a72a62Sopenharmony_ci        'UriInfo.java', 'UsedScene.java', 'Utility.java', 'Want.java', 'ProxyDataItem.java'
90b8a72a62Sopenharmony_ci    ]
91b8a72a62Sopenharmony_ci    compile_java = get_compile_str(src_path, java_sources)
92b8a72a62Sopenharmony_ci
93b8a72a62Sopenharmony_ci    shell_path = os.path.join(root_path, "unpackingTool.sh")
94b8a72a62Sopenharmony_ci    command = ['bash', shell_path, root_path, jar_output, out_path, big_version,
95b8a72a62Sopenharmony_ci               toolchain, compile_java, fastjson_jar]
96b8a72a62Sopenharmony_ci    print('command: ', command)
97b8a72a62Sopenharmony_ci    run_cmd('app_unpacking_tool', command, 5000)
98b8a72a62Sopenharmony_ci    return
99b8a72a62Sopenharmony_ci
100b8a72a62Sopenharmony_ci
101b8a72a62Sopenharmony_cidef compile_packing_tool(root_path, src_path, jar_output, out_path,
102b8a72a62Sopenharmony_ci                         toolchain, fastjson_jar, compress_jar, io_jar):
103b8a72a62Sopenharmony_ci    java_sources = [
104b8a72a62Sopenharmony_ci        'ApiVersion.java',
105b8a72a62Sopenharmony_ci        'BundleException.java',
106b8a72a62Sopenharmony_ci        'CommandParser.java',
107b8a72a62Sopenharmony_ci        'CompressEntrance.java',
108b8a72a62Sopenharmony_ci        'Compressor.java',
109b8a72a62Sopenharmony_ci        'CompressVerify.java',
110b8a72a62Sopenharmony_ci        'Constants.java',
111b8a72a62Sopenharmony_ci        'CountryCode.java',
112b8a72a62Sopenharmony_ci        'DependencyItem.java',
113b8a72a62Sopenharmony_ci        'DistroFilter.java',
114b8a72a62Sopenharmony_ci        'FileUtils.java',
115b8a72a62Sopenharmony_ci        'HapVerify.java',
116b8a72a62Sopenharmony_ci        'HapVerifyInfo.java',
117b8a72a62Sopenharmony_ci        'HQFInfo.java',
118b8a72a62Sopenharmony_ci        'HQFVerify.java',
119b8a72a62Sopenharmony_ci        'Log.java',
120b8a72a62Sopenharmony_ci        'ModuleApiVersion.java',
121b8a72a62Sopenharmony_ci        'ModuleJsonInfo.java',
122b8a72a62Sopenharmony_ci        'ModuleJsonUtil.java',
123b8a72a62Sopenharmony_ci        'ModuleMetadataInfo.java',
124b8a72a62Sopenharmony_ci        'MultiAppMode.java',
125b8a72a62Sopenharmony_ci        'PackageNormalize.java',
126b8a72a62Sopenharmony_ci        'PackageUtil.java',
127b8a72a62Sopenharmony_ci        'PackFormatter.java',
128b8a72a62Sopenharmony_ci        'PreloadItem.java',
129b8a72a62Sopenharmony_ci        'ScreenDensity.java',
130b8a72a62Sopenharmony_ci        'ScreenShape.java',
131b8a72a62Sopenharmony_ci        'ScreenWindow.java',
132b8a72a62Sopenharmony_ci        'ShowHelp.java',
133b8a72a62Sopenharmony_ci        'Utility.java',
134b8a72a62Sopenharmony_ci        'VerifyCollection.java',
135b8a72a62Sopenharmony_ci        'Version.java'
136b8a72a62Sopenharmony_ci    ]
137b8a72a62Sopenharmony_ci    compile_java = get_compile_str(src_path, java_sources)
138b8a72a62Sopenharmony_ci
139b8a72a62Sopenharmony_ci    shell_path = os.path.join(root_path, "packingTool.sh")
140b8a72a62Sopenharmony_ci    command = ['bash', shell_path, root_path, jar_output, out_path,
141b8a72a62Sopenharmony_ci               toolchain, compile_java, fastjson_jar, compress_jar, io_jar]
142b8a72a62Sopenharmony_ci    print('command: ', command)
143b8a72a62Sopenharmony_ci    run_cmd('app_packing_tool', command, 5000)
144b8a72a62Sopenharmony_ci    return
145b8a72a62Sopenharmony_ci
146b8a72a62Sopenharmony_ci
147b8a72a62Sopenharmony_cidef compile_check_tool(root_path, src_path, jar_output, out_path,
148b8a72a62Sopenharmony_ci                       toolchain, fastjson_jar):
149b8a72a62Sopenharmony_ci    java_sources = [
150b8a72a62Sopenharmony_ci        'BundleException.java',
151b8a72a62Sopenharmony_ci        'CommandParser.java',
152b8a72a62Sopenharmony_ci        'FileUtils.java',
153b8a72a62Sopenharmony_ci        'Log.java',
154b8a72a62Sopenharmony_ci        'PackFormatter.java',
155b8a72a62Sopenharmony_ci        'Scan.java',
156b8a72a62Sopenharmony_ci        'ScanEntrance.java',
157b8a72a62Sopenharmony_ci        'ScanErrorEnum.java',
158b8a72a62Sopenharmony_ci        'ScanStatDuplicate.java',
159b8a72a62Sopenharmony_ci        'ScanStatFileSize.java',
160b8a72a62Sopenharmony_ci        'ScanStatSuffix.java',
161b8a72a62Sopenharmony_ci        'ScanVerify.java',
162b8a72a62Sopenharmony_ci        'ShowHelp.java',
163b8a72a62Sopenharmony_ci        'Utility.java'
164b8a72a62Sopenharmony_ci    ]
165b8a72a62Sopenharmony_ci    compile_java = get_compile_str(src_path, java_sources)
166b8a72a62Sopenharmony_ci
167b8a72a62Sopenharmony_ci    shell_path = os.path.join(root_path, "checkTool.sh")
168b8a72a62Sopenharmony_ci    command = ['bash', shell_path, root_path, jar_output, out_path,
169b8a72a62Sopenharmony_ci               toolchain, compile_java, fastjson_jar]
170b8a72a62Sopenharmony_ci    print('command: ', command)
171b8a72a62Sopenharmony_ci    run_cmd('app_check_tool', command, 5000)
172b8a72a62Sopenharmony_ci    return
173b8a72a62Sopenharmony_ci
174b8a72a62Sopenharmony_ci
175b8a72a62Sopenharmony_cidef main():
176b8a72a62Sopenharmony_ci    parser = argparse.ArgumentParser()
177b8a72a62Sopenharmony_ci    parser.add_argument('--haptobin', required=True)
178b8a72a62Sopenharmony_ci    parser.add_argument('--haptobinOutput', required=True)
179b8a72a62Sopenharmony_ci    parser.add_argument('--unpackOutput', required=True)
180b8a72a62Sopenharmony_ci    parser.add_argument('--packOutput', required=True)
181b8a72a62Sopenharmony_ci    parser.add_argument('--checkOutput', required=True)
182b8a72a62Sopenharmony_ci    parser.add_argument('--outpath', required=True)
183b8a72a62Sopenharmony_ci    parser.add_argument('--toolchain', required=True)
184b8a72a62Sopenharmony_ci    parser.add_argument('--compileTarget', required=True)
185b8a72a62Sopenharmony_ci    args = parser.parse_args()
186b8a72a62Sopenharmony_ci    print('packingTool args: ', args)
187b8a72a62Sopenharmony_ci    root_dir = os.path.dirname(os.path.realpath(__file__))
188b8a72a62Sopenharmony_ci    src_dir = os.path.join(root_dir, "./adapter/ohos/")
189b8a72a62Sopenharmony_ci    toolchain = args.toolchain
190b8a72a62Sopenharmony_ci    tool_list = toolchain.split(':')
191b8a72a62Sopenharmony_ci    toolchain = tool_list[-1]
192b8a72a62Sopenharmony_ci    toolchain += "_" + args.compileTarget
193b8a72a62Sopenharmony_ci
194b8a72a62Sopenharmony_ci    fastjson_jar = os.path.join(root_dir, '../../prebuilts/packing_tool/fastjson-1.2.83.jar')
195b8a72a62Sopenharmony_ci    compress_jar = os.path.join(root_dir, '../../prebuilts/packing_tool/commons-compress-1.26.1.jar')
196b8a72a62Sopenharmony_ci    io_jar = os.path.join(root_dir, '../../prebuilts/packing_tool/commons-io-2.15.1.jar')
197b8a72a62Sopenharmony_ci    print('packingTool: ', toolchain, fastjson_jar, compress_jar, io_jar)
198b8a72a62Sopenharmony_ci
199b8a72a62Sopenharmony_ci    # compile haptobin_tool.jar
200b8a72a62Sopenharmony_ci    compile_haptobin_tool(root_dir, src_dir, args.haptobinOutput, args.outpath,
201b8a72a62Sopenharmony_ci                          toolchain, fastjson_jar)
202b8a72a62Sopenharmony_ci
203b8a72a62Sopenharmony_ci    # compile app_unpacking_tool.jar
204b8a72a62Sopenharmony_ci    version = subprocess.check_output(['javac', '-version'], stderr=subprocess.STDOUT)
205b8a72a62Sopenharmony_ci    version = version.decode('utf-8')
206b8a72a62Sopenharmony_ci    array = re.findall(r'\d+', version)
207b8a72a62Sopenharmony_ci    compatible_version = 8
208b8a72a62Sopenharmony_ci    big_version = ''
209b8a72a62Sopenharmony_ci    if int(array[0]) > compatible_version:
210b8a72a62Sopenharmony_ci        big_version = 'true'
211b8a72a62Sopenharmony_ci    else:
212b8a72a62Sopenharmony_ci        big_version = 'false'
213b8a72a62Sopenharmony_ci    compile_unpacking_tool(root_dir, src_dir, args.unpackOutput, args.outpath, big_version,
214b8a72a62Sopenharmony_ci                           toolchain, fastjson_jar)
215b8a72a62Sopenharmony_ci
216b8a72a62Sopenharmony_ci    #compile app_packing_tool.jar
217b8a72a62Sopenharmony_ci    compile_packing_tool(root_dir, src_dir, args.packOutput, args.outpath,
218b8a72a62Sopenharmony_ci                         toolchain, fastjson_jar, compress_jar, io_jar)
219b8a72a62Sopenharmony_ci
220b8a72a62Sopenharmony_ci    #compile app_check_tool.jar
221b8a72a62Sopenharmony_ci    compile_check_tool(root_dir, src_dir, args.checkOutput, args.outpath,
222b8a72a62Sopenharmony_ci                       toolchain, fastjson_jar)
223b8a72a62Sopenharmony_ci
224b8a72a62Sopenharmony_ciif __name__ == '__main__':
225b8a72a62Sopenharmony_ci    sys.exit(main())
226