feat: env & account construct

since stage has problems
almamedia-open-source · Mar 7, 2022 · 9a89806 · 9a89806
1 parent 87d207b
commit 9a89806
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/stage/account.ts b/src/stage/account.ts
@@ -38,3 +38,20 @@ export class AccountStage extends Stage {
 }
 
 
+export class AccountConstruct extends Construct {
+
+  constructor(scope: Construct) {
+
+    const accountType = AccountContext.getAccountType(scope);
+    const id = `${pascalCase(accountType)}Account`;
+
+    super(scope, id);
+
+    // Ensure no environment access from within Account Construct
+    this.node.setContext('environment-type', undefined);
+    this.node.setContext('environment', undefined);
+    this.node.setContext('env', undefined);
+  }
+}
+
+
diff --git a/src/stage/environment.ts b/src/stage/environment.ts
@@ -26,4 +26,12 @@ export class EnvironmentStage extends Stage {
   }
 }
 
+export class EnvironmentConstruct extends Construct {
+  constructor(scope: Construct) {
+    const type = EnvironmentContext.getName(scope);
+    const id = `${pascalCase(type)}Environment`;
+    super(scope, id);
+  }
+}
+