ApiGatewayV2: link resources for all routes

platform/src/components/aws/apigatewayv2-lambda-route.ts

@@ -21,6 +21,10 @@ export interface Args extends ApiGatewayV2BaseRouteArgs {
    * Takes the handler path, the function args, or a function ARN.
    */
   handler: Input<string | FunctionArgs | FunctionArn>;
+  /**
+   * The resources to link to the route function.
+   */
+  handlerLink?: FunctionArgs["link"];
   /**
    * @internal
    */
@@ -66,6 +70,7 @@ export class ApiGatewayV2LambdaRoute extends Component {
         args.handler,
         {
           description: interpolate`${api.name} route ${route}`,
+          link: args.handlerLink,
         },
         args.handlerTransform,
         { parent: self },

platform/src/components/aws/apigatewayv2.ts

@@ -130,6 +130,22 @@ interface ApiGatewayV2CorsArgs {
 }
 
 export interface ApiGatewayV2Args {
+  /**
+   * [Link resources](/docs/linking/) to all your API Gateway routes.
+   *
+   * Linked resources will be merged with the resources linked to each route.
+   *
+   * @example
+   *
+   * Takes a list of resources to link to all the routes.
+   *
+   * ```js
+   * {
+   *   link: [bucket, stripeKey]
+   * }
+   * ```
+   */
+  link?: FunctionArgs["link"];
   /**
    * Set a custom domain for your HTTP API.
    *
@@ -1061,6 +1077,7 @@ export class ApiGatewayV2 extends Component implements Link.Linkable {
         },
         route,
         handler,
+        handlerLink: this.constructorArgs.link,
         handlerTransform: this.constructorArgs.transform?.route?.handler,
         ...transformed[1],
       },

platform/src/components/aws/function.ts

@@ -436,7 +436,7 @@ export interface FunctionArgs {
    * [Link resources](/docs/linking/) to your function. This will:
    *
    * 1. Grant the permissions needed to access the resources.
-   * 2. Allow you to access it in your site using the [SDK](/docs/reference/sdk/).
+   * 2. Allow you to access it in your function using the [SDK](/docs/reference/sdk/).
    *
    * @example
    *

platform/src/components/aws/helpers/function-builder.ts

@@ -18,7 +18,7 @@ export type FunctionBuilder = Output<{
 export function functionBuilder(
   name: string,
   definition: Input<string | FunctionArn | FunctionArgs>,
-  override: Pick<FunctionArgs, "description" | "permissions">,
+  override: Pick<FunctionArgs, "description" | "link" | "permissions">,
   argsTransform?: Transform<FunctionArgs>,
   opts?: ComponentResourceOptions,
 ): FunctionBuilder {
@@ -64,6 +64,12 @@ export function functionBuilder(
           {
             ...definition,
             ...override,
+            link: all([definition.link, override?.link]).apply(
+              ([link, overrideLink]) => [
+                ...(link ?? []),
+                ...(overrideLink ?? []),
+              ],
+            ),
             permissions: all([
               definition.permissions,
               override?.permissions,