@go-avro/avro-js 0.0.2-beta.1 → 0.0.2-beta.100

package/dist/client/hooks/jobs.js

@@ -0,0 +1,178 @@
+import { useMutation, useQuery, useQueryClient, useInfiniteQuery } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetJobs = function (body, total = 0, onProgress) {
+    const queryClient = useQueryClient();
+    const result = useInfiniteQuery({
+        queryKey: ['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.last_event = (job.tasks || []).reduce((latest, task) => {
+                    return task.last_event && (!latest || task.last_event.time_started > (latest.time_started ?? 0)) ? task.last_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_created ?? 0)) ? 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 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 = useQueryClient();
+    return useMutation({
+        mutationFn: ({ companyId, jobData }) => {
+            return this.post(`/company/${companyId}/job`, JSON.stringify(jobData), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ companyId, 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] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateJob = function () {
+    const queryClient = useQueryClient();
+    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 = useQueryClient();
+    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, useQueryClient } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useGetMessages = function (chatId, body) {
+    const queryClient = useQueryClient();
+    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,92 @@
+import { useQueryClient, useInfiniteQuery, useMutation } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetMonths = function (body) {
+    const queryClient = useQueryClient();
+    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 = useQueryClient();
+    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'] });
+        },
+    });
+};

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

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

package/dist/client/hooks/plans.js

@@ -0,0 +1,9 @@
+import { useQuery, useQueryClient } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useGetPlans = function (code) {
+    const queryClient = useQueryClient();
+    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('/'),
+    });
+};

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

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

package/dist/client/hooks/routes.js

@@ -0,0 +1,168 @@
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.useGetRoutes = function (body, total, onProgress) {
+    return useQuery({
+        queryKey: ['routes', this.companyId, body.amt ?? 50, body.query ?? ""],
+        queryFn: async () => {
+            if (typeof total !== "number") {
+                return this.fetchRoutes({ ...body, offset: 0 });
+            }
+            onProgress?.(0);
+            const pageCount = body.amt ? Math.ceil(total / body.amt) : 0;
+            let completed = 0;
+            const promises = Array.from({ length: pageCount }, (_, i) => this.fetchRoutes({
+                ...body,
+                offset: i * (body.amt ?? 0),
+            }));
+            const trackedPromises = promises.map((promise) => promise.then((result) => {
+                completed++;
+                const fraction = completed / pageCount;
+                onProgress?.(fraction);
+                return result;
+            }));
+            const pages = await Promise.all(trackedPromises);
+            const routes = pages.flat();
+            return routes;
+        },
+    });
+};
+AvroQueryClient.prototype.useGetRoute = function (routeId) {
+    const queryClient = useQueryClient();
+    return useQuery({
+        queryKey: ['route', routeId],
+        queryFn: () => this.get(`/route/${routeId}`),
+        enabled: Boolean(routeId) && routeId.length > 0,
+    });
+};
+AvroQueryClient.prototype.useCreateRoute = function () {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: async ({ companyId, routeData }) => {
+            return this.post(`/company/${companyId}/route`, JSON.stringify(routeData), undefined, { "Content-Type": "application/json" });
+        },
+        onMutate: async ({ companyId, routeData }) => {
+            await queryClient.cancelQueries({ queryKey: ['routes'] });
+            const previousRoutes = queryClient.getQueryData(['routes']);
+            queryClient.setQueriesData({ queryKey: ['routes'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    return {
+                        ...oldData,
+                        pages: [
+                            [{ ...routeData, id: 'temp-id' }],
+                            ...oldData.pages,
+                        ],
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return [{ ...routeData, id: 'temp-id' }, ...oldData];
+                }
+                return oldData;
+            });
+            return { previousRoutes };
+        },
+        onError: (_err, _variables, context) => {
+            if (context?.previousRoutes) {
+                queryClient.setQueryData(['routes'], context.previousRoutes);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { companyId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['routes'] });
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+            queryClient.invalidateQueries({ queryKey: ['company', companyId] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateRoute = function () {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: async ({ routeId, updates, }) => {
+            return this.put(`/route/${routeId}`, JSON.stringify(updates), undefined, { "Content-Type": "application/json" });
+        },
+        onMutate: async ({ routeId, updates }) => {
+            await queryClient.cancelQueries({ queryKey: ['route', routeId] });
+            await queryClient.cancelQueries({ queryKey: ['routes'] });
+            const previousRoute = queryClient.getQueryData(['route', routeId]);
+            const previousRoutes = queryClient.getQueryData(['routes']);
+            queryClient.setQueryData(['route', routeId], (oldData) => oldData ? { ...oldData, ...updates } : undefined);
+            queryClient.setQueriesData({ queryKey: ['routes'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    return {
+                        ...oldData,
+                        pages: oldData.pages.map((page) => page.map((route) => route.id === routeId ? { ...route, ...updates } : route)),
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((route) => route.id === routeId ? { ...route, ...updates } : route);
+                }
+                return oldData;
+            });
+            return { previousRoute, previousRoutes };
+        },
+        onError: (_err, variables, context) => {
+            const { routeId } = variables;
+            if (context?.previousRoute) {
+                queryClient.setQueryData(['route', routeId], context.previousRoute);
+            }
+            if (context?.previousRoutes) {
+                queryClient.setQueryData(['routes'], context.previousRoutes);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { routeId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['routes'] });
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['route', routeId] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useDeleteRoute = function () {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: async ({ routeId, }) => {
+            return this.delete(`/route/${routeId}`);
+        },
+        onMutate: async ({ routeId }) => {
+            await queryClient.cancelQueries({ queryKey: ['routes'] });
+            await queryClient.cancelQueries({ queryKey: ['route', routeId] });
+            const previousRoutes = queryClient.getQueryData(['routes']);
+            const previousRoute = queryClient.getQueryData(['route', routeId]);
+            queryClient.setQueryData(['route', routeId], undefined);
+            queryClient.setQueriesData({ queryKey: ['routes'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.filter((route) => route.id !== routeId));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.filter((route) => route.id !== routeId);
+                }
+                return oldData;
+            });
+            return { previousRoutes, previousRoute };
+        },
+        onError: (_err, variables, context) => {
+            const { routeId } = variables;
+            if (context?.previousRoutes) {
+                queryClient.setQueryData(['routes'], context.previousRoutes);
+            }
+            if (context?.previousRoute) {
+                queryClient.setQueryData(['route', routeId], context.previousRoute);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { routeId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['routes'] });
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['route', routeId] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+        },
+    });
+};

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

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