feat: add methods to get schemas

CodyTseng · Sep 9, 2024 · 4f3ea84 · 4f3ea84
1 parent 3778e90
commit 4f3ea84
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/packages/validator/__test__/validator.spec.ts b/packages/validator/__test__/validator.spec.ts
@@ -242,4 +242,13 @@ describe('Validator', () => {
       await expect(specValidator.validateEvent(event2)).rejects.toThrow();
     });
   });
+
+  it('get schema', () => {
+    expect(validator.getIncomingMessageSchema()).toBe(
+      validator['incomingMessageSchema'],
+    );
+    expect(validator.getFilterSchema()).toBe(validator['filterSchema']);
+    expect(validator.getFiltersSchema()).toBe(validator['filtersSchema']);
+    expect(validator.getEventSchema()).toBe(validator['eventSchema']);
+  });
 });
diff --git a/packages/validator/src/validator.ts b/packages/validator/src/validator.ts
@@ -75,6 +75,34 @@ export class Validator {
     );
   }
 
+  /**
+   * Get incoming message schema
+   */
+  getIncomingMessageSchema(): z.ZodType<IncomingMessage> {
+    return this.incomingMessageSchema;
+  }
+
+  /**
+   * Get filter schema
+   */
+  getFilterSchema(): z.ZodType<Filter> {
+    return this.filterSchema;
+  }
+
+  /**
+   * Get filters schema
+   */
+  getFiltersSchema(): z.ZodType<Filter[]> {
+    return this.filtersSchema;
+  }
+
+  /**
+   * Get event schema
+   */
+  getEventSchema(): z.ZodType<Event> {
+    return this.eventSchema;
+  }
+
   private async errorHandler<T>(asyncFunc: () => Promise<T>): Promise<T> {
     try {
       return await asyncFunc();