infisical/backend/src/utils/authn/helpers/authDataExtractors.ts

import { AuthData } from "../../../interfaces/middleware";
import {
  ServiceAccount,
  ServiceTokenData,
  ServiceTokenDataV3,
  User
} from "../../../models";

/**
 * Returns an object containing the id of the authentication data payload
 * @param {AuthData} authData - authentication data object
 * @returns 
 */
 export const getAuthDataPayloadIdObj = (authData: AuthData) => {
    if (authData.authPayload instanceof User) {
        return { userId: authData.authPayload._id };
    }
  
    if (authData.authPayload instanceof ServiceAccount) {
        return { serviceAccountId: authData.authPayload._id };
    }
  
    if (authData.authPayload instanceof ServiceTokenData) {
        return { serviceTokenDataId: authData.authPayload._id };
    }
  
    if (authData.authPayload instanceof ServiceTokenDataV3) {
        return { serviceTokenDataId: authData.authPayload._id };
    }
};
  
/**
 * Returns an object containing the user associated with the authentication data payload
 * @param {AuthData} authData - authentication data object
 * @returns 
 */
export const getAuthDataPayloadUserObj = (authData: AuthData) => {
    if (authData.authPayload instanceof User) {
        return { user: authData.authPayload._id };
    }

    if (authData.authPayload instanceof ServiceAccount) {
        return { user: authData.authPayload.user };
    }

    if (authData.authPayload instanceof ServiceTokenData) {
        return { user: authData.authPayload.user };
    }

    if (authData.authPayload instanceof ServiceTokenDataV3) {
        return { user: authData.authPayload.user };
    }
}