Update local model stores in all mutations (#129)

src-web/hooks/useUpdateEnvironment.ts

@@ -1,10 +1,18 @@
 import { useMutation } from '@tanstack/react-query';
 import type { Environment } from '@yaakapp-internal/models';
+import { useSetAtom } from 'jotai/index';
 import { getEnvironment } from '../lib/store';
 import { invokeCmd } from '../lib/tauri';
+import { environmentsAtom } from './useEnvironments';
+import {updateModelList} from "./useSyncModelStores";
 
 export function useUpdateEnvironment(id: string | null) {
-  return useMutation<void, unknown, Partial<Environment> | ((r: Environment) => Environment)>({
+  const setEnvironments = useSetAtom(environmentsAtom);
+  return useMutation<
+    Environment,
+    unknown,
+    Partial<Environment> | ((r: Environment) => Environment)
+  >({
     mutationKey: ['update_environment', id],
     mutationFn: async (v) => {
       const environment = await getEnvironment(id);
@@ -13,7 +21,10 @@ export function useUpdateEnvironment(id: string | null) {
       }
 
       const newEnvironment = typeof v === 'function' ? v(environment) : { ...environment, ...v };
-      await invokeCmd('cmd_update_environment', { environment: newEnvironment });
+      return invokeCmd<Environment>('cmd_update_environment', { environment: newEnvironment });
+    },
+    onSuccess: async (environment) => {
+      setEnvironments(updateModelList(environment));
     },
   });
 }