@go-avro/avro-js 0.0.39 → 0.0.43

package/dist/client/AvroQueryClientProvider.js

@@ -17,7 +17,10 @@ export const AvroQueryClientProvider = ({ baseUrl, authManager, queryClient, con
         return () => {
             try {
                 client.teardownSocketInvalidation();
-                client.socket?.disconnect();
+                // Socket is NOT disconnected here. Its lifecycle is managed by the
+                // AvroQueryClient constructor and token refresh callbacks.
+                // Disconnecting here breaks React StrictMode — the constructor's
+                // async connect flow has already completed, so the socket stays dead.
             }
             catch (e) {
                 // ignore

package/dist/client/QueryClient.js

@@ -1,7 +1,7 @@
 import io from "socket.io-client";
 import { useMutation, useQueryClient, } from "@tanstack/react-query";
 import { v4 as uuidv4 } from "uuid";
-import { Job, LoginResponse, } from "../types/api";
+import { _Event, Job, LoginResponse, Route, } from "../types/api";
 import { AuthState } from "../types/auth";
 import { StandardError } from "../types/error";
 function isBulkEvent(c) {
@@ -53,11 +53,13 @@ const SOCKET_EVENT_CONFIG = {
         entityKey: "routes",
         action: "create",
         fetchPath: (id) => `/route/${id}`,
+        construct: (d) => new Route(d),
     },
     update_route: {
         entityKey: "routes",
         action: "update",
         fetchPath: (id) => `/route/${id}`,
+        construct: (d) => new Route(d),
     },
     delete_route: { entityKey: "routes", action: "delete", fetchPath: null },
     // ── Events (also refetch parent job — overdueness, last_event, etc.) ──
@@ -65,6 +67,7 @@ const SOCKET_EVENT_CONFIG = {
         entityKey: "events",
         action: "create",
         fetchPath: (id) => `/event/${id}`,
+        construct: (d) => new _Event(d),
         relatedRefetch: [
             {
                 entityKey: "jobs",
@@ -78,6 +81,7 @@ const SOCKET_EVENT_CONFIG = {
         entityKey: "events",
         action: "update",
         fetchPath: (id) => `/event/${id}`,
+        construct: (d) => new _Event(d),
         relatedRefetch: [
             {
                 entityKey: "jobs",
@@ -239,7 +243,6 @@ export class AvroQueryClient {
                 ])
                     .then(([id, token]) => {
                     this.companyId = id;
-                    console.log("Initializing socket connection with token:", token);
                     this.socket.auth = { token: token };
                     this.socket.connect();
                 })
@@ -267,7 +270,6 @@ export class AvroQueryClient {
         this.config.authManager.onTokenRefreshed((newAccessToken) => {
             if (this.socket && newAccessToken) {
                 this.setAuthState(AuthState.AUTHENTICATED);
-                console.log("Access token refreshed, updating socket auth...");
                 this.socket.auth = { token: newAccessToken };
                 this.socket.disconnect().connect();
             }
@@ -336,7 +338,6 @@ export class AvroQueryClient {
                 const { entityKey, action, fetchPath, idField, alsoInvalidate, relatedRefetch, } = config;
                 const handler = async (data) => {
                     const id = data?.[idField ?? "id"];
-                    // No id → old backend or malformed payload → full invalidation
                     if (!id || typeof id !== "string") {
                         invalidateEntity(entityKey);
                         alsoInvalidate?.forEach((k) => queryClient.invalidateQueries({ queryKey: k }));
@@ -378,6 +379,9 @@ export class AvroQueryClient {
             if (this.companyId) {
                 this.socket.emit("join_company", { company_id: this.companyId });
             }
+            else {
+                console.warn(`[socket-debug] joinCompanyRoom skipped — companyId is undefined`);
+            }
         };
         this.socket.on("connect", joinCompanyRoom);
         handlers.push({ event: "connect", handler: joinCompanyRoom });
@@ -632,7 +636,7 @@ export class AvroQueryClient {
         // ─── DELETE ─────────────────────────────────────────
         if (action === "delete") {
             queryClient.removeQueries({ queryKey: [entityKey, id], exact: true });
-            queryClient.setQueriesData({ predicate, type: "active" }, (old) => {
+            queryClient.setQueriesData({ predicate }, (old) => {
                 if (!old)
                     return old;
                 if (old.pages && Array.isArray(old.pages)) {
@@ -662,7 +666,7 @@ export class AvroQueryClient {
             const item = construct ? construct(raw) : raw;
             queryClient.setQueryData([entityKey, id], item);
             if (action === "create") {
-                queryClient.setQueriesData({ predicate, type: "active" }, (old) => {
+                queryClient.setQueriesData({ predicate }, (old) => {
                     if (!old)
                         return old;
                     if (old.pages && Array.isArray(old.pages)) {
@@ -683,7 +687,7 @@ export class AvroQueryClient {
             }
             else {
                 // UPDATE — replace in every active list / infinite-query cache
-                queryClient.setQueriesData({ predicate, type: "active" }, (old) => {
+                queryClient.setQueriesData({ predicate }, (old) => {
                     if (!old)
                         return old;
                     if (old.pages && Array.isArray(old.pages)) {
@@ -700,7 +704,8 @@ export class AvroQueryClient {
             }
             return item;
         }
-        catch {
+        catch (err) {
+            console.error(`[socket-debug] _syncEntity fetch FAILED for ${entityKey}/${id}:`, err);
             invalidate();
             return undefined;
         }

package/dist/client/hooks/companies.js

@@ -32,14 +32,17 @@ AvroQueryClient.prototype.useGetCurrentCompany = function () {
         queryKey: ["companies", "current"],
         queryFn: async () => {
             if (!this.companyId) {
-                this.companyId = await this.config.authManager.getCompanyId();
+                const storedId = await this.config.authManager.getCompanyId();
+                if (storedId) {
+                    await this.setCompanyId(storedId);
+                }
             }
             if (!this.companyId) {
                 const companyList = await this.get({
                     path: `/company/list`,
                 });
                 if (companyList.length > 0) {
-                    this.companyId = companyList[0].id;
+                    await this.setCompanyId(companyList[0].id);
                 }
                 else {
                     throw new Error("No company ID set and no companies available");

package/package.json

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