@go-avro/avro-js 0.0.2-beta.57 → 0.0.2-beta.58

package/dist/client/QueryClient.d.ts

@@ -1,6 +1,6 @@
 import { InfiniteData, UseInfiniteQueryResult, useMutation, UseQueryResult } from '@tanstack/react-query';
 import { AuthManager } from '../auth/AuthManager';
-import { _Event, ApiInfo, Bill, Company, Job, LineItem, Route, ServiceMonth, Session, User } from '../types/api';
+import { _Event, ApiInfo, Bill, Company, Job, LineItem, Route, ServiceMonth, Session, User, UserCompanyAssociation } from '../types/api';
 import { Tokens } from '../types/auth';
 import { CancelToken, RetryStrategy } from '../types/client';
 import { StandardError } from '../types/error';
@@ -147,6 +147,13 @@ declare module '../client/QueryClient' {
             })[];
             action: "billed" | "paid";
         }>>;
+        useUpdateUserCompany(): ReturnType<typeof useMutation<{
+            msg: string;
+        }, StandardError, {
+            company_id: string;
+            user_id: string;
+            data: Partial<UserCompanyAssociation>;
+        }>>;
         useDeleteJob(): ReturnType<typeof useMutation<{
             msg: string;
         }, StandardError, {

package/dist/client/hooks/users.js

@@ -25,6 +25,50 @@ AvroQueryClient.prototype.useCreateSelf = function () {
         },
     });
 };
+AvroQueryClient.prototype.useUpdateUserCompany = function () {
+    const queryClient = useQueryClient();
+    return useMutation({
+        mutationFn: async ({ company_id, user_id, data }) => {
+            return this.put(`/company/${company_id}/user/${user_id}`, JSON.stringify(data), undefined, { "Content-Type": "application/json" });
+        },
+        // Optimistically update the user data and company data in the cache
+        onMutate: async (data) => {
+            await queryClient.cancelQueries({ queryKey: ['user'] });
+            await queryClient.cancelQueries({ queryKey: ['company', data.company_id] });
+            const previousUser = queryClient.getQueryData(['user']);
+            const previousCompany = queryClient.getQueryData(['company', data.company_id]);
+            if (previousUser) {
+                let oldUserCompany = previousUser.companies?.find(c => c.company === data.company_id);
+                if (oldUserCompany) {
+                    oldUserCompany = { ...oldUserCompany, ...data.data };
+                    const newCompanies = previousUser.companies?.map(c => c.company === data.company_id ? oldUserCompany : c) ?? [];
+                    queryClient.setQueryData(['user'], { ...previousUser, companies: newCompanies });
+                }
+            }
+            if (previousCompany) {
+                let oldCompanyUser = previousCompany.users?.find((u) => u.user.id === data.user_id);
+                if (oldCompanyUser) {
+                    oldCompanyUser = { ...oldCompanyUser, ...data.data };
+                    const newUsers = previousCompany.users?.map((u) => u.user.id === data.user_id ? oldCompanyUser : u) ?? [];
+                    queryClient.setQueryData(['company', data.company_id], { ...previousCompany, users: newUsers });
+                }
+            }
+            return { previousUser, previousCompany };
+        },
+        onError: (err, _, context) => {
+            if (context?.previousUser) {
+                queryClient.setQueryData(['user'], context.previousUser);
+            }
+            if (context?.previousCompany) {
+                queryClient.setQueryData(['company'], context.previousCompany);
+            }
+        },
+        onSettled: (data, error, variables) => {
+            queryClient.invalidateQueries({ queryKey: ['user'] });
+            queryClient.invalidateQueries({ queryKey: ['company', variables.company_id] });
+        },
+    });
+};
 AvroQueryClient.prototype.useDeleteSelf = function () {
     const queryClient = useQueryClient();
     return useMutation({

package/dist/types/api.d.ts

@@ -257,6 +257,12 @@ export interface Group {
     time_created: number;
     time_updated: number;
 }
+export declare const NotificationLevel: Readonly<{
+    IN_APP: 0;
+    EMAIL: 1;
+    SMS: 2;
+    PUSH: 3;
+}>;
 export interface UserCompanyAssociation {
     id: string;
     user: User;
@@ -267,7 +273,7 @@ export interface UserCompanyAssociation {
     time_updated: number | null;
     notification_setting: number[];
     share_email_company_wide: boolean;
-    notifications: number[];
+    notifications: Notification[];
     groups: string[];
 }
 export interface Email {

package/dist/types/api.js

@@ -1 +1,6 @@
-export {};
+export const NotificationLevel = Object.freeze({
+    IN_APP: 0,
+    EMAIL: 1,
+    SMS: 2,
+    PUSH: 3
+});

package/package.json

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