@go-avro/avro-js 0.0.2-beta.13 → 0.0.2-beta.131

package/dist/client/hooks/events.js

@@ -0,0 +1,308 @@
+import { useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetEvents = function (body) {
+    const queryClient = this.getQueryClient();
+    const result = useInfiniteQuery({
+        queryKey: [
+            'events',
+            this.companyId,
+            body.amt ?? 50,
+            body.known_ids ?? [],
+            body.unknown_ids ?? [],
+            body.query ?? '',
+            body.unbilled ?? true,
+            body.billed ?? true,
+            body.paid ?? true,
+            body.jobId ?? '',
+        ],
+        initialPageParam: 0,
+        getNextPageParam: (lastPage, allPages) => {
+            if (lastPage.length < (body.amt ?? 50))
+                return undefined;
+            return allPages.flat().length; // next offset
+        },
+        queryFn: ({ pageParam = 0 }) => this.fetchEvents({ ...body, offset: pageParam }),
+    });
+    if (result.data) {
+        result.data.pages.forEach((data_page) => {
+            data_page.forEach((event) => {
+                queryClient.setQueryData(['event', event.id], event);
+            });
+        });
+    }
+    return result;
+};
+AvroQueryClient.prototype.useGetEvent = function (eventId) {
+    return useQuery({
+        queryKey: ['event', eventId],
+        queryFn: () => this.get(`/event/${eventId}`),
+        enabled: Boolean(eventId),
+    });
+};
+AvroQueryClient.prototype.useCreateEvent = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ eventData }) => {
+            return this.post(`/company/${this.companyId}/event`, JSON.stringify(eventData), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ eventData }) => {
+            await queryClient.cancelQueries({ queryKey: ['events'] });
+            const previousEvents = queryClient.getQueryData(['events']);
+            const previousJob = queryClient.getQueryData(['job', eventData.job_id]);
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            if (previousJob) {
+                const updatedJob = {
+                    ...previousJob,
+                    current_event: eventData,
+                    last_completed_event: (eventData.time_ended ?? -1) > -1 ? eventData : previousJob.last_completed_event,
+                };
+                updatedJob.tasks = previousJob.tasks.map((task) => {
+                    if (eventData.tasks?.includes(task.id)) {
+                        return {
+                            ...task,
+                            current_event: eventData,
+                            last_completed_event: (eventData.time_ended ?? -1) > -1 ? eventData : task.last_completed_event,
+                        };
+                    }
+                    return task;
+                });
+                const updatedJobs = previousJobs?.map((job) => job.id === updatedJob.id ? updatedJob : job);
+                queryClient.setQueryData(['job', previousJob.id], updatedJob);
+                queryClient.setQueryData(['jobs'], updatedJobs);
+            }
+            queryClient.setQueryData(['events'], (oldData) => {
+                if (!oldData)
+                    return [];
+                if (oldData.pages) {
+                    const firstPage = oldData.pages[0] || [];
+                    return {
+                        ...oldData,
+                        pages: [
+                            [
+                                {
+                                    ...eventData,
+                                    id: Math.random().toString(36).substring(2, 11),
+                                    company_id: this.companyId,
+                                },
+                                ...firstPage,
+                            ],
+                            ...oldData.pages.slice(1),
+                        ],
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return [
+                        {
+                            ...eventData,
+                            id: Math.random().toString(36).substring(2, 11),
+                            company_id: this.companyId,
+                        },
+                        ...oldData,
+                    ];
+                }
+                return oldData;
+            });
+            return { previousEvents, previousJob, previousJobs };
+        },
+        onError: (err, variables, context) => {
+            if (context?.previousEvents) {
+                queryClient.setQueryData(['events'], context.previousEvents);
+            }
+            if (context?.previousJob) {
+                queryClient.setQueryData(['job', context.previousJob.id], context.previousJob);
+            }
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+        },
+        onSettled: () => {
+            queryClient.invalidateQueries({ queryKey: ['events'] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateEvent = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ eventId, updates }) => {
+            return this.put(`/event/${eventId}`, JSON.stringify(updates), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ eventId, updates }) => {
+            await queryClient.cancelQueries({ queryKey: ['event', eventId] });
+            await queryClient.cancelQueries({ queryKey: ['events'] });
+            const previousEvent = queryClient.getQueryData(['event', eventId]);
+            const previousEvents = queryClient.getQueryData(['events']);
+            const previousJob = queryClient.getQueryData(['job', previousEvent?.job_id]);
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            if (previousJob) {
+                const updatedJob = {
+                    ...previousJob,
+                    current_event: previousEvent ? { ...previousEvent, ...updates } : previousJob.current_event,
+                    last_completed_event: (updates.time_ended ?? -1) > -1 && previousEvent ? { ...previousEvent, ...updates } : previousJob.last_completed_event,
+                };
+                updatedJob.tasks = previousJob.tasks.map((task) => {
+                    if (updates.tasks?.includes(task.id)) {
+                        return {
+                            ...task,
+                            current_event: previousEvent ? { ...previousEvent, ...updates } : task.current_event,
+                            last_completed_event: (updates.time_ended ?? -1) > -1 ? (previousEvent ? { ...previousEvent, ...updates } : task.last_completed_event) : task.last_completed_event,
+                            overdue_time: -task.frequency - task.delay,
+                        };
+                    }
+                    return task;
+                });
+                const updatedJobs = previousJobs?.map((job) => job.id === updatedJob.id ? updatedJob : job);
+                queryClient.setQueryData(['job', previousJob.id], updatedJob);
+                queryClient.setQueryData(['jobs'], updatedJobs);
+            }
+            queryClient.setQueryData(['event', eventId], (oldData) => oldData ? { ...oldData, ...updates } : undefined);
+            queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.map((event) => event.id === eventId ? { ...event, ...updates } : event));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((event) => event.id === eventId ? { ...event, ...updates } : event);
+                }
+                return oldData;
+            });
+            return { previousEvent, previousEvents, previousJob, previousJobs };
+        },
+        onError: (_err, variables, context) => {
+            const { eventId } = variables;
+            if (context?.previousEvent) {
+                queryClient.setQueryData(['event', eventId], context.previousEvent);
+            }
+            if (context?.previousEvents) {
+                queryClient.setQueryData(['events'], context.previousEvents);
+            }
+            if (context?.previousJob) {
+                queryClient.setQueryData(['job', context.previousJob.id], context.previousJob);
+            }
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { eventId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['event', eventId] });
+            queryClient.invalidateQueries({ queryKey: ['events'] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateEvents = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ events, action, }) => {
+            const eventIds = events.map(event => event.id);
+            return this.put(`/company/${this.companyId}/events`, JSON.stringify({
+                events: eventIds,
+                billed: true,
+                paid: action === "paid",
+            }), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ events, action }) => {
+            await queryClient.cancelQueries({ queryKey: ['events'] });
+            await queryClient.cancelQueries({ queryKey: ['event'] });
+            const previousEvents = queryClient.getQueryData(['events']);
+            const previousEventObjs = events.map(event => queryClient.getQueryData(['event', event.id]));
+            const eventIds = events.map(event => event.id);
+            // Optimistically update individual event cache
+            eventIds.forEach((eventId, idx) => {
+                queryClient.setQueryData(['event', eventId], (oldData) => {
+                    return oldData
+                        ? { ...oldData, billed: true, paid: action === "paid" }
+                        : oldData;
+                });
+            });
+            // Optimistically update events list cache
+            queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.map((event) => eventIds.includes(event.id)
+                        ? { ...event, billed: true, paid: action === "paid" }
+                        : event));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((event) => eventIds.includes(event.id)
+                        ? { ...event, billed: true, paid: action === "paid" }
+                        : event);
+                }
+                return oldData;
+            });
+            return { previousEvents, previousEventObjs };
+        },
+        onError: (err, variables, context) => {
+            if (context?.previousEvents) {
+                queryClient.setQueryData(['events'], context.previousEvents);
+            }
+            if (context?.previousEventObjs) {
+                context.previousEventObjs.forEach((eventObj) => {
+                    queryClient.setQueryData(['event', eventObj.id], eventObj);
+                });
+            }
+        },
+        onSettled: () => {
+            queryClient.invalidateQueries({ queryKey: ['events'] });
+            queryClient.invalidateQueries({ queryKey: ['event'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useDeleteEvent = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ eventId, }) => {
+            return this.delete(`/event/${eventId}`, undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ eventId }) => {
+            await queryClient.cancelQueries({ queryKey: ['events'] });
+            await queryClient.cancelQueries({ queryKey: ['event', eventId] });
+            const previousEvents = queryClient.getQueryData(['events']);
+            const previousEvent = queryClient.getQueryData(['event', eventId]);
+            queryClient.setQueryData(['event', eventId], undefined);
+            queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.filter((event) => event.id !== eventId));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.filter((event) => event.id !== eventId);
+                }
+                return oldData;
+            });
+            return { previousEvents, previousEvent };
+        },
+        onError: (_err, variables, context) => {
+            const { eventId } = variables;
+            if (context?.previousEvents) {
+                queryClient.setQueryData(['events'], context.previousEvents);
+            }
+            if (context?.previousEvent) {
+                queryClient.setQueryData(['event', eventId], context.previousEvent);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { eventId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['events'] });
+            queryClient.invalidateQueries({ queryKey: ['event', eventId] });
+        },
+    });
+};

package/dist/client/hooks/jobs.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/client/hooks/jobs.js

@@ -0,0 +1,215 @@
+import { useMutation, useQuery, useInfiniteQuery } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.getJobsFromCache = function () {
+    const queryClient = this.getQueryClient();
+    return (queryClient.getQueryData(['jobs', this.companyId, this.company?.num_jobs]) ?? []);
+};
+AvroQueryClient.prototype.useGetJobs = function (onProgress) {
+    const queryClient = this.getQueryClient();
+    const amt = 50;
+    return useQuery({
+        queryKey: ['jobs', this.companyId, this.company?.num_jobs],
+        queryFn: async () => {
+            if (this.company?.num_jobs === 0) {
+                onProgress?.(1);
+                return [];
+            }
+            onProgress?.(0);
+            const pageCount = amt ? Math.ceil((this.company?.num_jobs ?? 0) / amt) : 0;
+            let completed = 0;
+            const promises = Array.from({ length: pageCount }, (_, i) => this.fetchJobs({ offset: i * amt, amt }));
+            const trackedPromises = promises.map((promise) => promise.then((result) => {
+                completed++;
+                const fraction = completed / pageCount;
+                onProgress?.(fraction);
+                return result;
+            }));
+            const pages = await Promise.all(trackedPromises);
+            const jobs = pages.flat();
+            jobs.forEach((job) => {
+                job.current_event = job.tasks.reduce((latest, task) => {
+                    return task.current_event && (!latest || task.current_event.time_started > latest.time_started) ? task.current_event : latest;
+                }, null);
+                job.last_completed_event = job.tasks.reduce((latest, task) => {
+                    return task.last_completed_event && (!latest || task.last_completed_event.time_started > latest.time_started) ? task.last_completed_event : latest;
+                }, null);
+                job.overdue_time = job.tasks.reduce((maxOverdue, task) => {
+                    return task.overdue_time && task.overdue_time > maxOverdue ? task.overdue_time : maxOverdue;
+                }, 0);
+                queryClient.setQueryData(['job', job.id], job);
+            });
+            return jobs;
+        },
+    });
+};
+AvroQueryClient.prototype.useGetInfiniteJobs = function (body, onProgress) {
+    const queryClient = this.getQueryClient();
+    const result = useInfiniteQuery({
+        queryKey: ['infinite', 'jobs', this.companyId, body.amt ?? 50, body.query ?? "", body.routeId ?? ""],
+        initialPageParam: 0,
+        getNextPageParam: (lastPage, allPages) => {
+            if (lastPage.length < (body.amt ?? 50))
+                return undefined;
+            return allPages.flat().length;
+        },
+        queryFn: ({ pageParam = 0 }) => {
+            onProgress?.(0);
+            return this.fetchJobs({
+                ...body,
+                route_id: body.routeId,
+                offset: pageParam,
+            });
+        },
+    });
+    if (result.data) {
+        result.data.pages.forEach((data_page) => {
+            data_page.forEach((job) => {
+                job.overdue_time = (job.tasks || []).reduce((maxOverdue, task) => {
+                    return (task.overdue_time && task.overdue_time > maxOverdue) ? task.overdue_time : maxOverdue;
+                }, 0);
+                queryClient.setQueryData(['job', job.id], job);
+            });
+        });
+    }
+    return result;
+};
+AvroQueryClient.prototype.useGetJob = function (jobId) {
+    return useQuery({
+        queryKey: ['job', jobId],
+        queryFn: () => this.get(`/job/${jobId}`),
+        enabled: Boolean(jobId),
+    });
+};
+AvroQueryClient.prototype.useCreateJob = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ jobData }) => {
+            return this.post(`/company/${this.companyId}/job`, JSON.stringify(jobData), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ jobData }) => {
+            await queryClient.cancelQueries({ queryKey: ['jobs'] });
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            queryClient.setQueryData(['jobs'], (oldData) => {
+                if (!oldData)
+                    return [jobData];
+                if (oldData.pages) {
+                    const firstPage = oldData.pages[0] || [];
+                    return {
+                        ...oldData,
+                        pages: [[jobData, ...firstPage], ...oldData.pages.slice(1)],
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return [jobData, ...oldData];
+                }
+                return oldData;
+            });
+            return { previousJobs };
+        },
+        onError: (err, variables, context) => {
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+        },
+        onSettled: (data, error, variables) => {
+            const { id: jobId } = data ?? {};
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['job', jobId] });
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateJob = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ jobId, updates }) => {
+            return this.put(`/job/${jobId}`, JSON.stringify(updates), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ jobId, updates }) => {
+            await queryClient.cancelQueries({ queryKey: ['jobs'] });
+            await queryClient.cancelQueries({ queryKey: ['job', jobId] });
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            const previousJob = queryClient.getQueryData(['job', jobId]);
+            queryClient.setQueryData(['job', jobId], (oldData) => oldData ? { ...oldData, ...updates } : undefined);
+            queryClient.setQueriesData({ queryKey: ['jobs'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    return {
+                        ...oldData,
+                        pages: oldData.pages.map((page) => page.map((job) => job.id === jobId ? { ...job, ...updates } : job)),
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((job) => job.id === jobId ? { ...job, ...updates } : job);
+                }
+                return oldData;
+            });
+            return { previousJobs, previousJob };
+        },
+        onError: (err, variables, context) => {
+            const { jobId } = variables;
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+            if (context?.previousJob) {
+                queryClient.setQueryData(['job', jobId], context.previousJob);
+            }
+        },
+        onSettled: (data, error, variables) => {
+            const { jobId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['job', jobId] });
+        },
+    });
+};
+AvroQueryClient.prototype.useDeleteJob = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ jobId }) => {
+            return this.delete(`/job/${jobId}`, undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ jobId }) => {
+            await queryClient.cancelQueries({ queryKey: ['jobs'] });
+            await queryClient.cancelQueries({ queryKey: ['job', jobId] });
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            const previousJob = queryClient.getQueryData(['job', jobId]);
+            queryClient.setQueryData(['job', jobId], undefined);
+            queryClient.setQueriesData({ queryKey: ['jobs'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.filter((job) => job.id !== jobId));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.filter((job) => job.id !== jobId);
+                }
+                return oldData;
+            });
+            return { previousJobs, previousJob };
+        },
+        onError: (_err, variables, context) => {
+            const { jobId } = variables;
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+            if (context?.previousJob) {
+                queryClient.setQueryData(['job', jobId], context.previousJob);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { jobId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['routes'] });
+            queryClient.invalidateQueries({ queryKey: ['job', jobId] });
+            queryClient.invalidateQueries({ queryKey: ['route'] });
+        },
+    });
+};

package/dist/client/hooks/messages.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/client/hooks/messages.js

@@ -0,0 +1,30 @@
+import { useInfiniteQuery } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useGetMessages = function (chatId, body) {
+    const queryClient = this.getQueryClient();
+    const result = useInfiniteQuery({
+        queryKey: [
+            'messages',
+            chatId,
+            body.amt ?? 50,
+            body.known_ids ?? [],
+            body.unknown_ids ?? [],
+            body.query ?? '',
+        ],
+        initialPageParam: 0,
+        getNextPageParam: (lastPage, allPages) => {
+            if (lastPage.length < (body.amt ?? 50))
+                return undefined;
+            return allPages.flat().length; // next offset
+        },
+        queryFn: ({ pageParam = 0 }) => this.fetchMessages(chatId, { ...body, offset: pageParam }),
+    });
+    if (result.data) {
+        result.data.pages.forEach((data_page) => {
+            data_page.forEach((chat) => {
+                queryClient.setQueryData(['chat', chat.id], chat);
+            });
+        });
+    }
+    return result;
+};

package/dist/client/hooks/months.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/client/hooks/months.js

@@ -0,0 +1,93 @@
+import { useInfiniteQuery, useMutation } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetMonths = function (body) {
+    const queryClient = this.getQueryClient();
+    const result = useInfiniteQuery({
+        queryKey: [
+            'months',
+            this.companyId,
+            body.amt ?? 50,
+            body.known_ids ?? [],
+            body.unknown_ids ?? [],
+            body.query ?? '',
+            body.unbilled ?? true,
+            body.billed ?? true,
+            body.paid ?? true,
+            body.jobId ?? '',
+        ],
+        initialPageParam: 0,
+        getNextPageParam: (lastPage, allPages) => {
+            if (lastPage.length < (body.amt ?? 50))
+                return undefined;
+            return allPages.flat().length; // next offset
+        },
+        queryFn: ({ pageParam = 0 }) => this.fetchMonths({ ...body, offset: pageParam }),
+    });
+    if (result.data) {
+        result.data.pages.forEach((data_page) => {
+            data_page.forEach((month) => {
+                queryClient.setQueryData(['month', month.id], month);
+            });
+        });
+    }
+    return result;
+};
+AvroQueryClient.prototype.useUpdateMonths = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ months, action, }) => {
+            const monthIds = months.map(month => month.id);
+            return this.put(`/company/${this.companyId}/months`, JSON.stringify({
+                months: monthIds,
+                billed: true,
+                paid: action === "paid",
+            }), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ months, action }) => {
+            await queryClient.cancelQueries({ queryKey: ['months'] });
+            await queryClient.cancelQueries({ queryKey: ['month'] });
+            const previousMonths = queryClient.getQueryData(['months']);
+            const previousMonthObjs = months.map(month => queryClient.getQueryData(['month', month.id]));
+            const monthIds = months.map(month => month.id);
+            monthIds.forEach((monthId, idx) => {
+                queryClient.setQueryData(['month', monthId], (oldData) => {
+                    return oldData
+                        ? { ...oldData, billed: true, paid: action === "paid" }
+                        : oldData;
+                });
+            });
+            queryClient.setQueriesData({ queryKey: ['months'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.map((month) => monthIds.includes(month.id)
+                        ? { ...month, billed: true, paid: action === "paid" }
+                        : month));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((month) => monthIds.includes(month.id)
+                        ? { ...month, billed: true, paid: action === "paid" }
+                        : month);
+                }
+                return oldData;
+            });
+            return { previousMonths, previousMonthObjs };
+        },
+        onError: (err, variables, context) => {
+            if (context) {
+                queryClient.setQueryData(['months'], context.previousMonths);
+                context.previousMonthObjs.forEach((monthObj) => {
+                    queryClient.setQueryData(['month', monthObj.id], monthObj);
+                });
+            }
+        },
+        onSettled: () => {
+            queryClient.invalidateQueries({ queryKey: ['months'] });
+            queryClient.invalidateQueries({ queryKey: ['month'] });
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+        },
+    });
+};

package/dist/client/hooks/plans.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/client/hooks/plans.js

@@ -0,0 +1,8 @@
+import { useQuery } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useGetPlans = function (code) {
+    return useQuery({
+        queryKey: ['plans', code],
+        queryFn: async () => this.get(`/plans${code ? `?code=${code}` : ''}`),
+    });
+};

package/dist/client/hooks/root.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/client/hooks/root.js

@@ -0,0 +1,8 @@
+import { useQuery } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetRoot = function () {
+    return useQuery({
+        queryKey: ['health'],
+        queryFn: () => this.get('/'),
+    });
+};