ito-cli 0.1.23

Command-line interface for Ito
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

use std::path::Path;
use vergen_gitcl::{BuildBuilder, Emitter, GitclBuilder};

fn main() {
    println!("cargo:rerun-if-env-changed=ITO_LOCAL_VERSION_STAMP");
    println!("cargo:rerun-if-env-changed=ITO_LOCAL_VERSION");

    // Emit git and build metadata via vergen
    // This sets VERGEN_GIT_SHA, VERGEN_GIT_BRANCH, VERGEN_GIT_DIRTY, VERGEN_BUILD_TIMESTAMP
    let build = BuildBuilder::default().build_timestamp(true).build();
    let gitcl = GitclBuilder::default()
        .sha(true)
        .branch(true)
        .dirty(true)
        .build();

    // In bare/control-repo + worktree layouts, build scripts may run with a CWD
    // that is not a git worktree (e.g., the bare repo root). Force vergen-gitcl
    // to run git commands from the crate directory so branch/SHA metadata is
    // available during local development builds.
    let manifest_dir = std::env::var_os("CARGO_MANIFEST_DIR").map(std::path::PathBuf::from);

    let mut emitter = Emitter::default();
    if let Ok(build) = build {
        let _ = emitter.add_instructions(&build);
    }
    if let Ok(mut gitcl) = gitcl {
        if let Some(dir) = manifest_dir {
            let _ = gitcl.at_path(dir);
        }
        let _ = emitter.add_instructions(&gitcl);
    }
    let _ = emitter.emit();

    // Keep `ito --version` in sync with the workspace version.
    // This avoids touching `crates/ito-cli/Cargo.toml` (release-please managed)
    // while still reflecting local version bumps in the root workspace `Cargo.toml`.
    let workspace_manifest = Path::new("../../../Cargo.toml");
    println!("cargo:rerun-if-changed={}", workspace_manifest.display());

    let Ok(contents) = std::fs::read_to_string(workspace_manifest) else {
        return;
    };
    let Some(version) = workspace_package_version(&contents) else {
        return;
    };

    let version = append_local_suffix(version);
    println!("cargo:rustc-env=ITO_WORKSPACE_VERSION={version}");
}

fn append_local_suffix(version: String) -> String {
    let Ok(stamp) = std::env::var("ITO_LOCAL_VERSION_STAMP") else {
        return append_local_flag(version);
    };

    let stamp = stamp.trim();
    if stamp.is_empty() {
        return append_local_flag(version);
    }
    if !stamp.chars().all(|c| c.is_ascii_digit()) || stamp.len() != 12 {
        return append_local_flag(version);
    }

    format!("{version}-local.{stamp}")
}

fn append_local_flag(version: String) -> String {
    let Some(flag) = std::env::var_os("ITO_LOCAL_VERSION") else {
        return version;
    };
    if flag.is_empty() {
        return version;
    }
    format!("{version}-local")
}

fn workspace_package_version(contents: &str) -> Option<String> {
    let mut in_section = false;
    for line in contents.lines() {
        let trimmed = line.trim();

        if trimmed == "[workspace.package]" {
            in_section = true;
            continue;
        }

        if in_section && trimmed.starts_with('[') && trimmed.ends_with(']') {
            break;
        }

        if !in_section {
            continue;
        }

        let Some(rest) = trimmed.strip_prefix("version") else {
            continue;
        };
        let rest = rest.trim_start();
        let Some(rest) = rest.strip_prefix('=') else {
            continue;
        };
        let value = rest.trim();
        let value = value.strip_prefix('"')?.strip_suffix('"')?;
        if value.trim().is_empty() {
            return None;
        }
        return Some(value.to_string());
    }
    None
}