Moving proprietary schemas into ez namespace. (#850)

RobinTail · Mar 8, 2023 · 172690b · 172690b
1 parent e84b8e4
commit 172690b
Show file tree

Hide file tree

Showing 10 changed files with 65 additions and 60 deletions.
diff --git a/example/endpoints/update-user.ts b/example/endpoints/update-user.ts
@@ -1,4 +1,4 @@
-import { createHttpError, withMeta, z } from "../../src";
+import { createHttpError, ez, withMeta, z } from "../../src";
 import { keyAndTokenAuthenticatedEndpointsFactory } from "../factories";
 
 export const updateUserEndpoint =
@@ -17,7 +17,7 @@ export const updateUserEndpoint =
             "should be greater than or equal to 0"
           ),
         name: z.string().min(1),
-        birthday: z.dateIn(),
+        birthday: ez.dateIn(),
       })
     ).example({
       id: "12",
@@ -27,7 +27,7 @@ export const updateUserEndpoint =
     output: withMeta(
       z.object({
         name: z.string(),
-        createdAt: z.dateOut(),
+        createdAt: ez.dateOut(),
       })
     ).example({
       name: "John Doe",

diff --git a/example/endpoints/upload-avatar.ts b/example/endpoints/upload-avatar.ts
@@ -1,4 +1,4 @@
-import { z } from "../../src";
+import { ez, z } from "../../src";
 import crypto from "crypto";
 import { taggedEndpointsFactory } from "../factories";
 
@@ -8,7 +8,7 @@ export const uploadAvatarEndpoint = taggedEndpointsFactory.build({
   description: "Handles a file upload.",
   input: z
     .object({
-      avatar: z
+      avatar: ez
         .upload()
         .refine(
           (file) => file.mimetype.match(/image\/.+/),

diff --git a/example/factories.ts b/example/factories.ts
@@ -3,6 +3,7 @@ import {
   EndpointsFactory,
   createResultHandler,
   defaultResultHandler,
+  ez,
   z,
 } from "../src";
 import { config } from "./config";
@@ -46,7 +47,7 @@ export const fileStreamingEndpointsFactory = new EndpointsFactory({
   config,
   resultHandler: createResultHandler({
     getPositiveResponse: () => ({
-      schema: z.file().binary(),
+      schema: ez.file().binary(),
       mimeType: "image/*",
     }),
     getNegativeResponse: () => ({

diff --git a/src/extend-zod.ts b/src/extend-zod.ts
diff --git a/src/ez-namespace.ts b/src/ez-namespace.ts
@@ -0,0 +1,11 @@
+import { ZodDateIn } from "./date-in-schema";
+import { ZodDateOut } from "./date-out-schema";
+import { ZodFile } from "./file-schema";
+import { ZodUpload } from "./upload-schema";
+
+export namespace ez {
+  export const file = ZodFile.create;
+  export const upload = ZodUpload.create;
+  export const dateIn = ZodDateIn.create;
+  export const dateOut = ZodDateOut.create;
+}
diff --git a/src/index.ts b/src/index.ts
@@ -27,7 +27,8 @@ export { withMeta } from "./metadata";
 export { testEndpoint } from "./mock";
 export { Client } from "./client";
 
-import * as z from "./extend-zod";
-import createHttpError from "http-errors";
+export { z } from "zod";
+export { ez } from "./ez-namespace";
 
-export { createHttpError, z };
+import createHttpError from "http-errors";
+export { createHttpError };
diff --git a/tests/unit/common-helpers.spec.ts b/tests/unit/common-helpers.spec.ts
@@ -15,7 +15,13 @@ import {
   isValidDate,
   makeErrorFromAnything,
 } from "../../src/common-helpers";
-import { InputValidationError, createHttpError, withMeta, z } from "../../src";
+import {
+  InputValidationError,
+  createHttpError,
+  ez,
+  withMeta,
+  z,
+} from "../../src";
 import { Request } from "express";
 
 describe("Common Helpers", () => {
@@ -441,29 +447,29 @@ describe("Common Helpers", () => {
 
   describe("hasUpload()", () => {
     test("should return true for z.upload()", () => {
-      expect(hasUpload(z.upload())).toBeTruthy();
+      expect(hasUpload(ez.upload())).toBeTruthy();
     });
-    test("should return true for wrapped z.upload()", () => {
-      expect(hasUpload(z.object({ test: z.upload() }))).toBeTruthy();
-      expect(hasUpload(z.upload().or(z.boolean()))).toBeTruthy();
-      expect(
-        hasUpload(
-          z.object({ test: z.boolean() }).and(z.object({ test2: z.upload() }))
-        )
-      ).toBeTruthy();
-      expect(hasUpload(z.optional(z.upload()))).toBeTruthy();
-      expect(hasUpload(z.upload().nullable())).toBeTruthy();
-      expect(hasUpload(z.upload().default({} as UploadedFile))).toBeTruthy();
-      expect(hasUpload(z.record(z.upload()))).toBeTruthy();
-      expect(hasUpload(z.upload().refine(() => true))).toBeTruthy();
-      expect(hasUpload(z.array(z.upload()))).toBeTruthy();
-    });
-    test("should return false in other cases", () => {
-      expect(hasUpload(z.object({}))).toBeFalsy();
-      expect(hasUpload(z.any())).toBeFalsy();
-      expect(hasUpload(z.literal("test"))).toBeFalsy();
-      expect(hasUpload(z.boolean().and(z.literal(true)))).toBeFalsy();
-      expect(hasUpload(z.number().or(z.string()))).toBeFalsy();
+    test.each([
+      z.object({ test: ez.upload() }),
+      ez.upload().or(z.boolean()),
+      z.object({ test: z.boolean() }).and(z.object({ test2: ez.upload() })),
+      z.optional(ez.upload()),
+      ez.upload().nullable(),
+      ez.upload().default({} as UploadedFile),
+      z.record(ez.upload()),
+      ez.upload().refine(() => true),
+      z.array(ez.upload()),
+    ])("should return true for wrapped z.upload() %#", (subject) => {
+      expect(hasUpload(subject)).toBeTruthy();
+    });
+    test.each([
+      z.object({}),
+      z.any(),
+      z.literal("test"),
+      z.boolean().and(z.literal(true)),
+      z.number().or(z.string()),
+    ])("should return false in other cases %#", (subject) => {
+      expect(hasUpload(subject)).toBeFalsy();
     });
   });
 

diff --git a/tests/unit/endpoint.spec.ts b/tests/unit/endpoint.spec.ts
@@ -4,6 +4,7 @@ import {
   createResultHandler,
   defaultEndpointsFactory,
   defaultResultHandler,
+  ez,
   testEndpoint,
   z,
 } from "../../src";
@@ -755,7 +756,7 @@ describe("Endpoint", () => {
     test("should avoid double parsing, should not mutate input", async () => {
       const dateInputMiddleware = createMiddleware({
         input: z.object({
-          middleware_date_input: z.dateIn().optional(),
+          middleware_date_input: ez.dateIn().optional(),
         }),
         middleware: async ({ input: { middleware_date_input }, logger }) => {
           logger.debug("date in mw handler", typeof middleware_date_input);

diff --git a/tests/unit/open-api-helpers.spec.ts b/tests/unit/open-api-helpers.spec.ts
@@ -3,6 +3,7 @@ import { IOSchemaError } from "../../src/errors";
 import {
   OpenAPIError,
   defaultEndpointsFactory,
+  ez,
   withMeta,
   z,
 } from "../../src/index";
@@ -218,7 +219,7 @@ describe("Open API helpers", () => {
     test("should set format:binary and type:string", () => {
       expect(
         depictUpload({
-          schema: z.upload(),
+          schema: ez.upload(),
           ...requestContext,
           next: makeNext(requestContext),
         })
@@ -227,7 +228,7 @@ describe("Open API helpers", () => {
     test("should throw when using in response", () => {
       try {
         depictUpload({
-          schema: z.upload(),
+          schema: ez.upload(),
           ...responseContext,
           next: makeNext(responseContext),
         });
@@ -240,7 +241,7 @@ describe("Open API helpers", () => {
   });
 
   describe("depictFile()", () => {
-    test.each([z.file(), z.file().binary(), z.file().base64()])(
+    test.each([ez.file(), ez.file().binary(), ez.file().base64()])(
       "should set type:string and format accordingly %#",
       (schema) => {
         expect(
@@ -255,7 +256,7 @@ describe("Open API helpers", () => {
     test("should throw when using in input", () => {
       try {
         depictFile({
-          schema: z.file().binary(),
+          schema: ez.file().binary(),
           ...requestContext,
           next: makeNext(requestContext),
         });
@@ -768,7 +769,7 @@ describe("Open API helpers", () => {
     test("should set type:string, pattern and format", () => {
       expect(
         depictDateIn({
-          schema: z.dateIn(),
+          schema: ez.dateIn(),
           ...requestContext,
           next: makeNext(requestContext),
         })
@@ -777,7 +778,7 @@ describe("Open API helpers", () => {
     test("should throw when ZodDateIn in response", () => {
       try {
         depictDateIn({
-          schema: z.dateIn(),
+          schema: ez.dateIn(),
           ...responseContext,
           next: makeNext(responseContext),
         });
@@ -793,7 +794,7 @@ describe("Open API helpers", () => {
     test("should set type:string, description and format", () => {
       expect(
         depictDateOut({
-          schema: z.dateOut(),
+          schema: ez.dateOut(),
           ...responseContext,
           next: makeNext(responseContext),
         })
@@ -802,7 +803,7 @@ describe("Open API helpers", () => {
     test("should throw when ZodDateOut in request", () => {
       try {
         depictDateOut({
-          schema: z.dateOut(),
+          schema: ez.dateOut(),
           ...requestContext,
           next: makeNext(requestContext),
         });

diff --git a/tests/unit/open-api.spec.ts b/tests/unit/open-api.spec.ts
@@ -7,6 +7,7 @@ import {
   createMiddleware,
   createResultHandler,
   defaultEndpointsFactory,
+  ez,
   withMeta,
   z,
 } from "../../src";
@@ -266,11 +267,11 @@ describe("Open API generator", () => {
               input: z.object({
                 bigint: z.bigint(),
                 boolean: z.boolean(),
-                dateIn: z.dateIn(),
+                dateIn: ez.dateIn(),
               }),
               output: z.object({
                 null: z.null(),
-                dateOut: z.dateOut(),
+                dateOut: ez.dateOut(),
               }),
               handler: async () => ({
                 null: null,