aboutsummaryrefslogtreecommitdiffstats
path: root/src/bin/rbw/actions.rs
blob: 08446969736bc4f39a33e89fdf631d4b5d00fb56 (plain) (blame)
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
pub fn login() -> anyhow::Result<()> {
    simple_action(rbw::agent::Action::Login, "login")
}

pub fn unlock() -> anyhow::Result<()> {
    simple_action(rbw::agent::Action::Unlock, "unlock")
}

pub fn sync() -> anyhow::Result<()> {
    simple_action(rbw::agent::Action::Sync, "sync")
}

pub fn lock() -> anyhow::Result<()> {
    simple_action(rbw::agent::Action::Lock, "lock")
}

pub fn quit() -> anyhow::Result<()> {
    let mut sock = crate::sock::Sock::connect()?;
    sock.send(&rbw::agent::Request {
        tty: std::env::var("TTY").ok(),
        action: rbw::agent::Action::Quit,
    })?;
    Ok(())
}

pub fn decrypt(cipherstring: &str) -> anyhow::Result<String> {
    let mut sock = crate::sock::Sock::connect()?;
    sock.send(&rbw::agent::Request {
        tty: std::env::var("TTY").ok(),
        action: rbw::agent::Action::Decrypt {
            cipherstring: cipherstring.to_string(),
        },
    })?;

    let res = sock.recv()?;
    match res {
        rbw::agent::Response::Decrypt { plaintext } => Ok(plaintext),
        rbw::agent::Response::Error { error } => {
            Err(anyhow::anyhow!("failed to decrypt: {}", error))
        }
        _ => Err(anyhow::anyhow!("unexpected message: {:?}", res)),
    }
}

fn simple_action(
    action: rbw::agent::Action,
    desc: &str,
) -> anyhow::Result<()> {
    let mut sock = crate::sock::Sock::connect()?;

    sock.send(&rbw::agent::Request {
        tty: std::env::var("TTY").ok(),
        action,
    })?;

    let res = sock.recv()?;
    match res {
        rbw::agent::Response::Ack => Ok(()),
        rbw::agent::Response::Error { error } => {
            Err(anyhow::anyhow!("failed to {}: {}", desc, error))
        }
        _ => Err(anyhow::anyhow!("unexpected message: {:?}", res)),
    }
}