fix(mixin): use withMixin in Stack to set mixin metadata in its constructs

packages/aws-cdk-lib/core/lib/stack.ts

@@ -1,6 +1,6 @@
 import * as fs from 'fs';
 import * as path from 'path';
-import type { IConstruct } from 'constructs';
+import type { IConstruct, IMixin } from 'constructs';
 import { Construct, Node } from 'constructs';
 import { Annotations } from './annotations';
 import { App } from './app';
@@ -15,6 +15,7 @@ import { CfnResource, TagType } from './cfn-resource';
 import { ContextProvider } from './context-provider';
 import type { Environment, ResourceEnvironment } from './environment';
 import { FeatureFlags } from './feature-flags';
+import { withMixins } from './mixins/private/mixin-metadata';
 import type { PermissionsBoundary } from './permissions-boundary';
 import { PERMISSIONS_BOUNDARY_CONTEXT_KEY } from './permissions-boundary';
 import { CLOUDFORMATION_TOKEN_RESOLVER, CloudFormationLang } from './private/cloudformation-lang';
@@ -616,6 +617,10 @@ export class Stack extends Construct implements ITaggable {
     }
   }
 
+  public with(...mixins: IMixin[]): IConstruct {
+    return withMixins(this, ...mixins);
+  }
+
   /**
    * Resolve a tokenized value in the context of the current stack.
    */

packages/aws-cdk-lib/core/test/mixins/mixin-metadata.test.ts

@@ -68,4 +68,21 @@ describe('Mixin Metadata', () => {
     expect(metadata).toBeDefined();
     expect(metadata?.data).toEqual({ mixin: '*' });
   });
+
+  test('Stack.with() records mixin metadata', () => {
+    stack.with(new TestMixin());
+
+    const metadata = stack.node.metadata.find(m => m.type === MIXIN_METADATA_KEY);
+    expect(metadata).toBeDefined();
+    expect(metadata?.data).toEqual({ mixin: '@aws-cdk/mixins-preview.TestMixin' });
+  });
+
+  test('Stack.with() records metadata for child constructs', () => {
+    const child = new Construct(stack, 'Child');
+    stack.with(new TestMixin());
+
+    const childMetadata = child.node.metadata.find(m => m.type === MIXIN_METADATA_KEY);
+    expect(childMetadata).toBeDefined();
+    expect(childMetadata?.data).toEqual({ mixin: '@aws-cdk/mixins-preview.TestMixin' });
+  });
 });