prisma-extension-random/dist/index.cjs

Add some randomness to your favorite Prisma queries!

"use strict";
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);

// src/index.ts
var src_exports = {};
__export(src_exports, {
  default: () => src_default
});
module.exports = __toCommonJS(src_exports);
var import_client = require("@prisma/client");

// src/helpers.ts
var $findRandom = async (context, args) => {
  const numRows = await context.count({
    where: args?.where
  });
  return await context.findFirst({
    ...args,
    skip: Math.max(0, Math.floor(Math.random() * numRows))
  });
};
var $findManyRandom = async (context, num, args) => {
  const uniqueKey = args?.custom_uniqueKey ?? "id";
  const rows = [];
  const rowIds = [];
  const select = args?.select ?? {};
  select[uniqueKey] = true;
  const where = args?.where ?? {};
  where[uniqueKey] = { notIn: rowIds };
  let numRows = await context.count({ where });
  for (let i = 0; i < num && numRows > 0; ++i) {
    const row = await context.findFirst({
      select,
      where,
      skip: Math.max(0, Math.floor(Math.random() * numRows))
    });
    if (!row) {
      console.error(
        `get random row failed. Where clause: ${JSON.stringify(where)}`
      );
      break;
    }
    rows.push(row);
    rowIds.push(row[uniqueKey]);
    numRows--;
  }
  return rows;
};

// src/index.ts
var src_default = (_extensionArgs) => import_client.Prisma.getExtensionContext({
  name: "prisma-extension-random",
  model: {
    $allModels: {
      async findRandom(args) {
        const context = import_client.Prisma.getExtensionContext(this);
        return await $findRandom(
          context,
          args
        );
      },
      async findManyRandom(num, args) {
        const context = import_client.Prisma.getExtensionContext(this);
        return await $findManyRandom(
          context,
          num,
          args
        );
      }
    }
  }
});