reinfer-cli 0.38.5

Command line interface for Re:infer, the conversational data intelligence platform
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
76
77
78
79
80
81
82
83
84

use anyhow::{Context, Result};
use log::info;
use reinfer_client::{resources::source::StatisticsRequestParams, Client, SourceIdentifier};
use std::collections::HashMap;
use structopt::StructOpt;

use crate::printer::{PrintableSource, Printer};

#[derive(Debug, StructOpt)]
pub struct GetSourcesArgs {
    #[structopt(name = "source")]
    /// If specified, only list this source (name or id)
    source: Option<SourceIdentifier>,

    #[structopt(long = "stats")]
    /// Whether to include source statistics in response
    include_stats: bool,
}

pub fn get(client: &Client, args: &GetSourcesArgs, printer: &Printer) -> Result<()> {
    let GetSourcesArgs {
        source,
        include_stats,
    } = args;

    let sources = if let Some(source) = source {
        vec![client
            .get_source(source.clone())
            .context("Operation to list sources has failed.")?]
    } else {
        let mut sources = client
            .get_sources()
            .context("Operation to list sources has failed.")?;
        sources.sort_unstable_by(|lhs, rhs| {
            (&lhs.owner.0, &lhs.name.0).cmp(&(&rhs.owner.0, &rhs.name.0))
        });
        sources
    };

    let buckets: HashMap<_, _> = client
        .get_buckets()
        .context("Operation to list buckets has failed.")?
        .into_iter()
        .map(|bucket| (bucket.id.clone(), bucket))
        .collect();

    let mut source_stats: HashMap<_, _> = HashMap::new();
    if *include_stats {
        sources.iter().try_for_each(|source| -> Result<()> {
            info!("Getting statistics for source {}", source.full_name().0);
            let stats = client
                .get_source_statistics(
                    &source.full_name(),
                    &StatisticsRequestParams {
                        comment_filter: Default::default(),
                    },
                )
                .context("Could not get statistics for source")?;

            source_stats.insert(source.id.clone(), stats);
            Ok(())
        })?;
    };

    let printable_sources: Vec<PrintableSource> = sources
        .into_iter()
        .map(|source| {
            let bucket = source
                .bucket_id
                .as_ref()
                .and_then(|id| buckets.get(id))
                .cloned();

            let stats = source_stats.get(&source.id).cloned();
            PrintableSource {
                source,
                bucket,
                stats,
            }
        })
        .collect();

    printer.print_resources(&printable_sources)
}