buzz/mobile/lib/features/channels/channel_window.dart

import 'dart:convert';

import '../../shared/relay/relay.dart';
import 'channel_event_order.dart';

class ChannelPageCursor {
  final int createdAt;
  final String eventId;

  const ChannelPageCursor({required this.createdAt, required this.eventId});
}

class ChannelWindowThreadSummary {
  final int replyCount;
  final int descendantCount;
  final int? lastReplyAt;
  final List<String> participantPubkeys;

  const ChannelWindowThreadSummary({
    required this.replyCount,
    required this.descendantCount,
    required this.lastReplyAt,
    required this.participantPubkeys,
  });
}

/// A thread-summary snapshot received from the live channel subscription.
///
/// [createdAt] is the relay event timestamp used to reject delayed snapshots
/// after a fresher recount has already been displayed. Recounts created in the
/// same second retain their live delivery order, because relay timestamps have
/// second precision.
class ChannelWindowLiveThreadSummary {
  final ChannelWindowThreadSummary summary;
  final int createdAt;

  const ChannelWindowLiveThreadSummary({
    required this.summary,
    required this.createdAt,
  });
}

class ChannelWindowRow {
  final NostrEvent event;
  final ChannelWindowThreadSummary? thread;
  final int? threadSummaryCreatedAt;

  const ChannelWindowRow({
    required this.event,
    this.thread,
    this.threadSummaryCreatedAt,
  });

  ChannelWindowRow copyWith({
    ChannelWindowThreadSummary? thread,
    int? threadSummaryCreatedAt,
  }) => ChannelWindowRow(
    event: event,
    thread: thread ?? this.thread,
    threadSummaryCreatedAt:
        threadSummaryCreatedAt ?? this.threadSummaryCreatedAt,
  );
}

class ChannelWindowPage {
  final ChannelPageCursor? startCursor;
  final List<ChannelWindowRow> rows;
  final List<NostrEvent> aux;
  final ChannelPageCursor? nextCursor;
  final bool hasMore;

  const ChannelWindowPage({
    required this.startCursor,
    required this.rows,
    required this.aux,
    required this.nextCursor,
    required this.hasMore,
  });
}

class ChannelWindowStore {
  final List<ChannelWindowPage> pages;
  final List<NostrEvent> liveOverlay;
  final List<NostrEvent> liveAux;

  /// Thread summaries that arrived over the live socket, keyed by root event id.
  ///
  /// Kept beside the pages rather than folded into [ChannelWindowRow.thread]
  /// because a root can be in [liveOverlay] instead — a message you just sent
  /// has no row yet, and its reply count has to land somewhere.
  final Map<String, ChannelWindowLiveThreadSummary> liveThreadSummaries;

  const ChannelWindowStore({
    required this.pages,
    required this.liveThreadSummaries,
    required this.liveOverlay,
    required this.liveAux,
  });

  const ChannelWindowStore.empty()
    : pages = const [],
      liveOverlay = const [],
      liveAux = const [],
      liveThreadSummaries = const {};
}

ChannelWindowPage parseChannelWindowResponse(
  List<NostrEvent> events,
  String channelId,
  ChannelPageCursor? startCursor,
) {
  var rows = [
    for (final event in events)
      if (EventKind.channelTimelineContentKinds.contains(event.kind))
        ChannelWindowRow(event: event),
  ];
  final rowIndexesById = <String, int>{
    for (var i = 0; i < rows.length; i++) rows[i].event.id: i,
  };

  for (final event in events) {
    if (event.kind != EventKind.channelThreadSummary) continue;
    final rootId = event.getTagValue('e');
    final rowIndex = rootId == null ? null : rowIndexesById[rootId];
    if (rowIndex == null) continue;
    rows[rowIndex] = rows[rowIndex].copyWith(
      thread: parseChannelWindowThreadSummary(event),
      threadSummaryCreatedAt: event.createdAt,
    );
  }

  final boundsEvents = events
      .where((event) => event.kind == EventKind.channelWindowBounds)
      .toList();
  if (boundsEvents.length != 1) {
    throw Exception(
      'Channel window response must contain exactly one bounds event.',
    );
  }
  final boundsEvent = boundsEvents.single;
  if (boundsEvent.getTagValue('d') !=
      _expectedBoundsKey(channelId, startCursor)) {
    throw Exception('Channel window bounds do not match the request cursor.');
  }

  final bounds = _parseJsonMap(boundsEvent, 'window bounds');
  final hasMore = bounds['has_more'] as bool;
  final nextCursor = _parseCursor(bounds['next_cursor']);
  if (hasMore != (nextCursor != null)) {
    throw Exception('Channel window bounds has_more and next_cursor disagree.');
  }

  return ChannelWindowPage(
    startCursor: startCursor,
    rows: rows,
    aux: [
      for (final event in events)
        if (EventKind.channelAuxEventKinds.contains(event.kind)) event,
    ],
    nextCursor: nextCursor,
    hasMore: hasMore,
  );
}

ChannelWindowStore replaceNewestChannelWindow(
  ChannelWindowStore current,
  ChannelWindowPage page, {
  Set<String> retainLiveSummaryRootIds = const {},
}) {
  if (page.startCursor != null) {
    throw Exception('Newest channel page must have a null start cursor.');
  }
  _assertValidPage(page);
  final rowIds = page.rows.map((row) => row.event.id).toSet();
  final auxIds = page.aux.map((event) => event.id).toSet();
  return ChannelWindowStore(
    pages: [page],
    liveOverlay: current.liveOverlay
        .where((event) => !rowIds.contains(event.id))
        .toList(),
    liveAux: current.liveAux
        .where((event) => !auxIds.contains(event.id))
        .toList(),
    // A summary received while the page query was in flight can describe a
    // mutation after that query's snapshot. Retain just those summaries even
    // when the relay timestamp matches the page overlay; subscription replay
    // received before the query still compares timestamps normally.
    liveThreadSummaries: _retainLiveSummariesAfterPage(
      current,
      page.rows,
      retainLiveSummaryRootIds: retainLiveSummaryRootIds,
    ),
  );
}

Map<String, ChannelWindowLiveThreadSummary> _retainLiveSummariesAfterPage(
  ChannelWindowStore current,
  List<ChannelWindowRow> rows, {
  required Set<String> retainLiveSummaryRootIds,
}) {
  final rowsById = {for (final row in rows) row.event.id: row};
  return {
    for (final entry in current.liveThreadSummaries.entries)
      if (retainLiveSummaryRootIds.contains(entry.key) ||
          rowsById[entry.key] == null ||
          rowsById[entry.key]!.thread == null ||
          entry.value.createdAt >
              (rowsById[entry.key]!.threadSummaryCreatedAt ?? -1))
        entry.key: entry.value,
  };
}

ChannelWindowStore appendOlderChannelWindow(
  ChannelWindowStore current,
  ChannelWindowPage page,
) {
  _assertValidPage(page);
  if (current.pages.isEmpty) {
    throw Exception('Load the newest channel page first.');
  }
  final tail = current.pages.last;
  if (!tail.hasMore || tail.nextCursor == null) {
    throw Exception('The channel window is already complete.');
  }
  if (!_cursorsEqual(page.startCursor, tail.nextCursor)) {
    throw Exception(
      'Channel page does not continue the retained cursor chain.',
    );
  }
  final retainedIds = current.pages
      .expand((page) => page.rows)
      .map((row) => row.event.id)
      .toSet();
  for (final row in page.rows) {
    if (retainedIds.contains(row.event.id)) {
      throw Exception('Channel row ${row.event.id} overlaps a retained page.');
    }
  }
  final pageIds = page.rows.map((row) => row.event.id).toSet();
  return ChannelWindowStore(
    pages: [...current.pages, page],
    liveOverlay: current.liveOverlay
        .where((event) => !pageIds.contains(event.id))
        .toList(),
    liveAux: current.liveAux,
    // A live recount can arrive while this older page is in flight. Keep it
    // when it is fresher than the page's embedded snapshot, just as a newest
    // page replacement does.
    liveThreadSummaries: _retainLiveSummariesAfterPage(
      current,
      page.rows,
      retainLiveSummaryRootIds: const {},
    ),
  );
}

/// Decode a kind-39005 thread summary. Same payload whether it came down as a
/// channel-window page overlay or over the live socket — one contract, two doors.
ChannelWindowThreadSummary parseChannelWindowThreadSummary(NostrEvent event) {
  final payload = _parseJsonMap(event, 'thread summary');
  final participants = payload['participants'];
  return ChannelWindowThreadSummary(
    replyCount: (payload['reply_count'] as num).toInt(),
    descendantCount: (payload['descendant_count'] as num).toInt(),
    lastReplyAt: (payload['last_reply_at'] as num?)?.toInt(),
    participantPubkeys: participants is List
        ? participants.whereType<String>().toList()
        : const [],
  );
}

ChannelWindowStore mergeLiveChannelWindowEvent(
  ChannelWindowStore current,
  NostrEvent event, {
  required bool isTimelineRow,
}) {
  // A reply doesn't reach the main timeline itself — the root's "N replies" row
  // comes from this summary event, which the relay re-emits on every reply. It
  // has to be merged, not stored as an aux event: it is a replaceable snapshot
  // keyed by root, not another row in the timeline.
  if (event.kind == EventKind.channelThreadSummary) {
    final rootId = event.getTagValue('e');
    if (rootId == null) return current;
    final ChannelWindowThreadSummary summary;
    try {
      summary = parseChannelWindowThreadSummary(event);
    } catch (_) {
      return current;
    }
    final existing = current.liveThreadSummaries[rootId];
    // A newer timestamp wins. Equal timestamps are ordered by live delivery:
    // two mutations can produce relay-signed recounts in the same second, and
    // the later delivery is the authoritative snapshot for that root.
    if (existing != null && existing.createdAt > event.createdAt) {
      return current;
    }
    return ChannelWindowStore(
      pages: current.pages,
      liveOverlay: current.liveOverlay,
      liveAux: current.liveAux,
      liveThreadSummaries: {
        ...current.liveThreadSummaries,
        rootId: ChannelWindowLiveThreadSummary(
          summary: summary,
          createdAt: event.createdAt,
        ),
      },
    );
  }

  if (!isTimelineRow) {
    final alreadyKnown =
        current.liveAux.any((candidate) => candidate.id == event.id) ||
        current.pages.any(
          (page) => page.aux.any((candidate) => candidate.id == event.id),
        );
    if (alreadyKnown) return current;
    return ChannelWindowStore(
      pages: current.pages,
      liveOverlay: current.liveOverlay,
      liveAux: [...current.liveAux, event],
      liveThreadSummaries: current.liveThreadSummaries,
    );
  }

  final inPages = current.pages.any(
    (page) => page.rows.any((row) => row.event.id == event.id),
  );
  if (inPages) return current;
  final oldestPage = current.pages.isEmpty ? null : current.pages.last;
  final oldest = oldestPage?.rows.isEmpty ?? true
      ? null
      : oldestPage!.rows.last.event;
  if (oldest != null &&
      (event.createdAt < oldest.createdAt ||
          (oldestPage!.hasMore && _compareRelayOrder(event, oldest) >= 0))) {
    return current;
  }
  final overlay =
      current.liveOverlay
          .where((candidate) => candidate.id != event.id)
          .toList()
        ..add(event)
        ..sort(_compareRelayOrder);
  return ChannelWindowStore(
    pages: current.pages,
    liveOverlay: overlay,
    liveAux: current.liveAux,
    liveThreadSummaries: current.liveThreadSummaries,
  );
}

List<NostrEvent> flattenChannelWindowEvents(ChannelWindowStore store) {
  final byId = <String, NostrEvent>{};
  for (final page in store.pages) {
    for (final row in page.rows) {
      byId[row.event.id] = row.event;
    }
    for (final event in page.aux) {
      byId[event.id] = event;
    }
  }
  for (final event in store.liveOverlay) {
    byId[event.id] = event;
  }
  for (final event in store.liveAux) {
    byId[event.id] = event;
  }
  return byId.values.toList()
    ..sort(compareChannelTimelineEventsChronologically);
}

bool channelWindowHasMore(ChannelWindowStore store) =>
    store.pages.isNotEmpty && store.pages.last.hasMore;

ChannelPageCursor? channelWindowNextCursor(ChannelWindowStore store) =>
    store.pages.isEmpty ? null : store.pages.last.nextCursor;

Map<String, ChannelWindowThreadSummary> channelWindowThreadSummaries(
  ChannelWindowStore store,
) {
  return {
    for (final page in store.pages)
      for (final row in page.rows)
        if (row.thread != null) row.event.id: row.thread!,
    // Live entries last: they are the newer snapshot for any root they cover.
    for (final entry in store.liveThreadSummaries.entries)
      entry.key: entry.value.summary,
  };
}

Map<String, dynamic> _parseJsonMap(NostrEvent event, String label) {
  try {
    final decoded = jsonDecode(event.content);
    if (decoded is Map<String, dynamic>) return decoded;
  } catch (_) {}
  throw Exception('Invalid $label event ${event.id}.');
}

ChannelPageCursor? _parseCursor(Object? value) {
  if (value == null) return null;
  if (value is! Map<String, dynamic>) {
    throw Exception('Invalid channel window cursor.');
  }
  return ChannelPageCursor(
    createdAt: (value['created_at'] as num).toInt(),
    eventId: value['id'] as String,
  );
}

String _expectedBoundsKey(String channelId, ChannelPageCursor? cursor) {
  final suffix = cursor == null
      ? 'head'
      : '${cursor.createdAt}:${cursor.eventId.toLowerCase()}';
  return '${channelId.toLowerCase()}:$suffix';
}

bool _cursorsEqual(ChannelPageCursor? left, ChannelPageCursor? right) =>
    identical(left, right) ||
    (left != null &&
        right != null &&
        left.createdAt == right.createdAt &&
        left.eventId == right.eventId);

void _assertValidPage(ChannelWindowPage page) {
  if (page.hasMore != (page.nextCursor != null)) {
    throw Exception('Channel window hasMore and nextCursor disagree.');
  }
  final seen = <String>{};
  for (var index = 0; index < page.rows.length; index++) {
    final event = page.rows[index].event;
    if (!seen.add(event.id)) {
      throw Exception('Duplicate channel row ${event.id}.');
    }
    final startCursor = page.startCursor;
    if (startCursor != null && !_isStrictlyOlder(event, startCursor)) {
      throw Exception(
        'Channel row ${event.id} is outside its cursor interval.',
      );
    }
    if (index > 0 &&
        _compareRelayOrder(page.rows[index - 1].event, event) > 0) {
      throw Exception('Channel window rows are not in relay order.');
    }
  }
}

bool _isStrictlyOlder(NostrEvent event, ChannelPageCursor cursor) =>
    event.createdAt < cursor.createdAt ||
    (event.createdAt == cursor.createdAt &&
        event.id.compareTo(cursor.eventId) > 0);

int _compareRelayOrder(NostrEvent left, NostrEvent right) {
  if (left.createdAt != right.createdAt) {
    return right.createdAt - left.createdAt;
  }
  return left.id.compareTo(right.id);
}