Function getMessageContext

packages/core/src/query/message-queries.ts:583–607

(
  db: DatabaseAdapter,
  messageIds: number[],
  contextSize: number = 20
)

 * Uses simple id-based ordering (not session-aware).
 */
export function getMessageContext(
  db: DatabaseAdapter,
  messageIds: number[],
  contextSize: number = 20
): MappedMessage[] {
  if (messageIds.length === 0) return []

  const contextIds = new Set<number>()

  for (const messageId of messageIds) {
    contextIds.add(messageId)

    const beforeRows = db
      .prepare('SELECT id FROM message WHERE id < ? ORDER BY id DESC LIMIT ?')
      .all(messageId, contextSize) as { id: number }[]
    beforeRows.forEach((r) => contextIds.add(r.id))

    const afterRows = db
      .prepare('SELECT id FROM message WHERE id > ? ORDER BY id ASC LIMIT ?')
      .all(messageId, contextSize) as { id: number }[]
    afterRows.forEach((r) => contextIds.add(r.id))
  }

  return hydrateMessagesByIds(db, Array.from(contextIds))
}

/**
 * Get context messages around search results.