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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

mod audit_events;
mod buckets;
mod comments;
mod custom_label_trend_report;
mod datasets;
mod emails;
mod integrations;
mod keyed_sync_states;
mod projects;
mod quota;
mod sources;
mod streams;
mod users;

use anyhow::Result;
use custom_label_trend_report::GetCustomLabelTrendReportArgs;
use quota::GetQuotaArgs;
use reinfer_client::Client;
use scoped_threadpool::Pool;
use structopt::StructOpt;

use self::{
    audit_events::GetAuditEventsArgs,
    buckets::GetBucketsArgs,
    comments::{GetManyCommentsArgs, GetSingleCommentArgs},
    datasets::GetDatasetsArgs,
    emails::GetManyEmailsArgs,
    integrations::GetIntegrationsArgs,
    keyed_sync_states::GetKeyedSyncStatesArgs,
    projects::GetProjectsArgs,
    sources::GetSourcesArgs,
    streams::{GetStreamCommentsArgs, GetStreamStatsArgs, GetStreamsArgs},
    users::GetUsersArgs,
};
use crate::printer::Printer;

#[allow(clippy::large_enum_variant)]
#[derive(Debug, StructOpt)]
pub enum GetArgs {
    #[structopt(name = "buckets")]
    /// List the available buckets
    Buckets(GetBucketsArgs),

    #[structopt(name = "emails")]
    /// Download all emails from a source
    Emails(GetManyEmailsArgs),

    #[structopt(name = "comment")]
    /// Get a single comment from a source
    Comment(GetSingleCommentArgs),

    #[structopt(name = "comments")]
    /// Download all comments from a source
    Comments(GetManyCommentsArgs),

    #[structopt(name = "datasets")]
    /// List the available datasets
    Datasets(GetDatasetsArgs),

    #[structopt(name = "projects")]
    /// List the available projects
    Projects(GetProjectsArgs),

    #[structopt(name = "sources")]
    /// List the available sources
    Sources(GetSourcesArgs),

    #[structopt(name = "streams")]
    /// List the available streams for a dataset
    Streams(GetStreamsArgs),

    #[structopt(name = "stream-comments")]
    /// Fetch comments from a stream
    StreamComments(GetStreamCommentsArgs),

    #[structopt(name = "stream-stats")]
    /// Get the validation stats for a given stream
    StreamStats(GetStreamStatsArgs),

    #[structopt(name = "users")]
    /// List the available users
    Users(GetUsersArgs),

    #[structopt(name = "current-user")]
    /// Get the user associated with the API token in use
    CurrentUser,

    #[structopt(name = "quotas")]
    /// List all quotas for current tenant
    Quotas(GetQuotaArgs),

    #[structopt(name = "audit-events")]
    /// Get audit events for current tenant
    AuditEvents(GetAuditEventsArgs),

    #[structopt(name = "integrations")]
    /// Get integrations
    Integrations(GetIntegrationsArgs),

    #[structopt(name = "keyed-sync-states")]
    /// Get keyed sync states
    KeyedSyncStates(GetKeyedSyncStatesArgs),

    #[structopt(name = "custom-label-trend-report")]
    /// Get Custom Report
    CustomDatasetReport(GetCustomLabelTrendReportArgs),
}

pub fn run(args: &GetArgs, client: Client, printer: &Printer, pool: &mut Pool) -> Result<()> {
    match args {
        GetArgs::Buckets(args) => buckets::get(&client, args, printer),
        GetArgs::Emails(args) => emails::get_many(&client, args),
        GetArgs::Comment(args) => comments::get_single(&client, args),
        GetArgs::Comments(args) => comments::get_many(&client, args),
        GetArgs::Datasets(args) => datasets::get(&client, args, printer, pool),
        GetArgs::Projects(args) => projects::get(&client, args, printer),
        GetArgs::Sources(args) => sources::get(&client, args, printer),
        GetArgs::Streams(args) => streams::get(&client, args, printer),
        GetArgs::StreamComments(args) => streams::get_stream_comments(&client, args),
        GetArgs::StreamStats(args) => streams::get_stream_stats(&client, args, printer, pool),
        GetArgs::Users(args) => users::get(&client, args, printer),
        GetArgs::CurrentUser => users::get_current_user(&client, printer),
        GetArgs::Quotas(args) => quota::get(&client, args, printer),
        GetArgs::AuditEvents(args) => audit_events::get(&client, args, printer),
        GetArgs::Integrations(args) => integrations::get(&client, args, printer),
        GetArgs::KeyedSyncStates(args) => keyed_sync_states::get(&client, args, printer),
        GetArgs::CustomDatasetReport(args) => {
            custom_label_trend_report::get(&client, args, printer)
        }
    }
}