Merge pull request #3 from smallcase/feat/network-array

feat: added route 53 entry block
smallcase · Feb 17, 2022 · 02ae065 · 02ae065
2 parents 5d24aac + 7ffdcd4
commit 02ae065
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 2 deletions.
diff --git a/API.md b/API.md
@@ -1571,6 +1571,7 @@ const loadBalancerProps: LoadBalancerProps = { ... }
 | <code><a href="#@smallcase/aws-cdk-microservice.LoadBalancerProps.property.lbArn">lbArn</a></code> | <code>string</code> | *No description.* |
 | <code><a href="#@smallcase/aws-cdk-microservice.LoadBalancerProps.property.sslEnabled">sslEnabled</a></code> | <code>boolean</code> | *No description.* |
 | <code><a href="#@smallcase/aws-cdk-microservice.LoadBalancerProps.property.targetGroupArn">targetGroupArn</a></code> | <code>string</code> | *No description.* |
+| <code><a href="#@smallcase/aws-cdk-microservice.LoadBalancerProps.property.zoneName">zoneName</a></code> | <code>string</code> | *No description.* |
 
 ---
 
@@ -1624,6 +1625,16 @@ public readonly targetGroupArn: string;
 
 ---
 
+##### `zoneName`<sup>Required</sup> <a name="zoneName" id="@smallcase/aws-cdk-microservice.LoadBalancerProps.property.zoneName"></a>
+
+```typescript
+public readonly zoneName: string;
+```
+
+- *Type:* string
+
+---
+
 ### MicroServiceProps <a name="MicroServiceProps" id="@smallcase/aws-cdk-microservice.MicroServiceProps"></a>
 
 #### Initializer <a name="Initializer" id="@smallcase/aws-cdk-microservice.MicroServiceProps.Initializer"></a>
@@ -1870,6 +1881,7 @@ const networkProps: NetworkProps = { ... }
 | <code><a href="#@smallcase/aws-cdk-microservice.NetworkProps.property.port">port</a></code> | <code>number</code> | *No description.* |
 | <code><a href="#@smallcase/aws-cdk-microservice.NetworkProps.property.protocol">protocol</a></code> | <code>string</code> | *No description.* |
 | <code><a href="#@smallcase/aws-cdk-microservice.NetworkProps.property.sslEnabled">sslEnabled</a></code> | <code>boolean</code> | *No description.* |
+| <code><a href="#@smallcase/aws-cdk-microservice.NetworkProps.property.zoneName">zoneName</a></code> | <code>string</code> | *No description.* |
 
 ---
 
@@ -1933,6 +1945,16 @@ public readonly sslEnabled: boolean;
 
 ---
 
+##### `zoneName`<sup>Required</sup> <a name="zoneName" id="@smallcase/aws-cdk-microservice.NetworkProps.property.zoneName"></a>
+
+```typescript
+public readonly zoneName: string;
+```
+
+- *Type:* string
+
+---
+
 ### TargetGroupProps <a name="TargetGroupProps" id="@smallcase/aws-cdk-microservice.TargetGroupProps"></a>
 
 #### Initializer <a name="Initializer" id="@smallcase/aws-cdk-microservice.TargetGroupProps.Initializer"></a>

diff --git a/src/constructs/autoScalingGroup.ts b/src/constructs/autoScalingGroup.ts
@@ -19,6 +19,7 @@ export interface NetworkProps {
   readonly sslEnabled: boolean;
   readonly host: string;
   readonly lbArn: string;
+  readonly zoneName: string;
 }
 
 export interface IngressRule {
@@ -279,6 +280,7 @@ export class AutoScaler extends Resource {
         lbArn: t.lbArn,
         sslEnabled: t.sslEnabled,
         targetGroupArn: tg.ref,
+        zoneName: t.zoneName,
       });
 
     });

diff --git a/src/constructs/network.ts b/src/constructs/network.ts
@@ -1,5 +1,7 @@
 import { Resource } from 'aws-cdk-lib';
-import { ApplicationListener, ApplicationProtocol, CfnListenerRule } from 'aws-cdk-lib/aws-elasticloadbalancingv2';
+import { ApplicationListener, ApplicationLoadBalancer, ApplicationProtocol, CfnListenerRule } from 'aws-cdk-lib/aws-elasticloadbalancingv2';
+import { ARecord, HostedZone, RecordTarget } from 'aws-cdk-lib/aws-route53';
+import { LoadBalancerTarget } from 'aws-cdk-lib/aws-route53-targets';
 import { Construct } from 'constructs';
 
 export interface LoadBalancerProps {
@@ -8,6 +10,7 @@ export interface LoadBalancerProps {
   readonly targetGroupArn: string;
   readonly lbArn: string;
   readonly sslEnabled: boolean;
+  readonly zoneName: string;
 }
 
 export class BalancerEntry extends Resource {
@@ -53,6 +56,7 @@ export class BalancerEntry extends Resource {
         priority: Math.floor(Math.random() * (1000 - 200 + 1)) + 200, // this line is a flipping leap of faith
       });
     }
+    this.createRoute53Entry(props);
   }
 
   private getLoadBalancerListener(loadBalancerArn: string, sslEnabled: boolean, appName: string) {
@@ -73,4 +77,16 @@ export class BalancerEntry extends Resource {
     }
     return listeners;
   }
+
+  private createRoute53Entry(props: LoadBalancerProps) {
+    const lb = ApplicationLoadBalancer.fromLookup(this, 'lb-' + props.lbArn, {
+      loadBalancerArn: props.lbArn,
+    });
+    new ARecord(this, 'record-' + props.hostHeader.split('.')[-2], {
+      target: RecordTarget.fromAlias(new LoadBalancerTarget(lb)),
+      zone: new HostedZone(this, props.hostHeader + '-zone', {
+        zoneName: props.zoneName,
+      }),
+    });
+  }
 }
diff --git a/test/hello.test.ts b/test/hello.test.ts
@@ -4,7 +4,7 @@ describe('my suite', () => {
   });
 });
 
-// import { App, Stack, StackProps } from '@aws-cdk/core';
+// import { App, Stack, StackProps } from 'aws-cdk-lib';
 // import { Construct } from 'constructs';
 // import { MicroService } from '../src/constructs/microservice';
 
@@ -48,6 +48,7 @@ describe('my suite', () => {
 //           sslEnabled: false,
 //           port: 8000,
 //           protocol: 'HTTP',
+//           zoneName: 'smallcase.com',
 //         },
 //       ],
 //       createCodedeployApplication: true,