add option to add custom AWS tags

Signed-off-by: 🐼 Samrose Ahmed 🐼 <samroseahmed@gmail.com>
matanolabs · Nov 5, 2022 · 514b60a · 514b60a
1 parent e6f5eb4
commit 514b60a
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/example/matano.config.yml b/example/matano.config.yml
@@ -1,5 +1,7 @@
 aws_account: "946413832404"
 aws_region: "us-east-1"
+aws_tags:
+  cost_center: "80432"
 
 vpc:
   id: vpc-05175918865d89771 # vpc-0ea06dd2385eeb53c
diff --git a/infra/bin/app.ts b/infra/bin/app.ts
@@ -28,6 +28,12 @@ const dpMainStack = new DPMainStack(app, "DPMainStack", {
   realtimeBucketTopic: dpCommonStack.realtimeBucketTopic,
 });
 
+const userAwsTags = dpMainStack.userAwsTags ?? {};
+if (userAwsTags.constructor !== Object || !Object.values(userAwsTags).every((x) => typeof x === "string")) {
+  throw new Error("Custom AWS tags must be key value object of strings.");
+}
+
 tagResources(app, () => ({
+  ...userAwsTags,
   "matano:managed": "true",
 }));
diff --git a/infra/lib/MatanoStack.ts b/infra/lib/MatanoStack.ts
@@ -30,8 +30,10 @@ export function tagResources(scope: Construct, tags: (stack: cdk.Stack) => Recor
 export interface MatanoConfiguration {
   aws_account: string | undefined;
   aws_region: string | undefined;
+  aws_tags: object | undefined;
   project_label: string | undefined;
   is_production: boolean | undefined;
+  [key: string]: any;
 }
 
 export type MatanoConfig = MatanoConfiguration;
@@ -86,4 +88,8 @@ export class MatanoStack extends cdk.Stack {
   get matanoContext() {
     return JSON.parse(this.node.tryGetContext("matanoContext"));
   }
+
+  get userAwsTags(): object | undefined {
+    return this.matanoConfig.aws_tags;
+  }
 }
diff --git a/infra/src/DPCommonStack.ts b/infra/src/DPCommonStack.ts
@@ -22,7 +22,7 @@ export class DPCommonStack extends MatanoStack {
     this.matanoIngestionBucket = new S3BucketWithNotifications(this, "MatanoIngestionBucket", {
       bucketProps: {
         blockPublicAccess: BlockPublicAccess.BLOCK_ALL,
-      }
+      },
     });
 
     this.matanoLakeStorageBucket = new S3BucketWithNotifications(this, "MatanoLakeStorageBucket", {
@@ -79,5 +79,6 @@ export class DPCommonStack extends MatanoStack {
 
     // important: to prevent output deletion
     this.exportValue(this.matanoIngestionBucket.topic.topicArn);
+    this.exportValue(this.matanoIngestionBucket.bucket.bucketArn);
   }
 }