fix(servicecatalogappregistry): Associate an application with attribute group

packages/@aws-cdk/aws-servicecatalogappregistry/README.md

@@ -99,6 +99,41 @@ const associatedApp = new appreg.ApplicationAssociator(app, 'AssociatedApplicati
 });
 ```
 
+If you want to associate an Attribute Group with application created by `ApplicationAssociator`, then use as shown in the example below:
+
+```ts
+const app = new App();
+const customAttributeGroup = new CustomAppRegistryAttributeGroup(app, 'AppRegistryAttributeGroup');
+
+const associatedApp = new appreg.ApplicationAssociator(app, 'AssociatedApplication', {
+  applications: [appreg.TargetApplication.createApplicationStack({
+    applicationName: 'MyAssociatedApplication',
+    // 'Application containing stacks deployed via CDK.' is the default
+    applicationDescription: 'Associated Application description',
+    stackName: 'MyAssociatedApplicationStack',
+    // AWS Account and Region that are implied by the current CLI configuration is the default
+    env: { account: '123456789012', region: 'us-east-1' },
+  })],
+});
+
+// Associate application to the attribute group.
+customAttributeGroup.attributeGroup.associateApplicationWithAttributeGroup(associatedApp.appRegistryApplication());
+
+class CustomAppRegistryAttributeGroup extends cdk.Stack {
+  public readonly attributeGroup: appreg.AttributeGroup
+  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
+    super(scope, id, props);
+    const myAttributeGroup = new appreg.AttributeGroup(this, 'MyFirstAttributeGroup', {
+      attributeGroupName: 'MyAttributeGroupName',
+      description: 'Test attribute group',
+      attributes: {},
+    });
+
+    this.attributeGroup = myAttributeGroup;
+  }
+}
+```
+
 If you are using CDK Pipelines to deploy your application, the application stacks will be inside Stages, and
 ApplicationAssociator will not be able to find them. Call `associateStage` on each Stage object before adding it to the
 Pipeline, as shown in the example below:
@@ -191,6 +226,16 @@ declare const attributeGroup: appreg.AttributeGroup;
 application.associateAttributeGroup(attributeGroup);
 ```
 
+### Associating an attribute group with application
+
+You can associate an application with an attribute group with `associateApplicationWithAttributeGroup`:
+
+```ts
+declare const application: appreg.Application;
+declare const attributeGroup: appreg.AttributeGroup;
+attributeGroup.associateApplicationWithAttributeGroup(application);
+```
+
 ### Associating application with a Stack
 
 You can associate a stack with an application with the `associateStack()` API:

packages/@aws-cdk/aws-servicecatalogappregistry/lib/attribute-group.ts

@@ -2,9 +2,10 @@ import { CfnResourceShare } from '@aws-cdk/aws-ram';
 import * as cdk from '@aws-cdk/core';
 import { Names } from '@aws-cdk/core';
 import { Construct } from 'constructs';
+import { IApplication } from './application';
 import { getPrincipalsforSharing, hashValues, ShareOptions, SharePermission } from './common';
 import { InputValidator } from './private/validation';
-import { CfnAttributeGroup } from './servicecatalogappregistry.generated';
+import { CfnAttributeGroup, CfnAttributeGroupAssociation } from './servicecatalogappregistry.generated';
 
 const ATTRIBUTE_GROUP_READ_ONLY_RAM_PERMISSION_ARN = 'arn:aws:ram::aws:permission/AWSRAMPermissionServiceCatalogAppRegistryAttributeGroupReadOnly';
 const ATTRIBUTE_GROUP_ALLOW_ACCESS_RAM_PERMISSION_ARN = 'arn:aws:ram::aws:permission/AWSRAMPermissionServiceCatalogAppRegistryAttributeGroupAllowAssociation';
@@ -58,6 +59,22 @@ export interface AttributeGroupProps {
 abstract class AttributeGroupBase extends cdk.Resource implements IAttributeGroup {
   public abstract readonly attributeGroupArn: string;
   public abstract readonly attributeGroupId: string;
+  private readonly associatedApplications: Set<string> = new Set();
+
+  /**
+   * Associate an application with attribute group
+   * If the attribute group is already associated, it will ignore duplicate request.
+   */
+  public associateApplicationWithAttributeGroup(application: IApplication): void {
+    if (!this.associatedApplications.has(application.node.addr)) {
+      const hashId = this.generateUniqueHash(application.node.addr);
+      new CfnAttributeGroupAssociation(this, `ApplicationAttributeGroupAssociation${hashId}`, {
+        application: application.applicationId,
+        attributeGroup: this.attributeGroupId,
+      });
+      this.associatedApplications.add(application.node.addr);
+    }
+  }
 
   public shareAttributeGroup(shareOptions: ShareOptions): void {
     const principals = getPrincipalsforSharing(shareOptions);
@@ -85,6 +102,11 @@ abstract class AttributeGroupBase extends cdk.Resource implements IAttributeGrou
         return shareOptions.sharePermission ?? ATTRIBUTE_GROUP_READ_ONLY_RAM_PERMISSION_ARN;
     }
   }
+
+  /**
+   * Create a unique hash
+   */
+  protected abstract generateUniqueHash(resourceAddress: string): string;
 }
 
 /**
@@ -109,6 +131,10 @@ export class AttributeGroup extends AttributeGroupBase implements IAttributeGrou
     class Import extends AttributeGroupBase {
       public readonly attributeGroupArn = attributeGroupArn;
       public readonly attributeGroupId = attributeGroupId!;
+
+      protected generateUniqueHash(resourceAddress: string): string {
+        return hashValues(this.attributeGroupArn, resourceAddress);
+      }
     }
 
     return new Import(scope, id, {
@@ -118,6 +144,7 @@ export class AttributeGroup extends AttributeGroupBase implements IAttributeGrou
 
   public readonly attributeGroupArn: string;
   public readonly attributeGroupId: string;
+  private readonly nodeAddress: string;
 
   constructor(scope: Construct, id: string, props: AttributeGroupProps) {
     super(scope, id);
@@ -132,6 +159,11 @@ export class AttributeGroup extends AttributeGroupBase implements IAttributeGrou
 
     this.attributeGroupArn = attributeGroup.attrArn;
     this.attributeGroupId = attributeGroup.attrId;
+    this.nodeAddress = cdk.Names.nodeUniqueId(attributeGroup.node);
+  }
+
+  protected generateUniqueHash(resourceAddress: string): string {
+    return hashValues(this.nodeAddress, resourceAddress);
   }
 
   private validateAttributeGroupProps(props: AttributeGroupProps) {

packages/@aws-cdk/aws-servicecatalogappregistry/test/application-associator.test.ts

@@ -35,6 +35,38 @@ describe('Scope based Associations with Application within Same Account', () =>
     });
   });
 });
+
+describe('Associate attribute group with Application', () => {
+  let app: cdk.App;
+  beforeEach(() => {
+    app = new cdk.App({
+      context: {
+        '@aws-cdk/core:newStyleStackSynthesis': false,
+      },
+    });
+  });
+
+  test('Associate Attribute Group with application created by ApplicationAssociator', () => {
+
+    const customAttributeGroup = new CustomAppRegistryAttributeGroup(app, 'AppRegistryAttributeGroup');
+
+    const appAssociator = new appreg.ApplicationAssociator(app, 'TestApplication', {
+      applications: [appreg.TargetApplication.createApplicationStack({
+        applicationName: 'TestAssociatedApplication',
+        stackName: 'TestAssociatedApplicationStack',
+      })],
+    });
+
+    customAttributeGroup.attributeGroup.associateApplicationWithAttributeGroup(appAssociator.appRegistryApplication());
+    Template.fromStack(customAttributeGroup.attributeGroup.stack).resourceCountIs('AWS::ServiceCatalogAppRegistry::AttributeGroupAssociation', 1);
+    Template.fromStack(customAttributeGroup.attributeGroup.stack).hasResourceProperties('AWS::ServiceCatalogAppRegistry::AttributeGroupAssociation', {
+      Application: { 'Fn::ImportValue': 'TestAssociatedApplicationStack:ExportsOutputFnGetAttDefaultCdkApplication4573D5A3IdAEBA32E0' },
+      AttributeGroup: { 'Fn::GetAtt': ['MyFirstAttributeGroupDBC21379', 'Id'] },
+    });
+
+  });
+});
+
 describe('Scope based Associations with Application with Cross Region/Account', () => {
   let app: cdk.App;
   beforeEach(() => {
@@ -211,3 +243,18 @@ class AppRegistrySampleStack extends cdk.Stack {
     super(scope, id, props);
   }
 }
+
+class CustomAppRegistryAttributeGroup extends cdk.Stack {
+  public readonly attributeGroup: appreg.AttributeGroup;
+
+  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
+    super(scope, id, props);
+    const myAttributeGroup = new appreg.AttributeGroup(this, 'MyFirstAttributeGroup', {
+      attributeGroupName: 'MyFirstAttributeGroupName',
+      description: 'Test attribute group',
+      attributes: {},
+    });
+
+    this.attributeGroup = myAttributeGroup;
+  }
+}

packages/@aws-cdk/aws-servicecatalogappregistry/test/attribute-group.test.ts

@@ -176,6 +176,30 @@ describe('Attribute Group', () => {
     });
   });
 
+  describe('Associate application to an attribute group', () => {
+    let attributeGroup: appreg.AttributeGroup;
+
+    beforeEach(() => {
+      attributeGroup = new appreg.AttributeGroup(stack, 'MyAttributeGroupForAssociation', {
+        attributeGroupName: 'MyAttributeGroupForAssociation',
+        attributes: {},
+      });
+    });
+
+    test('Associate an application to an attribute group', () => {
+      const application = new appreg.Application(stack, 'MyApplication', {
+        applicationName: 'MyTestApplication',
+      });
+      attributeGroup.associateApplicationWithAttributeGroup(application);
+      Template.fromStack(stack).hasResourceProperties('AWS::ServiceCatalogAppRegistry::AttributeGroupAssociation', {
+        Application: { 'Fn::GetAtt': ['MyApplication5C63EC1D', 'Id'] },
+        AttributeGroup: { 'Fn::GetAtt': ['MyAttributeGroupForAssociation6B3E1329', 'Id'] },
+      });
+
+    });
+
+  });
+
   describe('Resource sharing of an attribute group', () => {
     let attributeGroup: appreg.AttributeGroup;
 

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/cdk.out

@@ -1 +1 @@
-{"version":"22.0.0"}
+{"version":"30.1.0"}

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/integ-servicecatalogappregistry-attribute-group.assets.json

@@ -1,15 +1,15 @@
 {
-  "version": "22.0.0",
+  "version": "30.1.0",
   "files": {
-    "3dece22dad73361a79cb380f2880362a20ffc5c0cc75ddc6707e26b5a88cf93f": {
+    "eceea7410a36f9390f3bde280de21a8cf782a5552f62b847978c3106e40e8f7c": {
       "source": {
         "path": "integ-servicecatalogappregistry-attribute-group.template.json",
         "packaging": "file"
       },
       "destinations": {
         "current_account-current_region": {
           "bucketName": "cdk-hnb659fds-assets-${AWS::AccountId}-${AWS::Region}",
-          "objectKey": "3dece22dad73361a79cb380f2880362a20ffc5c0cc75ddc6707e26b5a88cf93f.json",
+          "objectKey": "eceea7410a36f9390f3bde280de21a8cf782a5552f62b847978c3106e40e8f7c.json",
           "assumeRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-file-publishing-role-${AWS::AccountId}-${AWS::Region}"
         }
       }

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/integ-servicecatalogappregistry-attribute-group.template.json

@@ -15,8 +15,8 @@
       "beta": "time2"
      }
     },
-    "Name": "myAttributeGroupTest",
-    "Description": "my attribute group description"
+    "Name": "myAttributeGroup",
+    "Description": "test attribute group description"
    }
   },
   "TestAttributeGroupRAMSharec67f7d80e5baA10EFB4E": {

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/integ.json

@@ -1,5 +1,5 @@
 {
-  "version": "22.0.0",
+  "version": "30.1.0",
   "testCases": {
     "integ.attribute-group": {
       "stacks": [

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/manifest.json

@@ -1,5 +1,5 @@
 {
-  "version": "22.0.0",
+  "version": "30.1.0",
   "artifacts": {
     "integ-servicecatalogappregistry-attribute-group.assets": {
       "type": "cdk:asset-manifest",
@@ -17,7 +17,7 @@
         "validateOnSynth": false,
         "assumeRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-deploy-role-${AWS::AccountId}-${AWS::Region}",
         "cloudFormationExecutionRoleArn": "arn:${AWS::Partition}:iam::${AWS::AccountId}:role/cdk-hnb659fds-cfn-exec-role-${AWS::AccountId}-${AWS::Region}",
-        "stackTemplateAssetObjectUrl": "s3://cdk-hnb659fds-assets-${AWS::AccountId}-${AWS::Region}/3dece22dad73361a79cb380f2880362a20ffc5c0cc75ddc6707e26b5a88cf93f.json",
+        "stackTemplateAssetObjectUrl": "s3://cdk-hnb659fds-assets-${AWS::AccountId}-${AWS::Region}/eceea7410a36f9390f3bde280de21a8cf782a5552f62b847978c3106e40e8f7c.json",
         "requiresBootstrapStackVersion": 6,
         "bootstrapStackVersionSsmParameter": "/cdk-bootstrap/hnb659fds/version",
         "additionalDependencies": [

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.js.snapshot/tree.json

@@ -30,8 +30,8 @@
                         "beta": "time2"
                       }
                     },
-                    "name": "myAttributeGroupTest",
-                    "description": "my attribute group description"
+                    "name": "myAttributeGroup",
+                    "description": "test attribute group description"
                   }
                 },
                 "constructInfo": {
@@ -228,7 +228,7 @@
         "path": "Tree",
         "constructInfo": {
           "fqn": "constructs.Construct",
-          "version": "10.1.189"
+          "version": "10.1.252"
         }
       }
     },

packages/@aws-cdk/aws-servicecatalogappregistry/test/integ.attribute-group.ts

@@ -6,8 +6,8 @@ const app = new cdk.App();
 const stack = new cdk.Stack(app, 'integ-servicecatalogappregistry-attribute-group');
 
 const attributeGroup = new appreg.AttributeGroup(stack, 'TestAttributeGroup', {
-  attributeGroupName: 'myAttributeGroupTest',
-  description: 'my attribute group description',
+  attributeGroupName: 'myAttributeGroup',
+  description: 'test attribute group description',
   attributes: {
     stage: 'alpha',
     teamMembers: [