updating storage_outputs class for multi-bucket support

aws-amplify · Sep 17, 2024 · 13d6723 · 13d6723
1 parent 51d01c4
commit 13d6723
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 9 deletions.
diff --git a/packages/amplify_core/lib/src/config/amplify_outputs/storage/storage_outputs.dart b/packages/amplify_core/lib/src/config/amplify_outputs/storage/storage_outputs.dart
@@ -12,7 +12,7 @@ part 'storage_outputs.g.dart';
 class StorageOutputs
     with AWSEquatable<StorageOutputs>, AWSSerializable, AWSDebuggable {
   /// {@macro amplify_core.amplify_outputs.storage_outputs}
-  const StorageOutputs({required this.awsRegion, required this.bucketName});
+  const StorageOutputs({required this.awsRegion, required this.bucketName, this.buckets});
 
   factory StorageOutputs.fromJson(Map<String, Object?> json) =>
       _$StorageOutputsFromJson(json);
@@ -23,6 +23,9 @@ class StorageOutputs
   /// The Amazon S3 bucket name.
   final String bucketName;
 
+  /// The list of buckets if there are multiple buckets for the project
+  final List<Map<String,String>>? buckets;
+
   @override
   List<Object?> get props => [awsRegion, bucketName];
 

diff --git a/packages/amplify_core/lib/src/config/amplify_outputs/storage/storage_outputs.g.dart b/packages/amplify_core/lib/src/config/amplify_outputs/storage/storage_outputs.g.dart
diff --git a/packages/amplify_core/test/config/amplify_outputs/test_data.dart b/packages/amplify_core/test/config/amplify_outputs/test_data.dart
@@ -99,9 +99,26 @@ const amplifyoutputs = '''{
     ]
   },
   "storage": {
-    "aws_region": "oem dks",
-    "bucket_name": "dolor et esse"
-  },
+    "aws_region": "us-east-2",
+    "bucket_name": "amplifyTeamDrive-one-stora-testbucketgen2bucket0b8c-9ggcfqfunkjr",
+    "buckets": [
+        {
+            "name": "amplifyTeamDrive-one",
+            "bucket_name": "amplifyTeamDrive-one-stora-testbucketgen2bucket0b8c-9ggcfqfunkjr",
+            "aws_region": "us-east-2"
+        },
+        {
+            "name": "amplifyTeamDrive-two",
+            "bucket_name": "amplifyTeamDrive-two-stora-testbucketgen2bucket0b8c-2",
+            "aws_region": "us-east-2"
+        },
+        {
+            "name": "amplifyTeamDrive-three",
+            "bucket_name": "amplifyTeamDrive-three-stora-testbucketgen2bucket0b8c-3",
+            "aws_region": "us-east-2"
+        }
+    ]
+},
   "version": "1"
 }
 ''';