1use std::{2 cell::{Ref, RefCell, RefMut},3 env::current_dir,4 ffi::{OsStr, OsString},5 ops::Deref,6 path::PathBuf,7 process::Command,8 sync::Arc,9};1011use anyhow::Result;12use clap::Clap;1314use crate::{command::CommandExt, fleetdata::FleetData};1516pub struct FleetConfigInternals {17 pub directory: PathBuf,18 pub opts: FleetOpts,19 pub data: RefCell<FleetData>,20}2122#[derive(Clone)]23pub struct Config(Arc<FleetConfigInternals>);2425impl Deref for Config {26 type Target = FleetConfigInternals;2728 fn deref(&self) -> &Self::Target {29 &self.030 }31}3233impl Config {34 pub fn should_skip(&self, host: &str) -> bool {35 if !self.opts.skip.is_empty() {36 self.opts.skip.iter().any(|h| h as &str == host)37 } else if !self.opts.only.is_empty() {38 !self.opts.only.iter().any(|h| h as &str == host)39 } else {40 false41 }42 }43 pub fn is_local(&self, host: &str) -> bool {44 self.opts.localhost.as_ref().map(|s| s as &str) == Some(host)45 }4647 pub fn command_on(&self, host: &str, program: impl AsRef<OsStr>, sudo: bool) -> Command {48 if self.is_local(host) {49 if sudo {50 let mut cmd = Command::new("sudo");51 cmd.arg(program);52 cmd53 } else {54 Command::new(program)55 }56 } else {57 let mut cmd = Command::new("ssh");58 cmd.arg(host).arg("--");59 if sudo {60 cmd.arg("sudo");61 }62 cmd.arg(program);63 cmd64 }65 }6667 pub fn full_attr_name(&self, attr_name: &str) -> OsString {68 let mut str = self.directory.as_os_str().to_owned();69 str.push("#");70 str.push(attr_name);7172 println!("{:?}", str);73 str74 }7576 pub fn list_hosts(&self) -> Result<Vec<String>> {77 Command::new("nix")78 .arg("eval")79 .arg(self.full_attr_name("fleetConfigurations.default.configuredHosts"))80 .args(&["--apply", "builtins.attrNames", "--json", "--show-trace"])81 .inherit_stdio()82 .run_json()83 }8485 pub fn data(&self) -> Ref<FleetData> {86 self.data.borrow()87 }88 pub fn data_mut(&self) -> RefMut<FleetData> {89 self.data.borrow_mut()90 }9192 pub fn save(&self) -> Result<()> {93 let mut fleet_data_path = self.directory.clone();94 fleet_data_path.push("fleet.nix");95 let data = nixlike::serialize(&self.data() as &FleetData)?;96 std::fs::write(97 fleet_data_path,98 format!(99 "# This file contains fleet state and shouldn't be edited by hand\n\n{}\n",100 data101 ),102 )?;103 Ok(())104 }105}106107#[derive(Clap, Clone)]108#[clap(group = clap::ArgGroup::new("target_hosts"))]109pub struct FleetOpts {110 111 #[clap(long, number_of_values = 1, group = "target_hosts")]112 only: Vec<String>,113114 115 #[clap(long, number_of_values = 1, group = "target_hosts")]116 skip: Vec<String>,117118 119 #[clap(long)]120 pub localhost: Option<String>,121}122123impl FleetOpts {124 pub fn build(mut self) -> Result<Config> {125 if self.localhost.is_none() {126 self.localhost127 .replace(hostname::get().unwrap().to_str().unwrap().to_owned());128 }129 let directory = current_dir()?;130131 let mut fleet_data_path = directory.clone();132 fleet_data_path.push("fleet.nix");133 let bytes = std::fs::read_to_string(fleet_data_path)?;134 let data = nixlike::parse_str(&bytes)?;135136 Ok(Config(Arc::new(FleetConfigInternals {137 opts: self,138 directory,139 data,140 })))141 }142}