1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import java_types
def Generate(jni_obj, *, gen_jni_class, script_name):
proxy_class = java_types.JavaClass(
f'{jni_obj.java_class.full_name_with_slashes}Jni')
visibility = 'public ' if jni_obj.proxy_visibility == 'public' else ''
interface_name = jni_obj.proxy_interface.name_with_dots
gen_jni = gen_jni_class.name
type_resolver = java_types.TypeResolver(proxy_class)
type_resolver.imports = list(jni_obj.type_resolver.imports)
sb = []
sb.append(f"""\
//
// This file was generated by {script_name}
//
package {jni_obj.java_class.class_without_prefix.package_with_dots};
import org.jni_zero.CheckDiscard;
import org.jni_zero.JniStaticTestMocker;
import org.jni_zero.NativeLibraryLoadedStatus;
import {gen_jni_class.full_name_with_dots};
""")
# Copy over all imports (some will be unused, but oh well).
for c in type_resolver.imports:
sb.append(f'import {c.full_name_with_dots};\n')
sb.append(f"""
@CheckDiscard("crbug.com/993421")
{visibility}class {proxy_class.name} implements {interface_name} {{
private static {interface_name} testInstance;
public static final JniStaticTestMocker<{interface_name}> TEST_HOOKS =
new JniStaticTestMocker<{interface_name}>() {{
@Override
public void setInstanceForTesting({interface_name} instance) {{
if (!{gen_jni}.TESTING_ENABLED) {{
throw new RuntimeException(
"Tried to set a JNI mock when mocks aren't enabled!");
}}
testInstance = instance;
}}
}};
""")
for native in jni_obj.proxy_natives:
call_params = native.params.to_call_str()
sig_params = native.params.to_java_declaration(type_resolver)
return_type_str = native.return_type.to_java(type_resolver)
return_prefix = ''
if not native.return_type.is_void():
return_prefix = f'return ({return_type_str}) '
sb.append(f"""
@Override
public {return_type_str} {native.name}({sig_params}) {{
{return_prefix}{gen_jni}.{native.proxy_name}({call_params});
}}
""")
sb.append(f"""
public static {interface_name} get() {{
if ({gen_jni}.TESTING_ENABLED) {{
if (testInstance != null) {{
return testInstance;
}}
if ({gen_jni}.REQUIRE_MOCK) {{
throw new UnsupportedOperationException(
"No mock found for the native implementation of {interface_name}. "
+ "The current configuration requires implementations be mocked.");
}}
}}
NativeLibraryLoadedStatus.checkLoaded();
return new {proxy_class.name}();
}}
}}
""")
return ''.join(sb)
|