1 files changed, 38 insertions, 0 deletions
diff --git a/crates/system_action/src/action.rs b/crates/system_action/src/action.rs
index c8d7a18..e7d2d8c 100644
--- a/crates/system_action/src/action.rs
+++ b/crates/system_action/src/action.rs
@@ -22,6 +22,12 @@ pub struct ActionContext {
     /// Whether the action is executed locally or remotely
     local: bool,
 
+    /// The name of the action being executed
+    action_name: String,
+
+    /// The JSON-serialized arguments for the action
+    action_args_json: String,
+
     /// The connection instance in the current context,
     /// used to interact with the machine on the other end
     instance: Option<ConnectionInstance>,
@@ -53,6 +59,11 @@ impl ActionContext {
         self.instance = Some(instance);
         self
     }
+
+    /// Pop connection instance from context
+    pub fn pop_instance(&mut self) -> Option<ConnectionInstance> {
+        self.instance.take()
+    }
 }
 
 impl ActionContext {
@@ -70,4 +81,31 @@ impl ActionContext {
     pub fn instance(&self) -> &Option<ConnectionInstance> {
         &self.instance
     }
+
+    /// Get a mutable reference to the connection instance in the current context
+    pub fn instance_mut(&mut self) -> &mut Option<ConnectionInstance> {
+        &mut self.instance
+    }
+
+    /// Get the action name from the context
+    pub fn action_name(&self) -> &str {
+        &self.action_name
+    }
+
+    /// Get the action arguments from the context
+    pub fn action_args_json(&self) -> &String {
+        &self.action_args_json
+    }
+
+    /// Set the action name in the context
+    pub fn set_action_name(mut self, action_name: String) -> Self {
+        self.action_name = action_name;
+        self
+    }
+
+    /// Set the action arguments in the context
+    pub fn set_action_args_json(mut self, action_args: String) -> Self {
+        self.action_args_json = action_args;
+        self
+    }
 }