Signed-off-by: Christoph Heiss <c.he...@proxmox.com> --- Changes v2 -> v3: * new patch, split out from later patch
proxmox-auto-installer/src/answer.rs | 16 +++++++++++++++- .../src/bin/proxmox-auto-installer.rs | 11 +---------- 2 files changed, 16 insertions(+), 11 deletions(-) diff --git a/proxmox-auto-installer/src/answer.rs b/proxmox-auto-installer/src/answer.rs index d691da1..bada8a5 100644 --- a/proxmox-auto-installer/src/answer.rs +++ b/proxmox-auto-installer/src/answer.rs @@ -1,10 +1,11 @@ +use anyhow::{format_err, Result}; use clap::ValueEnum; use proxmox_installer_common::{ options::{BtrfsRaidLevel, FsType, ZfsChecksumOption, ZfsCompressOption, ZfsRaidLevel}, utils::{CidrAddress, Fqdn}, }; use serde::{Deserialize, Serialize}; -use std::{collections::BTreeMap, net::IpAddr}; +use std::{collections::BTreeMap, io::BufRead, net::IpAddr}; // BTreeMap is used to store filters as the order of the filters will be stable, compared to // storing them in a HashMap @@ -18,6 +19,19 @@ pub struct Answer { pub disks: Disks, } +impl Answer { + pub fn try_from_reader(reader: impl BufRead) -> Result<Self> { + let mut buffer = String::new(); + let lines = reader.lines(); + for line in lines { + buffer.push_str(&line.unwrap()); + buffer.push('\n'); + } + + toml::from_str(&buffer).map_err(|err| format_err!("Failed parsing answer file: {err}")) + } +} + #[derive(Clone, Deserialize, Debug)] #[serde(deny_unknown_fields)] pub struct Global { diff --git a/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs b/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs index bf6f8fb..b29228f 100644 --- a/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs +++ b/proxmox-auto-installer/src/bin/proxmox-auto-installer.rs @@ -42,16 +42,7 @@ fn auto_installer_setup(in_test_mode: bool) -> Result<(Answer, UdevInfo)> { .map_err(|err| format_err!("Failed to retrieve udev info details: {err}"))? }; - let mut buffer = String::new(); - let lines = std::io::stdin().lock().lines(); - for line in lines { - buffer.push_str(&line.unwrap()); - buffer.push('\n'); - } - - let answer: Answer = - toml::from_str(&buffer).map_err(|err| format_err!("Failed parsing answer file: {err}"))?; - + let answer = Answer::try_from_reader(std::io::stdin().lock())?; Ok((answer, udev_info)) } -- 2.45.2 _______________________________________________ pve-devel mailing list pve-devel@lists.proxmox.com https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel