Merge pull request #5 from cloudflare/main

Add custom constraint messages (cloudflare#214)

package.json

@@ -1,6 +1,6 @@
 {
   "name": "chanfana",
-  "version": "2.5.0",
+  "version": "2.5.1",
   "description": "OpenAPI 3 and 3.1 schema generator and validator for Hono, itty-router and more!",
   "main": "dist/index.js",
   "module": "dist/index.mjs",

src/endpoints/d1/create.ts

@@ -1,10 +1,11 @@
-import { ApiException, InputValidationException } from "../../exceptions";
+import { ApiException, type InputValidationException } from "../../exceptions";
 import { CreateEndpoint } from "../create";
 import type { Logger, O } from "../types";
 
 export class D1CreateEndpoint<HandleArgs extends Array<object> = Array<object>> extends CreateEndpoint<HandleArgs> {
   dbName = "DB";
   logger?: Logger;
+  constraintsMessages: Record<string, InputValidationException> = {};
 
   getDBBinding(): D1Database {
     const env = this.params.router.getBindings(this.args);
@@ -36,11 +37,9 @@ export class D1CreateEndpoint<HandleArgs extends Array<object> = Array<object>>
       if (this.logger)
         this.logger.error(`Caught exception while trying to create ${this.meta.model.tableName}: ${e.message}`);
       if (e.message.includes("UNIQUE constraint failed")) {
-        if (e.message.includes(this.meta.model.tableName) && e.message.includes(this.meta.model.primaryKeys[0])) {
-          throw new InputValidationException(`An object with this ${this.meta.model.primaryKeys[0]} already exists`, [
-            "body",
-            this.meta.model.primaryKeys[0],
-          ]);
+        const constraintMessage = e.message.split("UNIQUE constraint failed:")[1].split(":")[0].trim();
+        if (this.constraintsMessages[constraintMessage]) {
+          throw this.constraintsMessages[constraintMessage];
         }
       }
 

src/endpoints/d1/update.ts

@@ -1,10 +1,11 @@
-import { ApiException } from "../../exceptions";
+import { ApiException, type InputValidationException } from "../../exceptions";
 import type { Logger, O, UpdateFilters } from "../types";
 import { UpdateEndpoint } from "../update";
 
 export class D1UpdateEndpoint<HandleArgs extends Array<object> = Array<object>> extends UpdateEndpoint<HandleArgs> {
   dbName = "DB";
   logger?: Logger;
+  constraintsMessages: Record<string, InputValidationException> = {};
 
   getDBBinding(): D1Database {
     const env = this.params.router.getBindings(this.args);
@@ -74,6 +75,13 @@ export class D1UpdateEndpoint<HandleArgs extends Array<object> = Array<object>>
     } catch (e: any) {
       if (this.logger)
         this.logger.error(`Caught exception while trying to update ${this.meta.model.tableName}: ${e.message}`);
+      if (e.message.includes("UNIQUE constraint failed")) {
+        const constraintMessage = e.message.split("UNIQUE constraint failed:")[1].split(":")[0].trim();
+        if (this.constraintsMessages[constraintMessage]) {
+          throw this.constraintsMessages[constraintMessage];
+        }
+      }
+
       throw new ApiException(e.message);
     }