infisical/backend/src/ee/services/gateway-v2/gateway-v2-dal.ts

import { Knex } from "knex";

import { TDbClient } from "@app/db";
import { AccessScope, GatewaysV2Schema, TableName, TGatewaysV2 } from "@app/db/schemas";
import { DatabaseError } from "@app/lib/errors";
import { buildFindFilter, ormify, selectAllTableCols, TFindFilter, TFindOpt } from "@app/lib/knex";

export type TGatewayV2DALFactory = ReturnType<typeof gatewayV2DalFactory>;

export const gatewayV2DalFactory = (db: TDbClient) => {
  const orm = ormify(db, TableName.GatewayV2);

  const find = async (filter: TFindFilter<TGatewaysV2>, { offset, limit, sort, tx }: TFindOpt<TGatewaysV2> = {}) => {
    try {
      const query = (tx || db.replicaNode())(TableName.GatewayV2)
        // eslint-disable-next-line @typescript-eslint/no-misused-promises
        .where(buildFindFilter(filter, TableName.GatewayV2))
        .join(TableName.Identity, `${TableName.Identity}.id`, `${TableName.GatewayV2}.identityId`)
        .select(selectAllTableCols(TableName.GatewayV2))
        .select(db.ref("name").withSchema(TableName.Identity).as("identityName"));

      if (limit) void query.limit(limit);
      if (offset) void query.offset(offset);
      if (sort) {
        void query.orderBy(sort.map(([column, order, nulls]) => ({ column: column as string, order, nulls })));
      }

      const docs = await query;

      return docs.map((el) => ({
        ...GatewaysV2Schema.parse(el),
        identity: { id: el.identityId, name: el.identityName }
      }));
    } catch (error) {
      throw new DatabaseError({ error, name: `${TableName.GatewayV2}: Find` });
    }
  };

  const findById = async (id: string, tx?: Knex) => {
    try {
      const doc = await (tx || db.replicaNode())(TableName.GatewayV2)
        .join(TableName.Organization, `${TableName.GatewayV2}.orgId`, `${TableName.Organization}.id`)
        .where(`${TableName.GatewayV2}.id`, id)
        .select(selectAllTableCols(TableName.GatewayV2))
        .select(db.ref("name").withSchema(TableName.Organization).as("orgName"))
        .first();

      return doc;
    } catch (error) {
      throw new DatabaseError({ error, name: `${TableName.GatewayV2}: Find by id` });
    }
  };

  return { ...orm, find, findById };
};