@go-avro/avro-js 0.0.2-beta.117 → 0.0.2-beta.119

package/dist/client/QueryClient.d.ts

@@ -1,5 +1,5 @@
 import { Socket } from 'socket.io-client';
-import { InfiniteData, UseInfiniteQueryResult, useMutation, UseQueryResult } from '@tanstack/react-query';
+import { InfiniteData, QueryClient, UseInfiniteQueryResult, useMutation, UseQueryResult } from '@tanstack/react-query';
 import { AuthManager } from '../auth/AuthManager';
 import { _Event, ApiInfo, Avro, Bill, Break, Chat, Company, FinancialInsightData, Job, EventInsightData, LoginResponse, Message, Plan, Route, ServiceMonth, Session, Team, User, UserCompanyAssociation } from '../types/api';
 import { Tokens } from '../types/auth';
@@ -9,6 +9,7 @@ import { CacheData } from '../types/cache';
 export interface AvroQueryClientConfig {
     baseUrl: string;
     authManager: AuthManager;
+    queryClient: QueryClient;
     maxRetries?: number;
     retryStrategy?: RetryStrategy;
     timeout?: number;
@@ -20,7 +21,9 @@ declare module '../client/QueryClient' {
         getDelay(strategy: RetryStrategy, attempt: number): number;
         sleep(ms: number): Promise<void>;
         useGetRoot(): UseQueryResult<ApiInfo, StandardError>;
-        useGetJobs(body: {
+        useGetJobs(total: number, onProgress?: (fraction: number) => void): UseQueryResult<Job[], StandardError>;
+        getJobsFromCache(): Job[];
+        useGetInfiniteJobs(body: {
             amt?: number;
             query?: string;
             routeId?: string;
@@ -318,6 +321,7 @@ export declare class AvroQueryClient {
     clearCache(): Promise<void>;
     isAuthenticated(): boolean;
     isAuthenticatedAsync(): Promise<boolean>;
+    getQueryClient(): QueryClient;
     useLogout(): ReturnType<typeof useMutation<void, StandardError, CancelToken | undefined>>;
     fetchJobs(body?: {
         amt?: number;

package/dist/client/QueryClient.js

@@ -1,5 +1,5 @@
 import io from 'socket.io-client';
-import { useMutation, useQueryClient } from '@tanstack/react-query';
+import { useMutation } from '@tanstack/react-query';
 import { LoginResponse } from '../types/api';
 import { StandardError } from '../types/error';
 export class AvroQueryClient {
@@ -9,6 +9,7 @@ export class AvroQueryClient {
         this.config = {
             baseUrl: config.baseUrl,
             authManager: config.authManager,
+            queryClient: config.queryClient,
             maxRetries: config.maxRetries ?? 3,
             retryStrategy: config.retryStrategy ?? 'fixed',
             timeout: config.timeout ?? 0,
@@ -78,7 +79,7 @@ export class AvroQueryClient {
         return this._xhr('DELETE', path, null, cancelToken, headers, false, this.config.maxRetries, progressUpdateCallback);
     }
     useLogin() {
-        const queryClient = useQueryClient();
+        const queryClient = this.getQueryClient();
         return useMutation({
             mutationFn: async ({ username, password, code, cancelToken }) => {
                 const resp = await this.post('/login', JSON.stringify({ username, password, code }), cancelToken, { 'Content-Type': 'application/json' });
@@ -106,7 +107,7 @@ export class AvroQueryClient {
         });
     }
     useRequestCode() {
-        const queryClient = useQueryClient();
+        const queryClient = this.getQueryClient();
         return useMutation({
             mutationFn: async ({ username, email, cancelToken }) => {
                 const resp = await this.post('/code', JSON.stringify({ username, email }), cancelToken, { 'Content-Type': 'application/json' });
@@ -121,7 +122,7 @@ export class AvroQueryClient {
         });
     }
     useUpdatePassword() {
-        const queryClient = useQueryClient();
+        const queryClient = this.getQueryClient();
         return useMutation({
             mutationFn: async ({ username, email, code, newPassword, cancelToken }) => {
                 await this.post(`/user/${username ?? email}/password`, JSON.stringify({ code, password: newPassword }), cancelToken, { 'Content-Type': 'application/json' });
@@ -135,7 +136,7 @@ export class AvroQueryClient {
         });
     }
     useGoogleLogin() {
-        const queryClient = useQueryClient();
+        const queryClient = this.getQueryClient();
         return useMutation({
             mutationFn: async ({ token, cancelToken }) => {
                 const resp = await this._xhr('POST', `/google/authorize?token=${token}`, {}, cancelToken, { 'Content-Type': 'application/json' });
@@ -184,8 +185,11 @@ export class AvroQueryClient {
     isAuthenticatedAsync() {
         return this.config.authManager.isAuthenticated();
     }
+    getQueryClient() {
+        return this.config.queryClient;
+    }
     useLogout() {
-        const queryClient = useQueryClient();
+        const queryClient = this.getQueryClient();
         return useMutation({
             mutationFn: async (cancelToken) => {
                 await this.post('/logout', null, cancelToken);

package/dist/client/hooks/bills.js

@@ -1,7 +1,7 @@
-import { useQueryClient, useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
+import { useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
 import { AvroQueryClient } from '../../client/QueryClient';
 AvroQueryClient.prototype.useGetBills = function (body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: [
             'bills',
@@ -36,7 +36,7 @@ AvroQueryClient.prototype.useGetBill = function (billId) {
     });
 };
 AvroQueryClient.prototype.useCreateBill = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ data }) => {
             const body = {
@@ -70,7 +70,7 @@ AvroQueryClient.prototype.useCreateBill = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateBill = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ billId, updates }) => {
             return this.put(`/bill/${billId}`, JSON.stringify(updates), undefined, {
@@ -116,7 +116,7 @@ AvroQueryClient.prototype.useUpdateBill = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteBill = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ billId }) => {
             return this.delete(`/bill/${billId}`);
@@ -140,7 +140,7 @@ AvroQueryClient.prototype.useDeleteBill = function () {
     });
 };
 AvroQueryClient.prototype.useSyncBillToIntuit = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ billId }) => {
             return this.post(`/company/${this.companyId}/bill/${billId}/intuit/sync`, null);

package/dist/client/hooks/chats.js

@@ -1,7 +1,7 @@
-import { useInfiniteQuery, useQuery, useQueryClient } from "@tanstack/react-query";
+import { useInfiniteQuery, useQuery } from "@tanstack/react-query";
 import { AvroQueryClient } from "../../client/QueryClient";
 AvroQueryClient.prototype.useGetChats = function (body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: [
             'chats',

package/dist/client/hooks/companies.js

@@ -1,4 +1,4 @@
-import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { useMutation, useQuery } from '@tanstack/react-query';
 import { AvroQueryClient } from '../../client/QueryClient';
 AvroQueryClient.prototype.useGetCompanies = function (options = {}) {
     return useQuery({
@@ -32,7 +32,7 @@ AvroQueryClient.prototype.useGetCurrentCompany = function () {
     });
 };
 AvroQueryClient.prototype.useCreateCompany = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ companyData }) => {
             return this.post(`/company`, JSON.stringify(companyData), undefined, { "Content-Type": "application/json" });
@@ -43,7 +43,7 @@ AvroQueryClient.prototype.useCreateCompany = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateCompany = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ companyId, companyData, }) => {
             return this.put(`/company/${companyId}`, JSON.stringify(companyData), undefined, { "Content-Type": "application/json" });
@@ -78,7 +78,7 @@ AvroQueryClient.prototype.useUpdateCompany = function () {
     });
 };
 AvroQueryClient.prototype.useRemoveUserCompany = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ companyId, userId }) => {
             return this.delete(`/company/${companyId}/user/${userId}`);
@@ -109,7 +109,7 @@ AvroQueryClient.prototype.useRemoveUserCompany = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteCompany = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ companyId }) => {
             return this.delete(`/company/${companyId}`);

package/dist/client/hooks/events.js

@@ -1,7 +1,7 @@
-import { useQueryClient, useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
+import { useInfiniteQuery, useQuery, useMutation } from '@tanstack/react-query';
 import { AvroQueryClient } from '../../client/QueryClient';
 AvroQueryClient.prototype.useGetEvents = function (body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: [
             'events',
@@ -40,7 +40,7 @@ AvroQueryClient.prototype.useGetEvent = function (eventId) {
     });
 };
 AvroQueryClient.prototype.useCreateEvent = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ eventData }) => {
             return this.post(`/company/${this.companyId}/event`, JSON.stringify(eventData), undefined, {
@@ -125,7 +125,7 @@ AvroQueryClient.prototype.useCreateEvent = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateEvent = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ eventId, updates }) => {
             return this.put(`/event/${eventId}`, JSON.stringify(updates), undefined, {
@@ -200,7 +200,7 @@ AvroQueryClient.prototype.useUpdateEvent = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateEvents = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ events, action, }) => {
             const eventIds = events.map(event => event.id);
@@ -262,7 +262,7 @@ AvroQueryClient.prototype.useUpdateEvents = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteEvent = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ eventId, }) => {
             return this.delete(`/event/${eventId}`, undefined, {

package/dist/client/hooks/jobs.js

@@ -1,7 +1,54 @@
-import { useMutation, useQuery, useQueryClient, useInfiniteQuery } from '@tanstack/react-query';
+import { useMutation, useQuery, useInfiniteQuery } from '@tanstack/react-query';
 import { AvroQueryClient } from '../../client/QueryClient';
-AvroQueryClient.prototype.useGetJobs = function (body, total = 0, onProgress) {
-    const queryClient = useQueryClient();
+AvroQueryClient.prototype.getJobsFromCache = function () {
+    const queryClient = this.getQueryClient();
+    const cachedJobsData = queryClient.getQueryData(['jobs', this.companyId]);
+    if (cachedJobsData && cachedJobsData.pages) {
+        return cachedJobsData.pages.flat();
+    }
+    const cachedJobs = queryClient.getQueryData(['jobs', this.companyId]);
+    return cachedJobs || [];
+};
+AvroQueryClient.prototype.useGetJobs = function (total = 0, onProgress) {
+    const queryClient = this.getQueryClient();
+    const amt = 50;
+    return useQuery({
+        queryKey: ['jobs', this.companyId],
+        queryFn: async () => {
+            if (total === 0) {
+                onProgress?.(1);
+                return [];
+            }
+            onProgress?.(0);
+            const pageCount = amt ? Math.ceil(total / amt) : 0;
+            let completed = 0;
+            const promises = Array.from({ length: pageCount }, (_, i) => this.fetchJobs({ offset: i * 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: ['jobs', this.companyId, body.amt ?? 50, body.query ?? "", body.routeId ?? ""],
         initialPageParam: 0,
@@ -39,7 +86,7 @@ AvroQueryClient.prototype.useGetJob = function (jobId) {
     });
 };
 AvroQueryClient.prototype.useCreateJob = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ jobData }) => {
             return this.post(`/company/${this.companyId}/job`, JSON.stringify(jobData), undefined, {
@@ -79,7 +126,7 @@ AvroQueryClient.prototype.useCreateJob = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateJob = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ jobId, updates }) => {
             return this.put(`/job/${jobId}`, JSON.stringify(updates), undefined, {
@@ -125,7 +172,7 @@ AvroQueryClient.prototype.useUpdateJob = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteJob = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ jobId }) => {
             return this.delete(`/job/${jobId}`, undefined, {

package/dist/client/hooks/messages.js

@@ -1,7 +1,7 @@
-import { useInfiniteQuery, useQueryClient } from "@tanstack/react-query";
+import { useInfiniteQuery } from "@tanstack/react-query";
 import { AvroQueryClient } from "../../client/QueryClient";
 AvroQueryClient.prototype.useGetMessages = function (chatId, body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: [
             'messages',

package/dist/client/hooks/months.js

@@ -1,7 +1,7 @@
-import { useQueryClient, useInfiniteQuery, useMutation } from '@tanstack/react-query';
+import { useInfiniteQuery, useMutation } from '@tanstack/react-query';
 import { AvroQueryClient } from '../../client/QueryClient';
 AvroQueryClient.prototype.useGetMonths = function (body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: [
             'months',
@@ -33,7 +33,7 @@ AvroQueryClient.prototype.useGetMonths = function (body) {
     return result;
 };
 AvroQueryClient.prototype.useUpdateMonths = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ months, action, }) => {
             const monthIds = months.map(month => month.id);

package/dist/client/hooks/plans.js

@@ -1,7 +1,6 @@
-import { useQuery, useQueryClient } from "@tanstack/react-query";
+import { useQuery } 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/routes.js

@@ -1,8 +1,8 @@
-import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { useMutation, useQuery } 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 ?? ""],
+        queryKey: ['routes', this.companyId, body.amt ?? 50, body.query ?? "", total ?? "all"],
         queryFn: async () => {
             if (typeof total !== "number") {
                 return this.fetchRoutes({ ...body, offset: 0 });
@@ -27,7 +27,7 @@ AvroQueryClient.prototype.useGetRoutes = function (body, total, onProgress) {
     });
 };
 AvroQueryClient.prototype.useGetRoute = function (routeId) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useQuery({
         queryKey: ['route', routeId],
         queryFn: () => this.get(`/route/${routeId}`),
@@ -35,7 +35,7 @@ AvroQueryClient.prototype.useGetRoute = function (routeId) {
     });
 };
 AvroQueryClient.prototype.useCreateRoute = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ routeData }) => {
             return this.post(`/company/${this.companyId}/route`, JSON.stringify(routeData), undefined, { "Content-Type": "application/json" });
@@ -76,7 +76,7 @@ AvroQueryClient.prototype.useCreateRoute = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateRoute = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ routeId, updates, }) => {
             return this.put(`/route/${routeId}`, JSON.stringify(updates), undefined, { "Content-Type": "application/json" });
@@ -122,7 +122,7 @@ AvroQueryClient.prototype.useUpdateRoute = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteRoute = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ routeId, }) => {
             return this.delete(`/route/${routeId}`);

package/dist/client/hooks/sessions.js

@@ -1,4 +1,4 @@
-import { useMutation, useQuery, useQueryClient, useInfiniteQuery } from "@tanstack/react-query";
+import { useMutation, useQuery, useInfiniteQuery } from "@tanstack/react-query";
 import { AvroQueryClient } from "../../client/QueryClient";
 AvroQueryClient.prototype.useGetUserSessions = function () {
     return useQuery({
@@ -7,7 +7,7 @@ AvroQueryClient.prototype.useGetUserSessions = function () {
     });
 };
 AvroQueryClient.prototype.useCreateUserSession = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ sessionData }) => {
             return this.post(`/company/${this.companyId}/session`, JSON.stringify(sessionData), undefined, {
@@ -45,7 +45,7 @@ AvroQueryClient.prototype.useCreateUserSession = function () {
     });
 };
 AvroQueryClient.prototype.useCreateSessionBreak = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ sessionId, breakData }) => {
             return this.post(`/session/${sessionId}/break`, JSON.stringify(breakData), undefined, {
@@ -88,7 +88,7 @@ AvroQueryClient.prototype.useCreateSessionBreak = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateUserSession = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ sessionId, updates }) => {
             return this.put(`/session/${sessionId}`, JSON.stringify(updates), undefined, {
@@ -119,7 +119,7 @@ AvroQueryClient.prototype.useUpdateUserSession = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateSessionBreak = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: ({ breakId, updates }) => {
             return this.put(`/break/${breakId}`, JSON.stringify(updates), undefined, {
@@ -153,7 +153,7 @@ AvroQueryClient.prototype.useUpdateSessionBreak = function () {
     });
 };
 AvroQueryClient.prototype.useGetSessions = function (body) {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     const result = useInfiniteQuery({
         queryKey: ['sessions', this.companyId, body.amt ?? 50, body.query ?? ""],
         initialPageParam: 0,

package/dist/client/hooks/teams.js

@@ -1,8 +1,8 @@
-import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { useMutation, useQuery } from "@tanstack/react-query";
 import { AvroQueryClient } from "../../client/QueryClient";
 AvroQueryClient.prototype.useGetTeams = function (body, total, onProgress) {
     return useQuery({
-        queryKey: ['teams', this.companyId, body.amt ?? 50, body.query ?? ""],
+        queryKey: ['teams', this.companyId, body.amt ?? 50, body.query ?? "", total ?? "all"],
         queryFn: async () => {
             if (typeof total !== "number") {
                 return this.fetchTeams({ ...body, offset: 0 });
@@ -27,7 +27,7 @@ AvroQueryClient.prototype.useGetTeams = function (body, total, onProgress) {
     });
 };
 AvroQueryClient.prototype.useCreateTeam = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ teamData }) => {
             return this.post(`/company/${this.companyId}/team`, JSON.stringify(teamData), undefined, { "Content-Type": "application/json" });
@@ -38,7 +38,7 @@ AvroQueryClient.prototype.useCreateTeam = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateTeam = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ teamId, teamData }) => {
             return this.put(`/team/${teamId}`, JSON.stringify(teamData), undefined, { "Content-Type": "application/json" });
@@ -84,7 +84,7 @@ AvroQueryClient.prototype.useUpdateTeam = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteTeam = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ teamId, }) => {
             return this.delete(`/team/${teamId}`);

package/dist/client/hooks/users.js

@@ -1,4 +1,4 @@
-import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { useMutation, useQuery } from "@tanstack/react-query";
 import { AvroQueryClient } from "../../client/QueryClient";
 AvroQueryClient.prototype.useGetUser = function (userId) {
     return useQuery({
@@ -26,7 +26,7 @@ AvroQueryClient.prototype.useGetSelf = function () {
     });
 };
 AvroQueryClient.prototype.useCreateSelf = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async (data) => {
             return this.post('/user', JSON.stringify(data), undefined, { "Content-Type": "application/json" });
@@ -37,7 +37,7 @@ AvroQueryClient.prototype.useCreateSelf = function () {
     });
 };
 AvroQueryClient.prototype.useUpdateUserCompany = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async ({ user_id, data }) => {
             return this.put(`/company/${this.companyId}/user/${user_id}`, JSON.stringify(data), undefined, { "Content-Type": "application/json" });
@@ -81,7 +81,7 @@ AvroQueryClient.prototype.useUpdateUserCompany = function () {
     });
 };
 AvroQueryClient.prototype.useDeleteSelf = function () {
-    const queryClient = useQueryClient();
+    const queryClient = this.getQueryClient();
     return useMutation({
         mutationFn: async () => {
             return this.delete(`/user`);

package/package.json

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