ax 2.18.2

ax distributed event databank and command line tool
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
65
66
67
68
69
70
71
72
73
74
75

use ax_core::{
    swarm::StreamAlias,
    util::formats::{ActyxOSResult, ActyxOSResultExt},
};
use ax_sdk::types::StreamId;
use futures::{prelude::*, stream, Stream};
use ipfs_sqlite_block_store::BlockStore;
use libipld::{Cid, DefaultParams};
use std::{convert::TryFrom, path::PathBuf};

use crate::cmd::AxCliCommand;

#[derive(clap::ValueEnum, Debug, Clone)]
enum List {
    /// List all aliases that resolve to stream ids, and their respective root
    /// hashes
    Aliases,
    /// List all cids in the blockstore
    Cids,
    /// List blocks cids in the blockstore
    Blocks,
}

#[derive(clap::Parser, Clone, Debug)]
pub struct ExploreTreeOpts {
    /// Path to a sqlite blockstore (read-only access!)
    #[arg(long)]
    block_store: PathBuf,
    command: List,
}

fn run(opts: ExploreTreeOpts) -> anyhow::Result<()> {
    let mut bs = BlockStore::<DefaultParams>::open(opts.block_store.clone(), Default::default())?;
    match opts.command {
        List::Aliases => {
            let aliases: Vec<(Vec<u8>, Cid)> = bs.aliases()?;
            for (bytes, cid) in aliases {
                let x = StreamAlias::try_from(&bytes[..])?;
                if let Ok(x) = StreamId::try_from(x) {
                    println!("{} --> {}", x, cid);
                }
            }
        }
        List::Cids => {
            let cids: Vec<Cid> = bs.get_known_cids()?;
            for cid in cids {
                println!("{}", cid);
            }
        }
        List::Blocks => {
            let cids: Vec<Cid> = bs.get_block_cids()?;
            for cid in cids {
                println!("{}", cid);
            }
        }
    }

    Ok(())
}

pub struct ExploreTree;
impl AxCliCommand for ExploreTree {
    type Opt = ExploreTreeOpts;
    type Output = ();
    fn run(opts: ExploreTreeOpts) -> Box<dyn Stream<Item = ActyxOSResult<Self::Output>> + Unpin> {
        Box::new(stream::once(
            async move { run(opts).ax_err_ctx(ax_core::util::formats::ActyxOSCode::ERR_INTERNAL_ERROR, "run failed") }
                .boxed(),
        ))
    }

    fn pretty(_: Self::Output) -> String {
        "".into()
    }
}