logto/packages/core/src/middleware/koa-error-handler.ts

import { LogtoErrorCode } from '@logto/phrases';
import { RequestErrorBody } from '@logto/schemas';
import decamelize from 'decamelize';
import { Middleware } from 'koa';
import { errors } from 'oidc-provider';
import { NotFoundError } from 'slonik';

import RequestError from '@/errors/RequestError';

export default function koaErrorHandler<StateT, ContextT>(): Middleware<
  StateT,
  ContextT,
  RequestErrorBody
> {
  return async (ctx, next) => {
    try {
      await next();
    } catch (error: unknown) {
      if (error instanceof RequestError) {
        ctx.status = error.status;
        ctx.body = error.body;
        return;
      }

      if (error instanceof errors.OIDCProviderError) {
        ctx.status = error.status;
        ctx.body = {
          message: error.error_description ?? error.message,
          // Assert error type of OIDCProviderError, code key should all covered in @logto/phrases
          code: `oidc.${decamelize(error.name)}` as LogtoErrorCode,
          data: error.error_detail,
        };
        return;
      }

      if (error instanceof NotFoundError) {
        const error = new RequestError({ code: 'entity.not_found', status: 404 });
        ctx.status = error.status;
        ctx.body = error.body;
        return;
      }

      throw error;
    }
  };
}