u

src/backendmonitor/dbHelper.js

@@ -1,5 +1,7 @@
 import env from '../config/env.js';
 import { getProvider } from '../db/store.js';
+import { clearCachedUser } from '../api/middleware/auth.js';
+import { setUserCompany } from '../api/sse.js';
 
 /**
  * Returns dialect ('sqlite' or 'mssql')
@@ -284,6 +286,14 @@ export async function updateRow(tableName, { primaryKey, changes } = {}) {
     const sql = `UPDATE [${tableName}] SET ${setParts.join(', ')} WHERE ${whereParts.join(' AND ')};`;
     const result = await provider.query(sql, params);
 
+    if (tableName.toLowerCase() === 'google_auth' && Object.prototype.hasOwnProperty.call(changes, 'selected_company') && primaryKey.id != null) {
+        const userId = Number(primaryKey.id);
+        const rawCompany = changes.selected_company;
+        const companyId = rawCompany === null || rawCompany === undefined || rawCompany === '' ? null : Number(rawCompany);
+        clearCachedUser(userId);
+        setUserCompany(userId, companyId);
+    }
+
     return {
         success: true,
         rowsAffected: result.rowsAffected,