s7cmd 1.6.0

Reliable, flexible, and fast command-line tool for Amazon S3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

// Vendored from s3util-rs@1.1.0
//   src/bin/s3util/cli/put_bucket_versioning.rs
// Adjustments: no tests stripped; rewrote crate::cli → super;
//              replaced the process-exiting `validate_state_flag()` call
//              with the non-exiting `check_state_flag` invoked from
//              dispatch.rs (a `process::exit(2)` would kill a whole
//              batch-run instead of failing one line).
use anyhow::Result;
use clap::CommandFactory;
use tracing::info;

use s3util_rs::config::ClientConfig;
use s3util_rs::config::args::put_bucket_versioning::PutBucketVersioningArgs;
use s3util_rs::storage::s3::api;

/// Runtime entry for `s3util put-bucket-versioning s3://<BUCKET> --enabled|--suspended`.
///
/// Builds the SDK client from `client_config`, issues `PutBucketVersioning`
/// with `Status=Enabled` or `Status=Suspended` (determined by the mutually-
/// exclusive `--enabled` / `--suspended` flags), and exits silently on success.
pub async fn run_put_bucket_versioning(
    args: PutBucketVersioningArgs,
    client_config: ClientConfig,
) -> Result<()> {
    // --enabled/--suspended presence is enforced by `check_state_flag`,
    // which dispatch.rs runs before invoking this runner.
    let bucket = args
        .bucket_name()
        .map_err(|e| anyhow::anyhow!("{}", e.trim_end()))?;
    let status = args.versioning_status();
    let client = client_config.create_client().await;
    if args.dry_run {
        info!(bucket = %bucket, status = %status.as_str(), "[dry-run] would put bucket versioning.");
        return Ok(());
    }
    api::put_bucket_versioning(&client, &bucket, status.clone()).await?;
    info!(bucket = %bucket, status = %status.as_str(), "Bucket versioning set.");
    Ok(())
}

/// Non-exiting replacement for upstream's `validate_state_flag`, whose
/// `clap::Error::exit()` (`std::process::exit(2)`) cannot be contained by
/// batch-run's per-line error handling. dispatch.rs calls this before the
/// runner, prints the error, and returns exit code 2 — same message and
/// code as upstream, without terminating the process.
pub fn check_state_flag(args: &PutBucketVersioningArgs) -> Result<(), clap::Error> {
    if !args.enabled && !args.suspended {
        let mut cmd = PutBucketVersioningArgs::command();
        return Err(cmd.error(
            clap::error::ErrorKind::MissingRequiredArgument,
            "one of --enabled or --suspended must be specified",
        ));
    }
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use clap::Parser;

    fn parse(argv: &[&str]) -> PutBucketVersioningArgs {
        PutBucketVersioningArgs::try_parse_from(argv).expect("args should parse")
    }

    #[test]
    fn check_state_flag_rejects_neither_flag() {
        let args = parse(&["put-bucket-versioning", "s3://b"]);
        let err = check_state_flag(&args).expect_err("neither flag must be rejected");
        assert_eq!(err.kind(), clap::error::ErrorKind::MissingRequiredArgument);
        assert!(
            err.to_string()
                .contains("one of --enabled or --suspended must be specified")
        );
    }

    #[test]
    fn check_state_flag_accepts_enabled() {
        let args = parse(&["put-bucket-versioning", "s3://b", "--enabled"]);
        assert!(check_state_flag(&args).is_ok());
    }

    #[test]
    fn check_state_flag_accepts_suspended() {
        let args = parse(&["put-bucket-versioning", "s3://b", "--suspended"]);
        assert!(check_state_flag(&args).is_ok());
    }
}