refactor: replace tracking setup queries with custom hook (#78)

Co-authored-by: iza <[email protected]>

Author: sbansal1999

apps/dashboard/app/(main)/websites/[id]/funnels/page.tsx

@@ -17,7 +17,7 @@ import {
 	useFunnelAnalyticsByReferrer,
 	useFunnels,
 } from '@/hooks/use-funnels';
-import { trpc } from '@/lib/trpc';
+import { useTrackingSetup } from '@/hooks/use-tracking-setup';
 import {
 	dateRangeAtom,
 	formattedDateRangeAtom,
@@ -106,19 +106,7 @@ export default function FunnelsPage() {
 	const [, setDateRangeAction] = useAtom(setDateRangeAndAdjustGranularityAtom);
 	const [formattedDateRangeState] = useAtom(formattedDateRangeAtom);
 
-	// Check tracking setup
-	const { data: trackingSetupData, isLoading: isTrackingSetupLoading } =
-		trpc.websites.isTrackingSetup.useQuery(
-			{ websiteId },
-			{ enabled: !!websiteId }
-		);
-
-	const isTrackingSetup = useMemo(() => {
-		if (isTrackingSetupLoading) {
-			return null;
-		}
-		return trackingSetupData?.tracking_setup ?? false;
-	}, [isTrackingSetupLoading, trackingSetupData?.tracking_setup]);
+	const { isTrackingSetup } = useTrackingSetup(websiteId);
 
 	// Date picker helpers
 	const dayPickerSelectedRange: DayPickerRange | undefined = useMemo(

apps/dashboard/app/(main)/websites/[id]/goals/page.tsx

@@ -23,8 +23,8 @@ import {
 	useBulkGoalAnalytics,
 	useGoals,
 } from '@/hooks/use-goals';
+import { useTrackingSetup } from '@/hooks/use-tracking-setup';
 import { useWebsite } from '@/hooks/use-websites';
-import { trpc } from '@/lib/trpc';
 import {
 	dateRangeAtom,
 	formattedDateRangeAtom,
@@ -105,18 +105,7 @@ export default function GoalsPage() {
 	const [, setDateRangeAction] = useAtom(setDateRangeAndAdjustGranularityAtom);
 	const [formattedDateRangeState] = useAtom(formattedDateRangeAtom);
 
-	const { data: trackingSetupData, isLoading: isTrackingSetupLoading } =
-		trpc.websites.isTrackingSetup.useQuery(
-			{ websiteId },
-			{ enabled: !!websiteId }
-		);
-
-	const isTrackingSetup = useMemo(() => {
-		if (isTrackingSetupLoading) {
-			return null;
-		}
-		return trackingSetupData?.tracking_setup ?? false;
-	}, [isTrackingSetupLoading, trackingSetupData?.tracking_setup]);
+	const { isTrackingSetup } = useTrackingSetup(websiteId);
 
 	const dayPickerSelectedRange: DayPickerRange | undefined = useMemo(
 		() => ({

apps/dashboard/app/(main)/websites/[id]/page.tsx

@@ -17,8 +17,8 @@ import { Button } from '@/components/ui/button';
 import { Skeleton } from '@/components/ui/skeleton';
 import { Tabs, TabsContent, TabsList, TabsTrigger } from '@/components/ui/tabs';
 import { operatorOptions, useFilters } from '@/hooks/use-filters';
+import { useTrackingSetup } from '@/hooks/use-tracking-setup';
 import { useWebsite } from '@/hooks/use-websites';
-import { trpc } from '@/lib/trpc';
 import {
 	dateRangeAtom,
 	formattedDateRangeAtom,
@@ -165,17 +165,7 @@ function WebsiteDetailsPage() {
 		refetch: refetchWebsiteData,
 	} = useWebsite(id as string);
 
-	const { data: trackingSetupData, isLoading: isTrackingSetupLoading } =
-		trpc.websites.isTrackingSetup.useQuery(
-			{ websiteId: id as string },
-			{ enabled: !!id }
-		);
-	const isTrackingSetup = useMemo(() => {
-		if (!data || isTrackingSetupLoading) {
-			return null;
-		}
-		return trackingSetupData?.tracking_setup ?? false;
-	}, [data, isTrackingSetupLoading, trackingSetupData?.tracking_setup]);
+	const { isTrackingSetup } = useTrackingSetup(id as string);
 
 	useEffect(() => {
 		if (isTrackingSetup === false && activeTab === 'overview') {

apps/dashboard/hooks/use-tracking-setup.ts

@@ -0,0 +1,26 @@
+import { trpc } from '@/lib/trpc';
+
+export function useTrackingSetup(websiteId: string) {
+	const {
+		data: trackingSetupData,
+		isLoading: isTrackingSetupLoading,
+		isError: isTrackingSetupError,
+		error: trackingSetupError,
+		refetch: refetchTrackingSetup,
+	} = trpc.websites.isTrackingSetup.useQuery(
+		{ websiteId },
+		{ enabled: !!websiteId }
+	);
+
+	const isTrackingSetup = isTrackingSetupLoading
+		? null
+		: (trackingSetupData?.tracking_setup ?? false);
+
+	return {
+		isTrackingSetup,
+		isTrackingSetupLoading,
+		isTrackingSetupError,
+		trackingSetupError,
+		refetchTrackingSetup,
+	};
+}