feat: added loading-boolean to fetch-production hook

Eyevinn · Apr 23, 2024 · 5a678cf · 5a678cf
1 parent ca0f0b3
commit 5a678cf
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/components/landing-page/use-fetch-production.ts b/src/components/landing-page/use-fetch-production.ts
@@ -5,31 +5,35 @@ import { TProduction } from "../production-line/types";
 type TUseFetchProduction = (id: number | null) => {
   production: TProduction | null;
   error: Error | null;
+  loading: boolean;
 };
 
 export const useFetchProduction: TUseFetchProduction = (id) => {
   const [production, setProduction] = useState<TProduction | null>(null);
+  const [loading, setLoading] = useState<boolean>(false);
   const [error, setError] = useState<Error | null>(null);
 
   useEffect(() => {
     let aborted = false;
+    setLoading(true);
 
     if (id) {
       API.fetchProduction(id)
         .then((p) => {
           if (aborted) return;
 
           setError(null);
-
+          setLoading(false);
           setProduction(p);
         })
         .catch((e) => {
           setProduction(null);
-
+          setLoading(false);
           setError(e);
         });
     } else {
       setProduction(null);
+      setLoading(false);
     }
 
     return () => {
@@ -40,5 +44,6 @@ export const useFetchProduction: TUseFetchProduction = (id) => {
   return {
     error,
     production,
+    loading,
   };
 };