fix: broken context inference

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

Author: tunnckoCore

src-v4/ss-zod.ts

@@ -96,27 +96,36 @@ console.log(
   proc4("Carol", { user: "carol", role: "admin", paid: true }),
 );
 
-const ping = zagora()
+const pingProc = zagora()
+  .context<{ user: string }>({ user: "default" })
   .input(z.number())
-  .output(z.object({ pong: z.number() }))
+  .output(z.object({ pong: z.number(), userId: z.string() }).strict())
   .errors({
     NOT_FOUND: z.object({ type: z.literal("NOT_FOUND"), userId: z.string() }),
     AUTH_ERR: z.object({ type: z.literal("AUTH_ERR"), retryAfter: z.number() }),
   })
-  .handler(({ errors }, id) => {
+  .handler(({ errors, context }, id) => {
     if (id === 42) {
       throw errors.AUTH_ERR({ retryAfter: Date.now() + 1000 });
     }
     if (id <= 20) {
       throw errors.NOT_FOUND({ userId: "123" });
     }
 
-    return { pong: id + 1 };
-  })
-  .callable();
+    return { pong: id + 1, userId: context.user + "-foo" };
+  });
+
+// const router = {
+//   users: {
+//     ping: pingProc,
+//   },
+// };
+
+// router.users.ping.callable({ user: "sasa" });
 
 // Call synchronously, no try-catch needed
-const resPing = ping(42);
+const ping = pingProc.callable();
+const resPing = ping(30);
 console.log("Test 5 ping-pong:", resPing);
 
 if (

src-v4/sscore.ts

@@ -253,7 +253,7 @@ export interface ProcedureOptions<
   TContext,
   TErrorsMap extends Record<string, StandardSchemaV1> | undefined,
 > {
-  context: TContext | undefined;
+  context: TContext;
   errors: TErrorsMap extends Record<string, StandardSchemaV1>
     ? ErrorHelpers<TErrorsMap>
     : undefined;
@@ -362,7 +362,7 @@ class Builder<
   }
 
   callable<
-    TContext,
+    TNewContext extends TContext,
     TSpread extends SpreadTuple<
       InferSchemaInput<TInputSchema>,
       ZagoraResult<
@@ -378,7 +378,7 @@ class Builder<
           InferSchemaOutput<TOutputSchema>,
           ErrorsMapResolved<TErrorsMap>
         >,
-  >(context?: TContext): TProcReturn {
+  >(context?: TNewContext): TProcReturn {
     const { initialContext, errorsMap } = this.def;
     const handlerFn = this.def.handler as any; // todo: fix, return internal error if not defined (thru createResult)
     const inputSchema = this.def.inputSchema as TInputSchema;
@@ -392,7 +392,7 @@ class Builder<
     const errors = errorsMap ? createErrorHelpers(errorsMap as any) : undefined;
     const options = {
       errors: errors as ErrorHelpers<TErrorsMap>,
-      context: mergedContext,
+      context: mergedContext as TNewContext,
     };
 
     const wrapped = (...args: unknown[]) => {