add error register in rpc handler (#4356)

* add error register in rpc handler

* fix

* Create empty-pugs-prove.md

* fixes

* Update packages/blitz/src/errors.ts

Author: siddhsuresh

.changeset/empty-pugs-prove.md

@@ -0,0 +1,6 @@
+---
+"@blitzjs/rpc": patch
+"blitz": patch
+---
+
+fix: make sure blitz superjson custom error registers in rpc handler

packages/blitz-rpc/src/index-server.ts

@@ -1,4 +1,4 @@
-import {assert, Ctx, ResolverConfig} from "blitz"
+import {assert, Ctx, ResolverConfig, registerBlitzErrorClasses} from "blitz"
 import {NextApiRequest, NextApiResponse} from "next"
 import {resolve} from "path"
 import {deserialize, parse, serialize as superjsonSerialize} from "superjson"
@@ -239,6 +239,7 @@ interface RpcConfig {
 }
 
 export function rpcHandler(config?: RpcConfig) {
+  registerBlitzErrorClasses()
   return async function handleRpcRequest(req: NextApiRequest, res: NextApiResponse, ctx: Ctx) {
     const resolverMap = await getResolverMap()
     assert(resolverMap, "No query or mutation resolvers found")
@@ -362,6 +363,7 @@ export function rpcHandler(config?: RpcConfig) {
 type Params = Record<string, unknown>
 
 export function rpcAppHandler(config?: RpcConfig) {
+  registerBlitzErrorClasses()
   async function handleRpcRequest(req: Request, context: {params: Params}, ctx?: Ctx) {
     const session = ctx?.session
     const resolverMap = await getResolverMap()

packages/blitz/src/errors.ts

@@ -89,7 +89,10 @@ export class PaginationArgumentError extends Error {
   }
 }
 
-if (isNotInUserTestEnvironment() && !globalThis._BLITZ_ERROR_CLASS_REGISTERED) {
+let _blitzErrorClassRegistered = false
+
+export function registerBlitzErrorClasses() {
+  if (_blitzErrorClassRegistered || !isNotInUserTestEnvironment()) return
   SuperJson.registerClass(AuthenticationError, {
     identifier: "BlitzAuthenticationError",
     allowProps: errorProps,
@@ -125,5 +128,7 @@ if (isNotInUserTestEnvironment() && !globalThis._BLITZ_ERROR_CLASS_REGISTERED) {
     allowProps: errorProps,
   })
 
-  globalThis._BLITZ_ERROR_CLASS_REGISTERED = true
+  _blitzErrorClassRegistered = true
 }
+
+registerBlitzErrorClasses()

packages/blitz/src/index-browser.tsx

@@ -8,6 +8,7 @@ import {
   PaginationArgumentError,
   RedirectError,
   OAuthError,
+  registerBlitzErrorClasses,
 } from "./errors"
 import type {EventHooks, MiddlewareHooks} from "./types"
 export {
@@ -18,6 +19,7 @@ export {
   PaginationArgumentError,
   RedirectError,
   OAuthError,
+  registerBlitzErrorClasses,
 }
 export * from "./utils/enhance-prisma"