@go-avro/avro-js 0.0.2-beta.17 → 0.0.2-beta.170

package/dist/client/hooks/jobs.js

@@ -0,0 +1,296 @@
+import { useMutation, useQuery, useInfiniteQuery } from '@tanstack/react-query';
+import { AvroQueryClient } from '../../client/QueryClient';
+AvroQueryClient.prototype.getJobsFromCache = function (queryClient) {
+    if (!queryClient) {
+        queryClient = this.getQueryClient();
+    }
+    return (queryClient.getQueryData(['jobs', this.companyId, this.company?.num_jobs]) ?? []);
+};
+AvroQueryClient.prototype.useGetJobs = function (params) {
+    const queryClient = this.getQueryClient();
+    const amt = 50;
+    const company_id = params?.companyId ?? this.companyId;
+    const num_jobs = params?.numJobs ?? this.company?.num_jobs ?? 0;
+    return useQuery({
+        queryKey: ['jobs', company_id, num_jobs],
+        queryFn: async () => {
+            params?.onProgress?.(0);
+            const pageCount = amt ? Math.ceil((num_jobs ?? 0) / amt) + 1 : 1;
+            let completed = 0;
+            const promises = Array.from({ length: pageCount }, (_, i) => this.fetchJobs({ offset: i * (amt ?? 1), amt }, company_id));
+            const trackedPromises = promises.map((promise) => promise.then((result) => {
+                completed++;
+                const fraction = completed / pageCount;
+                params?.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;
+                }, job.current_event);
+                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;
+                }, job.last_completed_event);
+                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.useManageJobs = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ jobs }) => {
+            return this.post(`/company/${this.companyId}/jobs/manage`, JSON.stringify({ jobs }), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ jobs }) => {
+            await queryClient.cancelQueries({ queryKey: ['jobs'] });
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            queryClient.setQueryData(['jobs'], (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    return {
+                        ...oldData,
+                        pages: oldData.pages.map((page) => page.map((job) => {
+                            const updatedJob = jobs.find((j) => j.id === job.id);
+                            return updatedJob ? { ...job, ...updatedJob } : job;
+                        })),
+                    };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((job) => {
+                        const updatedJob = jobs.find((j) => j.id === job.id);
+                        return updatedJob ? { ...job, ...updatedJob } : job;
+                    });
+                }
+                return oldData;
+            });
+            return { previousJobs };
+        },
+        onError: (err, variables, context) => {
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+        },
+        onSettled: (_data, _error, _variables) => {
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+        },
+    });
+};
+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.useDeleteJobs = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: ({ ids }) => {
+            return this.post(`/company/${this.companyId}/jobs/delete`, JSON.stringify({ ids }), undefined, {
+                "Content-Type": "application/json",
+            });
+        },
+        onMutate: async ({ ids }) => {
+            await queryClient.cancelQueries({ queryKey: ['jobs'] });
+            const previousJobs = queryClient.getQueryData(['jobs']);
+            queryClient.setQueryData(['jobs'], (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.filter((job) => !ids.includes(job.id)));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.filter((job) => !ids.includes(job.id));
+                }
+                return oldData;
+            });
+            return { previousJobs };
+        },
+        onError: (_err, variables, context) => {
+            if (context?.previousJobs) {
+                queryClient.setQueryData(['jobs'], context.previousJobs);
+            }
+        },
+        onSettled: (_data, _error, _variables) => {
+            queryClient.invalidateQueries({ queryKey: ['jobs'] });
+            queryClient.invalidateQueries({ queryKey: ['routes'] });
+            queryClient.invalidateQueries({ queryKey: ['job'] });
+            queryClient.invalidateQueries({ queryKey: ['route'] });
+        },
+    });
+};
+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/labels.d.ts

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

package/dist/client/hooks/labels.js

@@ -0,0 +1,130 @@
+import { useMutation, useQuery } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useGetLabels = function (body, total, onProgress) {
+    return useQuery({
+        queryKey: ['labels', this.companyId, body.amt ?? 50, body.query ?? "", total ?? "all"],
+        queryFn: async () => {
+            if (typeof total !== "number") {
+                return this.fetchLabels({ ...body, offset: 0 });
+            }
+            onProgress?.(0);
+            const pageCount = body.amt ? Math.ceil(total / body.amt) + 1 : 1;
+            let completed = 0;
+            const promises = Array.from({ length: pageCount }, (_, i) => this.fetchLabels({
+                ...body,
+                offset: i * (body.amt ?? 1),
+            }));
+            const trackedPromises = promises.map((promise) => promise.then((result) => {
+                completed++;
+                const fraction = completed / pageCount;
+                onProgress?.(fraction);
+                return result;
+            }));
+            const pages = await Promise.all(trackedPromises);
+            const labels = pages.flat();
+            return labels;
+        },
+    });
+};
+AvroQueryClient.prototype.useCreateLabel = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ labelData }) => {
+            return this.post(`/company/${this.companyId}/label`, JSON.stringify(labelData), undefined, { "Content-Type": "application/json" });
+        },
+        onSettled: () => {
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+            queryClient.invalidateQueries({ queryKey: ['labels'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useUpdateLabel = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ labelId, labelData }) => {
+            return this.put(`/label/${labelId}`, JSON.stringify(labelData), undefined, { "Content-Type": "application/json" });
+        },
+        onMutate: async ({ labelId, labelData }) => {
+            await queryClient.cancelQueries({ queryKey: ['labels'] });
+            await queryClient.cancelQueries({ queryKey: ['label', labelId] });
+            const previousLabels = queryClient.getQueryData(['labels']);
+            const previousLabel = queryClient.getQueryData(['label', labelId]);
+            queryClient.setQueryData(['label', labelId], (oldData) => {
+                if (!oldData)
+                    return oldData;
+                return { ...oldData, ...labelData };
+            });
+            queryClient.setQueriesData({ queryKey: ['labels'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.map((label) => label.id === labelId ? { ...label, ...labelData } : label));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.map((label) => label.id === labelId ? { ...label, ...labelData } : label);
+                }
+                return oldData;
+            });
+            return { previousLabels, previousLabel };
+        },
+        onError: (_err, variables, context) => {
+            const { labelId } = variables;
+            if (context?.previousLabels) {
+                queryClient.setQueryData(['labels'], context.previousLabels);
+            }
+            if (context?.previousLabel) {
+                queryClient.setQueryData(['label', labelId], context.previousLabel);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { labelId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['labels'] });
+            queryClient.invalidateQueries({ queryKey: ['label', labelId] });
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+        },
+    });
+};
+AvroQueryClient.prototype.useDeleteLabel = function () {
+    const queryClient = this.getQueryClient();
+    return useMutation({
+        mutationFn: async ({ labelId, }) => {
+            return this.delete(`/label/${labelId}`);
+        },
+        onMutate: async ({ labelId }) => {
+            await queryClient.cancelQueries({ queryKey: ['labels'] });
+            await queryClient.cancelQueries({ queryKey: ['label', labelId] });
+            const previousLabels = queryClient.getQueryData(['labels']);
+            const previousLabel = queryClient.getQueryData(['label', labelId]);
+            queryClient.setQueryData(['label', labelId], undefined);
+            queryClient.setQueriesData({ queryKey: ['labels'] }, (oldData) => {
+                if (!oldData)
+                    return oldData;
+                if (oldData.pages) {
+                    const updatedPages = oldData.pages.map((page) => page.filter((label) => label.id !== labelId));
+                    return { ...oldData, pages: updatedPages };
+                }
+                if (Array.isArray(oldData)) {
+                    return oldData.filter((label) => label.id !== labelId);
+                }
+                return oldData;
+            });
+            return { previousLabels, previousLabel };
+        },
+        onError: (_err, variables, context) => {
+            const { labelId } = variables;
+            if (context?.previousLabels) {
+                queryClient.setQueryData(['labels'], context.previousLabels);
+            }
+            if (context?.previousLabel) {
+                queryClient.setQueryData(['label', labelId], context.previousLabel);
+            }
+        },
+        onSettled: (_data, _error, variables) => {
+            const { labelId } = variables;
+            queryClient.invalidateQueries({ queryKey: ['labels'] });
+            queryClient.invalidateQueries({ queryKey: ['label', labelId] });
+            queryClient.invalidateQueries({ queryKey: ['company'] });
+        },
+    });
+};

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/proposal.d.ts

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

package/dist/client/hooks/proposal.js

@@ -0,0 +1,22 @@
+import { useMutation, useQuery } from "@tanstack/react-query";
+import { AvroQueryClient } from "../../client/QueryClient";
+AvroQueryClient.prototype.useCreateProposal = function () {
+    return useMutation({
+        mutationFn: async ({ job_id, data }) => {
+            return this.post(`/job/${job_id}/proposal`, JSON.stringify(data), undefined, { "Content-Type": "application/json" });
+        },
+    });
+};
+AvroQueryClient.prototype.useAcceptProposal = function (proposal_id) {
+    return useMutation({
+        mutationFn: async (data) => {
+            return this.post(`/proposal/${proposal_id}/accept`, JSON.stringify(data), undefined, { "Content-Type": "application/json" });
+        },
+    });
+};
+AvroQueryClient.prototype.useGetProposal = function (proposal_id) {
+    return useQuery({
+        queryKey: ['proposal', proposal_id],
+        queryFn: async () => this.get(`/proposal/${proposal_id}`),
+    });
+};

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 {};