1#!/usr/bin/env python3
2# coding: utf-8
3
4"""
5Copyright (c) 2021 Huawei Device Co., Ltd.
6Licensed under the Apache License, Version 2.0 (the "License");
7you may not use this file except in compliance with the License.
8You may obtain a copy of the License at
9
10    http://www.apache.org/licenses/LICENSE-2.0
11
12Unless required by applicable law or agreed to in writing, software
13distributed under the License is distributed on an "AS IS" BASIS,
14WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15See the License for the specific language governing permissions and
16limitations under the License.
17
18Description: Generate javascript byte code using es2abc
19"""
20
21import os
22import subprocess
23import platform
24import argparse
25
26
27def parse_args():
28    parser = argparse.ArgumentParser()
29    parser.add_argument('--src-js',
30                        help='js source file')
31    parser.add_argument('--dst-file',
32                        help='the converted target file')
33    parser.add_argument('--frontend-tool-path',
34                        help='path of the frontend conversion tool')
35    parser.add_argument('--extension',
36                        help='source file extension')
37    parser.add_argument("--debug", action='store_true',
38                        help='whether add debuginfo')
39    parser.add_argument("--module", action='store_true',
40                        help='whether is module')
41    parser.add_argument("--commonjs", action='store_true',
42                        help='whether is commonjs')
43    parser.add_argument("--merge-abc", action='store_true',
44                        help='whether is merge abc')
45    parser.add_argument("--generate-patch", action='store_true',
46                        help='generate patch abc')
47    parser.add_argument("--dump-symbol-table",
48                        help='dump symbol table of base abc')
49    parser.add_argument("--input-symbol-table",
50                        help='input symbol table for patch abc')
51    parser.add_argument("--target-api-sub-version",
52                        help='input symbol table for patch abc')
53    parser.add_argument("--module-record-field-name",
54                        help='specify the field name of module record in unmerged abc. This argument is optional, ' +
55                             'its value will be the path of input file if not specified')
56    parser.add_argument("--source-file",
57                        help='specify the file path info recorded in generated abc. This argument is optional, ' +
58                             'its value will be the path of input file if not specified')
59    parser.add_argument("--enable-annotations", action='store_true',
60                        help='whether annotations are enabled or not')
61    arguments = parser.parse_args()
62    return arguments
63
64
65def run_command(cmd, execution_path):
66    print(" ".join(cmd) + " | execution_path: " + execution_path)
67    proc = subprocess.Popen(cmd, cwd=execution_path)
68    proc.wait()
69    if proc.returncode != 0:
70        raise subprocess.CalledProcessError(proc.returncode, cmd)
71
72
73def gen_abc_info(input_arguments):
74    frontend_tool_path = input_arguments.frontend_tool_path
75
76    (path, name) = os.path.split(frontend_tool_path)
77
78    cmd = [os.path.join("./", name, "es2abc"),
79           '--output', input_arguments.dst_file,
80           input_arguments.src_js]
81
82    if input_arguments.extension:
83        cmd += ['--extension', input_arguments.extension]
84    if input_arguments.dump_symbol_table:
85        cmd += ['--dump-symbol-table', input_arguments.dump_symbol_table]
86    if input_arguments.input_symbol_table:
87        cmd += ['--input-symbol-table', input_arguments.input_symbol_table]
88    if input_arguments.debug:
89        src_index = cmd.index(input_arguments.src_js)
90        cmd.insert(src_index, '--debug-info')
91    if input_arguments.module:
92        src_index = cmd.index(input_arguments.src_js)
93        cmd.insert(src_index, '--module')
94    if input_arguments.commonjs:
95        src_index = cmd.index(input_arguments.src_js)
96        cmd.insert(src_index, '--commonjs')
97    if input_arguments.merge_abc:
98        src_index = cmd.index(input_arguments.src_js)
99        cmd.insert(src_index, '--merge-abc')
100    if input_arguments.generate_patch:
101        src_index = cmd.index(input_arguments.src_js)
102        cmd.insert(src_index, '--generate-patch')
103    if input_arguments.module_record_field_name:
104        cmd += ["--module-record-field-name", input_arguments.module_record_field_name]
105    if input_arguments.source_file:
106        cmd += ["--source-file", input_arguments.source_file]
107    if input_arguments.enable_annotations:
108        src_index = cmd.index(input_arguments.src_js)
109        cmd.insert(src_index, '--enable-annotations')
110        # insert d.ts option to cmd later
111    cmd.append("--target-api-sub-version=beta3")
112
113    try:
114        run_command(cmd, path)
115    except subprocess.CalledProcessError as e:
116        exit(e.returncode)
117
118
119if __name__ == '__main__':
120    gen_abc_info(parse_args())
121