feat: add replace_hero hook to custom section

This commit is contained in:
gosangam
2026-06-21 23:36:25 +05:30
parent 5cd75c27f9
commit f92b41384e
4 changed files with 192 additions and 103 deletions
@@ -14,15 +14,14 @@
* stop when a page is short) — the same idiom as the home feed.
*
* `fetchPage(before, limit)` is supplied by the caller so the same hook serves
* the owner feed (posts.mine) and the public feed (profile.feed). `setFeed` is
* returned so the owner page can optimistically prune on delete/hide.
* the scroll feeds (posts.mine / profile.feed) AND the virtualised profile
* grids — anything that keyset-pages a user's posts. `setFeed` is returned so
* the owner views can optimistically prune on delete/hide.
*/
import { useCallback, useEffect, useRef, useState } from "react";
import { posts as postsApi, type FeedCursor, type Post } from "@/lib/social";
const PAGE = 10;
type FetchPage = (before: FeedCursor | null, limit: number) => Promise<Post[]>;
const cursorOf = (p: Post): FeedCursor => ({
@@ -30,7 +29,7 @@ const cursorOf = (p: Post): FeedCursor => ({
id: p.id,
});
export function usePostScrollFeed(fetchPage: FetchPage) {
export function usePostScrollFeed(fetchPage: FetchPage, pageSize = 10) {
const [feed, setFeed] = useState<Post[]>([]);
const [loading, setLoading] = useState(true);
const [hasMore, setHasMore] = useState(true);
@@ -51,7 +50,7 @@ export function usePostScrollFeed(fetchPage: FetchPage) {
if (fetchingRef.current || !hasMore) return;
fetchingRef.current = true;
try {
const rows = (await fetchPage(cursorRef.current, PAGE)) ?? [];
const rows = (await fetchPage(cursorRef.current, pageSize)) ?? [];
const fresh = rows.filter((p) => !idsRef.current.has(p.id));
if (fresh.length > 0) {
fresh.forEach((p) => idsRef.current.add(p.id));
@@ -60,7 +59,7 @@ export function usePostScrollFeed(fetchPage: FetchPage) {
setFeed((prev) => [...prev, ...fresh]);
}
// Stop on a short page OR a page that produced nothing new (stalled).
if (rows.length < PAGE || fresh.length === 0) setHasMore(false);
if (rows.length < pageSize || fresh.length === 0) setHasMore(false);
} catch {
setHasMore(false);
setError((e) => e ?? "Could not load posts.");
@@ -68,7 +67,7 @@ export function usePostScrollFeed(fetchPage: FetchPage) {
fetchingRef.current = false;
setLoading(false);
}
}, [fetchPage, hasMore]);
}, [fetchPage, hasMore, pageSize]);
// First load: anchor-first when arriving with #post-<id>, else newest page.
useEffect(() => {
@@ -0,0 +1,133 @@
"use client";
/**
* VirtualPostGrid — window-virtualised, infinite-scroll 2-up thumbnail grid for
* the profile Posts tabs (/u/me and /u/[handle]).
*
* Mirrors HomeFeed's window virtualization, but the unit of virtualization is a
* GRID ROW (two square cards) rather than a single card. Data (keyset paging +
* the bottom sentinel) comes from usePostScrollFeed; this component only owns
* rendering + windowing, so a profile with hundreds of posts stays cheap.
*/
import { useEffect, useMemo, useRef, useState } from "react";
import { useWindowVirtualizer } from "@tanstack/react-virtual";
import { Loader2 } from "lucide-react";
import type { Post } from "@/lib/social";
import { PostGridCard } from "./post-grid-card";
const COLS = 2;
export function VirtualPostGrid({
posts,
loading,
hasMore,
sentinelRef,
emptyMessage,
hrefFor,
onDelete,
onUpdated,
}: {
posts: Post[];
loading: boolean;
hasMore: boolean;
/** Bottom sentinel observed by usePostScrollFeed to load the next page. */
sentinelRef: React.RefObject<HTMLDivElement | null>;
emptyMessage?: string;
hrefFor?: (post: Post) => string;
onDelete?: (id: string) => void;
onUpdated?: (updated: Post) => void;
}) {
// Chunk the flat post list into rows of COLS for row-level virtualization.
const rows = useMemo(() => {
const out: Post[][] = [];
for (let i = 0; i < posts.length; i += COLS) out.push(posts.slice(i, i + COLS));
return out;
}, [posts]);
// Document-absolute top of the grid, so the window virtualizer can map window
// scroll → visible rows (the profile header/tabs sit above us). Re-measured
// on resize and any body reflow. Same approach as HomeFeed.
const parentRef = useRef<HTMLDivElement>(null);
const [offsetTop, setOffsetTop] = useState(0);
useEffect(() => {
const measure = () => {
const el = parentRef.current;
if (el) setOffsetTop(el.getBoundingClientRect().top + window.scrollY);
};
measure();
window.addEventListener("resize", measure);
const ro = new ResizeObserver(measure);
ro.observe(document.body);
return () => {
window.removeEventListener("resize", measure);
ro.disconnect();
};
}, []);
const virtualizer = useWindowVirtualizer({
count: rows.length,
estimateSize: () => 240, // square card + gap; measureElement corrects it
overscan: 6,
scrollMargin: offsetTop,
});
if (loading && posts.length === 0) {
return (
<div className="flex justify-center py-16">
<Loader2 className="size-6 animate-spin text-muted-foreground" />
</div>
);
}
if (!loading && posts.length === 0) {
return (
<p className="text-sm text-muted-foreground">
{emptyMessage ?? "Nothing here yet."}
</p>
);
}
const virtualRows = virtualizer.getVirtualItems();
return (
<div>
<div
ref={parentRef}
className="relative w-full"
style={{ height: virtualizer.getTotalSize() }}
>
{virtualRows.map((vr) => {
const row = rows[vr.index];
return (
<div
key={row[0]?.id ?? vr.index}
data-index={vr.index}
ref={virtualizer.measureElement}
className="absolute left-0 top-0 w-full"
style={{ transform: `translateY(${vr.start - offsetTop}px)` }}
>
<div className="grid grid-cols-2 gap-2 sm:gap-3 pb-2 sm:pb-3">
{row.map((p) => (
<PostGridCard
key={p.id}
post={p}
href={hrefFor?.(p)}
onDelete={onDelete}
onUpdated={onUpdated}
/>
))}
</div>
</div>
);
})}
</div>
{hasMore ? (
<div ref={sentinelRef} className="flex justify-center py-6">
<Loader2 className="size-5 animate-spin text-muted-foreground" />
</div>
) : null}
</div>
);
}