feat: add sha

Author: gadomski

Cargo.lock

@@ -41,6 +41,9 @@ tokio = { version = "1.44.0", features = ["rt-multi-thread"] }
 pyo3-log = "0.12.1"
 tracing = "0.1.41"
 
+[build-dependencies]
+cargo-lock = "10"
+
 [patch.crates-io]
 # TODO unpatch cql2 when we upgrade geoarrow to use geo v0.30
 cql2 = { git = "https://github.com/gadomski/cql2-rs", branch = "geo-v0.29" }

Cargo.toml

@@ -0,0 +1,12 @@
+use cargo_lock::Lockfile;
+
+fn main() {
+    let lockfile = Lockfile::load("Cargo.lock").unwrap();
+    let package = lockfile
+        .packages
+        .into_iter()
+        .find(|package| package.name.as_str() == "rustac")
+        .unwrap();
+    let precise = package.source.unwrap().precise().unwrap().to_string();
+    println!("cargo:rustc-env=RUSTAC_SHA={}", precise);
+}

build.rs

@@ -453,3 +453,12 @@ def version(
         >>> rustac.version("duckdb")
         "1.0.0"
     """
+
+def sha() -> str:
+    """
+    Returns the SHA of the underlying rustac crate.
+
+    Examples:
+        >>> rustac.sha()
+        "4d6a60a3df1386922285191aba95a76ec704a8b4"
+    """

rustac.pyi

@@ -31,6 +31,7 @@ fn rustac(py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(read::read, m)?)?;
     m.add_function(wrap_pyfunction!(search::search, m)?)?;
     m.add_function(wrap_pyfunction!(search::search_to, m)?)?;
+    m.add_function(wrap_pyfunction!(version::sha, m)?)?;
     m.add_function(wrap_pyfunction!(version::version, m)?)?;
     m.add_function(wrap_pyfunction!(walk::walk, m)?)?;
     m.add_function(wrap_pyfunction!(write::write, m)?)?;

src/lib.rs

@@ -17,3 +17,9 @@ pub fn version(name: Option<String>) -> Option<String> {
         Some(env!("CARGO_PKG_VERSION").to_string())
     }
 }
+
+#[pyfunction]
+pub fn sha() -> String {
+    // This environment variable is set in the build script
+    env!("RUSTAC_SHA").to_string()
+}

src/version.rs

@@ -6,3 +6,7 @@ def test_version() -> None:
     assert rustac.version("stac") is not None
     assert rustac.version("stac-api") is not None
     assert rustac.version("stac-duckdb") is not None
+
+
+def test_sha() -> None:
+    assert rustac.sha() is not None