s0me1newithhand7s/system-manager
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Be a bit more flexible regarding the flake-uri argument.

Browse source
This commit is contained in:
R-VdP 2023-03-15 19:51:56 +01:00
parent ed99b39f6c
commit 25ac812c53
No known key found for this signature in database
1 changed files with 32 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

51
src/generate.rs
View file

@ -1,5 +1,6 @@
use anyhow::{Context, Result}; use anyhow::{Context, Result};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use std::borrow::Cow;
use std::collections::HashMap; use std::collections::HashMap;
use std::fs::DirBuilder; use std::fs::DirBuilder;
use std::path::Path; use std::path::Path;
@ -52,39 +53,51 @@ fn create_gcroot(gcroot_path: &str, profile_path: &Path) -> Result<()> {
} }
pub fn build(flake_uri: &str) -> Result<StorePath> { pub fn build(flake_uri: &str) -> Result<StorePath> {
let flake_attr = find_flake_attr(flake_uri)?; let full_flake_uri = find_flake_attr(flake_uri)?;
log::info!("Building new system-manager generation..."); log::info!("Building new system-manager generation...");
log::info!("Running nix build..."); log::info!("Running nix build...");
let store_path = run_nix_build(flake_uri, &flake_attr).and_then(get_store_path)?; let store_path = run_nix_build(full_flake_uri.as_ref()).and_then(get_store_path)?;
log::info!("Build system-manager profile {store_path}"); log::info!("Build system-manager profile {store_path}");
Ok(store_path) Ok(store_path)
} }
fn find_flake_attr(flake_uri: &str) -> Result<String> { fn find_flake_attr(flake_uri: &str) -> Result<Cow<'_, str>> {
let hostname = nix::unistd::gethostname()?; if flake_uri.contains('#') {
let flake_attr = format!("{FLAKE_ATTR}.{}", hostname.to_string_lossy()); let status = try_flake_attr(flake_uri)?;
return if status {
Ok(flake_uri.into())
} else {
anyhow::bail!(
"Explicitly provided flake URI {} does not seem to point to a valid system-manager configuration.",
flake_uri
)
};
}
let status = try_flake_attr(flake_uri, &flake_attr)?; let hostname = nix::unistd::gethostname()?;
let full_uri = format!("{flake_uri}#{FLAKE_ATTR}.{}", hostname.to_string_lossy());
let status = try_flake_attr(&full_uri)?;
if status { if status {
return Ok(flake_attr); return Ok(full_uri.into());
} else { } else {
let flake_attr = format!("{FLAKE_ATTR}.default"); let full_uri = format!("{flake_uri}#{FLAKE_ATTR}.default");
let status = try_flake_attr(flake_uri, &flake_attr)?; let status = try_flake_attr(&full_uri)?;
if status { if status {
return Ok(flake_attr); return Ok(full_uri.into());
}; };
}; };
anyhow::bail!("No suitable flake attribute found, giving up."); anyhow::bail!("No suitable flake attribute found, giving up.");
} }
fn try_flake_attr(flake_uri: &str, flake_attr: &str) -> Result<bool> { fn try_flake_attr(flake_uri: &str) -> Result<bool> {
log::info!("Trying flake attribute: {flake_uri}#{flake_attr}..."); log::info!("Trying flake URI: {flake_uri}...");
let status = try_nix_eval(flake_uri, flake_attr)?; let status = try_nix_eval(flake_uri)?;
if status.success() { if status.success() {
log::info!("Success, using {flake_uri}#{flake_attr}"); log::info!("Success, using {flake_uri}");
} else { } else {
log::info!("Attribute {flake_uri}#{flake_attr} not found in flake."); log::info!("Attribute {flake_uri} not found in flake.");
}; };
Ok(status.success()) Ok(status.success())
} }
@ -118,10 +131,10 @@ fn parse_nix_build_output(output: String) -> Result<StorePath> {
anyhow::bail!("Multiple build results were returned, we cannot handle that yet.") anyhow::bail!("Multiple build results were returned, we cannot handle that yet.")
} }
fn run_nix_build(flake_uri: &str, flake_attr: &str) -> Result<process::Output> { fn run_nix_build(flake_uri: &str) -> Result<process::Output> {
let output = process::Command::new("nix") let output = process::Command::new("nix")
.arg("build") .arg("build")
.arg(format!("{flake_uri}#{flake_attr}")) .arg(flake_uri)
.arg("--json") .arg("--json")
// Nix outputs progress info on stderr and the final output on stdout, // Nix outputs progress info on stderr and the final output on stdout,
// so we inherit and output stderr directly to the terminal, but we // so we inherit and output stderr directly to the terminal, but we
@ -131,10 +144,10 @@ fn run_nix_build(flake_uri: &str, flake_attr: &str) -> Result<process::Output> {
Ok(output) Ok(output)
} }
fn try_nix_eval(flake_uri: &str, flake_attr: &str) -> Result<process::ExitStatus> { fn try_nix_eval(flake_uri: &str) -> Result<process::ExitStatus> {
let status = process::Command::new("nix") let status = process::Command::new("nix")
.arg("eval") .arg("eval")
.arg(format!("{flake_uri}#{flake_attr}")) .arg(flake_uri)
.arg("--json") .arg("--json")
.stdout(process::Stdio::null()) .stdout(process::Stdio::null())
.stderr(process::Stdio::null()) .stderr(process::Stdio::null())