infisical/backend/src/services/project-key/project-key-dal.ts

import { TDbClient } from "@app/db";
import { TableName, TProjectKeys } from "@app/db/schemas";
import { DatabaseError } from "@app/lib/errors";
import { ormify } from "@app/lib/knex";

export type TProjectKeyDALFactory = ReturnType<typeof projectKeyDALFactory>;

export const projectKeyDALFactory = (db: TDbClient) => {
  const projectKeyOrm = ormify(db, TableName.ProjectKeys);

  const findLatestProjectKey = async (
    userId: string,
    projectId: string
  ): Promise<TProjectKeys & { sender: { publicKey: string } }> => {
    try {
      const projectKey = await db(TableName.ProjectKeys)
        .where({ projectId, receiverId: userId })
        .join(TableName.Users, `${TableName.ProjectKeys}.senderId`, `${TableName.Users}.id`)
        .join(
          TableName.UserEncryptionKey,
          `${TableName.UserEncryptionKey}.userId`,
          `${TableName.Users}.id`
        )
        .orderBy("createdAt", "desc", "last")
        .select(`${TableName.ProjectKeys}.*`, `${TableName.UserEncryptionKey}.publicKey`)
        .first();
      if (projectKey) {
        projectKey.sender = {
          publicKey: projectKey.publicKey
        };
      }
      return projectKey;
    } catch (error) {
      throw new DatabaseError({ error, name: "Find latest project key" });
    }
  };

  const findAllProjectUserPubKeys = async (projectId: string) => {
    try {
      const pubKeys = await db(TableName.ProjectMembership)
        .where({ projectId })
        .join(TableName.Users, `${TableName.ProjectMembership}.userId`, `${TableName.Users}.id`)
        .join(
          TableName.UserEncryptionKey,
          `${TableName.Users}.id`,
          `${TableName.UserEncryptionKey}.userId`
        )
        .select("userId", "publicKey");
      return pubKeys;
    } catch (error) {
      throw new DatabaseError({ error, name: "Find all workspace pub keys" });
    }
  };

  return {
    ...projectKeyOrm,
    findAllProjectUserPubKeys,
    findLatestProjectKey
  };
};