clerie/flake-tracker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Move flake scanning to module

Browse Source
This commit is contained in:
clerie 2025-02-10 17:30:24 +01:00
parent 7a4fbaf74a
commit f6733f86a7
3 changed files with 87 additions and 69 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
src/bin/scan-flake.rs
View File

@ -2,25 +2,17 @@ use anyhow::{
Context,
Result,
};
use chrono::{
Utc,
};
use clap::{
Parser,
};
use flake_tracker::{
flake::{
FlakeLocksNodeInputs,
FlakeMetadata,
FlakeUri,
scan::{
scan_flake,
},
storage::{
InputRow,
RevisionRow,
Storage,
},
};
use std::process::Command;
#[derive(Parser)]
#[command(version, about, long_about = None)]
@ -32,69 +24,11 @@ struct Cli {
async fn main() -> Result<()> {
let cli = Cli::parse();
let scan_time = Utc::now().timestamp();
let storage = Storage::connect("sqlite://flake-tracker.db")
.await
.context("Failed to connect to database")?;
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")?;
let revision_row = RevisionRow {
revision_uri: flake_metadata.locked.flake_uri()?.clone(),
flake_uri: Some(flake_metadata.resolved.flake_uri()?.clone()),
nix_store_path: Some(flake_metadata.path.clone()),
nar_hash: Some(flake_metadata.locked.narHash.clone()),
last_modified: Some(flake_metadata.locked.lastModified.clone()),
tracker_last_scanned: Some(scan_time.clone()),
};
storage.set_revision(revision_row)
.await?;
let locks_root_name = &flake_metadata.locks.root;
let locks_root_node = flake_metadata.locks.nodes.get(locks_root_name)
.context("Failed to get locks root node")?;
for (input_name, locks_input_name) in locks_root_node.inputs.clone().context("No inputs found for flake")? {
if let FlakeLocksNodeInputs::String(locks_input_name) = locks_input_name {
let locks_input_node = flake_metadata.locks.nodes.get(&locks_input_name)
.context("Failed to find lock of input")?;
let input_row = InputRow {
revision_uri: flake_metadata.locked.flake_uri()?.clone(),
input_name: input_name.clone(),
locked_revision_uri: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.flake_uri()?),
locked_flake_uri: Some(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?),
locked_nar_hash: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.narHash),
last_modified: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.lastModified),
};
storage.set_input(input_row)
.await?;
let revision_row = RevisionRow {
revision_uri: locks_input_node.locked.clone().context("Unexpected missing lock")?.flake_uri()?.clone(),
flake_uri: Some(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?.clone()),
nix_store_path: None,
nar_hash: None,
last_modified: None,
tracker_last_scanned: None,
};
storage.set_revision_exist(revision_row)
.await?;
}
}
scan_flake(storage, &cli.flake_uri).await?;
Ok(())
}

1
src/lib.rs
View File

@ -1,4 +1,5 @@
pub mod flake;
pub mod scan;
pub mod storage;
pub mod templates;
pub mod utils;

83
src/scan.rs Normal file
View File

@ -0,0 +1,83 @@
use anyhow::{
Context,
Result,
};
use chrono::{
Utc,
};
use crate::{
flake::{
FlakeLocksNodeInputs,
FlakeMetadata,
FlakeUri,
},
storage::{
InputRow,
RevisionRow,
Storage,
},
};
use std::process::Command;
pub async fn scan_flake(storage: Storage, flake_uri: &str) -> Result<()> {
let scan_time = Utc::now().timestamp();
let flake_metadata_raw = Command::new("nix")
.arg("flake")
.arg("metadata")
.arg("--json")
.arg(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")?;
let revision_row = RevisionRow {
revision_uri: flake_metadata.locked.flake_uri()?.clone(),
flake_uri: Some(flake_metadata.resolved.flake_uri()?.clone()),
nix_store_path: Some(flake_metadata.path.clone()),
nar_hash: Some(flake_metadata.locked.narHash.clone()),
last_modified: Some(flake_metadata.locked.lastModified.clone()),
tracker_last_scanned: Some(scan_time.clone()),
};
storage.set_revision(revision_row)
.await?;
let locks_root_name = &flake_metadata.locks.root;
let locks_root_node = flake_metadata.locks.nodes.get(locks_root_name)
.context("Failed to get locks root node")?;
for (input_name, locks_input_name) in locks_root_node.inputs.clone().context("No inputs found for flake")? {
if let FlakeLocksNodeInputs::String(locks_input_name) = locks_input_name {
let locks_input_node = flake_metadata.locks.nodes.get(&locks_input_name)
.context("Failed to find lock of input")?;
let input_row = InputRow {
revision_uri: flake_metadata.locked.flake_uri()?.clone(),
input_name: input_name.clone(),
locked_revision_uri: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.flake_uri()?),
locked_flake_uri: Some(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?),
locked_nar_hash: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.narHash),
last_modified: Some(locks_input_node.locked.clone().context("Unexpected missing lock")?.lastModified),
};
storage.set_input(input_row)
.await?;
let revision_row = RevisionRow {
revision_uri: locks_input_node.locked.clone().context("Unexpected missing lock")?.flake_uri()?.clone(),
flake_uri: Some(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?.clone()),
nix_store_path: None,
nar_hash: None,
last_modified: None,
tracker_last_scanned: None,
};
storage.set_revision_exist(revision_row)
.await?;
}
}
Ok(())
}