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
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

// Copyright 2016 LambdaStack All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#![allow(unused_imports)]
#![allow(unused_variables)]
#![allow(unused_must_use)]

use clap::ArgMatches;
use aws_sdk_rust::aws::errors::s3::S3Error;
use aws_sdk_rust::aws::common::credentials::AwsCredentialsProvider;
use aws_sdk_rust::aws::common::request::DispatchSignedRequest;
use aws_sdk_rust::aws::s3::acl::*;
use aws_sdk_rust::aws::s3::bucket::*;

use term;
use Client;
use bucket::get::get_bucket_acl;

/// All PUT requests pass through this function.
pub fn commands<P, D>(matches: &ArgMatches,
                      client: &mut Client<P, D>)
                      -> Result<(), S3Error>
                      where P: AwsCredentialsProvider,
                            D: DispatchSignedRequest {
    let bucket = matches.value_of("name").unwrap_or("");

    match matches.subcommand() {
        /// acl command.
        ("acl", Some(sub_matches)) => {
            let result = put_bucket_acl(sub_matches, bucket, client);
          },
        ("versioning", Some(sub_matches)) => {
            let result = put_bucket_versioning(sub_matches, bucket, client);
        },
        (e, _) => {
            if e.is_empty() && bucket.is_empty() {

                println!("what?");

            } else {
                let error = format!("incorrect or missing request {}", e);
                println_color_quiet!(client.is_quiet, term::color::RED, "{}", error);
            }
        },
    }

    Ok(())
}

fn put_bucket_versioning<P, D>(sub_matches: &ArgMatches,
                               bucket: &str,
                               client: &Client<P, D>)
                               -> Result<(), S3Error>
                               where P: AwsCredentialsProvider,
                                     D: DispatchSignedRequest {
    let version = PutBucketVersioningRequest{
        bucket: bucket.to_string(),
        versioning_configuration: VersioningConfiguration {
            status: "Enabled".to_string(),
            mfa_delete: "".to_string(),
        },
        mfa: None,
        content_md5: None,
    };

    match client.s3client.put_bucket_versioning(&version) {
        Ok(()) => {
          println_color_quiet!(client.is_quiet, client.output.color, "Success");
          Ok(())
        },
        Err(e) => {
          println_color_quiet!(client.is_quiet, client.error.color, "{:#?}", e);
          Err(e)
        },
    }
}

fn put_bucket_acl<P, D>(sub_matches: &ArgMatches,
                        bucket: &str,
                        client: &Client<P, D>)
                        -> Result<(), S3Error>
                        where P: AwsCredentialsProvider,
                              D: DispatchSignedRequest {

    // Default to Private
    let mut acl: CannedAcl = CannedAcl::Private;

    match sub_matches.subcommand() {
        ("public-read", _) => acl = CannedAcl::PublicRead,
        ("public-rw", _) => acl = CannedAcl::PublicReadWrite,
        ("public-readwrite", _) => acl = CannedAcl::PublicReadWrite,
        ("private", _) => acl = CannedAcl::Private,
        (e, _) => println_color_quiet!(client.is_quiet, client.error.color, "Something {:?}", e),
    }

    let mut bucket_acl = PutBucketAclRequest::default();
    bucket_acl.bucket = bucket.to_string();

    // get acl option...
    bucket_acl.acl = Some(acl);

    match client.s3client.put_bucket_acl(&bucket_acl) {
        Ok(val) => {
          let acl = get_bucket_acl(bucket, client);
          if let Ok(acl) = acl {
              println_color_quiet!(client.is_quiet, client.output.color, "{:#?}", acl);
          }
        },
        Err(e) => {
            println_color_quiet!(client.is_quiet, client.error.color, "Something {:?}", e);
            return Err(e)
        },
    }
    
    Ok(())
}