update

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "atri_bot"
-version = "0.3.0"
+version = "0.3.1"
 edition = "2021"
 authors = ["LaoLittle"]
 description = "A simple bot"

src/plugin/ffi/env.rs

@@ -0,0 +1,32 @@
+use std::path::PathBuf;
+use atri_ffi::RustString;
+use tracing::error;
+use crate::plugin::cast_ref;
+use crate::service::plugin_manager::PluginManager;
+
+pub extern "C" fn env_get_workspace(handle: usize, manager: *const ()) -> RustString {
+    let manager: &PluginManager = cast_ref(manager);
+    manager.find_plugin(handle)
+        .map(|p| {
+            let name = p.name();
+            let mut path = std::env::current_dir()
+                .ok()
+                .and_then(|p| p.to_str().map(|str| {
+                    let mut p = String::from(str);
+                    p.push('/');
+                    p
+                }))
+                .unwrap_or_else(String::new);
+
+            path.push_str("workspaces/");
+            path.push_str(name);
+
+            if let Err(e) = std::fs::create_dir_all(&path) {
+                error!("为{}创建Workspace失败: {}", p, e);
+            }
+
+            path
+        })
+        .unwrap_or_else(String::new)
+        .into()
+}

src/plugin/ffi/mod.rs

@@ -6,6 +6,7 @@ mod listener;
 mod log;
 mod member;
 mod message;
+mod env;
 
 use crate::plugin::ffi::client::{
     client_find_friend, client_find_group, client_get_friends, client_get_groups, client_get_id,
@@ -36,10 +37,11 @@ use crate::plugin::ffi::message::{image_get_id, image_get_url};
 use crate::PluginManager;
 use atri_ffi::future::FFIFuture;
 use atri_ffi::Managed;
+use crate::plugin::ffi::env::env_get_workspace;
 
 pub extern "C" fn plugin_get_function(sig: u16) -> *const () {
     extern "C" fn not_impl() {
-        panic!("No such sig");
+        panic!("No such operation");
     }
 
     macro_rules! match_function {
@@ -116,6 +118,9 @@ pub extern "C" fn plugin_get_function(sig: u16) -> *const () {
 
         // log
         20000 => log,
+
+        // env
+        30000 => env_get_workspace,
     }
 }