Function fetchMessageContext

packages/core/src/query/message-query-functions.ts:154–188

(
  executor: AsyncSqlExecutor,
  messageIds: number | number[],
  contextSize: number = 20
)

 * Uses simple id-based ordering (not session-aware).
 */
export async function fetchMessageContext(
  executor: AsyncSqlExecutor,
  messageIds: number | number[],
  contextSize: number = 20
): Promise<MappedMessage[]> {
  const ids = Array.isArray(messageIds) ? messageIds : [messageIds]
  if (ids.length === 0) return []

  const allIds = new Set<number>()

  for (const id of ids) {
    allIds.add(id)
    if (contextSize > 0) {
      const before = await executor.all<{ id: number }>(
        'SELECT id FROM message WHERE id < ? ORDER BY id DESC LIMIT ?',
        [id, contextSize]
      )
      before.forEach((r) => allIds.add(r.id))

      const after = await executor.all<{ id: number }>('SELECT id FROM message WHERE id > ? ORDER BY id ASC LIMIT ?', [
        id,
        contextSize,
      ])
      after.forEach((r) => allIds.add(r.id))
    }
  }

  const idList = Array.from(allIds).sort((a, b) => a - b)
  if (idList.length === 0) return []

  const placeholders = idList.map(() => '?').join(', ')
  const sql = `${FULL_MSG_SELECT} WHERE msg.id IN (${placeholders}) ORDER BY msg.id ASC`
  const rows = await executor.all<FullMessageRow>(sql, idList)
  return rows.map(mapMessageRow)
}

/**
 * Get context messages around search results.