Something went wrong. Try again.
leaflet fork for copy edit / bug fixes forked from leaflet.pub/leaflet
Something went wrong. Try again.
2.2 kB · 68 lines
TSX
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869"use client";import { createContext, useContext } from "react";import useSWR from "swr";import { callRPC } from "app/api/rpc/client";import type { StandardSitePublicationData } from "app/api/rpc/[command]/get_standard_site_publications";
type Resolver = (data: StandardSitePublicationData | null) => void;let pendingURIs = new Set<string>();let pendingResolvers = new Map<string, Resolver[]>();let flushScheduled = false;
function scheduleFlush() { if (flushScheduled) return; flushScheduled = true; queueMicrotask(async () => { const uris = Array.from(pendingURIs); pendingURIs = new Set(); const resolvers = pendingResolvers; pendingResolvers = new Map(); flushScheduled = false; if (uris.length === 0) return; try { const res = await callRPC("get_standard_site_publications", { uris }); const byUri = new Map( (res?.result.publications || []).map((p) => [p.uri, p] as const), ); for (const uri of uris) { const pub = byUri.get(uri) ?? null; for (const r of resolvers.get(uri) || []) r(pub); } } catch { for (const uri of uris) { for (const r of resolvers.get(uri) || []) r(null); } } });}
function batchFetchStandardSitePublication( uri: string,): Promise<StandardSitePublicationData | null> { return new Promise((resolve) => { pendingURIs.add(uri); const existing = pendingResolvers.get(uri) || []; existing.push(resolve); pendingResolvers.set(uri, existing); scheduleFlush(); });}
const StandardSitePublicationDataContext = createContext<Record< string, StandardSitePublicationData> | null>(null);
export function useStandardSitePublication(uri: string | null | undefined) { const ctx = useContext(StandardSitePublicationDataContext); const fromContext = uri ? ctx?.[uri] : undefined; const swr = useSWR<StandardSitePublicationData | null>( !uri || fromContext ? null : `standard-site-publication:${uri}`, () => batchFetchStandardSitePublication(uri!), { revalidateOnFocus: false, revalidateIfStale: false }, ); return { data: fromContext ?? swr.data, isLoading: !fromContext && swr.isLoading, };}