Commit f33b11a3c0e6206b106b2cc655632d14ea2650c6 | react-mutation

jsr.json+1-1

...	@@ -1,6 +1,6 @@	...	@@ -1,6 +1,6 @@
1	{	1	{
2	"name": "@clo/react-mutation",	2	"name": "@clo/react-mutation",
3	"version": "1.0.0-beta.8",	3	"version": "1.0.0-beta.9",
4	"exports": {	4	"exports": {
5	".": "./src/mod.ts",	5	".": "./src/mod.ts",
6	"./tanstack-query.ts": "./src/tanstack-query.ts",	6	"./tanstack-query.ts": "./src/tanstack-query.ts",

src/blocking.ts+23-3

 @@ -237,12 +237,12 @@ export class BlockingMutation<
     }
     const args = array.slice() as Args;
-    const { onSuccess, onSuccessDataOnly, onError, onSettled } = args
+    const { onSuccess, onSuccessDataOnly, onError, onSettled, onRestore } = args
       .pop() as RunOptions<Result>;
     const suppressGlobalSuccess = onSuccess !== undefined;
     const suppressGlobalError = onError !== undefined;
-    const promise = this.runAsPromise(...args);
+    const promise = this.#runAsPromiseWithOptions(args, { onRestore });
     promise.then((result) => {
       // Call user handlers
       onSuccess?.(result);
 @@ -275,6 +275,13 @@ export class BlockingMutation<
   /** Calls the mutation, treating the errors as promise rejection. */
   runAsPromise(...args: Args): Promise<Result> {
+    return this.#runAsPromiseWithOptions(args, {});
+  }
+  #runAsPromiseWithOptions(
+    args: Args,
+    { onRestore: userOnRestore }: Pick<RunOptions<Result>, "onRestore">,
+  ): Promise<Result> {
     if (!this.#client.enabled) {
       throw new Error(
         "MutationClient was passed enabled: false. Are you trying to perform a mutation from SSR?",
 @@ -285,7 +292,7 @@ export class BlockingMutation<
     // Check if debouncing is enabled
     if (this.#options.debounceMs !== undefined) {
-      return this.#runDebouncedAndReturn(args, key, channel);
+      return this.#runDebouncedAndReturn(args, key, channel, userOnRestore);
     }
     // Create shared optimistic helpers instance for the channel if it doesn't exist
 @@ -315,6 +322,12 @@ export class BlockingMutation<
       rollbacks += 1;
     };
+    // Register user's onRestore callback if provided
+    if (userOnRestore) {
+      channel.rollbacks.push(userOnRestore);
+      rollbacks += 1;
+    }
     try {
       this.#options.optimistic({
         args,
 @@ -452,6 +465,7 @@ export class BlockingMutation<
     args: Args,
     key: string,
     channel: Channel<Args, Result, Config["optimisticHelpers"]>,
+    userOnRestore?: () => void,
   ): Promise<Result> {
     // If there's a pending debounced call, roll it back
     if (channel.pendingDebounced) {
 @@ -485,6 +499,12 @@ export class BlockingMutation<
       rollbacks += 1;
     };
+    // Register user's onRestore callback if provided
+    if (userOnRestore) {
+      channel.rollbacks.push(userOnRestore);
+      rollbacks += 1;
+    }
     try {
       this.#options.optimistic({
         args,

src/debounced.ts+14-5

 @@ -285,7 +285,7 @@ export class DebouncedMutation<
         "MutationClient was passed enabled: false. Are you trying to perform a mutation from SSR?",
       );
     }
-    this.#runAndReturn(args, true).catch((error) => {
+    this.#runAndReturn(args, true, undefined).catch((error) => {
       const message = `Failed to ${this.describe(...args)}: ${
         errMessage(error)
       }`;
 @@ -300,12 +300,12 @@ export class DebouncedMutation<
       );
     }
     const args = array.slice() as Args;
-    const { onSuccess, onSuccessDataOnly, onError, onSettled } = args
+    const { onSuccess, onSuccessDataOnly, onError, onSettled, onRestore } = args
       .pop() as RunOptions<Result>;
     const suppressGlobalSuccess = onSuccess !== undefined;
     const suppressGlobalError = onError !== undefined;
-    const promise = this.#runAndReturn(args, !suppressGlobalSuccess);
+    const promise = this.#runAndReturn(args, !suppressGlobalSuccess, onRestore);
     promise.then((result) => {
       // Call user handlers
 @@ -334,10 +334,14 @@ export class DebouncedMutation<
         "MutationClient was passed enabled: false. Are you trying to perform a mutation from SSR?",
       );
     }
-    return this.#runAndReturn(args, false);
+    return this.#runAndReturn(args, false, undefined);
   }
-  #runAndReturn(args: Args, reportSuccessGlobally: boolean): Promise<Result> {
+  #runAndReturn(
+    args: Args,
+    reportSuccessGlobally: boolean,
+    userOnRestore?: () => void,
+  ): Promise<Result> {
     const key = this.key(args);
     const channel = this.#getOrPutChannel(key);
 @@ -374,6 +378,11 @@ export class DebouncedMutation<
       channel.rollbacks.push(cb);
     };
+    // Register user's onRestore callback if provided
+    if (userOnRestore) {
+      channel.rollbacks.push(userOnRestore);
+    }
     try {
       this.#options.optimistic(
         {

src/react.ts+24-18

 @@ -45,6 +45,7 @@ export interface UseMutateResultBase<Args extends unknown[], Result> {
     ..._: [...args: Args, options: RunOptions<Result>]
   ) => Promise<Result>;
   clear: () => void;
+  setError: (error: unknown) => void;
   args: Args | undefined;
 }
 @@ -98,7 +99,7 @@ export interface UseMutateIdle {
   isOptimisticData: boolean;
 }
-type AnyMutationStateWithoutRun<Result> =
+type AnyMutationStateWithoutRun<Args extends unknown[], Result> =
   & Omit<
     UseMutateIdle,
     "status" | "result" | "error" | "isSuccess" | "isError" | "errorMessage"
 @@ -110,10 +111,11 @@ type AnyMutationStateWithoutRun<Result> =
     errorMessage: undefined | string;
     isSuccess: boolean;
     isError: boolean;
+    args: Args | undefined;
   };
 export type AnyMutationState<Args extends unknown[], Result> =
-  & AnyMutationStateWithoutRun<Result>
+  & AnyMutationStateWithoutRun<Args, Result>
   & UseMutateResultBase<Args, Result>;
 function initialState() {
 @@ -127,6 +129,7 @@ function initialState() {
     isSuccess: false,
     isError: false,
     isOptimisticData: false,
+    args: undefined,
   } as const;
 }
 @@ -135,15 +138,14 @@ class Observer<Args extends unknown[], Result> {
   mutation: Mutation<Args, Result> | null = null;
   unsubscribe: (() => void) | null = null;
   currentKey: string | null = null;
-  currentArgs: Args | null = null;
   constructor(setRerender: (fn: number) => void) {
     this.setRerender = setRerender;
   }
   watched: Set<string> = new Set();
-  state: AnyMutationStateWithoutRun<Result> = initialState();
+  state: AnyMutationStateWithoutRun<Args, Result> = initialState();
-  setState(newState: Partial<AnyMutationStateWithoutRun<Result>>) {
+  setState(newState: Partial<AnyMutationStateWithoutRun<Args, Result>>) {
     let updateUi = false;
     const current: Record<string, unknown> = this.state;
     for (const [key, value] of Object.entries(newState)) {
 @@ -167,8 +169,8 @@ class Observer<Args extends unknown[], Result> {
   computeErrorMessage(error: unknown): string | undefined {
     if (!error) return undefined;
     const mutation = this.mutation;
-    if (!mutation || !this.currentArgs) return errMessage(error);
+    if (!mutation || !this.state.args) return errMessage(error);
-    return `Failed to ${mutation.describe(...this.currentArgs)}: ${
+    return `Failed to ${mutation.describe(...this.state.args)}: ${
       errMessage(error)
     }`;
   }
 @@ -176,11 +178,7 @@ class Observer<Args extends unknown[], Result> {
   run(...args: Args) {
     const mutation = this.mutation;
     if (!mutation) return;
-    const argsChanged = this.currentArgs !== args;
+    this.setState({ args });
-    this.currentArgs = args;
-    if (argsChanged && this.watched.has("args")) {
-      this.setRerender(Math.random());
-    }
     const key = mutation.key(args);
     if (key !== this.currentKey) {
       this.currentKey = key;
 @@ -216,6 +214,7 @@ class Observer<Args extends unknown[], Result> {
             isError: hasError,
             isOptimisticData: status === "waiting" || status === "mutating" ||
               status === "refetching",
+            args: hasError || hasResult ? undefined : this.state.args,
           });
         },
       );
 @@ -254,11 +253,7 @@ class Observer<Args extends unknown[], Result> {
     const args = array.slice() as Args;
     const options = args.pop() as RunOptions<Result>;
-    const argsChanged = this.currentArgs !== args;
+    this.setState({ args });
-    this.currentArgs = args;
-    if (argsChanged && this.watched.has("args")) {
-      this.setRerender(Math.random());
-    }
     const key = mutation.key(args);
     // Set up subscription if key changed
 @@ -296,6 +291,7 @@ class Observer<Args extends unknown[], Result> {
             isError: hasError,
             isOptimisticData: status === "waiting" || status === "mutating" ||
               status === "refetching",
+            args: hasError || hasResult ? undefined : this.state.args,
           });
         },
       );
 @@ -320,6 +316,16 @@ class Observer<Args extends unknown[], Result> {
         result: undefined,
       });
     },
+    setError(error: unknown) {
+      self.setState({
+        status: "error",
+        error,
+        errorMessage: errMessage(error),
+        isError: true,
+        isSuccess: false,
+        result: undefined,
+      });
+    },
     get status() {
       self.watched.add("status");
       return self.state.status;
 @@ -358,7 +364,7 @@ class Observer<Args extends unknown[], Result> {
     },
     get args() {
       self.watched.add("args");
-      return self.currentArgs ?? undefined;
+      return self.state.args;
     },
   } as UseMutateResult<Args, Result>))(this);
 }

src/types.ts+2

...	@@ -33,6 +33,8 @@ export interface RunOptions<Result> {	...	@@ -33,6 +33,8 @@ export interface RunOptions<Result> {
33	\| { status: "success"; result: Result }	33	\| { status: "success"; result: Result }
34	\| { status: "error"; error: unknown },	34	\| { status: "error"; error: unknown },
35	) => void;	35	) => void;
		36	/** Called when optimistic state is being restored/rolled back */
		37	onRestore?: () => void;
36	}	38	}
37		39
38	export interface MutationEvent<Result> {	40	export interface MutationEvent<Result> {

author	clover caruso <git@paperclover.net> 2026-01-29 17:22:15-08:00
committer	clover caruso <git@paperclover.net> 2026-01-29 22:05:19-08:00
log	f33b11a3c0e6206b106b2cc655632d14ea2650c6
tree	1c69bc4109f33148dccab6459071a6804dd8a134
parent	d5c11b294ef2257d31778609caddf9980e24e6f6
signature	Commit is signed but in an unrecognized format.

chore: some more stuff