feat : support kclvm cli capi call

kclvm/Cargo.toml

@@ -46,6 +46,7 @@ lto = true
 
 [workspace]
 members = [
+    "capi",
     "ast",
     "compiler",
     "config",

kclvm/capi/Cargo.toml

@@ -0,0 +1,35 @@
+[package]
+name = "kclvm-capi"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib"]
+path = "src/lib.rs"
+name = "kclvm_capi"
+
+[dependencies]
+protobuf = "3.1.0"
+serde_json = "1.0"
+serde_yaml = "0.9.2"
+protobuf-json-mapping = "3.1.0"
+serde = { version = "1", features = ["derive"] }
+
+kclvm-runner = {path = "../runner", version = "0.1.0"}
+kclvm-parser = {path = "../parser", version = "0.1.0"}
+kclvm-ast = {path = "../ast", version = "0.1.0"}
+kclvm-runtime = {path = "../runtime", version = "0.1.0"}
+kclvm-tools = {path = "../tools", version= "0.1.0" }
+
+[dev-dependencies]
+criterion = "0.3"
+
+[build_dependencies]
+protobuf-codegen= "3.1.0"
+protoc-bin-vendored = "3.0.0"
+
+[profile.release]
+rpath = true
+panic = "unwind"
+opt-level = "z"  # Optimize for size.
+lto = true

kclvm/capi/build.rs

@@ -0,0 +1,12 @@
+/// According to the file KCLVM/internal/kclvm_py/spec/gpyrpc/gpyrpc.proto, automatically generate
+/// the corresponding rust source file to the directory src/model
+fn main() {
+    protobuf_codegen::Codegen::new()
+        .protoc()
+        .protoc_path(&protoc_bin_vendored::protoc_bin_path().unwrap())
+        .out_dir("src/model")
+        .include("../../internal/kclvm_py/spec/gpyrpc")
+        .inputs(&["../../internal/kclvm_py/spec/gpyrpc/gpyrpc.proto"])
+        .run()
+        .expect("Running protoc failed.");
+}

kclvm/capi/src/api_test.rs

@@ -0,0 +1,79 @@
+use protobuf::Message;
+
+use crate::model::gpyrpc::*;
+use crate::service::api::*;
+use crate::service::util::*;
+use std::ffi::{CStr, CString};
+use std::fs;
+use std::path::Path;
+const TEST_DATA_PATH: &str = "./src/testdata";
+
+#[test]
+fn test_c_api_call_exec_program() {
+    let serv = kclvm_service_new(0);
+    let input_path = Path::new(TEST_DATA_PATH).join("exec-program.json");
+    let input = fs::read_to_string(&input_path)
+        .expect(format!("Something went wrong reading {}", input_path.display()).as_str());
+    let args = unsafe {
+        CString::from_vec_unchecked(
+            parse_message_from_json::<ExecProgram_Args>(&input)
+                .unwrap()
+                .write_to_bytes()
+                .unwrap(),
+        )
+    };
+    let call = CString::new("KclvmService.ExecProgram").unwrap();
+    let result_ptr = kclvm_service_call(serv, call.as_ptr(), args.as_ptr()) as *mut i8;
+    let result = unsafe { CStr::from_ptr(result_ptr) };
+
+    let result = parse_message_from_protobuf::<ExecProgram_Result>(result.to_bytes()).unwrap();
+    let except_result_path = Path::new(TEST_DATA_PATH).join("exec-program.response.json");
+    let except_result_json = fs::read_to_string(&except_result_path).expect(
+        format!(
+            "Something went wrong reading {}",
+            except_result_path.display()
+        )
+        .as_str(),
+    );
+    let except_result = parse_message_from_json::<ExecProgram_Result>(&except_result_json).unwrap();
+    assert_eq!(result.json_result, except_result.json_result);
+    assert_eq!(result.yaml_result, except_result.yaml_result);
+
+    kclvm_service_delete(serv);
+    kclvm_service_free_string(result_ptr);
+}
+
+#[test]
+fn test_c_api_call_override_file() {
+    let serv = kclvm_service_new(0);
+    let input_path = Path::new(TEST_DATA_PATH).join("override-file.json");
+    let input = fs::read_to_string(&input_path)
+        .expect(format!("Something went wrong reading {}", input_path.display()).as_str());
+    let args = unsafe {
+        CString::from_vec_unchecked(
+            parse_message_from_json::<OverrideFile_Args>(&input)
+                .unwrap()
+                .write_to_bytes()
+                .unwrap(),
+        )
+    };
+    let call = CString::new("KclvmService.OverrideFile").unwrap();
+    let result_ptr = kclvm_service_call(serv, call.as_ptr(), args.as_ptr()) as *mut i8;
+    let result = unsafe { CStr::from_ptr(result_ptr) };
+
+    let result = parse_message_from_protobuf::<OverrideFile_Result>(result.to_bytes()).unwrap();
+    let except_result_path = Path::new(TEST_DATA_PATH).join("override-file.response.json");
+    let except_result_json = fs::read_to_string(&except_result_path).expect(
+        format!(
+            "Something went wrong reading {}",
+            except_result_path.display()
+        )
+        .as_str(),
+    );
+    let except_result =
+        parse_message_from_json::<OverrideFile_Result>(&except_result_json).unwrap();
+    assert_eq!(result.result, except_result.result);
+
+    kclvm_service_delete(serv);
+    kclvm_service_free_string(result_ptr);
+}

kclvm/capi/src/lib.rs

@@ -0,0 +1,5 @@
+pub mod model;
+pub mod service;
+
+#[cfg(test)]
+pub mod api_test;