diff options
Diffstat (limited to 'src/bin/rbw-agent/actions.rs')
-rw-r--r-- | src/bin/rbw-agent/actions.rs | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/bin/rbw-agent/actions.rs b/src/bin/rbw-agent/actions.rs index 52c34fa..afe76c8 100644 --- a/src/bin/rbw-agent/actions.rs +++ b/src/bin/rbw-agent/actions.rs @@ -211,9 +211,10 @@ pub async fn decrypt( sock: &mut crate::sock::Sock, state: std::sync::Arc<tokio::sync::RwLock<crate::agent::State>>, cipherstring: &str, + org_id: Option<&str>, ) -> anyhow::Result<()> { let state = state.read().await; - let keys = if let Some(keys) = &state.priv_key { + let keys = if let Some(keys) = state.key(org_id) { keys } else { return Err(anyhow::anyhow!( @@ -238,9 +239,10 @@ pub async fn encrypt( sock: &mut crate::sock::Sock, state: std::sync::Arc<tokio::sync::RwLock<crate::agent::State>>, plaintext: &str, + org_id: Option<&str>, ) -> anyhow::Result<()> { let state = state.read().await; - let keys = if let Some(keys) = &state.priv_key { + let keys = if let Some(keys) = state.key(org_id) { keys } else { return Err(anyhow::anyhow!( |