wip: use enum dispatch in image mold

Author: cilki

Cargo.lock

@@ -36,3 +36,5 @@ fn impl_prompt(ast: &syn::DeriveInput) -> TokenStream {
     };
     gen.into()
 }
+
+// TODO probably need a macro for ImageMold and Fabricator

goldboot-macros/src/lib.rs

@@ -18,3 +18,4 @@ rustls = "0.21.8"
 rustls-pemfile = "1.0.3"
 simple-error = "0.3.0"
 tokio = { version = "1.34.0", features = ["full"] }
+goldboot = { path="../goldboot", version = "0.0.1" }

goldboot-registry/Cargo.toml

@@ -44,6 +44,7 @@ validator = { version = "0.16.1", features = ["derive"] }
 vnc = "0.4.0"
 whoami = "1.4.1"
 zstd = "0.13.0"
+enum_dispatch = "0.3.12"
 
 [dev-dependencies]
 test-env-log = "0"

goldboot/Cargo.toml

@@ -1,31 +1,25 @@
-use crate::{
-    build::BuildConfig,
-    cmd::Commands,
-    templates::{Template, TemplateMetadata},
-    Architecture,
-};
 use console::Style;
 use dialoguer::{theme::ColorfulTheme, Confirm, Input, Select};
+use goldboot_image::ImageArch;
 use simple_error::bail;
 use std::{error::Error, path::Path};
 use strum::IntoEnumIterator;
 
-#[rustfmt::skip]
 fn print_banner() {
     if console::colors_enabled() {
         let style = Style::new().yellow();
 
         println!("{}", "");
         for line in fossable::goldboot_word() {
-        println!("  {}", style.apply_to(line));
-                    }
+            println!("  {}", style.apply_to(line));
+        }
         println!("{}", "");
     }
 }
 
-pub fn run(cmd: crate::cmd::Commands) -> Result<(), Box<dyn Error>> {
+pub fn run(cmd: super::Commands) -> Result<(), Box<dyn Error>> {
     match cmd {
-        Commands::Init {
+        super::Commands::Init {
             name,
             template,
             mimic_hardware,
@@ -91,7 +85,7 @@ pub fn run(cmd: crate::cmd::Commands) -> Result<(), Box<dyn Error>> {
 
                 // Prompt image architecture
                 {
-                    let architectures: Vec<Architecture> = Architecture::iter().collect();
+                    let architectures: Vec<ImageArch> = ImageArch::iter().collect();
                     let arch_index = Select::with_theme(&theme)
                         .with_prompt("Choose image architecture")
                         .default(0)
@@ -128,7 +122,7 @@ pub fn run(cmd: crate::cmd::Commands) -> Result<(), Box<dyn Error>> {
             }
 
             // Finally write out the config
-            std::fs::write(config_path, serde_yaml::to_string(&config)?)?;
+            std::fs::write(config_path, ron::to_string(&config)?)?;
             Ok(())
         }
         _ => panic!(),

goldboot/src/cli/cmd/init.rs

@@ -1,19 +1,17 @@
-use std::error::Error;
-
-use crate::cmd::Commands;
 use console::Style;
 use dialoguer::{theme::ColorfulTheme, Confirm, Input, Select};
+use std::error::Error;
 
 use super::RegistryCommands;
 
-pub fn run(cmd: crate::cmd::Commands) -> Result<(), Box<dyn Error>> {
+pub fn run(cmd: super::Commands) -> Result<(), Box<dyn Error>> {
     let theme = ColorfulTheme {
         values_style: Style::new().yellow().dim(),
         ..ColorfulTheme::default()
     };
 
     match cmd {
-        Commands::Registry { command } => match &command {
+        super::Commands::Registry { command } => match &command {
             RegistryCommands::Push { url } => todo!(),
             RegistryCommands::Pull { url } => todo!(),
             RegistryCommands::Login {} => {

goldboot/src/cli/cmd/registry.rs

@@ -1,11 +1,10 @@
-use crate::{cmd::Commands, library::ImageLibrary};
 use console::Style;
 use dialoguer::{theme::ColorfulTheme, Confirm, Input, Select};
 use std::{error::Error, path::Path};
 
-pub fn run(cmd: crate::cmd::Commands) -> Result<(), Box<dyn Error>> {
+pub fn run(cmd: super::Commands) -> Result<(), Box<dyn Error>> {
     match cmd {
-        Commands::Write {
+        super::Commands::Write {
             image,
             output,
             confirm,

goldboot/src/cli/cmd/write.rs

@@ -1,2 +1,3 @@
 pub mod cmd;
 pub mod progress;
+pub mod prompt;

goldboot/src/cli/mod.rs

@@ -0,0 +1,18 @@
+use std::error::Error;
+
+pub trait Prompt {
+    fn prompt(
+        &mut self,
+        config: &BuildConfig,
+        theme: Box<dyn dialoguer::theme::Theme>,
+    ) -> Result<(), Box<dyn Error>>;
+}
+
+pub trait PromptNew {
+    fn prompt(
+        config: &BuildConfig,
+        theme: Box<dyn dialoguer::theme::Theme>,
+    ) -> Result<Self, Box<dyn Error>>
+    where
+        Self: Sized;
+}

goldboot/src/cli/prompt.rs

@@ -1,13 +1,10 @@
-use std::{error::Error, path::Path, process::Command};
-
-use dialoguer::theme::ColorfulTheme;
+use crate::{cli::prompt::Prompt, foundry::ssh::SshConnection};
 use log::info;
 use serde::{Deserialize, Serialize};
 use simple_error::bail;
+use std::{error::Error, path::Path, process::Command};
 use validator::Validate;
 
-use crate::{foundry::ssh::SshConnection, PromptMut};
-
 /// Runs an Ansible playbook on the image remotely.
 #[derive(Clone, Serialize, Deserialize, Validate, Debug)]
 pub struct AnsibleFabricator {
@@ -47,18 +44,18 @@ impl AnsibleFabricator {
     }
 }
 
-impl PromptMut for AnsibleFabricator {
+impl Prompt for AnsibleFabricator {
     fn prompt(
         &mut self,
         config: &BuildConfig,
-        theme: &ColorfulTheme,
+        theme: Box<dyn dialoguer::theme::Theme>,
     ) -> Result<(), Box<dyn Error>> {
-        self.playbook = dialoguer::Input::with_theme(theme)
+        self.playbook = dialoguer::Input::with_theme(&theme)
             .with_prompt("Enter the playbook path relative to the current directory")
             .interact()?;
 
         if !Path::new(&self.playbook).exists() {
-            if !dialoguer::Confirm::with_theme(theme)
+            if !dialoguer::Confirm::with_theme(&theme)
                 .with_prompt("The path does not exist. Add anyway?")
                 .interact()?
             {