Merge pull request #19 from JustEnoughVCS/jvcs_dev

Jvcs dev

2 files changed, 50 insertions, 5 deletions

diff --git a/crates/system_action/action_macros/src/lib.rs b/crates/system_action/action_macros/src/lib.rs
index 5d274ad..a7de9b6 100644
--- a/crates/system_action/action_macros/src/lib.rs
+++ b/crates/system_action/action_macros/src/lib.rs
@@ -55,10 +55,38 @@ fn generate_action_struct(input_fn: ItemFn, _is_local: bool) -> proc_macro2::Tok
             }
         }
 
+        impl #struct_name {
+            #fn_vis fn register_to_pool(pool: &mut action_system::action_pool::ActionPool) {
+                pool.register::<#struct_name, #arg_type, #return_type>();
+            }
+
+            #fn_vis async fn process_at_pool<'a>(
+                pool: &'a action_system::action_pool::ActionPool,
+                ctx: action_system::action::ActionContext,
+                #arg_param_name: #arg_type
+            ) -> Result<#return_type, tcp_connection::error::TcpTargetError> {
+                pool.process::<#arg_type, #return_type>(
+                    Box::leak(string_proc::snake_case!(stringify!(#action_name_ident)).into_boxed_str()),
+                    ctx,
+                    #arg_param_name
+                ).await
+            }
+        }
+
+        #[allow(dead_code)]
         #[deprecated = "This function is used by #[action_gen] as a template."]
-        #[doc = " This function is used by #[action_gen] as a template to generate the struct. "]
-        #[doc = " It is forbidden to call it anywhere."]
-        #[doc = " and call it using the function name."]
+        #[doc = "Template function for #[[action_gen]] - do not call directly."]
+        #[doc = "Use the generated struct instead."]
+        #[doc = ""]
+        #[doc = "Register the action to the pool."]
+        #[doc = "```rust"]
+        #[doc = "YourActionPascalName::register_to_pool(&mut pool);"]
+        #[doc = "```"]
+        #[doc = ""]
+        #[doc = "Process the action at the pool."]
+        #[doc = "```rust"]
+        #[doc = "let result = YourActionPascalName::process_at_pool(&pool, ctx, arg).await?;"]
+        #[doc = "```"]
         #fn_vis #fn_sig #fn_block
     }
 }
diff --git a/crates/system_action/src/action.rs b/crates/system_action/src/action.rs
index 14f1148..562a142 100644
--- a/crates/system_action/src/action.rs
+++ b/crates/system_action/src/action.rs
@@ -11,13 +11,30 @@ pub trait Action<Args, Return> {
     ) -> impl std::future::Future<Output = Result<Return, TcpTargetError>> + Send;
 }
 
+#[derive(Default)]
 pub struct ActionContext {
     // Whether the action is executed locally or remotely
     local: bool,
 
     /// The connection instance in the current context,
     /// used to interact with the machine on the other end
-    instance: ConnectionInstance,
+    instance: Option<ConnectionInstance>,
+}
+
+impl ActionContext {
+    /// Generate local context
+    pub fn local() -> Self {
+        let mut ctx = ActionContext::default();
+        ctx.local = true;
+        ctx
+    }
+
+    /// Generate remote context
+    pub fn remote() -> Self {
+        let mut ctx = ActionContext::default();
+        ctx.local = false;
+        ctx
+    }
 }
 
 impl ActionContext {
@@ -32,7 +49,7 @@ impl ActionContext {
     }
 
     /// Get the connection instance in the current context
-    pub fn instance(&self) -> &ConnectionInstance {
+    pub fn instance(&self) -> &Option<ConnectionInstance> {
         &self.instance
     }
 }