import type { EventContext, MemberSummary, Sender, TimelineItem } from './types';

export type RoomProfiles = Record<string, Record<string, MemberSummary>>;

export function lacksProfile(sender: Sender): boolean {
  return !sender.display_name && !sender.avatar;
}

/** Rooms holding a sender or reply sender that came back with no name and no avatar. */
export function roomsLackingProfiles(items: { roomId: string; item: TimelineItem }[]): Set<string> {
  const rooms = new Set<string>();
  for (const { roomId, item } of items) {
    if (item.kind !== 'event') continue;
    const reply = item.event.reply_to?.details;
    if (lacksProfile(item.event.sender) || (reply?.kind === 'Ready' && lacksProfile(reply.sender))) rooms.add(roomId);
  }
  return rooms;
}

function filled(sender: Sender, members: Record<string, MemberSummary> | undefined): Sender {
  const member = members?.[sender.user_id];
  if (!member || !lacksProfile(sender) || (!member.display_name && !member.avatar)) return sender;
  return { ...sender, display_name: member.display_name, avatar: member.avatar, name_ambiguous: member.name_ambiguous };
}

export function withProfiles(item: TimelineItem, members: Record<string, MemberSummary> | undefined): TimelineItem {
  if (item.kind !== 'event' || !members) return item;
  const { reply_to: reply } = item.event;
  const details = reply?.details;
  const sender = filled(item.event.sender, members);
  const replySender = details?.kind === 'Ready' ? filled(details.sender, members) : null;
  if (sender === item.event.sender && (details?.kind !== 'Ready' || replySender === details.sender)) return item;
  const reply_to = reply && details?.kind === 'Ready' && replySender ? { ...reply, details: { ...details, sender: replySender } } : reply;
  return { ...item, event: { ...item.event, sender, reply_to } };
}

export function contextWithProfiles(context: EventContext, members: Record<string, MemberSummary> | undefined): EventContext {
  if (!members) return context;
  return {
    ...context,
    events_before: context.events_before.map((item) => withProfiles(item, members)),
    event: withProfiles(context.event, members),
    events_after: context.events_after.map((item) => withProfiles(item, members))
  };
}