getAllStateUsers.ts

import {
  CognitoIdentityProviderClient,
  ListUsersCommand,
  ListUsersCommandInput,
  ListUsersCommandOutput,
} from "@aws-sdk/client-cognito-identity-provider";

export type StateUser = {
  firstName: string;
  lastName: string;
  email: string;
  formattedEmailAddress: string;
};

export const getAllStateUsers = async ({
  userPoolId,
  state,
}: {
  userPoolId: string;
  state: string;
}): Promise<StateUser[]> => {
  try {
    const params: ListUsersCommandInput = {
      UserPoolId: userPoolId,
      Limit: 60,
    };
    const command = new ListUsersCommand(params);
    const cognitoClient = new CognitoIdentityProviderClient({
      region: process.env.region,
    });
    const response: ListUsersCommandOutput = await cognitoClient.send(command);

    if (!response.Users || response.Users.length === 0) {
      return [];
    }
    const filteredStateUsers = response.Users.filter((user) => {
      const stateAttribute = user.Attributes?.find((attr) => attr.Name === "custom:state");
      return stateAttribute?.Value?.split(",").includes(state);
    }).map((user) => {
      const attributes = user.Attributes?.reduce((acc, attr) => {
        acc[attr.Name as any] = attr.Value;
        return acc;
      }, {} as Record<string, string | undefined>);
      return {
        firstName: attributes?.["given_name"],
        lastName: attributes?.["family_name"],
        email: attributes?.["email"],
        formattedEmailAddress: `${attributes?.["given_name"]} ${attributes?.["family_name"]} <${attributes?.["email"]}>`,
      };
    });

    return filteredStateUsers as StateUser[];
  } catch (error) {
    console.error("Error fetching users:", error);
    throw new Error("Error fetching users");
  }
};