git.delta.rocks / jrsonnet / refs/commits / 2e7f8f868321

1use std::{collections::BTreeMap, process::Command};23use anyhow::Result;4use log::*;56use crate::{command::CommandExt, nix::HOSTS_ATTRIBUTE};78use serde::{Deserialize, Serialize};910use super::db::DbData;1112pub fn list_hosts() -> Result<Vec<String>> {13	Ok(Command::new("nix")14		.inherit_stdio()15		.arg("eval")16		.arg(HOSTS_ATTRIBUTE)17		.arg("--apply")18		.arg("builtins.attrNames")19		.run_json()?)20}2122#[derive(Serialize, Deserialize, Default)]23pub struct KeyDb {24	host_keys: BTreeMap<String, String>,25}26impl DbData for KeyDb {27	const DB_NAME: &'static str = "keys";28}2930impl KeyDb {31	pub fn fetch_key(&mut self, host: &str) -> Result<()> {32		info!("Fetching key for {}", host);33		let key = Command::ssh_on(host, "cat")34			.arg("/etc/ssh/ssh_host_ed25519_key.pub")35			.run_string()?;36		self.host_keys.insert(host.to_owned(), key);37		Ok(())38	}3940	pub fn ensure_key_loaded(&mut self, host: &str, force: bool) -> Result<()> {41		if !self.host_keys.contains_key(host) || force {42			self.fetch_key(host)?;43		}44		Ok(())45	}4647	pub fn get_host_key(&self, host: &str) -> Result<String> {48		Ok(self49			.host_keys50			.get(host)51			.ok_or_else(|| anyhow::anyhow!("no host key for {}", host))?52			.to_owned())53	}5455	pub fn has_key(&self, key: &str) -> bool {56		self.host_keys.contains_key(key)57	}5859	pub fn remove_key(&mut self, host: &str) {60		self.host_keys.remove(host);61	}62}