cow-test-cli 0.5.3

Development and testing CLI for driving the CoW Protocol Solana settlement program.
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

//! Reading signer keypairs from CLI arguments.

use std::ops::Deref;

use solana_sdk::signer::keypair::Keypair;

use super::parsing::KeypairArg;

/// A keypair the caller either owns or borrows.
///
/// This wouldn't be needed if we could just `.clone()` a keypair ref, however
/// this is explicitly discouraged to avoid keeping key material in memory.
#[allow(clippy::large_enum_variant)]
pub enum MaybeKeypair<'a> {
    Owned(Keypair),
    Borrowed(&'a Keypair),
}

impl Deref for MaybeKeypair<'_> {
    type Target = Keypair;

    fn deref(&self) -> &Keypair {
        match self {
            MaybeKeypair::Owned(keypair) => keypair,
            MaybeKeypair::Borrowed(keypair) => keypair,
        }
    }
}

/// The keypair `arg` names, or `default` borrowed when no `arg` is given.
pub fn read_keypair_or(
    arg: Option<KeypairArg>,
    default: &Keypair,
) -> anyhow::Result<MaybeKeypair<'_>> {
    match arg {
        Some(arg) => arg.signer().map(MaybeKeypair::Owned),
        None => Ok(MaybeKeypair::Borrowed(default)),
    }
}