git.delta.rocks / jrsonnet / refs/commits / 81e3c77f96ff

1use std::{env::current_dir, time::Duration};23use crate::{command::CommandExt, host::Config};4use anyhow::Result;5use structopt::StructOpt;6use tokio::{process::Command, task::LocalSet, time::sleep};7use tracing::{error, field, info, info_span, warn, Instrument};89#[derive(StructOpt, Clone)]10pub struct BuildSystems {11	/// --builders arg for nix12	#[structopt(long)]13	builders: Option<String>,14	/// Jobs to run locally15	#[structopt(long)]16	jobs: Option<usize>,17	/// Do not continue on error18	#[structopt(long)]19	fail_fast: bool,20	#[structopt(long)]21	privileged_build: bool,22	#[structopt(subcommand)]23	subcommand: Subcommand,24	#[structopt(long)]25	show_trace: bool,26}2728enum UploadAction {29	Test,30	Boot,31	Switch,32}33impl UploadAction {34	fn name(&self) -> &'static str {35		match self {36			UploadAction::Test => "test",37			UploadAction::Boot => "boot",38			UploadAction::Switch => "switch",39		}40	}4142	pub(crate) fn should_switch_profile(&self) -> bool {43		matches!(self, Self::Switch | Self::Test)44	}45}4647enum PackageAction {48	SdImage,49}5051enum Action {52	Upload(Option<UploadAction>),53	Package(PackageAction),54}5556impl From<Subcommand> for Action {57	fn from(s: Subcommand) -> Self {58		match s {59			Subcommand::Upload => Self::Upload(None),60			Subcommand::Test => Self::Upload(Some(UploadAction::Test)),61			Subcommand::Boot => Self::Upload(Some(UploadAction::Boot)),62			Subcommand::Switch => Self::Upload(Some(UploadAction::Switch)),63			Subcommand::SdImage => Self::Package(PackageAction::SdImage),64		}65	}66}6768#[derive(StructOpt, Clone)]69enum Subcommand {70	/// Upload, but do not switch71	Upload,72	/// Upload + switch to built system until reboot73	Test,74	/// Upload + switch to built system after reboot75	Boot,76	/// Upload + test + boot77	Switch,7879	/// Build sd image80	SdImage,81}8283impl BuildSystems {84	async fn build_task(self, config: Config, host: String) -> Result<()> {85		info!("building");86		let built = {87			let dir = tempfile::tempdir()?;88			dir.path().to_owned()89		};9091		let mut nix_build = if self.privileged_build {92			let mut out = Command::new("sudo");93			out.arg("nix");94			out95		} else {96			Command::new("nix")97		};98		nix_build99			.args(&[100				"build",101				"--impure",102				"--json",103				// "--show-trace",104				"--no-link",105				"--out-link",106			])107			.arg(&built)108			.arg(config.configuration_attr_name(&format!(109				"configuredSystems.{}.config.system.build.toplevel",110				host111			)));112113		if self.show_trace {114			nix_build.arg("--show-trace");115		}116		if let Some(builders) = &self.builders {117			nix_build.arg("--builders").arg(builders);118		}119		if let Some(jobs) = &self.jobs {120			nix_build.arg("--max-jobs");121			nix_build.arg(format!("{}", jobs));122		}123		if !self.fail_fast {124			nix_build.arg("--keep-going");125		}126127		nix_build.run_nix().await?;128		let built = std::fs::canonicalize(built)?;129130		let action = Action::from(self.subcommand.clone());131132		match action {133			Action::Upload(action) => {134				if !config.is_local(&host) {135					info!("uploading system closure");136					let mut tries = 0;137					loop {138						match Command::new("nix")139							.args(&["copy", "--to"])140							.arg(format!("ssh://root@{}", host))141							.arg(&built)142							.inherit_stdio()143							.run_nix()144							.await145						{146							Ok(()) => break,147							Err(e) if tries < 3 => {148								tries += 1;149								warn!("Copy failure ({}/3): {}", tries, e);150								sleep(Duration::from_millis(5000)).await;151							}152							Err(e) => return Err(e),153						}154					}155				}156				if let Some(action) = action {157					if action.should_switch_profile() {158						info!("switching generation");159						config160							.command_on(&host, "nix-env", true)161							.args(&["-p", "/nix/var/nix/profiles/system", "--set"])162							.arg(&built)163							.inherit_stdio()164							.run()165							.await?;166					}167					info!("executing activation script");168					let mut switch_script = built.clone();169					switch_script.push("bin");170					switch_script.push("switch-to-configuration");171					config172						.command_on(&host, switch_script, true)173						.arg(action.name())174						.inherit_stdio()175						.run()176						.await?;177				}178			}179			Action::Package(PackageAction::SdImage) => {180				let mut out = current_dir()?;181				out.push(format!("sd-image-{}", host));182183				info!("building sd image to {:?}", out);184				let mut nix_build = if self.privileged_build {185					let mut out = Command::new("sudo");186					out.arg("nix");187					out188				} else {189					Command::new("nix")190				};191				nix_build192					.args(&["build", "--impure", "--no-link", "--out-link"])193					.arg(&out)194					.arg(config.configuration_attr_name(&format!(195						"configuredSystems.{}.config.system.build.sdImage",196						host,197					)));198				if let Some(builders) = &self.builders {199					nix_build.arg("--builders").arg(builders);200				}201				if let Some(jobs) = &self.jobs {202					nix_build.arg("--max-jobs");203					nix_build.arg(format!("{}", jobs));204				}205				if !self.fail_fast {206					nix_build.arg("--keep-going");207				}208209				nix_build.inherit_stdio().run_nix().await?;210			}211		};212		Ok(())213	}214215	pub async fn run(self, config: &Config) -> Result<()> {216		let hosts = config.list_hosts().await?;217		let set = LocalSet::new();218		let this = &self;219		for host in hosts.iter() {220			if config.should_skip(host) {221				continue;222			}223			let config = config.clone();224			let host = host.clone();225			let this = this.clone();226			let span = info_span!("deployment", host = field::display(&host));227			set.spawn_local(228				(async move {229					match this.build_task(config, host).await {230						Ok(_) => {}231						Err(e) => {232							error!("failed to deploy host: {}", e)233						}234					}235				})236				.instrument(span),237			);238		}239		set.await;240		Ok(())241	}242}