feat: support function as params skip ci

zhangyuang · Oct 25, 2023 · 1776184 · 1776184
1 parent 2f11b74
commit 1776184
Show file tree

Hide file tree

Showing 6 changed files with 394 additions and 319 deletions.
diff --git a/cpp/sum.cpp b/cpp/sum.cpp
@@ -3,6 +3,7 @@
 #include <iostream>
 #include <string>
 #include <vector>
+#include <xlocale/_stdio.h>
 
 extern "C" int sum(int a, int b) { return a + b; }
 
@@ -98,6 +99,7 @@ typedef void (*FunctionPointer)(double a);
 extern "C" void callFunction(FunctionPointer func) {
   printf("callFunction");
   double a = 100.11;
+  // bool a = false;
   // char *a = "Hello, World!";
   // char *str = (char *)malloc(14 * sizeof(char));
   // strcpy(str, "Hello, World!");

diff --git a/scripts/types.d.ts b/scripts/types.d.ts
@@ -33,6 +33,7 @@ export type ArrayConstructorOptions<T extends DataType> = {
 
 export type FuncConstructorOptions<T extends DataType> = {
   paramsType: Array<DataFieldType<T>>
+  retType: DataFieldType<T>
 }
 
 export function arrayConstructor<T extends DataType>(options: ArrayConstructorOptions<T>): ArrayConstructorOptions<T>

diff --git a/src/ffi_macro.rs b/src/ffi_macro.rs
@@ -0,0 +1,30 @@
+macro_rules! match_args_len {
+     ($args_len:expr, $func_args_type:expr, $js_function:expr, $env:expr, $($num:literal => $closure:ident, $($arg:ident),*),*) => {
+         match $args_len {
+             $(
+                 $num => {
+                     use libffi::high::$closure;
+                     let lambda = |$($arg: *mut c_void),*| {
+                         let value: Vec<JsUnknown> = (0..$args_len)
+                             .map(|index| {
+                                 let c_param = match index {
+                                     $(
+                                         idx if idx == index => $arg,
+                                     )*
+                                     _ => unreachable!(),
+                                 };
+                                 let arg_type = $func_args_type.get_element::<JsUnknown>(index).unwrap();
+                                 let param = get_js_function_call_value($env, arg_type, c_param);
+                                 param
+                             })
+                             .collect();
+                         $js_function.call(None, &value).unwrap();
+                     };
+                     let closure = Box::into_raw(Box::new($closure::new(&lambda)));
+                     return std::mem::transmute((*closure).code_ptr());
+                 }
+             )*
+             _ => panic!("func_args get array error"),
+         }
+     };
+ }