From fefba1acafb4a6e5d3aab22eeb8a03a0df978c50 Mon Sep 17 00:00:00 2001
From: Vitalii Kiiko <mr.psiho@gmail.com>
Date: Tue, 1 Oct 2024 15:05:48 +0200
Subject: [PATCH] - removed unwanted call for companies for admin user;

---
 src/App.js      | 16 +++++++++++-----
 src/tempData.js |  4 ++--
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/App.js b/src/App.js
index 451b994..5844a59 100644
--- a/src/App.js
+++ b/src/App.js
@@ -7,7 +7,7 @@ import './assets/scss/theme.scss';
 import AuthenticationService from './service/authentication-service';
 
 // store
-import { useStore, storeSet } from './store';
+import { useStore, storeSet, storeGet } from './store';
 import CompanyService from './service/company-service';
 
 const i18n = createI18n({}, 'gepafin');
@@ -16,17 +16,15 @@ function App() {
     const role = useStore().main.getRole();
 
     const callback = (data) => {
+        storeSet.main.unsetAsyncRequest();
         if (data.status === 'SUCCESS') {
             storeSet.main.userData(data.data);
-            storeSet.main.setAsyncRequest();
-            CompanyService.getCompanyForUser(data.data.id, companyCallback, errCompanyCallback)
         } else {
             storeSet.main.doLogout();
         }
-        storeSet.main.unsetAsyncRequest();
     }
 
-    const errCallback = (data) => {
+    const errCallback = () => {
         storeSet.main.doLogout();
         storeSet.main.unsetAsyncRequest();
     }
@@ -43,6 +41,14 @@ function App() {
         storeSet.main.unsetAsyncRequest();
     }
 
+    useEffect(() => {
+        if (role === 'ROLE_BENEFICIARY') {
+            storeSet.main.setAsyncRequest();
+            const userData = storeGet.main.userData();
+            CompanyService.getCompanyForUser(userData.id, companyCallback, errCompanyCallback)
+        }
+    }, [role])
+
     useEffect(() => {
         storeSet.main.setAsyncRequest();
         AuthenticationService.me(callback, errCallback);
diff --git a/src/tempData.js b/src/tempData.js
index 04a0648..5df1b1a 100644
--- a/src/tempData.js
+++ b/src/tempData.js
@@ -734,7 +734,7 @@ export const elementItems = [
         ],
         validators: {}
     },
-    /*{
+    {
         id: 20,
         sortOrder: 20,
         name: 'table',
@@ -747,5 +747,5 @@ export const elementItems = [
             }
         ],
         validators: {}
-    }*/
+    }
 ]