Working on refactoring command stuff

Author: mkeeter

Cargo.lock

@@ -13,10 +13,9 @@ use anyhow::Result;
 use clap::{CommandFactory, Parser};
 use colored::Colorize;
 use humility_cli::ExecutionContext;
-use humility_cmd::CommandKind;
 
 use humility_auxflash::AuxFlashHandler;
-use humility_cmd::{Archive, Attach, Command, Validate};
+use humility_cmd::Command;
 
 #[derive(Parser, Debug)]
 #[clap(name = "auxflash", about = env!("CARGO_PKG_DESCRIPTION"))]
@@ -99,9 +98,9 @@ fn auxflash_status(mut worker: AuxFlashHandler, verbose: bool) -> Result<()> {
 }
 
 fn auxflash(context: &mut ExecutionContext) -> Result<()> {
-    let core = &mut **context.core.as_mut().unwrap();
     let subargs = AuxFlashArgs::try_parse_from(&context.cli.cmd)?;
-    let hubris = context.archive.as_ref().unwrap();
+    let hubris = &context.cli.archive()?;
+    let core = &mut *context.cli.attach_live_booted(hubris)?;
     let mut worker = AuxFlashHandler::new(hubris, core, subargs.timeout)?;
 
     match subargs.cmd {
@@ -132,14 +131,5 @@ fn auxflash(context: &mut ExecutionContext) -> Result<()> {
 }
 
 pub fn init() -> Command {
-    Command {
-        app: AuxFlashArgs::command(),
-        name: "auxflash",
-        run: auxflash,
-        kind: CommandKind::Attached {
-            archive: Archive::Required,
-            attach: Attach::LiveOnly,
-            validate: Validate::Booted,
-        },
-    }
+    Command { app: AuxFlashArgs::command(), name: "auxflash", run: auxflash }
 }

cmd/auxflash/src/lib.rs

@@ -10,7 +10,7 @@ use std::path::PathBuf;
 
 use clap::{CommandFactory, Parser};
 
-use humility_cmd::{Archive, Attach, Command, CommandKind, Validate};
+use humility_cmd::Command;
 
 #[cfg(not(windows))]
 mod posix;
@@ -113,10 +113,5 @@ pub fn init() -> Command {
         app: UartConsoleArgs::command(),
         name: "console-proxy",
         run: console_proxy,
-        kind: CommandKind::Attached {
-            archive: Archive::Required,
-            attach: Attach::LiveOnly,
-            validate: Validate::Booted,
-        },
     }
 }

cmd/console-proxy/src/lib.rs

@@ -314,9 +314,9 @@ impl UnrawTermiosGuard {
 }
 
 pub(super) fn console_proxy(context: &mut ExecutionContext) -> Result<()> {
-    let core = &mut **context.core.as_mut().unwrap();
     let subargs = UartConsoleArgs::try_parse_from(&context.cli.cmd)?;
-    let hubris = context.archive.as_ref().unwrap();
+    let hubris = &context.cli.archive()?;
+    let core = &mut *context.cli.attach_live_booted(hubris)?;
     let mut worker = UartConsoleHandler::new(
         hubris,
         core,

cmd/console-proxy/src/posix.rs

@@ -206,7 +206,7 @@ use humility::core::Core;
 use humility::hubris::*;
 use humility::reflect::{self, Load, Value};
 use humility_cli::ExecutionContext;
-use humility_cmd::{Archive, Attach, Command, CommandKind, Validate};
+use humility_cmd::Command;
 use humility_doppel::{CountedRingbuf, CounterVariant, Counters};
 use indexmap::IndexMap;
 use std::collections::BTreeMap;
@@ -324,12 +324,11 @@ const LIST_HINT: &str = "use `humility counters list` to list all \
     available counters";
 
 fn counters(context: &mut ExecutionContext) -> Result<()> {
-    let core = &mut **context.core.as_mut().unwrap();
-    let hubris = context.archive.as_ref().unwrap();
-
+    let hubris = &context.cli.archive()?;
     let subargs = CountersArgs::try_parse_from(&context.cli.cmd)?;
 
     if let Some(Subcmd::Ipc(ipc)) = subargs.command {
+        let core = &mut *context.cli.attach_live_or_dump_match(hubris)?;
         return ipc.ipc_counter_dump(hubris, core);
     }
     let name = subargs.name();
@@ -428,6 +427,7 @@ fn counters(context: &mut ExecutionContext) -> Result<()> {
     }
 
     let mut json: IndexMap<&str, IndexMap<_, _>> = IndexMap::new();
+    let core = &mut *context.cli.attach_live_or_dump_match(hubris)?;
     for (t, ctrs) in counters {
         // Try not to use `?` here, because it causes one bad counter to make
         // them all unavailable. Instead, construct an iterator of
@@ -623,14 +623,5 @@ fn hint() -> impl std::fmt::Display {
 }
 
 pub fn init() -> Command {
-    Command {
-        app: CountersArgs::command(),
-        name: "counters",
-        run: counters,
-        kind: CommandKind::Attached {
-            archive: Archive::Required,
-            attach: Attach::Any,
-            validate: Validate::Match,
-        },
-    }
+    Command { app: CountersArgs::command(), name: "counters", run: counters }
 }

cmd/counters/src/lib.rs

@@ -29,7 +29,7 @@ use hif::*;
 use humility::core::Core;
 use humility::hubris::*;
 use humility_cli::ExecutionContext;
-use humility_cmd::{Archive, Attach, Command, CommandKind, Validate};
+use humility_cmd::Command;
 use humility_hiffy::*;
 use humility_idol::{self as idol, HubrisIdol};
 use ratatui::{
@@ -736,11 +736,10 @@ where
 }
 
 fn dashboard(context: &mut ExecutionContext) -> Result<()> {
-    let hubris = context.archive.as_ref().unwrap();
-
-    let core = &mut **context.core.as_mut().unwrap();
-
     let subargs = DashboardArgs::try_parse_from(&context.cli.cmd)?;
+    let hubris = &context.cli.archive()?;
+    let core = &mut *context.cli.attach_live_booted(hubris)?;
+
     let dashboard = Dashboard::new(hubris, core, &subargs)?;
 
     // setup terminal
@@ -767,16 +766,7 @@ fn dashboard(context: &mut ExecutionContext) -> Result<()> {
 }
 
 pub fn init() -> Command {
-    Command {
-        app: DashboardArgs::command(),
-        name: "dashboard",
-        run: dashboard,
-        kind: CommandKind::Attached {
-            archive: Archive::Required,
-            attach: Attach::LiveOnly,
-            validate: Validate::Booted,
-        },
-    }
+    Command { app: DashboardArgs::command(), name: "dashboard", run: dashboard }
 }
 
 fn sensor_ops(

cmd/dashboard/src/lib.rs

@@ -32,7 +32,7 @@ use anyhow::{Context, Result, bail};
 use byteorder::{ByteOrder, LittleEndian, WriteBytesExt};
 use clap::{CommandFactory, Parser};
 use humility_cli::ExecutionContext;
-use humility_cmd::{Archive, Command, CommandKind};
+use humility_cmd::Command;
 use probe_rs::{
     DebugProbeError, DebugProbeSelector, Probe,
     architecture::arm::{ApAddress, ArmProbeInterface, DapError, DpAddress},
@@ -464,6 +464,5 @@ pub fn init() -> Command {
         app: DebugMailboxArgs::command(),
         name: "debugmailbox",
         run: debugmailboxcmd,
-        kind: CommandKind::Unattached { archive: Archive::Ignored },
     }
 }

cmd/debugmailbox/src/lib.rs

@@ -19,24 +19,14 @@ use clap::{CommandFactory, Parser};
 use humility::core::Core;
 use humility::hubris::*;
 use humility_cli::ExecutionContext;
-use humility_cmd::CommandKind;
-use humility_cmd::{Archive, Attach, Command, Validate};
+use humility_cmd::Command;
 use humility_doppel::{GenOrRestartCount, Task, TaskDesc, TaskState};
 use std::num::NonZeroU32;
 use std::time::Duration;
 
 /// Command registration.
 pub fn init() -> Command {
-    Command {
-        app: DiagnoseArgs::command(),
-        name: "diagnose",
-        run: diagnose,
-        kind: CommandKind::Attached {
-            archive: Archive::Required,
-            attach: Attach::Any,
-            validate: Validate::Booted,
-        },
-    }
+    Command { app: DiagnoseArgs::command(), name: "diagnose", run: diagnose }
 }
 
 #[derive(Parser, Debug)]
@@ -83,10 +73,9 @@ fn section(title: &str) {
 }
 
 fn diagnose(context: &mut ExecutionContext) -> Result<()> {
-    let core = &mut **context.core.as_mut().unwrap();
-    let hubris = context.archive.as_ref().unwrap();
-
     let subargs = DiagnoseArgs::try_parse_from(&context.cli.cmd)?;
+    let hubris = &context.cli.archive()?;
+    let core = &mut *context.cli.attach_live_booted(hubris)?;
 
     section("Initial Inspection");
 

cmd/diagnose/src/lib.rs

@@ -45,7 +45,7 @@ use colored::Colorize;
 use hubpack::SerializedSize;
 use humility::net::decode_iface;
 use humility_cli::ExecutionContext;
-use humility_cmd::{Archive, Command, CommandKind};
+use humility_cmd::Command;
 use serde::{Deserialize, Serialize};
 
 #[derive(Parser, Debug)]
@@ -313,9 +313,9 @@ fn discover_dump(seen: BTreeSet<Target>, image_id: Option<&[u8]>) {
 
 fn discover_run(context: &mut ExecutionContext) -> Result<()> {
     let subargs = DiscoverArgs::try_parse_from(&context.cli.cmd)?;
-    let hubris = context.archive.as_ref().unwrap();
+    let hubris = &context.cli.try_archive()?;
 
-    let image_id = hubris.image_id();
+    let image_id = hubris.as_ref().and_then(|h| h.image_id());
     if image_id.is_none() {
         humility::warn!("no archive provided; not checking for compatibility");
     }
@@ -330,6 +330,5 @@ pub fn init() -> Command {
         app: DiscoverArgs::command(),
         name: "discover",
         run: discover_run,
-        kind: CommandKind::Detached { archive: Archive::Optional },
     }
 }

cmd/discover/src/lib.rs

@@ -13,7 +13,7 @@
 use anyhow::{Result, bail};
 use clap::{CommandFactory, Parser};
 use humility_cli::ExecutionContext;
-use humility_cmd::{Archive, Command, CommandKind};
+use humility_cmd::Command;
 use std::collections::HashMap;
 use termimad::*;
 
@@ -64,10 +64,5 @@ fn doc(context: &mut ExecutionContext) -> Result<()> {
 }
 
 pub fn init() -> Command {
-    Command {
-        app: DocArgs::command(),
-        name: "doc",
-        run: doc,
-        kind: CommandKind::Unattached { archive: Archive::Ignored },
-    }
+    Command { app: DocArgs::command(), name: "doc", run: doc }
 }