supermemory/apps/web/hooks/use-org-summaries.ts

53 lines
1.3 KiB
TypeScript

"use client"
import { useQuery } from "@tanstack/react-query"
import { useEffect, useState } from "react"
import { useAuth } from "@lib/auth-context"
import { normalizePlanType, type PlanType } from "@/hooks/use-token-usage"
const API_BASE =
process.env.NEXT_PUBLIC_BACKEND_URL ?? "https://api.supermemory.ai"
export type OrgSummary = {
orgId: string
plan: PlanType
activeConnectors: number
containerTagCount: number
documentCount: number
}
export function useOrgSummaries() {
const { user } = useAuth()
const [enabled, setEnabled] = useState(false)
useEffect(() => {
if (!user?.id) {
setEnabled(false)
return
}
setEnabled(false)
const timeout = window.setTimeout(() => setEnabled(true), 1200)
return () => window.clearTimeout(timeout)
}, [user?.id])
return useQuery({
queryKey: ["account", "org-summaries"],
queryFn: async (): Promise<OrgSummary[]> => {
const res = await fetch(`${API_BASE}/v3/auth/org-summaries`, {
credentials: "include",
headers: { "X-App-Source": "nova" },
})
if (!res.ok) {
throw new Error("Failed to load organization plans")
}
const data = (await res.json()) as { summaries: OrgSummary[] }
return (data.summaries ?? []).map((s) => ({
...s,
plan: normalizePlanType(s.plan),
}))
},
enabled,
staleTime: 60 * 1000,
})
}