crates/vcs_actions/src/actions/local_actions.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use std::net::SocketAddr;

use action_system::{action::ActionContext, macros::action_gen};
use log::info;
use tcp_connection::error::TcpTargetError;

#[action_gen]
pub async fn set_upstream_vault_action(
    ctx: ActionContext,
    _upstream: SocketAddr,
) -> Result<(), TcpTargetError> {
    // Ensure the instance is available
    let Some(instance) = ctx.instance() else {
        return Err(TcpTargetError::NotFound(
            "Connection Instance Lost.".to_string(),
        ));
    };

    if ctx.is_proc_on_local() {
        // Invoke on local
        // Send the message to the server
        let _ = instance.lock().await.write_text("Hello World!").await;
    } else if ctx.is_proc_on_remote() {
        // Remote execution - read the message from the client
        let read = instance.lock().await.read_text().await?;
        info!("Received: {}", read)
    }

    Ok(())
}