expose internal constructs

Author: joshbalfour

packages/constructs/edge-api/src/edge-api.ts

@@ -4,9 +4,14 @@ import { EdgeAPIProps, Endpoint } from './types'
 import { ProductionEdgeAPI } from './production-edge-api'
 import { RecordTarget } from 'aws-cdk-lib/aws-route53'
 import { DevEdgeAPI } from './dev-edge-api'
+import { Distribution } from 'aws-cdk-lib/aws-cloudfront'
+import { HttpApi } from '@aws-cdk/aws-apigatewayv2-alpha'
+
 export class EdgeAPI extends Construct {
   private api: ProductionEdgeAPI | DevEdgeAPI
   route53Target: RecordTarget
+  _distribution?: Distribution
+  _httpApi?: HttpApi
 
   constructor(scope: Construct, id: string, props: EdgeAPIProps) {
     super(scope, id)
@@ -22,6 +27,12 @@ export class EdgeAPI extends Construct {
     }
     this.api = props.devMode ? new DevEdgeAPI(this, 'api', props) : new ProductionEdgeAPI(this, 'api', props)
     this.route53Target = this.api.r53Target
+    if (this.api instanceof ProductionEdgeAPI) {
+      this._distribution = this.api.distribution
+    }
+    if (this.api instanceof DevEdgeAPI) {
+      this._httpApi = this.api.api
+    }
   }
 
   addEndpoint(endpoint: Endpoint) {