Prisma 4 Migration

README.md

This codemod can be used to migrate rejectOnNotFound usages to the new Prisma 4 findUniqueOrThrow and findFirstOrThrow methods.

rejectOnNotFound.ts

import {
  API,
  FileInfo,
  Identifier,
  MemberExpression,
  ObjectExpression,
  ObjectProperty,
} from "jscodeshift";
export const parser = "ts";

export default function (fileInfo: FileInfo, api: API, options: any) {
  const j = api.jscodeshift;
  const ast = j(fileInfo.source);

  // 1. Find any keys with rejectOnNotFound
  // ASSUMPTION: rejectOnNotFound is always set to true
  const calls = ast
    .find(j.CallExpression, (node) =>
      ["findUnique", "findFirst"].includes(node.callee.property?.name)
    )
    .filter((path) => {
      // path.node.arguments?.length === 1 && path.node.arguments[0].properties
      // console.log((path.node.arguments[0] as ObjectExpression).properties)
      return (path.node.arguments[0] as ObjectExpression).properties.some(
        (property) =>
          ((property as ObjectProperty).key as Identifier)?.name ===
          "rejectOnNotFound"
      );
    });

  calls.forEach((path) => {
    const name = ((path.node.callee as MemberExpression).property as Identifier)
      .name;
    if (name === "findUnique") {
      ((path.node.callee as MemberExpression).property as Identifier).name =
        "findUniqueOrThrow";
    } else if (name === "findFirst") {
      ((path.node.callee as MemberExpression).property as Identifier).name =
        "findFirstOrThrow";
    }
  });

  calls
    .find(
      j.ObjectProperty,
      (node) => node.key.name === "rejectOnNotFound"
    )
    .remove();

  return ast.toSource();
}