seedtool-cli 0.4.0

A command-line tool for generating and transforming cryptographic seeds.
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

use anyhow::{Result, bail};
use bc_envelope::Envelope;
use bc_ur::{MultipartDecoder, MultipartEncoder, URDecodable, UREncodable};

use super::{Format, InputFormat, OutputFormat};
use crate::{cli::Cli, seed::Seed};

pub struct MultipartFormat;

impl Format for MultipartFormat {
    fn name(&self) -> &str { "multipart" }

    fn round_trippable(&self) -> bool { true }
}

impl InputFormat for MultipartFormat {
    fn process_input(&self, mut state: Cli) -> Result<Cli> {
        let input = state.expect_input()?;
        let shares: Vec<&str> = input.split_whitespace().collect();

        let mut decoder = MultipartDecoder::new();
        for share in shares {
            decoder.receive(share)?;
            if decoder.is_complete() {
                break;
            }
        }

        if !decoder.is_complete() {
            bail!("Insufficient SSKR shares");
        }
        let ur = decoder.message().unwrap().unwrap();
        let envelope = Envelope::from_ur(&ur)?;
        let seed = Seed::try_from(envelope)?;
        state.seed = Some(seed);
        Ok(state)
    }
}

impl OutputFormat for MultipartFormat {
    fn process_output(&self, state: Cli) -> Result<String> {
        let ur = state.to_envelope().ur();
        let mut encoder = MultipartEncoder::new(&ur, state.max_fragment_len)?;
        let parts_count = encoder.parts_count() + state.additional_parts;
        let parts = (0..parts_count)
            .map(|_| encoder.next_part().map_err(anyhow::Error::from))
            .collect::<Result<Vec<String>>>()?;
        Ok(parts.join("\n"))
    }
}