Affiliate Stats and Referral Hooks

Author: youngkidwarrior

packages/app/features/affiliate/utils/useAffiliateReferrals.ts

@@ -0,0 +1,35 @@
+import { useInfiniteQuery } from '@tanstack/react-query'
+import { useSupabase } from 'app/utils/supabase/useSupabase'
+import { throwIf } from 'app/utils/throwIf'
+
+/**
+/**
+ * Infinite query to fetch referrals
+ * @param pageSize - number of items to fetch per page
+ */
+export function useAffiliateReferrals({ pageSize = 10 }: { pageSize?: number } = {}) {
+  const supabase = useSupabase()
+  async function fetchAffiliateReferrals({ pageParam }: { pageParam: number }) {
+    const from = pageParam * pageSize
+    const to = (pageParam + 1) * pageSize - 1
+    const request = supabase.from('affiliate_referrals').select('*').range(from, to)
+    const { data, error } = await request
+    throwIf(error)
+    return data
+  }
+  return useInfiniteQuery({
+    queryKey: ['affiliate_referrals'],
+    initialPageParam: 0,
+    getNextPageParam: (lastPage, _allPages, lastPageParam) => {
+      if (lastPage !== null && lastPage.length < pageSize) return undefined
+      return lastPageParam + 1
+    },
+    getPreviousPageParam: (_firstPage, _allPages, firstPageParam) => {
+      if (firstPageParam <= 1) {
+        return undefined
+      }
+      return firstPageParam - 1
+    },
+    queryFn: fetchAffiliateReferrals,
+  })
+}

packages/app/features/affiliate/utils/useAffiliateStats.ts

@@ -1,22 +1,25 @@
-import type { Tables } from '@my/supabase/database-generated.types'
-import type { PostgrestError } from '@supabase/postgrest-js'
-import { useQuery, type UseQueryResult } from '@tanstack/react-query'
+import { useQuery } from '@tanstack/react-query'
 import { useSupabase } from 'app/utils/supabase/useSupabase'
 
-export const useAffiliateStats = (): UseQueryResult<Tables<'affiliate_stats'>, PostgrestError> => {
+export function useAffiliateStats() {
   const supabase = useSupabase()
-  return useQuery({
-    queryKey: ['affiliate_stats'],
-    queryFn: async () => {
-      const { data, error } = await supabase.from('affiliate_stats').select('*').single()
-      if (error) {
-        // no rows in receipts table
-        if (error.code === 'PGRST116') {
-          return null
-        }
-        throw new Error(error.message)
+  async function fetchAffiliateStats() {
+    const request = supabase.from('affiliate_stats_summary').select('*').single()
+
+    const { data, error } = await request
+    if (error) {
+      // no rows in receipts table
+      if (error.code === 'PGRST116') {
+        return null
       }
-      return data
-    },
+      throw new Error(error.message)
+    }
+
+    return data
+  }
+
+  return useQuery({
+    queryKey: ['affiliate_stats_summary'],
+    queryFn: fetchAffiliateStats,
   })
 }