use anyhow::Result;
use clap::Parser;
use std::path::{Path, PathBuf};
use super::verify_commit::{VerifyCommitCommand, handle_verify_commit};
use crate::commands::device::verify_attestation::{VerifyCommand, handle_verify};
pub enum VerifyTarget {
GitRef(String),
Attestation(String),
}
pub fn parse_verify_target(raw_target: &str) -> VerifyTarget {
if raw_target == "-" {
return VerifyTarget::Attestation(raw_target.to_string());
}
let path = Path::new(raw_target);
if path.exists() {
return VerifyTarget::Attestation(raw_target.to_string());
}
if raw_target.contains("..") {
return VerifyTarget::GitRef(raw_target.to_string());
}
if raw_target.eq_ignore_ascii_case("HEAD") {
return VerifyTarget::GitRef(raw_target.to_string());
}
let is_hex = raw_target.len() >= 4
&& raw_target.len() <= 40
&& raw_target.chars().all(|c| c.is_ascii_hexdigit());
if is_hex {
return VerifyTarget::GitRef(raw_target.to_string());
}
VerifyTarget::GitRef(raw_target.to_string())
}
#[derive(Parser, Debug, Clone)]
#[command(about = "Verify a signed commit or attestation.")]
pub struct UnifiedVerifyCommand {
#[arg(default_value = "HEAD")]
pub target: String,
#[arg(long, default_value = ".auths/allowed_signers")]
pub allowed_signers: PathBuf,
#[arg(long, value_parser)]
pub identity_bundle: Option<PathBuf>,
#[arg(long = "issuer-pk")]
pub issuer_pk: Option<String>,
#[arg(long = "issuer-did", visible_alias = "issuer")]
pub issuer_did: Option<String>,
#[arg(long)]
pub witness_receipts: Option<PathBuf>,
#[arg(long, default_value = "1")]
pub witness_threshold: usize,
#[arg(long, num_args = 1..)]
pub witness_keys: Vec<String>,
}
pub async fn handle_verify_unified(cmd: UnifiedVerifyCommand) -> Result<()> {
match parse_verify_target(&cmd.target) {
VerifyTarget::GitRef(ref_str) => {
let commit_cmd = VerifyCommitCommand {
commit: ref_str,
allowed_signers: cmd.allowed_signers,
identity_bundle: cmd.identity_bundle,
witness_receipts: cmd.witness_receipts,
witness_threshold: cmd.witness_threshold,
witness_keys: cmd.witness_keys,
};
handle_verify_commit(commit_cmd).await
}
VerifyTarget::Attestation(path_str) => {
let verify_cmd = VerifyCommand {
attestation: path_str,
issuer_pk: cmd.issuer_pk,
issuer_did: cmd.issuer_did,
trust: None,
roots_file: None,
require_capability: None,
witness_receipts: cmd.witness_receipts,
witness_threshold: cmd.witness_threshold,
witness_keys: cmd.witness_keys,
};
handle_verify(verify_cmd).await
}
}
}
impl crate::commands::executable::ExecutableCommand for UnifiedVerifyCommand {
fn execute(&self, _ctx: &crate::config::CliConfig) -> anyhow::Result<()> {
let rt = tokio::runtime::Runtime::new()?;
rt.block_on(handle_verify_unified(self.clone()))
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_parse_verify_target_git_ref() {
assert!(matches!(
parse_verify_target("HEAD"),
VerifyTarget::GitRef(_)
));
assert!(matches!(
parse_verify_target("abc1234"),
VerifyTarget::GitRef(_)
));
assert!(matches!(
parse_verify_target("main..HEAD"),
VerifyTarget::GitRef(_)
));
}
#[test]
fn test_parse_verify_target_stdin() {
assert!(matches!(
parse_verify_target("-"),
VerifyTarget::Attestation(_)
));
}
#[test]
fn test_parse_verify_target_nonexistent_defaults_to_git_ref() {
let target = parse_verify_target("/nonexistent/attestation.json");
assert!(matches!(target, VerifyTarget::GitRef(_)));
}
#[test]
fn test_parse_verify_target_file() {
use std::fs::File;
use tempfile::tempdir;
let dir = tempdir().unwrap();
let f = dir.path().join("attestation.json");
File::create(&f).unwrap();
let target = parse_verify_target(f.to_str().unwrap());
assert!(matches!(target, VerifyTarget::Attestation(_)));
}
}