cogsbox-state 0.5.449 → 0.5.450

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cogsbox-state",
-  "version": "0.5.449",
+  "version": "0.5.450",
   "description": "React state management library with form controls and server sync",
   "type": "module",
   "main": "./dist/index.js",

package/src/CogsState.tsx

@@ -683,6 +683,47 @@ export const createCogsState = <State extends Record<StateKeys, unknown>>(
 
   return { useCogsState, setCogsOptions } as CogsApi<State>;
 };
+
+// First, let's create a proper type to extract API params from the sync schema
+type ExtractApiParamsFromSchema<T> = T extends {
+  apiParamsSchema: z.ZodObject<infer U>;
+}
+  ? { [K in keyof U]: z.infer<U[K]> }
+  : never;
+
+// Type to check if a schema has API params
+type HasApiParams<T> = T extends { apiParamsSchema: z.ZodObject<any> }
+  ? true
+  : false;
+
+// Create the proper return type for useCogsState
+type UseCogsStateFromSync<
+  TSyncSchema extends {
+    schemas: Record<
+      string,
+      { schemas: { defaultValues: any }; apiParamsSchema?: z.ZodObject<any> }
+    >;
+  },
+> = {
+  // For schemas WITH API params - require apiParams
+  <K extends keyof TSyncSchema['schemas']>(
+    stateKey: K,
+    options: HasApiParams<TSyncSchema['schemas'][K]> extends true
+      ? OptionsType<TSyncSchema['schemas'][K]['schemas']['defaultValues']> & {
+          apiParams: ExtractApiParamsFromSchema<TSyncSchema['schemas'][K]>;
+        }
+      : never
+  ): StateObject<TSyncSchema['schemas'][K]['schemas']['defaultValues']>;
+
+  // For schemas WITHOUT API params - apiParams is optional
+  <K extends keyof TSyncSchema['schemas']>(
+    stateKey: K,
+    options?: HasApiParams<TSyncSchema['schemas'][K]> extends true
+      ? never
+      : OptionsType<TSyncSchema['schemas'][K]['schemas']['defaultValues']>
+  ): StateObject<TSyncSchema['schemas'][K]['schemas']['defaultValues']>;
+};
+// Simplified approach that actually works
 export function createCogsStateFromSync<
   TSyncSchema extends {
     schemas: Record<
@@ -711,12 +752,31 @@ export function createCogsStateFromSync<
     __syncNotifications: syncSchema.notifications,
   });
 
-  // Override the useCogsState function to handle apiParams properly
+  // Create a wrapper that validates API params at runtime
   const useCogsState = <K extends keyof TSyncSchema['schemas']>(
     stateKey: K,
-    options?: any // Just make this flexible
-  ) => {
-    return baseApi.useCogsState(stateKey as any, options);
+    options?: OptionsType<
+      TSyncSchema['schemas'][K]['schemas']['defaultValues']
+    > & {
+      apiParams?: any; // We'll validate this at runtime
+    }
+  ): StateObject<TSyncSchema['schemas'][K]['schemas']['defaultValues']> => {
+    // Runtime validation of API params
+    const schemaEntry = schemas[stateKey as keyof typeof schemas];
+    if (schemaEntry?.apiParamsSchema && options?.apiParams) {
+      // Validate the API params at runtime
+      const result = schemaEntry.apiParamsSchema.safeParse(options.apiParams);
+      if (!result.success) {
+        throw new Error(
+          `Invalid API params for ${String(stateKey)}: ${result.error.message}`
+        );
+      }
+    }
+
+    // Cast to the correct type since we know it's properly typed
+    return baseApi.useCogsState(stateKey as any, options as any) as StateObject<
+      TSyncSchema['schemas'][K]['schemas']['defaultValues']
+    >;
   };
 
   return {
@@ -724,7 +784,6 @@ export function createCogsStateFromSync<
     setCogsOptions: baseApi.setCogsOptions,
   };
 }
-
 const {
   getInitialOptions,
   getValidationErrors,