malice

import type { RoomSummary } from './types';

export interface PaneBadge {

  count: number;

  mention: boolean;

}

export interface PaneBadges {

  rail: PaneBadge;

  channels: PaneBadge;

}

/** DMs and mentions, each counted on whichever folded pane reaches its room in the fewest clicks. */

export function paneBadges(

  rooms: RoomSummary[],

  channelIds: Set<string>,

  openRoomId: string | null,

  muted: (room: RoomSummary) => boolean

): PaneBadges {

  const rail = { count: 0, mention: false };

  const channels = { count: 0, mention: false };

  for (const room of rooms) {

    if (room.space || room.membership !== 'Joined' || room.room_id === openRoomId || muted(room)) continue;

    const count = room.direct

      ? room.notification_mode === 'MentionsAndKeywordsOnly' ? room.unread_mentions : room.unread_messages + room.unread_mentions

      : room.unread_mentions;

    if (count === 0) continue;

    const badge = channelIds.has(room.room_id) ? channels : rail;

    badge.count += count;

    badge.mention ||= room.unread_mentions > 0;

  }

  return { rail, channels };

}