@go-avro/avro-js 0.0.2-beta.72 → 0.0.2-beta.73

package/dist/client/QueryClient.d.ts

@@ -1,7 +1,7 @@
 import { Socket } from 'socket.io-client';
 import { InfiniteData, UseInfiniteQueryResult, useMutation, UseQueryResult } from '@tanstack/react-query';
 import { AuthManager } from '../auth/AuthManager';
-import { _Event, ApiInfo, Avro, Bill, Break, Chat, Company, Job, LineItem, LoginResponse, Message, Plan, Route, ServiceMonth, Session, User, UserCompanyAssociation } from '../types/api';
+import { _Event, ApiInfo, Avro, Bill, Break, Chat, Company, Job, LineItem, LoginResponse, Message, Plan, Route, ServiceMonth, Session, Team, User, UserCompanyAssociation } from '../types/api';
 import { Tokens } from '../types/auth';
 import { CancelToken, RetryStrategy } from '../types/client';
 import { StandardError } from '../types/error';
@@ -172,6 +172,12 @@ declare module '../client/QueryClient' {
             })[];
             action: "billed" | "paid";
         }>>;
+        useUpdateTeam(): ReturnType<typeof useMutation<{
+            msg: string;
+        }, StandardError, {
+            teamId: string;
+            teamData: Partial<Team>;
+        }>>;
         useUpdateMonths(): ReturnType<typeof useMutation<void, StandardError, {
             companyId: string;
             months: (ServiceMonth & {

package/dist/client/hooks/teams.js

@@ -21,12 +21,58 @@ AvroQueryClient.prototype.useGetTeams = function (companyGuid, body, total, onPr
                 return result;
             }));
             const pages = await Promise.all(trackedPromises);
-            const routes = pages.flat();
-            return routes;
+            const teams = pages.flat();
+            return teams;
         },
         enabled: Boolean(companyGuid) && companyGuid.length > 0,
     });
 };
+AvroQueryClient.prototype.useUpdateTeam = function () {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: async ({ teamId, teamData }) => {
+            return this.put(`/team/${teamId}`, JSON.stringify(teamData), undefined, { "Content-Type": "application/json" });
+        },
+        onMutate: async ({ teamId, teamData }) => {
+            await queryClient.cancelQueries({ queryKey: ['teams'] });
+            await queryClient.cancelQueries({ queryKey: ['team', teamId] });
+            const previousTeams = queryClient.getQueryData(['teams']);
+            const previousTeam = queryClient.getQueryData(['team', teamId]);
+            queryClient.setQueryData(['team', teamId], (oldData) => {
+                if (!oldData)
+                    return oldData;
+                return { ...oldData, ...teamData };
+            });
+            queryClient.setQueriesData({ queryKey: ['teams'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.map((team) => team.id === teamId ? { ...team, ...teamData } : team));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((team) => team.id === teamId ? { ...team, ...teamData } : team);
+                }
+                return oldData;
+            });
+            return { previousTeams, previousTeam };
+        },
+        onError: (_err, variables, context) => {
+            const { teamId } = variables;
+            if (context?.previousTeams) {
+                queryClient.setQueryData(['teams'], context.previousTeams);
+            }
+            if (context?.previousTeam) {
+                queryClient.setQueryData(['team', teamId], context.previousTeam);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { teamId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['teams'] });
+            queryClient.invalidateQueries({ queryKey: ['team', teamId] });
+        },
+    });
+};
 AvroQueryClient.prototype.useDeleteTeam = function () {
     const queryClient = useQueryClient();
     return useMutation({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@go-avro/avro-js",
-  "version": "0.0.2-beta.72",
+  "version": "0.0.2-beta.73",
   "description": "JS client for Avro backend integration.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",