malice

import assert from 'node:assert/strict';

import { test } from 'node:test';

import { contextWithProfiles, roomsLackingProfiles, withProfiles } from '../src/lib/mention-profiles.ts';

import type { MemberSummary, Sender, TimelineItem } from '../src/lib/types.ts';

const bare = (user_id: string): Sender => ({ user_id, display_name: null, avatar: null, name_ambiguous: false });

function message(sender: Sender, replySender: Sender | null = null): TimelineItem {

  return {

    kind: 'event',

    event: {

      item_id: `i-${sender.user_id}`,

      event_id: `$e-${sender.user_id}`,

      transaction_id: null,

      sender,

      timestamp_ms: 1,

      own: false,

      send_state: { kind: 'Sent' },

      content: { kind: 'Redacted' },

      reply_to: replySender ? { event_id: '$target', details: { kind: 'Ready', sender: replySender, content: { kind: 'Redacted' }, timestamp_ms: 0 } } : null,

      thread_root: null,

      thread_summary: null,

      reactions: [],

      read_by: [],

      editable: false

    }

  };

}

function member(user_id: string, display_name: string | null, avatar: string | null = null): MemberSummary {

  return { user_id, name: display_name ?? user_id, display_name, avatar, membership: 'Join', own: false, name_ambiguous: false, ignored: false, service: false };

}

const members = { '@carol:x': member('@carol:x', 'Carol', 'mxc://x/c'), '@dave:x': member('@dave:x', 'Dave') };

test('lists only rooms with a sender or reply sender missing both name and avatar', () => {

  const named = { ...bare('@carol:x'), display_name: 'Carol' };

  const rooms = roomsLackingProfiles([

    { roomId: '!a:x', item: message(named) },

    { roomId: '!b:x', item: message(bare('@carol:x')) },

    { roomId: '!c:x', item: message(named, bare('@dave:x')) },

    { roomId: '!d:x', item: { kind: 'readMarker' } }

  ]);

  assert.deepEqual([...rooms].sort(), ['!b:x', '!c:x']);

});

test('fills the sender and reply sender from the room members', () => {

  const filled = withProfiles(message(bare('@carol:x'), bare('@dave:x')), members);

  assert.equal(filled.kind, 'event');

  if (filled.kind !== 'event') return;

  assert.equal(filled.event.sender.display_name, 'Carol');

  assert.equal(filled.event.sender.avatar, 'mxc://x/c');

  assert.equal(filled.event.reply_to?.details.kind === 'Ready' && filled.event.reply_to.details.sender.display_name, 'Dave');

});

test('keeps an engine profile and the same item when nothing changes', () => {

  const named = message({ ...bare('@carol:x'), display_name: 'Carol from the engine' });

  assert.equal(withProfiles(named, members), named);

  const unknown = message(bare('@erin:x'));

  assert.equal(withProfiles(unknown, members), unknown);

  const context = { events_before: [unknown], event: named, events_after: [], start_token: null, end_token: null };

  assert.equal(contextWithProfiles(context, undefined), context);

});