pretty cli output

Cargo.toml

@@ -6,3 +6,9 @@ members = ["crates/*"]
 license = "MIT"
 edition = "2021"
 authors = ["Miguel Piedrafita <[email protected]>"]
+
+[profile.release]
+lto = true
+strip = true
+opt-level = 3
+panic = "abort"

Dockerfile.cli

@@ -5,6 +5,6 @@ RUN cargo build --release --bin orbit-cli
 
 FROM --platform=linux/amd64 alpine AS runtime
 WORKDIR /orbit-cli
-COPY --from=builder /orbit-cli/target/x86_64-unknown-linux-musl/release/orbit-cli /usr/local/bin/orbit
+COPY --from=builder /orbit-cli/target/x86_64-unknown-linux-musl/release/orbit /usr/local/bin
 
 ENTRYPOINT ["/usr/local/bin/orbit"]

crates/cli/Cargo.toml

@@ -1,14 +1,26 @@
 [package]
 version = "0.1.0"
 name = "orbit-cli"
+categories = ["command-line-utilities"]
+
 edition.workspace = true
 license.workspace = true
 authors.workspace = true
 
 [dependencies]
 url = "2.5.2"
+log = "0.4.22"
+anyhow = "1.0.86"
+console = "0.15.8"
+pin-utils = "0.1.0"
 futures-util = "0.3.30"
-tokio = { version = "1.38.1", features = ["full"] }
+tokio = { version = "=1.29", features = ["full"] }
+fern = { version = "0.6.2", features = ["colored"] }
 orbit-types = { version = "0.1.0", path = "../types" }
 orbit-client = { version = "0.1.0", path = "../client" }
 clap = { version = "4.5.9", features = ["derive", "env"] }
+
+
+[[bin]]
+name = "orbit"
+path = "./src/main.rs"

crates/cli/src/main.rs

@@ -1,17 +1,31 @@
 #![warn(clippy::all, clippy::pedantic, clippy::nursery)]
 
+use anyhow::Result;
 use clap::{Parser, Subcommand};
 use futures_util::StreamExt;
 use orbit_client::Client;
 use orbit_types::{Log, Progress, Stage};
+use pin_utils::pin_mut;
 use url::Url;
 
+mod utils;
+
 #[derive(Debug, Parser)]
+#[clap(
+	name = "orbit",
+	about = "🪐 Trigger Orbit deploys from the command line.",
+	version,
+	author
+)]
 struct Cli {
 	/// URL to the Orbit instance.
 	#[arg(short, long, env = "ORBIT_URL")]
 	url: Url,
 
+	/// Enable debug mode
+	#[clap(short = 'D', long)]
+	pub debug: bool,
+
 	#[clap(subcommand)]
 	command: Commands,
 }
@@ -30,37 +44,51 @@ enum Commands {
 }
 
 #[tokio::main]
-async fn main() {
+async fn main() -> Result<()> {
 	let cli = Cli::parse();
+
+	// setup panic hook
+	utils::set_hook();
+	utils::logs(cli.debug);
+
 	let client = Client::new(cli.url);
 
-	match cli.command {
-		Commands::Deploy { slug, r#ref } => run_deploy(slug, r#ref, &client).await,
+	if let Err(error) = handle_command(cli.command, &client).await {
+		log::error!("{error}");
+		log::debug!("{error:#?}");
+		std::process::exit(1);
 	}
+
+	utils::clean_term();
+
+	Ok(())
 }
 
-async fn run_deploy(slug: String, r#ref: Option<String>, client: &Client) {
+async fn handle_command(commands: Commands, client: &Client) -> Result<()> {
+	match commands {
+		Commands::Deploy { slug, r#ref } => run_deploy(slug, r#ref, client).await,
+	}
+}
+
+async fn run_deploy(slug: String, r#ref: Option<String>, client: &Client) -> Result<()> {
 	let stream = client.deploy(&slug, r#ref.as_deref());
+	pin_mut!(stream);
+
+	while let Some(event) = stream.next().await {
+		match event? {
+			Ok(Progress::Log(log)) => match log {
+				Log::Info(message) => println!("{message}"),
+				Log::Error(message) => eprintln!("{message}"),
+			},
+			Ok(Progress::Stage(stage)) => match stage {
+				Stage::Starting => log::info!("Starting deployment"),
+				Stage::Downloaded => log::info!("Downloaded repository"),
+				Stage::DepsInstalled => log::info!("Installed dependencies"),
+				Stage::Deployed => log::info!("Deployed site"),
+			},
+			Err(error) => return Err(error.into()),
+		}
+	}
 
-	stream
-		.map(|result| match result {
-			Err(err) => panic!("{err}"),
-			Ok(event) => event,
-		})
-		.for_each(|event| async {
-			match event {
-				Ok(Progress::Log(log)) => match log {
-					Log::Info(message) => println!("{message}"),
-					Log::Error(message) => eprintln!("{message}"),
-				},
-				Ok(Progress::Stage(stage)) => match stage {
-					Stage::Starting => println!("Starting deployment"),
-					Stage::Downloaded => println!("Downloaded repository"),
-					Stage::DepsInstalled => println!("Installed dependencies"),
-					Stage::Deployed => println!("Deployed site"),
-				},
-				Err(error) => eprintln!("{error}"),
-			}
-		})
-		.await;
+	Ok(())
 }