From 715db25581d6af538532033ee6f761b4671a22c4 Mon Sep 17 00:00:00 2001 From: clerie Date: Sat, 1 Feb 2025 00:55:10 +0100 Subject: [PATCH] Init repo --- .gitignore | 1 + Cargo.lock | 303 ++++++++++++++++++++++++++++++++++++++++++ Cargo.toml | 10 ++ src/bin/scan-flake.rs | 189 ++++++++++++++++++++++++++ 4 files changed, 503 insertions(+) create mode 100644 .gitignore create mode 100644 Cargo.lock create mode 100644 Cargo.toml create mode 100644 src/bin/scan-flake.rs diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ea8c4bf --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +/target diff --git a/Cargo.lock b/Cargo.lock new file mode 100644 index 0000000..b97b0f2 --- /dev/null +++ b/Cargo.lock @@ -0,0 +1,303 @@ +# This file is automatically @generated by Cargo. +# It is not intended for manual editing. +version = 3 + +[[package]] +name = "anstream" +version = "0.6.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8acc5369981196006228e28809f761875c0327210a891e941f4c683b3a99529b" +dependencies = [ + "anstyle", + "anstyle-parse", + "anstyle-query", + "anstyle-wincon", + "colorchoice", + "is_terminal_polyfill", + "utf8parse", +] + +[[package]] +name = "anstyle" +version = "1.0.10" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "55cc3b69f167a1ef2e161439aa98aed94e6028e5f9a59be9a6ffb47aef1651f9" + +[[package]] +name = "anstyle-parse" +version = "0.2.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "3b2d16507662817a6a20a9ea92df6652ee4f94f914589377d69f3b21bc5798a9" +dependencies = [ + "utf8parse", +] + +[[package]] +name = "anstyle-query" +version = "1.1.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "79947af37f4177cfead1110013d678905c37501914fba0efea834c3fe9a8d60c" +dependencies = [ + "windows-sys", +] + +[[package]] +name = "anstyle-wincon" +version = "3.0.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ca3534e77181a9cc07539ad51f2141fe32f6c3ffd4df76db8ad92346b003ae4e" +dependencies = [ + "anstyle", + "once_cell", + "windows-sys", +] + +[[package]] +name = "anyhow" +version = "1.0.95" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "34ac096ce696dc2fcabef30516bb13c0a68a11d30131d3df6f04711467681b04" + +[[package]] +name = "clap" +version = "4.5.27" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "769b0145982b4b48713e01ec42d61614425f27b7058bda7180a3a41f30104796" +dependencies = [ + "clap_builder", + "clap_derive", +] + +[[package]] +name = "clap_builder" +version = "4.5.27" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1b26884eb4b57140e4d2d93652abfa49498b938b3c9179f9fc487b0acc3edad7" +dependencies = [ + "anstream", + "anstyle", + "clap_lex", + "strsim", +] + +[[package]] +name = "clap_derive" +version = "4.5.24" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "54b755194d6389280185988721fffba69495eed5ee9feeee9a599b53db80318c" +dependencies = [ + "heck", + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "clap_lex" +version = "0.7.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f46ad14479a25103f283c0f10005961cf086d8dc42205bb44c46ac563475dca6" + +[[package]] +name = "colorchoice" +version = "1.0.3" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5b63caa9aa9397e2d9480a9b13673856c78d8ac123288526c37d7839f2a86990" + +[[package]] +name = "flake-tracker" +version = "0.1.0" +dependencies = [ + "anyhow", + "clap", + "serde", + "serde_json", +] + +[[package]] +name = "heck" +version = "0.5.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea" + +[[package]] +name = "is_terminal_polyfill" +version = "1.70.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7943c866cc5cd64cbc25b2e01621d07fa8eb2a1a23160ee81ce38704e97b8ecf" + +[[package]] +name = "itoa" +version = "1.0.14" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d75a2a4b1b190afb6f5425f10f6a8f959d2ea0b9c2b1d79553551850539e4674" + +[[package]] +name = "memchr" +version = "2.7.4" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3" + +[[package]] +name = "once_cell" +version = "1.20.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1261fe7e33c73b354eab43b1273a57c8f967d0391e80353e51f764ac02cf6775" + +[[package]] +name = "proc-macro2" +version = "1.0.93" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "60946a68e5f9d28b0dc1c21bb8a97ee7d018a8b322fa57838ba31cc878e22d99" +dependencies = [ + "unicode-ident", +] + +[[package]] +name = "quote" +version = "1.0.38" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0e4dccaaaf89514f546c693ddc140f729f958c247918a13380cccc6078391acc" +dependencies = [ + "proc-macro2", +] + +[[package]] +name = "ryu" +version = "1.0.19" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "6ea1a2d0a644769cc99faa24c3ad26b379b786fe7c36fd3c546254801650e6dd" + +[[package]] +name = "serde" +version = "1.0.217" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "02fc4265df13d6fa1d00ecff087228cc0a2b5f3c0e87e258d8b94a156e984c70" +dependencies = [ + "serde_derive", +] + +[[package]] +name = "serde_derive" +version = "1.0.217" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "5a9bf7cf98d04a2b28aead066b7496853d4779c9cc183c440dbac457641e19a0" +dependencies = [ + "proc-macro2", + "quote", + "syn", +] + +[[package]] +name = "serde_json" +version = "1.0.138" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d434192e7da787e94a6ea7e9670b26a036d0ca41e0b7efb2676dd32bae872949" +dependencies = [ + "itoa", + "memchr", + "ryu", + "serde", +] + +[[package]] +name = "strsim" +version = "0.11.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f" + +[[package]] +name = "syn" +version = "2.0.96" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "d5d0adab1ae378d7f53bdebc67a39f1f151407ef230f0ce2883572f5d8985c80" +dependencies = [ + "proc-macro2", + "quote", + "unicode-ident", +] + +[[package]] +name = "unicode-ident" +version = "1.0.16" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "a210d160f08b701c8721ba1c726c11662f877ea6b7094007e1ca9a1041945034" + +[[package]] +name = "utf8parse" +version = "0.2.2" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "06abde3611657adf66d383f00b093d7faecc7fa57071cce2578660c9f1010821" + +[[package]] +name = "windows-sys" +version = "0.59.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "1e38bc4d79ed67fd075bcc251a1c39b32a1776bbe92e5bef1f0bf1f8c531853b" +dependencies = [ + "windows-targets", +] + +[[package]] +name = "windows-targets" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973" +dependencies = [ + "windows_aarch64_gnullvm", + "windows_aarch64_msvc", + "windows_i686_gnu", + "windows_i686_gnullvm", + "windows_i686_msvc", + "windows_x86_64_gnu", + "windows_x86_64_gnullvm", + "windows_x86_64_msvc", +] + +[[package]] +name = "windows_aarch64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3" + +[[package]] +name = "windows_aarch64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469" + +[[package]] +name = "windows_i686_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b" + +[[package]] +name = "windows_i686_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66" + +[[package]] +name = "windows_i686_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66" + +[[package]] +name = "windows_x86_64_gnu" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78" + +[[package]] +name = "windows_x86_64_gnullvm" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d" + +[[package]] +name = "windows_x86_64_msvc" +version = "0.52.6" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec" diff --git a/Cargo.toml b/Cargo.toml new file mode 100644 index 0000000..d82a1da --- /dev/null +++ b/Cargo.toml @@ -0,0 +1,10 @@ +[package] +name = "flake-tracker" +version = "0.1.0" +edition = "2021" + +[dependencies] +anyhow = "1.0.95" +clap = { version = "4.5.23", features = ["derive"] } +serde = { version = "1.0.217", features = ["derive"] } +serde_json = "1.0.138" diff --git a/src/bin/scan-flake.rs b/src/bin/scan-flake.rs new file mode 100644 index 0000000..fa079f6 --- /dev/null +++ b/src/bin/scan-flake.rs @@ -0,0 +1,189 @@ +use anyhow::{ + anyhow, + Context, + Result, +}; +use clap::{ + Parser, +}; +use serde::{ + Deserialize, +}; +use std::collections::HashMap; +use std::process::Command; + +#[derive(Deserialize, Debug)] +struct FlakeMetadata { + lastModified: i64, + locked: FlakeLockedInfo, + locks: FlakeLocks, + original: FlakeSource, + path: String, + resolved: FlakeSource, + revision: String, +} + +#[derive(Deserialize, Debug)] +struct FlakeSource { + owner: Option, + repo: Option, + r#ref: Option, + rev: Option, + r#type: String, + url: Option, +} + +fn assemble_flake_git_uri(flake_source: &FlakeSource) -> Result { + let mut out = String::new(); + out.push_str("git+"); + out.push_str(&flake_source.url.clone() + .context("Flake git uri does not contain an url")?); + + let mut query: Vec = Vec::new(); + + if let Some(r#ref) = &flake_source.r#ref { + query.push(format!("ref={}", r#ref)); + } + + if let Some(rev) = &flake_source.rev { + query.push(format!("rev={}", rev)); + } + + if query.len() > 0 { + out.push_str("?"); + out.push_str(&query.join("&")); + } + + Ok(out) +} + +fn assemble_flake_github_uri(flake_source: &FlakeSource) -> Result { + let mut out = String::new(); + out.push_str("github:"); + out.push_str(&flake_source.owner.clone() + .context("Flake github uri does not contain an owner")?); + out.push_str("/"); + out.push_str(&flake_source.repo.clone() + .context("Flake github uri does not contain an repo")?); + + if let Some(rev) = &flake_source.rev { + out.push_str("/"); + out.push_str(&rev); + } + + Ok(out) +} + +fn assemble_flake_tarball_uri(flake_source: &FlakeSource) -> Result { + let mut out = String::new(); + out.push_str(&flake_source.url.clone() + .context("Flake tarball uri does not contain an url")?); + + if let Some(rev) = &flake_source.rev { + out.push_str("?rev="); + out.push_str(rev); + } + + Ok(out) +} + +trait FlakeUri { + fn flake_uri(&self) -> Result; +} + +impl FlakeUri for FlakeSource { + fn flake_uri(&self) -> Result { + match self.r#type.as_str() { + "git" => assemble_flake_git_uri(self), + "github" => assemble_flake_github_uri(self), + "tarball" => assemble_flake_tarball_uri(self), + other => Err(anyhow!("Unsupported flake uri type {}", other)), + } + } +} + +#[derive(Deserialize, Debug)] +struct FlakeLockedInfo { + lastModified: i64, + narHash: String, + owner: Option, + repo: Option, + r#ref: Option, + rev: Option, + r#type: String, + url: Option, +} + +impl FlakeUri for FlakeLockedInfo { + fn flake_uri(&self) -> Result { + Into::::into(self).flake_uri() + } +} + +impl Into for &FlakeLockedInfo { + fn into(self) -> FlakeSource { + FlakeSource { + owner: self.owner.clone(), + repo: self.repo.clone(), + r#ref: self.r#ref.clone(), + rev: self.rev.clone(), + r#type: self.r#type.clone(), + url: self.url.clone(), + } + } +} + +#[derive(Deserialize, Debug)] +struct FlakeLocks { + nodes: HashMap, + root: String, + version: u64, +} + +#[derive(Deserialize, Debug)] +#[serde(untagged)] +enum FlakeLocksNodeInputs { + String(String), + Array(Vec), +} + +#[derive(Deserialize, Debug)] +struct FlakeLocksNode { + inputs: Option>, + locked: Option, + original: Option, +} + +#[derive(Parser)] +#[command(version, about, long_about = None)] +struct Cli { + flake_uri: String, +} + +fn main() -> Result<()> { + let cli = Cli::parse(); + + let flake_metadata_raw = Command::new("nix") + .arg("flake") + .arg("metadata") + .arg("--json") + .arg(cli.flake_uri) + .output() + .context("Failed to fetch flake metadata")?; + + let flake_metadata: FlakeMetadata = serde_json::from_slice(&flake_metadata_raw.stdout) + .context("Failed to parse flake metadata")?; + + println!("{}", flake_metadata.original.flake_uri()?); + + for (input_name, input_data) in &flake_metadata.locks.nodes { + if let Some(original_flake_source) = &input_data.original { + println!("{}", original_flake_source.flake_uri()?); + } + if let Some(locked_flake_source) = &input_data.locked { + println!("{}", locked_flake_source.flake_uri()?); + } + } + + Ok(()) +}