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

package/dist/auth/AuthManager.d.ts

@@ -2,6 +2,7 @@ import { TokenStorage, Tokens } from '../types/auth';
 export declare class AuthManager {
     private storages;
     private baseUrl;
+    private tokenRefreshedCallbacks;
     constructor({ baseUrl, storage, }: {
         baseUrl: string;
         storage: TokenStorage | TokenStorage[];
@@ -9,6 +10,7 @@ export declare class AuthManager {
     isAuthenticated(): Promise<boolean>;
     fetchNewTokens(): Promise<Tokens>;
     accessToken(): Promise<string | undefined>;
+    onTokenRefreshed(callback: (accessToken: string) => void): void;
     refreshTokens(): Promise<Tokens | null>;
     setTokens(tokens: Tokens): Promise<void>;
     clearTokens(): Promise<void>;

package/dist/auth/AuthManager.js

@@ -1,5 +1,6 @@
 export class AuthManager {
     constructor({ baseUrl, storage, }) {
+        this.tokenRefreshedCallbacks = [];
         this.storages = Array.isArray(storage) ? storage : [storage];
         if (this.storages.length === 0) {
             throw new Error('At least one token storage must be provided');
@@ -88,10 +89,17 @@ export class AuthManager {
         const newToken = await this.refreshTokens();
         return newToken?.access_token;
     }
+    onTokenRefreshed(callback) {
+        this.tokenRefreshedCallbacks.push(callback);
+    }
     async refreshTokens() {
         try {
             const newToken = await this.fetchNewTokens();
             await Promise.all(this.storages.map(s => s.set(newToken)));
+            this.tokenRefreshedCallbacks.forEach(cb => {
+                if (newToken?.access_token)
+                    cb(newToken.access_token);
+            });
             return newToken;
         }
         catch (error) {

package/dist/client/QueryClient.d.ts

@@ -4,6 +4,7 @@ import { _Event, ApiInfo, Bill, Break, Company, Job, LineItem, Route, ServiceMon
 import { Tokens } from '../types/auth';
 import { CancelToken, RetryStrategy } from '../types/client';
 import { StandardError } from '../types/error';
+import { Socket } from 'socket.io-client';
 export interface AvroQueryClientConfig {
     baseUrl: string;
     authManager: AuthManager;
@@ -107,10 +108,7 @@ declare module '../client/QueryClient' {
             id: string;
         }, StandardError, {
             sessionId: string;
-            breakData: {
-                start_time: string;
-                end_time?: string;
-            };
+            breakData: Partial<Break>;
         }>>;
         useUpdateEvent(): ReturnType<typeof useMutation<{
             msg: string;
@@ -204,7 +202,11 @@ declare module '../client/QueryClient' {
 }
 export declare class AvroQueryClient {
     protected config: Required<AvroQueryClientConfig>;
+    readonly socket: Socket;
     constructor(config: AvroQueryClientConfig);
+    emit(eventName: string, data: unknown): void;
+    on<T>(eventName: string, callback: (data: T) => void): void;
+    off(eventName: string, callback?: Function): void;
     get<T>(path: string, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;
     post<T>(path: string, data: any, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;
     put<T>(path: string, data: any, cancelToken?: CancelToken, headers?: Record<string, string>, progressUpdateCallback?: (loaded: number, total: number) => void): Promise<T>;

package/dist/client/QueryClient.js

@@ -1,5 +1,6 @@
 import { useMutation, useQueryClient } from '@tanstack/react-query';
 import { StandardError } from '../types/error';
+import io from 'socket.io-client';
 export class AvroQueryClient {
     constructor(config) {
         this.config = {
@@ -9,6 +10,43 @@ export class AvroQueryClient {
             retryStrategy: config.retryStrategy ?? 'fixed',
             timeout: config.timeout ?? 0,
         };
+        this.socket = io(config.baseUrl, { autoConnect: false, transports: ["websocket"], });
+        if (!this.socket.connected) {
+            this.config.authManager.accessToken().then(token => {
+                console.log('Initializing socket connection with token...');
+                this.socket.auth = { token: token };
+                this.socket.connect();
+            });
+        }
+        this.socket.on('connect', () => {
+            console.log(`Socket connected with ID: ${this.socket?.id}`);
+        });
+        this.socket.on('disconnect', (reason) => {
+            console.log(`Socket disconnected: ${reason}`);
+        });
+        this.socket.on('connect_error', (err) => {
+            console.error(`Socket connection error: ${err.message}`);
+        });
+        this.config.authManager.onTokenRefreshed((newAccessToken) => {
+            if (this.socket && newAccessToken) {
+                console.log('Access token refreshed, updating socket auth...');
+                this.socket.auth = { token: newAccessToken };
+                this.socket.disconnect().connect();
+            }
+        });
+    }
+    emit(eventName, data) {
+        if (!this.socket?.connected) {
+            console.error('Socket is not connected. Cannot emit event.');
+            return;
+        }
+        this.socket.emit(eventName, data);
+    }
+    on(eventName, callback) {
+        this.socket?.on(eventName, callback);
+    }
+    off(eventName, callback) {
+        this.socket?.off(eventName, callback);
     }
     get(path, cancelToken, headers = {}, progressUpdateCallback) {
         return this._xhr('GET', path, null, cancelToken, headers, true, this.config.maxRetries, progressUpdateCallback);
@@ -28,6 +66,10 @@ export class AvroQueryClient {
             if (!tokens || !tokens.access_token || !tokens.refresh_token) {
                 throw new StandardError(401, 'Invalid login response');
             }
+            this.socket.auth = { token: tokens.access_token };
+            if (!this.socket.connected) {
+                this.socket.connect();
+            }
             return this.config.authManager.setTokens(tokens).then(() => true);
         })
             .catch(err => {
@@ -44,8 +86,13 @@ export class AvroQueryClient {
     useLogout() {
         const queryClient = useQueryClient();
         return useMutation({
-            mutationFn: (cancelToken) => this._fetch('POST', '/logout', null, cancelToken)
-                .then(() => this.config.authManager.clearTokens()),
+            mutationFn: async (cancelToken) => {
+                await this._fetch('POST', '/logout', null, cancelToken);
+                await this.config.authManager.clearTokens();
+                if (this.socket && this.socket.connected) {
+                    this.socket.disconnect();
+                }
+            },
             onSettled: () => {
                 queryClient.invalidateQueries();
             },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@go-avro/avro-js",
-  "version": "0.0.2-beta.60",
+  "version": "0.0.2-beta.61",
   "description": "JS client for Avro backend integration.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -53,6 +53,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@tanstack/react-query": "^5.90.2"
+    "@tanstack/react-query": "^5.90.2",
+    "socket.io-client": "^4.8.1"
   }
 }