canic_cli/
lib.rs

1mod admission;
2mod apps;
3mod auth;
4mod backup;
5mod blob_storage;
6mod build;
7mod cli;
8mod cycles;
9mod diagnostic;
10mod endpoints;
11mod evidence;
12mod evidence_support;
13mod fleet;
14mod info;
15mod info_env;
16mod info_subnets;
17mod inspect;
18mod list;
19mod medic;
20mod metrics;
21mod network;
22mod output;
23mod replica;
24mod restore;
25mod scaffold;
26mod state;
27mod status;
28mod support;
29#[cfg(test)]
30mod test_support;
31mod token;
32mod toolchain;
33
34use crate::cli::{
35    argv::trace_if_enabled,
36    clap::parse_matches,
37    globals::{
38        DISPATCH_ARGS, apply_global_environment, apply_global_icp, global_environment_conflict,
39        misplaced_global_option,
40    },
41};
42use clap::error::ErrorKind;
43pub use cli::top_level_command;
44use std::ffi::OsString;
45use thiserror::Error as ThisError;
46
47const VERSION_TEXT: &str = concat!("canic ", env!("CARGO_PKG_VERSION"));
48
49///
50/// CliError
51///
52
53#[derive(Debug, ThisError)]
54pub enum CliError {
55    #[error("admission: {0}")]
56    Admission(#[source] Box<admission::AdmissionCommandError>),
57
58    #[error(transparent)]
59    Clap(#[from] clap::Error),
60
61    #[error("build: {0}")]
62    Build(#[from] build::BuildCommandError),
63
64    #[error("cycles: {0}")]
65    Cycles(#[source] Box<cycles::CyclesCommandError>),
66
67    #[error("backup: {0}")]
68    Backup(#[source] Box<backup::BackupCommandError>),
69
70    #[error("auth: {0}")]
71    Auth(#[source] Box<auth::AuthCommandError>),
72
73    #[error("blob-storage: {0}")]
74    BlobStorage(#[source] Box<blob_storage::BlobStorageCommandError>),
75
76    #[error("diagnostic: {0}")]
77    Diagnostic(#[from] diagnostic::DiagnosticCommandError),
78
79    #[error("evidence: {0}")]
80    Evidence(#[from] evidence::EvidenceCommandError),
81
82    #[error("fleet: {0}")]
83    Fleet(#[source] Box<fleet::FleetCommandError>),
84
85    #[error("info: {0}")]
86    Info(#[from] info::InfoCommandError),
87
88    #[error("inspect: {0}")]
89    Inspect(#[source] Box<inspect::InspectCommandError>),
90
91    #[error("medic: {0}")]
92    Medic(#[source] Box<medic::MedicCommandError>),
93
94    #[error("network: {0}")]
95    Network(#[from] network::NetworkCommandError),
96
97    #[error("app: {0}")]
98    Apps(#[source] Box<apps::AppCommandError>),
99
100    #[error("state: {0}")]
101    State(#[from] state::StateCommandError),
102
103    #[error("status: {0}")]
104    Status(#[from] status::StatusCommandError),
105
106    #[error("token: {0}")]
107    Token(#[source] Box<token::TokenCommandError>),
108
109    #[error("toolchain: {0}")]
110    Toolchain(#[from] toolchain::ToolchainCommandError),
111
112    #[error("scaffold: {0}")]
113    Scaffold(#[from] scaffold::ScaffoldCommandError),
114
115    #[error("replica: {0}")]
116    Replica(#[from] replica::ReplicaCommandError),
117
118    #[error("restore: {0}")]
119    Restore(#[from] restore::RestoreCommandError),
120}
121
122impl From<apps::AppCommandError> for CliError {
123    fn from(error: apps::AppCommandError) -> Self {
124        Self::Apps(Box::new(error))
125    }
126}
127
128impl From<admission::AdmissionCommandError> for CliError {
129    fn from(error: admission::AdmissionCommandError) -> Self {
130        Self::Admission(Box::new(error))
131    }
132}
133
134impl From<auth::AuthCommandError> for CliError {
135    fn from(error: auth::AuthCommandError) -> Self {
136        Self::Auth(Box::new(error))
137    }
138}
139
140impl From<backup::BackupCommandError> for CliError {
141    fn from(error: backup::BackupCommandError) -> Self {
142        Self::Backup(Box::new(error))
143    }
144}
145
146impl From<blob_storage::BlobStorageCommandError> for CliError {
147    fn from(error: blob_storage::BlobStorageCommandError) -> Self {
148        Self::BlobStorage(Box::new(error))
149    }
150}
151
152impl From<cycles::CyclesCommandError> for CliError {
153    fn from(error: cycles::CyclesCommandError) -> Self {
154        Self::Cycles(Box::new(error))
155    }
156}
157
158impl From<fleet::FleetCommandError> for CliError {
159    fn from(error: fleet::FleetCommandError) -> Self {
160        Self::Fleet(Box::new(error))
161    }
162}
163
164impl From<inspect::InspectCommandError> for CliError {
165    fn from(error: inspect::InspectCommandError) -> Self {
166        Self::Inspect(Box::new(error))
167    }
168}
169
170impl From<medic::MedicCommandError> for CliError {
171    fn from(error: medic::MedicCommandError) -> Self {
172        Self::Medic(Box::new(error))
173    }
174}
175
176impl From<token::TokenCommandError> for CliError {
177    fn from(error: token::TokenCommandError) -> Self {
178        Self::Token(Box::new(error))
179    }
180}
181
182/// Run the CLI from process arguments.
183pub fn run_from_env() -> Result<(), CliError> {
184    let argv = std::env::args_os().collect::<Vec<_>>();
185    trace_if_enabled(&argv);
186    run(argv.into_iter().skip(1))
187}
188
189/// Run the CLI from an argument iterator.
190pub fn run<I>(args: I) -> Result<(), CliError>
191where
192    I: IntoIterator<Item = OsString>,
193{
194    let args = args.into_iter().collect::<Vec<_>>();
195    if let Some(option) = misplaced_global_option(&args) {
196        let error = top_level_command().error(
197            ErrorKind::UnknownArgument,
198            format!("unexpected argument '{option}' found; put it before the command"),
199        );
200        return Err(CliError::Clap(error));
201    }
202    let matches = match parse_matches(top_level_command(), args) {
203        Ok(matches) => matches,
204        Err(error) if !error.use_stderr() => {
205            let _ = error.print();
206            return Ok(());
207        }
208        Err(error) => return Err(CliError::Clap(error)),
209    };
210    let global_icp = cli::clap::string_option(&matches, "icp");
211    let global_environment = cli::clap::string_option(&matches, "environment");
212
213    let (command, subcommand_matches) = matches
214        .subcommand()
215        .unwrap_or_else(|| unreachable!("Clap requires one top-level command"));
216    let mut tail = subcommand_matches
217        .get_many::<OsString>(DISPATCH_ARGS)
218        .map(|values| values.cloned().collect::<Vec<_>>())
219        .unwrap_or_default();
220    if let Some(conflict) =
221        global_environment_conflict(command, &tail, global_environment.as_deref())
222    {
223        return Err(CliError::Clap(
224            top_level_command().error(ErrorKind::InvalidValue, conflict.to_string()),
225        ));
226    }
227    apply_global_icp(command, &mut tail, global_icp);
228    apply_global_environment(command, &mut tail, global_environment);
229    let tail = tail.into_iter();
230
231    match command {
232        "admission" => admission::run(tail).map_err(CliError::from),
233        "app" => apps::run(tail).map_err(CliError::from),
234        "auth" => auth::run(tail).map_err(CliError::from),
235        "backup" => backup::run(tail).map_err(CliError::from),
236        "blob-storage" => blob_storage::run(tail).map_err(CliError::from),
237        "build" => build::run(tail).map_err(CliError::from),
238        "cycles" => cycles::run(tail).map_err(CliError::from),
239        "diagnostic" => diagnostic::run(tail).map_err(CliError::from),
240        "evidence" => evidence::run(tail).map_err(CliError::from),
241        "fleet" => fleet::run(tail).map_err(CliError::from),
242        "info" => info::run(tail).map_err(CliError::from),
243        "inspect" => inspect::run(tail).map_err(CliError::from),
244        "medic" => medic::run(tail).map_err(CliError::from),
245        "network" => network::run(tail).map_err(CliError::from),
246        "replica" => replica::run(tail).map_err(CliError::from),
247        "restore" => restore::run(tail).map_err(CliError::from),
248        "scaffold" => scaffold::run(tail).map_err(CliError::from),
249        "state" => state::run(tail).map_err(CliError::from),
250        "status" => status::run(tail).map_err(CliError::from),
251        "token" => token::run(tail).map_err(CliError::from),
252        "toolchain" => toolchain::run(tail).map_err(CliError::from),
253        _ => unreachable!("top-level dispatch command only defines known commands"),
254    }
255}
256
257#[must_use]
258pub const fn version_text() -> &'static str {
259    VERSION_TEXT
260}
261
262#[must_use]
263pub fn render_cli_error(error: &CliError) -> String {
264    match error {
265        CliError::BlobStorage(err) => err.json_error_report().unwrap_or_else(|| error.to_string()),
266        CliError::Build(build::BuildCommandError::Clap(err)) | CliError::Clap(err) => {
267            err.to_string().trim_end().to_string()
268        }
269        CliError::Inspect(err) if err.suppress_stderr() => String::new(),
270        CliError::Medic(err) if err.suppress_stderr() => String::new(),
271        CliError::State(err) if err.suppress_stderr() => String::new(),
272        _ => error.to_string(),
273    }
274}
275
276#[must_use]
277pub fn cli_error_exit_code(err: &CliError) -> i32 {
278    match err {
279        CliError::Auth(err) => i32::from(err.exit_code()),
280        CliError::BlobStorage(err) => i32::from(err.exit_code()),
281        CliError::Build(err) => err.exit_code(),
282        CliError::Clap(err) => err.exit_code(),
283        CliError::Info(err) => i32::from(err.exit_code()),
284        CliError::Inspect(err) => i32::from(err.exit_code()),
285        CliError::Medic(err) => i32::from(err.exit_code()),
286        CliError::State(err) => i32::from(err.exit_code()),
287        _ => 1,
288    }
289}
290
291#[cfg(test)]
292mod tests;
canic_cli/lib.rs

canic_cli/
lib.rs
