Update src/hooks/useProduct.ts

src/hooks/useProduct.ts

@@ -1,45 +1,32 @@
-"use client";
+'use client';
 
-import { useEffect, useState } from "react";
-import { Product, fetchProduct } from "@/lib/api/product";
+import { useState, useEffect } from 'react';
+import { fetchProductById, Product } from '@/lib/api/product';
 
-export function useProduct(productId: string) {
-    const [product, setProduct] = useState<Product | null>(null);
-    const [isLoading, setIsLoading] = useState(true);
-    const [error, setError] = useState<Error | null>(null);
+const useProduct = (productId?: string) => {
+  const [product, setProduct] = useState<Product | null>(null);
+  const [loading, setLoading] = useState(false);
+  const [error, setError] = useState<string | null>(null);
 
-    useEffect(() => {
-        let isMounted = true;
+  useEffect(() => {
+    if (!productId) return;
 
-        async function loadProduct() {
-            if (!productId) {
-                setIsLoading(false);
-                return;
-            }
+    const loadProduct = async () => {
+      setLoading(true);
+      setError(null);
+      const response = await fetchProductById(productId);
+      if (response.success && response.data) {
+        setProduct(response.data);
+      } else {
+        setError(response.message || 'Failed to fetch product');
+      }
+      setLoading(false);
+    };
 
-            try {
-                setIsLoading(true);
-                const data = await fetchProduct(productId);
-                if (isMounted) {
-                    setProduct(data);
-                }
-            } catch (err) {
-                if (isMounted) {
-                    setError(err instanceof Error ? err : new Error("Failed to fetch product"));
-                }
-            } finally {
-                if (isMounted) {
-                    setIsLoading(false);
-                }
-            }
-        }
+    loadProduct();
+  }, [productId]);
 
-        loadProduct();
+  return { product, loading, error };
+};
 
-        return () => {
-            isMounted = false;
-        };
-    }, [productId]);
-
-    return { product, isLoading, error };
-}
+export default useProduct;