import { useOffscreenActivityChannelIds } from "@/features/sidebar/lib/useOffscreenActivityChannelIds";
import { useUnreadOverflow } from "@/features/sidebar/lib/useUnreadOverflow";

type ActivityOptions = Parameters<typeof useOffscreenActivityChannelIds>[0];
type ScrollRef = Parameters<typeof useUnreadOverflow>[0]["scrollRef"];

export function getSidebarActivityOverflowLabel({
  activityCount,
  messageCount,
}: {
  activityCount: number;
  messageCount: number;
}) {
  return activityCount === messageCount
    ? undefined
    : `${activityCount} new activity`;
}

export function useSidebarActivityOverflow({
  scrollRef,
  ...activityOptions
}: ActivityOptions & { scrollRef: ScrollRef }) {
  const { channelIds, messageChannelIds } =
    useOffscreenActivityChannelIds(activityOptions);
  const activityOverflow = useUnreadOverflow({
    scrollRef,
    unreadChannelIds: channelIds,
  });
  const messageOverflow = useUnreadOverflow({
    scrollRef,
    unreadChannelIds: messageChannelIds,
  });

  return {
    ...activityOverflow,
    unreadAboveLabel: getSidebarActivityOverflowLabel({
      activityCount: activityOverflow.unreadAboveCount,
      messageCount: messageOverflow.unreadAboveCount,
    }),
    unreadBelowLabel: getSidebarActivityOverflowLabel({
      activityCount: activityOverflow.unreadBelowCount,
      messageCount: messageOverflow.unreadBelowCount,
    }),
  };
}