Exposing stack parameter at higher level for backend and related resources

.changeset/chatty-beans-begin.md

@@ -0,0 +1,7 @@
+---
+'@aws-amplify/backend-function': minor
+'@aws-amplify/backend-auth': minor
+'@aws-amplify/backend': minor
+---
+
+expose stack parameter for backend, function resource, and auth resource

packages/backend-auth/src/factory.ts

@@ -28,10 +28,15 @@ import {
   Expand,
 } from './types.js';
 import { UserPoolAccessPolicyFactory } from './userpool_access_policy_factory.js';
-import { Tags } from 'aws-cdk-lib';
+import { Stack, Tags } from 'aws-cdk-lib';
 
 export type BackendAuth = ResourceProvider<AuthResources> &
-  ResourceAccessAcceptorFactory<AuthRoleName | string>;
+  ResourceAccessAcceptorFactory<AuthRoleName | string> &
+  AuthStackFactory;
+
+export type AuthStackFactory = {
+  stack: Stack;
+};
 
 export type AmplifyAuthProps = Expand<
   Omit<AuthProps, 'outputStorageStrategy' | 'loginWith'> & {
@@ -115,7 +120,10 @@ export class AmplifyAuthFactory implements ConstructFactory<BackendAuth> {
   };
 }
 
-class AmplifyAuthGenerator implements ConstructContainerEntryGenerator {
+class AmplifyAuthGenerator
+  implements ConstructContainerEntryGenerator, AuthStackFactory
+{
+  stack: Stack;
   readonly resourceGroupName = 'auth';
   private readonly name: string;
 
@@ -195,6 +203,7 @@ class AmplifyAuthGenerator implements ConstructContainerEntryGenerator {
           policy.attachToRole(role);
         },
       }),
+      stack: Stack.of(this.generateContainerEntry.prototype),
     };
     if (!this.props.access) {
       return authConstructMixin;

packages/backend-function/src/factory.ts

@@ -44,6 +44,10 @@ export type AddEnvironmentFactory = {
   addEnvironment: (key: string, value: string | BackendSecret) => void;
 };
 
+export type FuncStackFactory = {
+  stack: Stack;
+};
+
 export type CronSchedule =
   | `${string} ${string} ${string} ${string} ${string}`
   | `${string} ${string} ${string} ${string} ${string} ${string}`;
@@ -64,7 +68,8 @@ export const defineFunction = (
 ): ConstructFactory<
   ResourceProvider<FunctionResources> &
     ResourceAccessAcceptorFactory &
-    AddEnvironmentFactory
+    AddEnvironmentFactory &
+    FuncStackFactory
 > => new FunctionFactory(props, new Error().stack);
 
 export type FunctionProps = {
@@ -305,8 +310,10 @@ class AmplifyFunction
   implements
     ResourceProvider<FunctionResources>,
     ResourceAccessAcceptorFactory,
-    AddEnvironmentFactory
+    AddEnvironmentFactory,
+    FuncStackFactory
 {
+  stack: Stack;
   readonly resources: FunctionResources;
   private readonly functionEnvironmentTranslator: FunctionEnvironmentTranslator;
   constructor(
@@ -443,6 +450,8 @@ class AmplifyFunction
       functionStackType,
       fileURLToPath(new URL('../package.json', import.meta.url))
     );
+
+    this.stack = Stack.of(this);
   }
 
   addEnvironment = (key: string, value: string | BackendSecret) => {

packages/backend/src/backend.ts

@@ -12,6 +12,7 @@ export type BackendBase = {
   addOutput: (
     clientConfigPart: DeepPartialAmplifyGeneratedConfigs<ClientConfig>
   ) => void;
+  stack: () => Stack;
 };
 
 // Type that allows construct factories to be defined using any keys except those used in BackendHelpers

packages/backend/src/backend_factory.ts

@@ -143,6 +143,10 @@ export class BackendFactory<
     }
     this.customOutputsAccumulator.addOutput(clientConfigPart);
   };
+
+  rootStack = (): Stack => {
+    return this.stackResolver.getStackFor('root');
+  };
 }
 
 /**
@@ -157,5 +161,6 @@ export const defineBackend = <T extends DefineBackendProps>(
     ...backend.resources,
     createStack: backend.createStack,
     addOutput: backend.addOutput,
+    stack: backend.rootStack,
   };
 };