feat: colorized output + progress bars for HN launch (#18)

Author: markovejnovic

.github/workflows/ci.yml

@@ -37,7 +37,7 @@ jobs:
         run: cargo clippy --all-targets -p harmont-cli -- -D warnings
 
   python-lint:
-    name: harmont-py (pytest + ruff + mypy)
+    name: harmont-py (pytest + ruff + ty)
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
@@ -59,9 +59,9 @@ jobs:
         working-directory: dsls/harmont-py
         run: ruff check .
 
-      - name: mypy
+      - name: ty
         working-directory: dsls/harmont-py
-        run: mypy harmont
+        run: ty check harmont
 
       - name: pytest
         working-directory: dsls/harmont-py
@@ -97,7 +97,7 @@ jobs:
   dogfood:
     name: dogfood (hm run ci)
     runs-on: ubuntu-latest
-    timeout-minutes: 30
+    timeout-minutes: 45
     steps:
       - uses: actions/checkout@v4
 

.gitignore

@@ -3,6 +3,9 @@ __pycache__/
 *.py[codz]
 *$py.class
 
+# Node
+node_modules/
+
 # C extensions
 *.so
 

.harmont/ci.py

@@ -21,7 +21,7 @@ def shared_base() -> hm.Step:
 def rust_project(shared_base: hm.Target[hm.Step]) -> tuple[hm.Step, ...]:
     project = hm.rust.project(path=".", base=shared_base)
     return hm.group([
-        project.test(flags=("--lib",)),
+        project.test(flags=("--lib",), packages=("harmont-cli",)),
         project.clippy(),
         project.fmt(),
     ])

.harmont/ci.ts

@@ -27,7 +27,7 @@ const pipelines: PipelineDefinition[] = [
     slug: "ci",
     triggers: [push({ branch: "main" }), pullRequest({ branches: ["main"] })],
     pipeline: pipeline(
-      rustProject.test({ flags: ["--lib"] }),
+      rustProject.test({ flags: ["--lib"], packages: ["harmont-cli"] }),
       rustProject.clippy(),
       rustProject.fmt(),
       pyProject.lint(),

Cargo.lock

@@ -40,6 +40,12 @@ pub enum BuildEvent {
         step_id: Uuid,
         key: String,
         chain_idx: usize,
+        /// Key of this step's `BuildsIn` parent, if any. Lets renderers
+        /// nest progress bars to reflect the pipeline's DAG structure.
+        parent_key: Option<String>,
+        /// Human-readable name for display. Falls back to a truncated
+        /// command when no explicit label was set in the pipeline DSL.
+        display_name: String,
     },
     StepStart {
         step_id: Uuid,

crates/hm-plugin-protocol/src/events.rs

@@ -64,6 +64,8 @@ fn build_event_round_trip_all_variants() {
             step_id: Uuid::nil(),
             key: "a".into(),
             chain_idx: 0,
+            parent_key: None,
+            display_name: "a".into(),
         },
         BuildEvent::StepStart {
             step_id: Uuid::nil(),

crates/hm-plugin-protocol/tests/round_trip.rs

@@ -30,11 +30,13 @@ reqwest = { version = "0.13", default-features = false, features = ["rustls", "h
 serde = { version = "1", features = ["derive"] }
 serde_json = "1"
 toml = "0.8"
-owo-colors = { version = "4", features = ["supports-colors"] }
 dialoguer = { version = "0.11", features = ["fuzzy-select"] }
-indicatif = "0.17"
+owo-colors = "4"
+console = "0.16"
+indicatif = "0.18"
+tracing-indicatif = "0.3"
 comfy-table = { version = "7", features = ["custom_styling"] }
-console = "0.15"
+terminal_size = "0.4"
 tar = "0.4"
 flate2 = "1"
 fs2 = "0.4"
@@ -70,6 +72,7 @@ semver = { workspace = true }
 smart-default = { workspace = true }
 once_cell = "1"
 hex = "0.4"
+is_ci = "1.2.0"
 
 [features]
 default = []

crates/hm/Cargo.toml

@@ -37,4 +37,9 @@ pub struct RunArgs {
     /// `name`). Built-ins: `human`, `json`. Default: `human`.
     #[arg(long, value_name = "NAME", default_value = "human", global = false)]
     pub format: String,
+
+    /// Stream full build logs instead of showing progress bars.
+    /// Has no effect with `--format json`.
+    #[arg(long)]
+    pub logs: bool,
 }

crates/hm/src/cli/run.rs

@@ -22,7 +22,7 @@ fn decode_plan_to_wire(bytes: &[u8]) -> anyhow::Result<hm_pipeline_ir::PipelineG
 /// the resulting plan does not decode, the Docker daemon is unreachable,
 /// or the orchestrator surfaces an internal scheduler error. Non-zero
 /// step exit codes are returned as the `i32`, not as an Err.
-pub async fn handle(args: RunArgs, _ctx: RunContext) -> Result<i32> {
+pub async fn handle(args: RunArgs, ctx: RunContext) -> Result<i32> {
     let repo_root = match args.dir.clone() {
         Some(p) => p,
         None => std::env::current_dir().context("cannot determine current directory")?,
@@ -68,9 +68,19 @@ pub async fn handle(args: RunArgs, _ctx: RunContext) -> Result<i32> {
     runner_registry.register(Arc::new(DockerRunner), true);
     let runner_registry = Arc::new(runner_registry);
 
+    let use_logs = args.logs || std::env::var_os("CI").is_some_and(|v| !v.is_empty());
+
     let renderer: Box<dyn crate::runner::OutputRenderer> = match args.format.as_str() {
         "json" => Box::new(crate::output::json::JsonRenderer::new(std::io::stdout())),
-        _ => Box::new(crate::output::human::HumanRenderer::new(std::io::stderr())),
+        "human" if use_logs => Box::new(crate::output::human::HumanRenderer::new(
+            std::io::stderr(),
+            ctx.output.color_enabled(),
+        )),
+        "human" => Box::new(crate::output::progress::ProgressRenderer::new(
+            std::io::stderr(),
+            ctx.output.color_enabled(),
+        )),
+        other => anyhow::bail!("unknown --format '{other}'\n  available: human, json"),
     };
 
     let exit_code =