add template tags for marketplace constructs

awslabs · Jun 3, 2024 · 96e54a1 · 96e54a1
1 parent d58674e
commit 96e54a1
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/src/control-plane/aws-marketplace/entitlement-logic.ts b/src/control-plane/aws-marketplace/entitlement-logic.ts
@@ -12,6 +12,7 @@ import * as subscriptions from 'aws-cdk-lib/aws-sns-subscriptions';
 import * as sqs from 'aws-cdk-lib/aws-sqs';
 import { NagSuppressions } from 'cdk-nag';
 import { Construct } from 'constructs';
+import { addTemplateTag } from '../../utils';
 
 export interface EntitlementLogicProps {
   readonly assetBucket: s3.IBucket;
@@ -23,6 +24,7 @@ export interface EntitlementLogicProps {
 export class EntitlementLogic extends Construct {
   constructor(scope: Construct, id: string, props: EntitlementLogicProps) {
     super(scope, id);
+    addTemplateTag(this, 'EntitlementLogic');
 
     const entitlementSQSQueueEncryptionKey = new cdk.aws_kms.Key(
       this,

diff --git a/src/control-plane/aws-marketplace/saas-product.ts b/src/control-plane/aws-marketplace/saas-product.ts
@@ -18,7 +18,7 @@ import { NagSuppressions } from 'cdk-nag';
 import { Construct } from 'constructs';
 import { EntitlementLogic } from './entitlement-logic';
 import { SubscriptionLogic } from './subscription-logic';
-import { DetailType, IEventManager, generateAWSManagedRuleSet } from '../../utils';
+import { DetailType, IEventManager, addTemplateTag, generateAWSManagedRuleSet } from '../../utils';
 
 /**
  * Enum representing the pricing models for an AWS Marketplace SaaS product.
@@ -100,6 +100,7 @@ export class AWSMarketplaceSaaSProduct extends Construct {
    */
   constructor(scope: Construct, id: string, props: AWSMarketplaceSaaSProductProps) {
     super(scope, id);
+    addTemplateTag(this, 'AWSMarketplaceSaaSProduct');
 
     const quickstartBucket = s3.Bucket.fromBucketName(this, 'CodeBucket', 'aws-quickstart');
 

diff --git a/src/control-plane/aws-marketplace/sample-registration-web-page.ts b/src/control-plane/aws-marketplace/sample-registration-web-page.ts
@@ -10,7 +10,7 @@ import * as s3 from 'aws-cdk-lib/aws-s3';
 import { BucketDeployment, DeployTimeSubstitutedFile, Source } from 'aws-cdk-lib/aws-s3-deployment';
 import { NagSuppressions } from 'cdk-nag';
 import { Construct } from 'constructs';
-import { generateAWSManagedRuleSet } from '../../utils';
+import { addTemplateTag, generateAWSManagedRuleSet } from '../../utils';
 
 /**
  * Properties for the SampleRegistrationWebPage construct.
@@ -55,6 +55,8 @@ export interface SampleRegistrationWebPageProps {
 export class SampleRegistrationWebPage extends Construct {
   constructor(scope: Construct, id: string, props: SampleRegistrationWebPageProps) {
     super(scope, id);
+    addTemplateTag(this, 'SampleRegistrationWebPage');
+
     const websiteBucket = new s3.Bucket(this, 'WebsiteS3Bucket', {
       enforceSSL: true,
       autoDeleteObjects: props.autoDeleteBucketObjects,

diff --git a/src/control-plane/aws-marketplace/subscription-logic.ts b/src/control-plane/aws-marketplace/subscription-logic.ts
@@ -12,6 +12,7 @@ import * as s3 from 'aws-cdk-lib/aws-s3';
 import * as sqs from 'aws-cdk-lib/aws-sqs';
 import { NagSuppressions } from 'cdk-nag';
 import { Construct } from 'constructs';
+import { addTemplateTag } from '../../utils';
 
 export interface SubscriptionLogicProps {
   readonly productCode: string;
@@ -21,6 +22,7 @@ export interface SubscriptionLogicProps {
 export class SubscriptionLogic extends Construct {
   constructor(scope: Construct, id: string, props: SubscriptionLogicProps) {
     super(scope, id);
+    addTemplateTag(this, 'SubscriptionLogic');
 
     const meteringRecordsTable = new dynamodb.Table(this, 'AWSMarketplaceMeteringRecords', {
       partitionKey: { name: 'customerIdentifier', type: dynamodb.AttributeType.STRING },