patch(feat): octo-aws-cdk security-group overlay.

Author: rash805115

packages/octo-aws-cdk/src/index.ts

@@ -53,6 +53,9 @@ import './models/subnet/actions/update-subnet-association.model.action.js';
 export { AwsRegionSharedEfsModule } from './modules/aws-region-shared-efs.module.js';
 export { S3WebsiteSaveManifestModule } from './modules/s3-website-save-manifest.module.js';
 
+import './overlays/security-group/actions/add-security-group.overlay.action.js';
+import './overlays/security-group/actions/update-security-group.overlay.action.js';
+
 import './resources/ecr/actions/add-ecr-image.resource.action.js';
 import './resources/ecr/actions/delete-ecr-image.resource.action.js';
 

packages/octo-aws-cdk/src/overlays/security-group/actions/add-security-group.overlay.action.ts

@@ -0,0 +1,57 @@
+import { Action, ActionInputs, ActionOutputs, Diff, DiffAction, Factory, IModelAction, ModelType } from '@quadnix/octo';
+import { SecurityGroupAnchor } from '../../../anchors/security-group.anchor.js';
+import { SecurityGroup } from '../../../resources/security-group/security-group.resource.js';
+import { Vpc } from '../../../resources/vpc/vpc.resource.js';
+import { ISecurityGroupOverlayProperties } from '../security-group.overlay.interface.js';
+import { SecurityGroupOverlay } from '../security-group.overlay.js';
+
+@Action(ModelType.OVERLAY)
+export class AddSecurityGroupOverlayAction implements IModelAction {
+  readonly ACTION_NAME: string = 'AddSecurityGroupOverlayAction';
+
+  collectInput(diff: Diff): string[] {
+    const securityGroupOverlay = diff.model as SecurityGroupOverlay;
+    const properties = securityGroupOverlay.properties as unknown as ISecurityGroupOverlayProperties;
+
+    return [`resource.vpc-${properties.regionId}`];
+  }
+
+  filter(diff: Diff): boolean {
+    return (
+      diff.action === DiffAction.ADD && diff.model.MODEL_NAME === 'security-group-overlay' && diff.field === 'overlayId'
+    );
+  }
+
+  async handle(diff: Diff, actionInputs: ActionInputs): Promise<ActionOutputs> {
+    const securityGroupOverlay = diff.model as SecurityGroupOverlay;
+    const properties = securityGroupOverlay.properties as unknown as ISecurityGroupOverlayProperties;
+    const anchor = diff.value as SecurityGroupAnchor;
+
+    const vpc = actionInputs[`resource.vpc-${properties.regionId}`] as Vpc;
+
+    const securityGroup = new SecurityGroup(
+      `sec-grp-${properties.regionId}-${anchor.anchorId}`,
+      {
+        awsRegionId: properties.awsRegionId,
+        rules: anchor.rules,
+      },
+      [vpc],
+    );
+
+    const output: ActionOutputs = {};
+    output[securityGroup.resourceId] = securityGroup;
+
+    return output;
+  }
+
+  async revert(): Promise<ActionOutputs> {
+    return {};
+  }
+}
+
+@Factory<AddSecurityGroupOverlayAction>(AddSecurityGroupOverlayAction)
+export class AddSecurityGroupOverlayActionFactory {
+  static async create(): Promise<AddSecurityGroupOverlayAction> {
+    return new AddSecurityGroupOverlayAction();
+  }
+}

packages/octo-aws-cdk/src/overlays/security-group/actions/update-security-group.overlay.action.ts

@@ -0,0 +1,51 @@
+import { Action, ActionInputs, ActionOutputs, Diff, DiffAction, Factory, IModelAction, ModelType } from '@quadnix/octo';
+import { SecurityGroupAnchor } from '../../../anchors/security-group.anchor.js';
+import { SecurityGroup } from '../../../resources/security-group/security-group.resource.js';
+import { ISecurityGroupOverlayProperties } from '../security-group.overlay.interface.js';
+import { SecurityGroupOverlay } from '../security-group.overlay.js';
+
+@Action(ModelType.OVERLAY)
+export class UpdateSecurityGroupOverlayAction implements IModelAction {
+  readonly ACTION_NAME: string = 'UpdateSecurityGroupOverlayAction';
+
+  collectInput(diff: Diff): string[] {
+    const securityGroupOverlay = diff.model as SecurityGroupOverlay;
+    const properties = securityGroupOverlay.properties as unknown as ISecurityGroupOverlayProperties;
+    const anchor = diff.value as SecurityGroupAnchor;
+
+    return [`resource.sec-grp-${properties.regionId}-${anchor.anchorId}`];
+  }
+
+  filter(diff: Diff): boolean {
+    return (
+      diff.action === DiffAction.UPDATE &&
+      diff.model.MODEL_NAME === 'security-group-overlay' &&
+      diff.field === 'overlayId'
+    );
+  }
+
+  async handle(diff: Diff, actionInputs: ActionInputs): Promise<ActionOutputs> {
+    const securityGroupOverlay = diff.model as SecurityGroupOverlay;
+    const properties = securityGroupOverlay.properties as unknown as ISecurityGroupOverlayProperties;
+    const anchor = diff.value as SecurityGroupAnchor;
+
+    const securityGroup = actionInputs[`resource.sec-grp-${properties.regionId}-${anchor.anchorId}`] as SecurityGroup;
+    securityGroup.properties.rules = anchor.rules;
+
+    const output: ActionOutputs = {};
+    output[securityGroup.resourceId] = securityGroup;
+
+    return output;
+  }
+
+  async revert(): Promise<ActionOutputs> {
+    return {};
+  }
+}
+
+@Factory<UpdateSecurityGroupOverlayAction>(UpdateSecurityGroupOverlayAction)
+export class UpdateSecurityGroupOverlayActionFactory {
+  static async create(): Promise<UpdateSecurityGroupOverlayAction> {
+    return new UpdateSecurityGroupOverlayAction();
+  }
+}

packages/octo-aws-cdk/src/overlays/security-group/security-group.overlay.interface.ts

@@ -0,0 +1,4 @@
+export interface ISecurityGroupOverlayProperties {
+  awsRegionId: string;
+  regionId: string;
+}

packages/octo-aws-cdk/src/overlays/security-group/security-group.overlay.ts

@@ -0,0 +1,37 @@
+import { AOverlay, Diff, DiffAction, DiffUtility, IOverlay, IResource, Overlay } from '@quadnix/octo';
+import { SecurityGroupAnchor } from '../../anchors/security-group.anchor.js';
+import { ISecurityGroupOverlayProperties } from './security-group.overlay.interface.js';
+
+@Overlay()
+export class SecurityGroupOverlay extends AOverlay<SecurityGroupOverlay> {
+  override readonly MODEL_NAME: string = 'security-group-overlay';
+
+  constructor(
+    overlayId: IOverlay['overlayId'],
+    properties: ISecurityGroupOverlayProperties,
+    anchors: [SecurityGroupAnchor, SecurityGroupAnchor] | [SecurityGroupAnchor],
+  ) {
+    super(overlayId, properties as unknown as IResource['properties'], anchors);
+  }
+
+  override async diff(previous?: SecurityGroupOverlay): Promise<Diff[]> {
+    const diffs: Diff[] = [];
+
+    // Include diff of each anchor rules.
+    if (previous) {
+      for (let i = 0; i < this.anchors.length; i++) {
+        const anchor = this.anchors[i] as SecurityGroupAnchor;
+
+        if (!DiffUtility.isObjectDeepEquals((previous.anchors[i] as SecurityGroupAnchor).rules, anchor.rules)) {
+          diffs.push(new Diff(this, DiffAction.UPDATE, 'overlayId', anchor));
+        }
+      }
+    } else {
+      for (const anchor of this.anchors) {
+        diffs.push(new Diff(this, DiffAction.ADD, 'overlayId', anchor));
+      }
+    }
+
+    return diffs;
+  }
+}