Skip to content

Commit

Permalink
WIP - Consume BucketName request attribute
Browse files Browse the repository at this point in the history 
Trying out if this works (BucketNameFilter will take name from path),
assert that bucket name is the same that is passed in the path to the
handler method.

Fixes #144
  • Loading branch information
@afranken
afranken committed Nov 21, 2022
1 parent 7fd5d2c commit 7550de3
Showing 1 changed file with 34 additions and 10 deletions.
44 changes: 34 additions & 10 deletions server/src/main/java/com/adobe/testing/s3mock/BucketController.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@

package com.adobe.testing.s3mock;

import static com.adobe.testing.s3mock.BucketNameFilter.BUCKET_ATTRIBUTE;
import static com.adobe.testing.s3mock.util.AwsHttpHeaders.X_AMZ_BUCKET_OBJECT_LOCK_ENABLED;
import static com.adobe.testing.s3mock.util.AwsHttpParameters.CONTINUATION_TOKEN;
import static com.adobe.testing.s3mock.util.AwsHttpParameters.ENCODING_TYPE;
Expand All @@ -31,6 +32,7 @@
import static org.springframework.http.MediaType.APPLICATION_XML_VALUE;

import com.adobe.testing.s3mock.dto.BucketLifecycleConfiguration;
import com.adobe.testing.s3mock.dto.BucketName;
import com.adobe.testing.s3mock.dto.ListAllMyBucketsResult;
import com.adobe.testing.s3mock.dto.ListBucketResult;
import com.adobe.testing.s3mock.dto.ListBucketResultV2;
Expand All @@ -39,6 +41,7 @@
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.CrossOrigin;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestAttribute;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
Expand Down Expand Up @@ -105,7 +108,10 @@ public ResponseEntity<ListAllMyBucketsResult> listBuckets() {
)
public ResponseEntity<Void> createBucket(@PathVariable final String bucketName,
@RequestHeader(value = X_AMZ_BUCKET_OBJECT_LOCK_ENABLED,
required = false, defaultValue = "false") boolean objectLockEnabled) {
required = false, defaultValue = "false") boolean objectLockEnabled,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
//TODO: does subdomain access work for #createBucket in S3?
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketNameIsAllowed(bucketName);
bucketService.verifyBucketDoesNotExist(bucketName);
bucketService.createBucket(bucketName, objectLockEnabled);
Expand All @@ -124,7 +130,9 @@ public ResponseEntity<Void> createBucket(@PathVariable final String bucketName,
value = "/{bucketName:[a-z0-9.-]+}",
method = RequestMethod.HEAD
)
public ResponseEntity<Void> headBucket(@PathVariable final String bucketName) {
public ResponseEntity<Void> headBucket(@PathVariable final String bucketName,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
return ResponseEntity.ok().build();
}
Expand All @@ -144,7 +152,9 @@ public ResponseEntity<Void> headBucket(@PathVariable final String bucketName) {
},
method = RequestMethod.DELETE
)
public ResponseEntity<Void> deleteBucket(@PathVariable String bucketName) {
public ResponseEntity<Void> deleteBucket(@PathVariable String bucketName,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.verifyBucketIsEmpty(bucketName);
bucketService.deleteBucket(bucketName);
Expand All @@ -171,7 +181,9 @@ public ResponseEntity<Void> deleteBucket(@PathVariable String bucketName) {
}
)
public ResponseEntity<ObjectLockConfiguration> getObjectLockConfiguration(
@PathVariable String bucketName) {
@PathVariable String bucketName,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
ObjectLockConfiguration configuration = bucketService.getObjectLockConfiguration(bucketName);
return ResponseEntity.ok(configuration);
Expand All @@ -195,7 +207,9 @@ public ResponseEntity<ObjectLockConfiguration> getObjectLockConfiguration(
)
public ResponseEntity<Void> putObjectLockConfiguration(
@PathVariable String bucketName,
@RequestBody ObjectLockConfiguration configuration) {
@RequestBody ObjectLockConfiguration configuration,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.setObjectLockConfiguration(bucketName, configuration);
return ResponseEntity.ok().build();
Expand All @@ -221,7 +235,9 @@ public ResponseEntity<Void> putObjectLockConfiguration(
}
)
public ResponseEntity<BucketLifecycleConfiguration> getBucketLifecycleConfiguration(
@PathVariable String bucketName) {
@PathVariable String bucketName,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
BucketLifecycleConfiguration configuration =
bucketService.getBucketLifecycleConfiguration(bucketName);
Expand All @@ -246,7 +262,9 @@ public ResponseEntity<BucketLifecycleConfiguration> getBucketLifecycleConfigurat
)
public ResponseEntity<Void> putBucketLifecycleConfiguration(
@PathVariable String bucketName,
@RequestBody BucketLifecycleConfiguration configuration) {
@RequestBody BucketLifecycleConfiguration configuration,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.setBucketLifecycleConfiguration(bucketName, configuration);
return ResponseEntity.ok().build();
Expand All @@ -268,7 +286,9 @@ public ResponseEntity<Void> putBucketLifecycleConfiguration(
method = RequestMethod.DELETE
)
public ResponseEntity<Void> deleteBucketLifecycleConfiguration(
@PathVariable String bucketName) {
@PathVariable String bucketName,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.deleteBucketLifecycleConfiguration(bucketName);
return ResponseEntity.noContent().build();
Expand Down Expand Up @@ -305,7 +325,9 @@ public ResponseEntity<ListBucketResult> listObjects(
@RequestParam(required = false) String delimiter,
@RequestParam(required = false) String marker,
@RequestParam(name = ENCODING_TYPE, required = false) String encodingType,
@RequestParam(name = MAX_KEYS, defaultValue = "1000", required = false) Integer maxKeys) {
@RequestParam(name = MAX_KEYS, defaultValue = "1000", required = false) Integer maxKeys,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.verifyMaxKeys(maxKeys);
bucketService.verifyEncodingType(encodingType);
Expand Down Expand Up @@ -343,7 +365,9 @@ public ResponseEntity<ListBucketResultV2> listObjectsV2(
@RequestParam(name = ENCODING_TYPE, required = false) String encodingType,
@RequestParam(name = START_AFTER, required = false) String startAfter,
@RequestParam(name = MAX_KEYS, defaultValue = "1000", required = false) Integer maxKeys,
@RequestParam(name = CONTINUATION_TOKEN, required = false) String continuationToken) {
@RequestParam(name = CONTINUATION_TOKEN, required = false) String continuationToken,
@RequestAttribute(BUCKET_ATTRIBUTE) BucketName bucket) {
assert bucketName.equals(bucket.getName());
bucketService.verifyBucketExists(bucketName);
bucketService.verifyMaxKeys(maxKeys);
bucketService.verifyEncodingType(encodingType);
Expand Down

0 comments on commit 7550de3

Please sign in to comment.