All files / lib/lambda/middleware utils.ts

100% Statements 14/14
100% Branches 4/4
100% Functions 4/4
100% Lines 14/14

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69                              8x         30x   30x 13x                 8x     28x     28x                 8x         16x   16x 2x                 8x 14x 14x    
import { Request } from "@middy/core";
import { getInternal } from "@middy/util";
import { Context } from "aws-lambda";
import { FullUser } from "shared-types";
import { main } from "shared-types/opensearch";
 
export type ContextWithPackage = Context & { packageResult?: main.ItemResult };
export type ContextWithCurrUser = Context & { currUser?: FullUser };
 
/**
 * Stores the package in the request internal storage
 * @param {main.ItemResult} packageResult package to store
 * @param {Request} request request to store the package in
 * @param {boolean} setToContext [false] if the package should also be stored in the request context so that it is available in the handler
 */
export const storePackageInRequest = (
  packageResult: main.ItemResult | undefined,
  request: Request,
  setToContext?: boolean,
): void => {
  Object.assign(request.internal, { packageResult });
 
  if (setToContext) {
    Object.assign(request.context, { packageResult });
  }
};
 
/**
 * Gets the package from the request internal storage
 * @param {Request} request request the package is stored in
 * @returns {main.ItemResult} the package or undefined if not found
 */
export const getPackageFromRequest = async (
  request: Request,
): Promise<main.ItemResult | undefined> => {
  const { packageResult } = (await getInternal("packageResult", request)) as {
    packageResult: main.ItemResult;
  };
  return packageResult;
};
 
/**
 * Stores the authenticated user in the request internal storage
 * @param {MiddyUser} currUser authenticated user to store in internal storage
 * @param {Request} request request to store the user in
 * @param {boolean} setToContext [false] if the user should also be stored in the request context so that it is available in the handler
 */
export const storeAuthUserInRequest = (
  currUser: FullUser,
  request: Request,
  setToContext?: boolean,
): void => {
  Object.assign(request.internal, { currUser });
 
  if (setToContext) {
    Object.assign(request.context, { currUser });
  }
};
 
/**
 * Gets the authenticated user from the request internal storage
 * @param {Request} request request the user is stored in
 * @returns {MiddyUser} the user or undefined if not found
 */
export const getAuthUserFromRequest = async (request: Request): Promise<FullUser | undefined> => {
  const { currUser } = (await getInternal("currUser", request)) as { currUser: FullUser };
  return currUser;
};