import assert from 'node:assert/strict';
import { test } from 'node:test';
import { paneBadges } from '../src/lib/pane-badges.ts';
import type { RoomSummary } from '../src/lib/types.ts';

function room(id: string, extra: Partial<RoomSummary> = {}): RoomSummary {
  return {
    room_id: id,
    name: id,
    membership: 'Joined',
    space: false,
    direct: false,
    unread_messages: 0,
    unread_mentions: 0,
    notification_mode: 'AllMessages',
    notification_mode_custom: false,
    parent_spaces: [],
    children: [],
    ...extra
  } as RoomSummary;
}

const isMuted = (r: RoomSummary) => r.notification_mode === 'Mute';

const rooms = [
  room('!a', { unread_messages: 5, unread_mentions: 1 }),
  room('!b', { unread_messages: 4, unread_mentions: 2 }),
  room('!quiet', { unread_messages: 9 }),
  room('!dm', { direct: true, unread_messages: 3 }),
  room('!muted', { unread_mentions: 4, notification_mode: 'Mute' }),
  room('!left', { unread_mentions: 4, membership: 'Left' })
];

test('mentions in the shown space go on channels, others and DMs on the rail', () => {
  const badges = paneBadges(rooms, new Set(['!a', '!quiet']), null, isMuted);
  assert.deepEqual(badges.channels, { count: 1, mention: true });
  assert.deepEqual(badges.rail, { count: 5, mention: true });
});

test('home lists every room, so a DM goes on channels', () => {
  const badges = paneBadges(rooms, new Set(rooms.map((r) => r.room_id)), null, isMuted);
  assert.deepEqual(badges.channels, { count: 6, mention: true });
  assert.deepEqual(badges.rail, { count: 0, mention: false });
});

test('the open room, muted rooms and plain unread channels count for nothing', () => {
  const badges = paneBadges(rooms, new Set(), '!b', (r) => isMuted(r) || r.room_id === '!a');
  assert.deepEqual(badges.rail, { count: 3, mention: false });
});