Add update functionality to useLoading

packages/frontend/src/hooks/useLoading.ts

@@ -1,4 +1,4 @@
-import { useStore } from "@/hooks/useStore";
+import { mapProp, Updater, useStore } from "@/hooks/useStore";
 import { Treaty } from "@elysiajs/eden";
 import { ELYSIA_FORM_DATA } from "elysia";
 import { useEffect, useState } from "react";
@@ -15,6 +15,7 @@ export type LoadingResult<R extends Record<number, unknown>> =
 		isLoading: false,
 		data: FormDataFix<R[200]>,
 		error: null,
+		setData: Updater<FormDataFix<R[200]>>,
 	} | {
 		isLoading: false,
 		data: null,
@@ -46,7 +47,12 @@ export function useLoading<R extends Record<number, unknown>>(fn: () => Promise<
 				}
 			}
 
-			setResult({ isLoading: false, error, data } as LoadingResult<R>);
+			setResult({
+				isLoading: false,
+				error,
+				data,
+				setData: (action) => (setResult as any)(mapProp("data", action)),
+			} as LoadingResult<R>);
 		});
 
 		return () => { cancelled = true; };