@go-avro/avro-js 0.0.2-beta.114 → 0.0.2-beta.116

package/dist/client/QueryClient.d.ts

@@ -94,6 +94,7 @@ declare module '../client/QueryClient' {
             periods: number[][];
         }): UseQueryResult<EventInsightData[], StandardError>;
         useGetCompany(companyId: string): UseQueryResult<Company, StandardError>;
+        useGetCurrentCompany(): UseQueryResult<Company, StandardError>;
         useGetJob(jobId: string): UseQueryResult<Job, StandardError>;
         useGetEvent(eventId: string): UseQueryResult<_Event, StandardError>;
         useGetUser(userId: string): UseQueryResult<User, StandardError>;
@@ -312,7 +313,7 @@ export declare class AvroQueryClient {
     }>>;
     setTokens(tokens: Tokens): Promise<void>;
     setCache(data: Partial<CacheData>): Promise<void>;
-    getCache(key: keyof CacheData | undefined): Promise<CacheData | string | null>;
+    getCache(key?: keyof CacheData | undefined): Promise<CacheData | string | null>;
     setCompanyId(companyId: string): Promise<void[]>;
     clearCache(): Promise<void>;
     isAuthenticated(): boolean;

package/dist/client/QueryClient.js

@@ -145,6 +145,7 @@ export class AvroQueryClient {
                     }
                     throw new StandardError(401, 'Invalid Google login response');
                 }
+                this._isAuthenticated = true;
                 this.socket.auth = { token: resp.access_token };
                 if (!this.socket.connected) {
                     this.socket.connect();
@@ -195,10 +196,11 @@ export class AvroQueryClient {
             },
             onSettled: () => {
                 this._isAuthenticated = false;
+                this.clearCache();
                 queryClient.invalidateQueries();
             },
             onError: (err) => {
-                this.config.authManager.clearCache();
+                this.clearCache();
                 console.error('Logout failed:', err);
                 throw new StandardError(500, 'Logout failed');
             }
@@ -220,7 +222,7 @@ export class AvroQueryClient {
         })
             .catch(err => {
             console.error('Failed to fetch jobs:', err);
-            throw new StandardError(500, 'Failed to fetch jobs');
+            throw new StandardError(500, `Failed to fetch jobs: ${err.message ?? err}`);
         });
     }
     fetchChats(body = {}, cancelToken, headers = {}) {

package/dist/client/hooks/companies.js

@@ -14,6 +14,24 @@ AvroQueryClient.prototype.useGetCompany = function (companyId) {
         enabled: Boolean(companyId),
     });
 };
+AvroQueryClient.prototype.useGetCurrentCompany = function () {
+    return useQuery({
+        queryKey: ['company', 'current'],
+        queryFn: async () => {
+            let companyId = this.companyId;
+            if (!companyId) {
+                const companyList = await this.get(`/company/list`);
+                if (companyList.length > 0) {
+                    companyId = companyList[0].id;
+                }
+                else {
+                    throw new Error("No company ID set and no companies available");
+                }
+            }
+            return this.get(`/company/${companyId}`);
+        },
+    });
+};
 AvroQueryClient.prototype.useCreateCompany = function () {
     const queryClient = useQueryClient();
     return useMutation({

package/package.json

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