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

Connect to db via storage module

Browse Source
This commit is contained in:
clerie 2025-02-09 21:24:57 +01:00
parent fa489bf7bc
commit 3d8867d218
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,9 +14,9 @@ use flake_tracker::{
FlakeMetadata,
FlakeUri,
},
};
use sqlx::{
sqlite::SqlitePoolOptions,
storage::{
Storage,
},
};
use std::process::Command;
@ -32,7 +32,10 @@ async fn main() -> Result<()> {
let scan_time = Utc::now().timestamp();
let db = SqlitePoolOptions::new().connect("sqlite://flake-tracker.db").await?;
let storage = Storage::connect("sqlite://flake-tracker.db")
.await
.context("Failed to connect to database")?;
let flake_metadata_raw = Command::new("nix")
.arg("flake")
@ -60,7 +63,7 @@ async fn main() -> Result<()> {
.bind(&flake_metadata.locked.narHash)
.bind(&flake_metadata.locked.lastModified)
.bind(&scan_time)
.execute(&db).await?;
.execute(&storage.db).await?;
let locks_root_name = &flake_metadata.locks.root;
let locks_root_node = flake_metadata.locks.nodes.get(locks_root_name)
@ -86,7 +89,7 @@ async fn main() -> Result<()> {
.bind(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?)
.bind(locks_input_node.locked.clone().context("Unexpected missing lock")?.narHash)
.bind(locks_input_node.locked.clone().context("Unexpected missing lock")?.lastModified)
.execute(&db).await?;
.execute(&storage.db).await?;
sqlx::query("INSERT INTO revisions (revision_uri, flake_uri)
VALUES (?, ?)
@ -94,7 +97,7 @@ async fn main() -> Result<()> {
")
.bind(locks_input_node.locked.clone().context("Unexpected missing lock")?.flake_uri()?)
.bind(locks_input_node.original.clone().context("Unexpected missing lock")?.flake_uri()?)
.execute(&db).await?;
.execute(&storage.db).await?;
}
}