From aa8c54c511463b5afa98a6bb5a40d90e455f97e4 Mon Sep 17 00:00:00 2001
From: abdou6666 <sfayhi.abderrahmen@esprit.tn>
Date: Mon, 3 Feb 2025 11:09:23 +0100
Subject: [PATCH] feat: zod pattern

---
 api/src/chat/schemas/types/pattern.ts | 49 +++++++++++++++++----------
 1 file changed, 31 insertions(+), 18 deletions(-)

diff --git a/api/src/chat/schemas/types/pattern.ts b/api/src/chat/schemas/types/pattern.ts
index 8b0ceab1..9116a9ec 100644
--- a/api/src/chat/schemas/types/pattern.ts
+++ b/api/src/chat/schemas/types/pattern.ts
@@ -1,29 +1,42 @@
 /*
- * Copyright © 2024 Hexastack. All rights reserved.
+ * Copyright © 2025 Hexastack. All rights reserved.
  *
  * Licensed under the GNU Affero General Public License v3.0 (AGPLv3) with the following additional terms:
  * 1. The name "Hexabot" is a trademark of Hexastack. You may not use this name in derivative works without express written permission.
  * 2. All derivative works must include clear attribution to the original creator and software, Hexastack and Hexabot, in a prominent location (e.g., in the software's "About" section, documentation, and README file).
  */
 
+import { z } from 'zod';
+
 import { PayloadType } from './message';
 
-export interface PayloadPattern {
-  label: string;
-  value: string;
-  // @todo : rename 'attachment' to 'attachments'
-  type?: PayloadType;
-}
+export const PayloadPatternSchema = z.object({
+  label: z.string(),
+  value: z.string(),
+  type: z.nativeEnum(PayloadType).optional(),
+});
 
-export type NlpPattern =
-  | {
-      entity: string;
-      match: 'entity';
-    }
-  | {
-      entity: string;
-      match: 'value';
-      value: string;
-    };
+export type PayloadPattern = z.infer<typeof PayloadPatternSchema>;
 
-export type Pattern = string | RegExp | PayloadPattern | NlpPattern[];
+export const nlpPatternSchema = z.discriminatedUnion('match', [
+  z.object({
+    entity: z.string(),
+    match: z.literal('entity'),
+  }),
+  z.object({
+    entity: z.string(),
+    match: z.literal('value'),
+    value: z.string(),
+  }),
+]);
+
+export type NlpPattern = z.infer<typeof nlpPatternSchema>;
+
+export const PatternSchema = z.union([
+  z.string(),
+  z.instanceof(RegExp),
+  PayloadPatternSchema,
+  z.array(nlpPatternSchema),
+]);
+
+export type Pattern = z.infer<typeof PatternSchema>;