supermemory/apps/web/hooks/use-memories-usage.ts

import {
	DEFAULT_SUBSCRIPTION_STATUS,
	fetchMemoriesFeature,
	fetchSubscriptionStatus,
	isAllowedFrom,
} from "@lib/queries"
import type { useCustomer } from "autumn-js/react"

export function useMemoriesUsage(autumn: ReturnType<typeof useCustomer>) {
	const {
		data: status = DEFAULT_SUBSCRIPTION_STATUS,
		isLoading: isCheckingStatus,
	} = fetchSubscriptionStatus(autumn, !autumn.isLoading)

	const hasProProduct = isAllowedFrom(status, "api_pro")

	const { data: memoriesCheck, isLoading: isLoadingMemories } =
		fetchMemoriesFeature(autumn, !isCheckingStatus && !autumn.isLoading)

	const memoriesUsed = memoriesCheck?.usage ?? 0
	const memoriesLimit = memoriesCheck?.included_usage ?? 0

	const isLoading = autumn.isLoading || isCheckingStatus || isLoadingMemories

	const usagePercent =
		memoriesLimit <= 0
			? 0
			: Math.min(Math.max((memoriesUsed / memoriesLimit) * 100, 0), 100)

	return {
		memoriesUsed,
		memoriesLimit,
		hasProProduct,
		isLoading,
		usagePercent,
	}
}