fix: add edge cases testing for errors when no input/output schemas defined

Signed-off-by: tunnckoCore <5038030+tunnckoCore@users.noreply.github.com>

Author: tunnckoCore

bun.lock

@@ -7,7 +7,9 @@ import {
 import type {
   AnySchema,
   InferSchemaInput,
+  InferSchemaInputSafe,
   InferSchemaOutput,
+  InferSchemaOutputSafe,
   IsPromise,
   Prettify,
   ProcedureOptions,
@@ -31,8 +33,8 @@ export class Zagora<
   TIsHandlerAsync,
   THandlerFn,
   TContext extends any | undefined = undefined,
-  TInputSchema extends AnySchema = AnySchema,
-  TOutputSchema extends AnySchema = AnySchema,
+  TInputSchema extends AnySchema | undefined = undefined,
+  TOutputSchema extends AnySchema | undefined = undefined,
   TErrorsMap extends Record<string, AnySchema> | undefined = undefined,
 > {
   constructor(
@@ -106,18 +108,20 @@ export class Zagora<
   }
 
   handler<
-    TFn extends InferSchemaOutput<TInputSchema> extends readonly any[]
-      ? SpreadTuple<
-          [
-            Prettify<ProcedureOptions<TContext, TErrorsMap>>,
-            ...InferSchemaOutput<TInputSchema>,
-          ],
-          any
-        >
-      : (
-          options: Prettify<ProcedureOptions<TContext, TErrorsMap>>,
-          arg: InferSchemaOutput<TInputSchema>,
-        ) => any,
+    TFn extends TInputSchema extends AnySchema
+      ? InferSchemaOutput<TInputSchema> extends readonly any[]
+        ? SpreadTuple<
+            [
+              Prettify<ProcedureOptions<TContext, TErrorsMap>>,
+              ...InferSchemaOutput<TInputSchema>,
+            ],
+            any
+          >
+        : (
+            options: Prettify<ProcedureOptions<TContext, TErrorsMap>>,
+            arg: InferSchemaOutput<TInputSchema>,
+          ) => any
+      : (options: Prettify<ProcedureOptions<TContext, TErrorsMap>>) => any,
     TReturn = ReturnType<TFn>,
     TIsAsync extends boolean = IsPromise<TReturn>,
   >(
@@ -167,19 +171,21 @@ export class Zagora<
     type TResult = TIsHandlerAsync extends true
       ? Promise<
           ZagoraResult<
-            InferSchemaOutput<TOutputSchema>,
+            InferSchemaOutputSafe<TOutputSchema>,
             ErrorsMapResolved<TErrorsMap>,
             TResolvedResult
           >
         >
       : ZagoraResult<
-          InferSchemaOutput<TOutputSchema>,
+          InferSchemaOutputSafe<TOutputSchema>,
           ErrorsMapResolved<TErrorsMap>,
           TResolvedResult
         >;
 
-    return procedure as InferSchemaInput<TInputSchema> extends readonly any[]
-      ? SpreadTuple<InferSchemaInput<TInputSchema>, TResult>
-      : (arg: InferSchemaOutput<TInputSchema>) => TResult;
+    return procedure as TInputSchema extends AnySchema
+      ? InferSchemaInput<TInputSchema> extends readonly any[]
+        ? SpreadTuple<InferSchemaInput<TInputSchema>, TResult>
+        : (arg: InferSchemaOutput<TInputSchema>) => TResult
+      : () => TResult;
   }
 }

new-src/index.ts

@@ -27,6 +27,14 @@ export type InferSchemaInput<T extends AnySchema> = T extends StandardSchemaV1<
   ? UInput
   : never;
 
+export type InferSchemaOutputSafe<T> = T extends AnySchema
+  ? InferSchemaOutput<T>
+  : any;
+
+export type InferSchemaInputSafe<T> = T extends AnySchema
+  ? InferSchemaInput<T>
+  : any;
+
 export type UppercaseKeys<T> = {
   [K in keyof T as Uppercase<string & K>]: T[K];
 };

new-src/types.ts

@@ -9,7 +9,7 @@ import {
 } from "../new-src/errors";
 import { zagora } from "../new-src/index";
 
-test("typed errors should be in options", () => {
+test("typed errors should be in options - when input schema is defined", () => {
   const errorSchemas = {
     BAR_ERR: z.object({
       msg: z.string(),
@@ -18,7 +18,7 @@ test("typed errors should be in options", () => {
   };
 
   const func = zagora()
-    // .input(z.string())
+    .input(z.string())
     // .output(z.string())
     .errors(errorSchemas)
     // .errors({ FOO: z.object({ msg: z.string() }) })
@@ -310,3 +310,53 @@ test("Zod Tuple without defaults - missing required arg should fail", async () =
   expect(res.ok).toBe(false);
   expect(res.error?.message).toContain("Input validation failed");
 });
+
+test("Handler without input schema should work", () => {
+  const errorSchemas = {
+    BARRY_ERR: z.object({
+      msg: z.string(),
+      foo: z.number().min(100).max(999).default(500),
+    }),
+  };
+
+  const func = zagora()
+    // no .input() call
+    .errors(errorSchemas)
+    .handler(({ errors }) => {
+      throw errors.BARRY_ERR({ msg: "Some custom error" });
+      // biome-ignore lint/correctness/noUnreachable: bruh
+      return "foo";
+    })
+    .callable();
+
+  const res = func();
+  expect(res.ok).toBe(false);
+
+  if (res.error) {
+    expect(Object.keys(errorSchemas)[0]).toBe(res.error.kind);
+    expect(res.error.kind).toBe("BARRY_ERR");
+    if (res.error.kind === "BARRY_ERR") {
+      expect(res.error.msg).toBe("Some custom error");
+      expect(res.error.foo).toBe(500);
+    }
+  } else {
+    throw new Error(
+      "Expected BARRY_ERR, but got: " + JSON.stringify(res.error),
+    );
+  }
+});
+
+test("Handler without input schema and no errors should work", () => {
+  const func = zagora()
+    // no .input() call
+    .handler(({ context }) => {
+      return { result: "success", ctx: context };
+    })
+    .callable();
+
+  const res = func();
+  expect(res.ok).toBe(true);
+  if (res.ok) {
+    expect(res.data).toEqual({ result: "success", ctx: undefined });
+  }
+});

new-test/edge-cases.test.ts

@@ -47,9 +47,10 @@
   },
   "devDependencies": {
     "@biomejs/biome": "^2",
-    "@types/bun": "latest",
+    "@types/bun": "^1.3.3",
     "arktype": "2.1.23",
     "biome-config-xaxa": "^0.2.1",
+    "bun-types": "^1.3.3",
     "expect-type": "^0.20.0",
     "tsdown": "^0.16.6",
     "tsgo-dev": "^2025.11.22",

package.json

@@ -23,8 +23,8 @@
     "noUnusedParameters": false,
     "noPropertyAccessFromIndexSignature": false,
     "strictNullChecks": true,
-    "types": []
+    "types": ["bun-types"]
   },
-  "include": ["new-src"],
-  "exclude": ["node_modules", "dist", "**/*.test.ts", "**/*.spec.ts"]
+  "include": ["new-src", "new-test"],
+  "exclude": ["node_modules", "dist"]
 }