@go-avro/avro-js 0.0.2-beta.6 → 0.0.2-beta.61

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,141 @@
1
+ import { useQueryClient, useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
2
+ import { AvroQueryClient } from '../../client/QueryClient';
3
+ AvroQueryClient.prototype.useGetBills = function (companyGuid, body) {
4
+ const queryClient = useQueryClient();
5
+ const result = useInfiniteQuery({
6
+ queryKey: [
7
+ 'bills',
8
+ companyGuid,
9
+ body.query ?? "",
10
+ body.known_ids ?? [],
11
+ body.unknown_ids ?? [],
12
+ body.paid ?? false,
13
+ ],
14
+ initialPageParam: 0,
15
+ getNextPageParam: (lastPage, allPages) => {
16
+ if (lastPage.length < (body.amt ?? 50))
17
+ return undefined;
18
+ return allPages.flat().length; // next offset
19
+ },
20
+ queryFn: ({ pageParam = 0 }) => this.fetchBills(companyGuid, { ...body, offset: pageParam }),
21
+ });
22
+ if (result.data) {
23
+ result.data.pages.forEach((data_page) => {
24
+ data_page.forEach((bill) => {
25
+ queryClient.setQueryData(['bill', bill.id], bill);
26
+ });
27
+ });
28
+ }
29
+ return result;
30
+ };
31
+ AvroQueryClient.prototype.useGetBill = function (billId) {
32
+ return useQuery({
33
+ queryKey: ['bill', billId],
34
+ queryFn: () => this.get(`/bill/${billId}`),
35
+ enabled: Boolean(billId),
36
+ });
37
+ };
38
+ AvroQueryClient.prototype.useCreateBill = function () {
39
+ const queryClient = useQueryClient();
40
+ return useMutation({
41
+ mutationFn: async ({ companyId, data, }) => {
42
+ const body = {
43
+ events: data.line_items.filter(item => item.line_item_type === 'EVENT').map(item => item.id),
44
+ months: data.line_items.filter(item => item.line_item_type === 'SERVICE_MONTH').map(item => item.id),
45
+ line_items: data.line_items.filter(item => item.line_item_type === 'CUSTOM'),
46
+ manual_emails: data.custom_emails,
47
+ users: data.users,
48
+ due_date: data.due_date,
49
+ };
50
+ return this.post(`/company/${companyId}/bill`, JSON.stringify(body), undefined, {
51
+ 'Content-Type': 'application/json',
52
+ });
53
+ },
54
+ onMutate: async ({ companyId }) => {
55
+ await queryClient.cancelQueries({ queryKey: ['bills', companyId] });
56
+ const previousBills = queryClient.getQueryData(['bills', companyId]);
57
+ // TODO: Create a fake bill object for optimistic update and update events and months accordingly
58
+ return { previousBills };
59
+ },
60
+ onError: (_err, _variables, context) => {
61
+ if (context?.previousBills) {
62
+ queryClient.setQueryData(['bills'], context.previousBills);
63
+ }
64
+ },
65
+ onSettled: (_data, _error, variables) => {
66
+ queryClient.invalidateQueries({ queryKey: ['bills', variables.companyId] });
67
+ queryClient.invalidateQueries({ queryKey: ['events', variables.companyId] });
68
+ queryClient.invalidateQueries({ queryKey: ['months', variables.companyId] });
69
+ },
70
+ });
71
+ };
72
+ AvroQueryClient.prototype.useUpdateBill = function () {
73
+ const queryClient = useQueryClient();
74
+ return useMutation({
75
+ mutationFn: ({ billId, updates }) => {
76
+ return this.put(`/bill/${billId}`, JSON.stringify(updates), undefined, {
77
+ "Content-Type": "application/json",
78
+ });
79
+ },
80
+ onMutate: async ({ billId, updates }) => {
81
+ await queryClient.cancelQueries({ queryKey: ['bills'] });
82
+ await queryClient.cancelQueries({ queryKey: ['bill', billId] });
83
+ const previousBills = queryClient.getQueryData(['bills']);
84
+ const previousBill = queryClient.getQueryData(['bill', billId]);
85
+ queryClient.setQueryData(['bill', billId], (oldData) => oldData ? { ...oldData, ...updates } : undefined);
86
+ queryClient.setQueriesData({ queryKey: ['bills'] }, (oldData) => {
87
+ if (!oldData)
88
+ return oldData;
89
+ if (oldData.pages) {
90
+ return {
91
+ ...oldData,
92
+ pages: oldData.pages.map((page) => page.map((bill) => bill.id === billId ? { ...bill, ...updates } : bill)),
93
+ };
94
+ }
95
+ if (Array.isArray(oldData)) {
96
+ return oldData.map((bill) => bill.id === billId ? { ...bill, ...updates } : bill);
97
+ }
98
+ return oldData;
99
+ });
100
+ return { previousBills, previousBill };
101
+ },
102
+ onError: (err, variables, context) => {
103
+ const { billId } = variables;
104
+ if (context?.previousBills) {
105
+ queryClient.setQueryData(['bills'], context.previousBills);
106
+ }
107
+ if (context?.previousBill) {
108
+ queryClient.setQueryData(['bill', billId], context.previousBill);
109
+ }
110
+ },
111
+ onSettled: (data, error, variables) => {
112
+ const { billId } = variables;
113
+ queryClient.invalidateQueries({ queryKey: ['bills'] });
114
+ queryClient.invalidateQueries({ queryKey: ['bill', billId] });
115
+ },
116
+ });
117
+ };
118
+ AvroQueryClient.prototype.useDeleteBill = function () {
119
+ const queryClient = useQueryClient();
120
+ return useMutation({
121
+ mutationFn: async ({ billId }) => {
122
+ return this.delete(`/bill/${billId}`);
123
+ },
124
+ onMutate: async ({ billId }) => {
125
+ await queryClient.cancelQueries({ queryKey: ['bills'] });
126
+ const previousBills = queryClient.getQueryData(['bills']);
127
+ // TODO: Create a fake bill object for optimistic update and update events and months accordingly
128
+ return { previousBills };
129
+ },
130
+ onError: (_err, _variables, context) => {
131
+ if (context?.previousBills) {
132
+ queryClient.setQueryData(['bills'], context.previousBills);
133
+ }
134
+ },
135
+ onSettled: (_data, _error) => {
136
+ queryClient.invalidateQueries({ queryKey: ['bills'] });
137
+ queryClient.invalidateQueries({ queryKey: ['events'] });
138
+ queryClient.invalidateQueries({ queryKey: ['months'] });
139
+ },
140
+ });
141
+ };
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,51 @@
1
+ import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
2
+ import { AvroQueryClient } from '../../client/QueryClient';
3
+ AvroQueryClient.prototype.useGetCompanies = function (options = {}) {
4
+ return useQuery({
5
+ queryKey: ['/company/list'],
6
+ queryFn: () => this.get('/company/list'),
7
+ ...options,
8
+ });
9
+ };
10
+ AvroQueryClient.prototype.useGetCompany = function (companyId) {
11
+ return useQuery({
12
+ queryKey: ['company', companyId],
13
+ queryFn: () => this.get(`/company/${companyId}`),
14
+ enabled: Boolean(companyId),
15
+ });
16
+ };
17
+ AvroQueryClient.prototype.useUpdateCompany = function () {
18
+ const queryClient = useQueryClient();
19
+ return useMutation({
20
+ mutationFn: async ({ companyId, companyData, }) => {
21
+ return this.put(`/company/${companyId}`, JSON.stringify(companyData), undefined, { "Content-Type": "application/json" });
22
+ },
23
+ onMutate: async ({ companyId, companyData }) => {
24
+ await queryClient.cancelQueries({ queryKey: ['company', companyId] });
25
+ await queryClient.cancelQueries({ queryKey: ['/company/list'] });
26
+ const previousCompany = queryClient.getQueryData(['company', companyId]);
27
+ const previousCompanyList = queryClient.getQueryData(['/company/list']);
28
+ queryClient.setQueryData(['company', companyId], (oldData) => oldData ? { ...oldData, ...companyData } : undefined);
29
+ queryClient.setQueryData(['/company/list'], (oldList) => {
30
+ if (!oldList)
31
+ return oldList;
32
+ return oldList.map((company) => company.id === companyId ? { ...company, ...companyData } : company);
33
+ });
34
+ return { previousCompany, previousCompanyList };
35
+ },
36
+ onError: (_err, variables, context) => {
37
+ const { companyId } = variables;
38
+ if (context?.previousCompany) {
39
+ queryClient.setQueryData(['company', companyId], context.previousCompany);
40
+ }
41
+ if (context?.previousCompanyList) {
42
+ queryClient.setQueryData(['/company/list'], context.previousCompanyList);
43
+ }
44
+ },
45
+ onSettled: (_data, _error, variables) => {
46
+ const { companyId } = variables;
47
+ queryClient.invalidateQueries({ queryKey: ['company', companyId] });
48
+ queryClient.invalidateQueries({ queryKey: ['/company/list'] });
49
+ },
50
+ });
51
+ };
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,307 @@
1
+ import { useQueryClient, useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
2
+ import { AvroQueryClient } from '../../client/QueryClient';
3
+ AvroQueryClient.prototype.useGetEvents = function (companyGuid, body) {
4
+ const queryClient = useQueryClient();
5
+ const result = useInfiniteQuery({
6
+ queryKey: [
7
+ 'events',
8
+ companyGuid,
9
+ body.amt ?? 50,
10
+ body.known_ids ?? [],
11
+ body.unknown_ids ?? [],
12
+ body.query ?? '',
13
+ body.unbilled ?? true,
14
+ body.billed ?? true,
15
+ body.paid ?? true,
16
+ body.jobId ?? '',
17
+ ],
18
+ initialPageParam: 0,
19
+ getNextPageParam: (lastPage, allPages) => {
20
+ if (lastPage.length < (body.amt ?? 50))
21
+ return undefined;
22
+ return allPages.flat().length; // next offset
23
+ },
24
+ queryFn: ({ pageParam = 0 }) => this.fetchEvents(companyGuid, { ...body, offset: pageParam }),
25
+ });
26
+ if (result.data) {
27
+ result.data.pages.forEach((data_page) => {
28
+ data_page.forEach((event) => {
29
+ queryClient.setQueryData(['event', event.id], event);
30
+ });
31
+ });
32
+ }
33
+ return result;
34
+ };
35
+ AvroQueryClient.prototype.useGetEvent = function (eventId) {
36
+ return useQuery({
37
+ queryKey: ['event', eventId],
38
+ queryFn: () => this.get(`/event/${eventId}`),
39
+ enabled: Boolean(eventId),
40
+ });
41
+ };
42
+ AvroQueryClient.prototype.useCreateEvent = function () {
43
+ const queryClient = useQueryClient();
44
+ return useMutation({
45
+ mutationFn: ({ companyId, eventData }) => {
46
+ return this.post(`/company/${companyId}/event`, JSON.stringify(eventData), undefined, {
47
+ "Content-Type": "application/json",
48
+ });
49
+ },
50
+ onMutate: async ({ companyId, eventData }) => {
51
+ await queryClient.cancelQueries({ queryKey: ['events'] });
52
+ const previousEvents = queryClient.getQueryData(['events']);
53
+ const previousJob = queryClient.getQueryData(['job', eventData.job_id]);
54
+ const previousJobs = queryClient.getQueryData(['jobs']);
55
+ if (previousJob) {
56
+ const updatedJob = {
57
+ ...previousJob,
58
+ last_event: eventData,
59
+ last_completed_event: (eventData.time_ended ?? -1) > -1 ? eventData : previousJob.last_completed_event,
60
+ };
61
+ updatedJob.tasks = previousJob.tasks.map((task) => {
62
+ if (eventData.tasks?.includes(task.id)) {
63
+ return {
64
+ ...task,
65
+ last_event: eventData,
66
+ last_completed_event: (eventData.time_ended ?? -1) > -1 ? eventData : task.last_completed_event,
67
+ };
68
+ }
69
+ return task;
70
+ });
71
+ const updatedJobs = previousJobs?.map((job) => job.id === updatedJob.id ? updatedJob : job);
72
+ queryClient.setQueryData(['job', previousJob.id], updatedJob);
73
+ queryClient.setQueryData(['jobs'], updatedJobs);
74
+ }
75
+ queryClient.setQueryData(['events'], (oldData) => {
76
+ if (!oldData)
77
+ return [];
78
+ if (oldData.pages) {
79
+ const firstPage = oldData.pages[0] || [];
80
+ return {
81
+ ...oldData,
82
+ pages: [
83
+ [
84
+ {
85
+ ...eventData,
86
+ id: Math.random().toString(36).substring(2, 11),
87
+ company_id: companyId,
88
+ },
89
+ ...firstPage,
90
+ ],
91
+ ...oldData.pages.slice(1),
92
+ ],
93
+ };
94
+ }
95
+ if (Array.isArray(oldData)) {
96
+ return [
97
+ {
98
+ ...eventData,
99
+ id: Math.random().toString(36).substring(2, 11),
100
+ company_id: companyId,
101
+ },
102
+ ...oldData,
103
+ ];
104
+ }
105
+ return oldData;
106
+ });
107
+ return { previousEvents, previousJob, previousJobs };
108
+ },
109
+ onError: (err, variables, context) => {
110
+ if (context?.previousEvents) {
111
+ queryClient.setQueryData(['events'], context.previousEvents);
112
+ }
113
+ if (context?.previousJob) {
114
+ queryClient.setQueryData(['job', context.previousJob.id], context.previousJob);
115
+ }
116
+ if (context?.previousJobs) {
117
+ queryClient.setQueryData(['jobs'], context.previousJobs);
118
+ }
119
+ },
120
+ onSettled: () => {
121
+ queryClient.invalidateQueries({ queryKey: ['events'] });
122
+ queryClient.invalidateQueries({ queryKey: ['job'] });
123
+ queryClient.invalidateQueries({ queryKey: ['jobs'] });
124
+ },
125
+ });
126
+ };
127
+ AvroQueryClient.prototype.useUpdateEvent = function () {
128
+ const queryClient = useQueryClient();
129
+ return useMutation({
130
+ mutationFn: ({ eventId, updates }) => {
131
+ return this.put(`/event/${eventId}`, JSON.stringify(updates), undefined, {
132
+ "Content-Type": "application/json",
133
+ });
134
+ },
135
+ onMutate: async ({ eventId, updates }) => {
136
+ await queryClient.cancelQueries({ queryKey: ['event', eventId] });
137
+ await queryClient.cancelQueries({ queryKey: ['events'] });
138
+ const previousEvent = queryClient.getQueryData(['event', eventId]);
139
+ const previousEvents = queryClient.getQueryData(['events']);
140
+ const previousJob = queryClient.getQueryData(['job', previousEvent?.job_id]);
141
+ const previousJobs = queryClient.getQueryData(['jobs']);
142
+ if (previousJob) {
143
+ const updatedJob = {
144
+ ...previousJob,
145
+ last_event: previousEvent ? { ...previousEvent, ...updates } : previousJob.last_event,
146
+ last_completed_event: (updates.time_ended ?? -1) > -1 && previousEvent ? { ...previousEvent, ...updates } : previousJob.last_completed_event,
147
+ };
148
+ updatedJob.tasks = previousJob.tasks.map((task) => {
149
+ if (updates.tasks?.includes(task.id)) {
150
+ return {
151
+ ...task,
152
+ last_event: previousEvent ? { ...previousEvent, ...updates } : task.last_event,
153
+ last_completed_event: (updates.time_ended ?? -1) > -1 ? (previousEvent ? { ...previousEvent, ...updates } : task.last_completed_event) : task.last_completed_event,
154
+ overdue_time: -task.frequency - task.delay,
155
+ };
156
+ }
157
+ return task;
158
+ });
159
+ const updatedJobs = previousJobs?.map((job) => job.id === updatedJob.id ? updatedJob : job);
160
+ queryClient.setQueryData(['job', previousJob.id], updatedJob);
161
+ queryClient.setQueryData(['jobs'], updatedJobs);
162
+ }
163
+ queryClient.setQueryData(['event', eventId], (oldData) => oldData ? { ...oldData, ...updates } : undefined);
164
+ queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
165
+ if (!oldData)
166
+ return oldData;
167
+ if (oldData.pages) {
168
+ const updatedPages = oldData.pages.map((page) => page.map((event) => event.id === eventId ? { ...event, ...updates } : event));
169
+ return { ...oldData, pages: updatedPages };
170
+ }
171
+ if (Array.isArray(oldData)) {
172
+ return oldData.map((event) => event.id === eventId ? { ...event, ...updates } : event);
173
+ }
174
+ return oldData;
175
+ });
176
+ return { previousEvent, previousEvents, previousJob, previousJobs };
177
+ },
178
+ onError: (_err, variables, context) => {
179
+ const { eventId } = variables;
180
+ if (context?.previousEvent) {
181
+ queryClient.setQueryData(['event', eventId], context.previousEvent);
182
+ }
183
+ if (context?.previousEvents) {
184
+ queryClient.setQueryData(['events'], context.previousEvents);
185
+ }
186
+ if (context?.previousJob) {
187
+ queryClient.setQueryData(['job', context.previousJob.id], context.previousJob);
188
+ }
189
+ if (context?.previousJobs) {
190
+ queryClient.setQueryData(['jobs'], context.previousJobs);
191
+ }
192
+ },
193
+ onSettled: (_data, _error, variables) => {
194
+ const { eventId } = variables;
195
+ queryClient.invalidateQueries({ queryKey: ['event', eventId] });
196
+ queryClient.invalidateQueries({ queryKey: ['events'] });
197
+ queryClient.invalidateQueries({ queryKey: ['job'] });
198
+ queryClient.invalidateQueries({ queryKey: ['jobs'] });
199
+ },
200
+ });
201
+ };
202
+ AvroQueryClient.prototype.useUpdateEvents = function () {
203
+ const queryClient = useQueryClient();
204
+ return useMutation({
205
+ mutationFn: async ({ companyId, events, action, }) => {
206
+ const eventIds = events.map(event => event.id);
207
+ return this.put(`/company/${companyId}/events`, JSON.stringify({
208
+ events: eventIds,
209
+ billed: true,
210
+ paid: action === "paid",
211
+ }), undefined, {
212
+ "Content-Type": "application/json",
213
+ });
214
+ },
215
+ onMutate: async ({ events, action }) => {
216
+ await queryClient.cancelQueries({ queryKey: ['events'] });
217
+ await queryClient.cancelQueries({ queryKey: ['event'] });
218
+ const previousEvents = queryClient.getQueryData(['events']);
219
+ const previousEventObjs = events.map(event => queryClient.getQueryData(['event', event.id]));
220
+ const eventIds = events.map(event => event.id);
221
+ // Optimistically update individual event cache
222
+ eventIds.forEach((eventId, idx) => {
223
+ queryClient.setQueryData(['event', eventId], (oldData) => {
224
+ return oldData
225
+ ? { ...oldData, billed: true, paid: action === "paid" }
226
+ : oldData;
227
+ });
228
+ });
229
+ // Optimistically update events list cache
230
+ queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
231
+ if (!oldData)
232
+ return oldData;
233
+ if (oldData.pages) {
234
+ const updatedPages = oldData.pages.map((page) => page.map((event) => eventIds.includes(event.id)
235
+ ? { ...event, billed: true, paid: action === "paid" }
236
+ : event));
237
+ return { ...oldData, pages: updatedPages };
238
+ }
239
+ if (Array.isArray(oldData)) {
240
+ return oldData.map((event) => eventIds.includes(event.id)
241
+ ? { ...event, billed: true, paid: action === "paid" }
242
+ : event);
243
+ }
244
+ return oldData;
245
+ });
246
+ return { previousEvents, previousEventObjs };
247
+ },
248
+ onError: (err, variables, context) => {
249
+ if (context?.previousEvents) {
250
+ queryClient.setQueryData(['events'], context.previousEvents);
251
+ }
252
+ if (context?.previousEventObjs) {
253
+ context.previousEventObjs.forEach((eventObj) => {
254
+ queryClient.setQueryData(['event', eventObj.id], eventObj);
255
+ });
256
+ }
257
+ },
258
+ onSettled: () => {
259
+ queryClient.invalidateQueries({ queryKey: ['events'] });
260
+ queryClient.invalidateQueries({ queryKey: ['event'] });
261
+ },
262
+ });
263
+ };
264
+ AvroQueryClient.prototype.useDeleteEvent = function () {
265
+ const queryClient = useQueryClient();
266
+ return useMutation({
267
+ mutationFn: async ({ eventId, }) => {
268
+ return this.delete(`/event/${eventId}`, undefined, {
269
+ "Content-Type": "application/json",
270
+ });
271
+ },
272
+ onMutate: async ({ eventId }) => {
273
+ await queryClient.cancelQueries({ queryKey: ['events'] });
274
+ await queryClient.cancelQueries({ queryKey: ['event', eventId] });
275
+ const previousEvents = queryClient.getQueryData(['events']);
276
+ const previousEvent = queryClient.getQueryData(['event', eventId]);
277
+ queryClient.setQueryData(['event', eventId], undefined);
278
+ queryClient.setQueriesData({ queryKey: ['events'] }, (oldData) => {
279
+ if (!oldData)
280
+ return oldData;
281
+ if (oldData.pages) {
282
+ const updatedPages = oldData.pages.map((page) => page.filter((event) => event.id !== eventId));
283
+ return { ...oldData, pages: updatedPages };
284
+ }
285
+ if (Array.isArray(oldData)) {
286
+ return oldData.filter((event) => event.id !== eventId);
287
+ }
288
+ return oldData;
289
+ });
290
+ return { previousEvents, previousEvent };
291
+ },
292
+ onError: (_err, variables, context) => {
293
+ const { eventId } = variables;
294
+ if (context?.previousEvents) {
295
+ queryClient.setQueryData(['events'], context.previousEvents);
296
+ }
297
+ if (context?.previousEvent) {
298
+ queryClient.setQueryData(['event', eventId], context.previousEvent);
299
+ }
300
+ },
301
+ onSettled: (_data, _error, variables) => {
302
+ const { eventId } = variables;
303
+ queryClient.invalidateQueries({ queryKey: ['events'] });
304
+ queryClient.invalidateQueries({ queryKey: ['event', eventId] });
305
+ },
306
+ });
307
+ };
@@ -0,0 +1 @@
1
+ export {};