Class CoreDataProvider

packages/tools/src/providers/core-data-provider.ts:43–172

} from '../types'

export class CoreDataProvider implements ToolDataProvider {
  constructor(private db: DatabaseAdapter) {}

  async searchMessages(
    keywords: string[],
    options?: { timeFilter?: ToolTimeRange; limit?: number; senderId?: number }
  ): Promise<SearchMessagesResult> {
    const result = searchMessagesByKeywords(this.db, keywords, {
      startTs: options?.timeFilter?.startTs,
      endTs: options?.timeFilter?.endTs,
      senderId: options?.senderId,
      limit: options?.limit ?? 50,
    })
    return {
      messages: result.messages.map((m) => ({
        id: m.id,
        senderId: m.senderId,
        senderName: m.senderName,
        senderPlatformId: m.senderPlatformId,
        content: m.content,
        timestamp: m.timestamp,
      })),
      total: result.total ?? result.messages.length,
    }
  }

  async deepSearchMessages(
    keywords: string[],
    options?: { timeFilter?: ToolTimeRange; limit?: number; senderId?: number }
  ): Promise<SearchMessagesResult> {
    return this.searchMessages(keywords, options)
  }

  async getSearchMessageContext(
    messageIds: number[],
    contextBefore: number,
    contextAfter: number
  ): Promise<RawMessage[]> {
    return coreGetSearchMessageContext(this.db, messageIds, contextBefore, contextAfter)
  }

  async getRecentMessages(options?: { timeFilter?: ToolTimeRange; limit?: number }): Promise<SearchMessagesResult> {
    const messages = coreGetRecentMessages(this.db, { limit: options?.limit ?? 50 })
    return {
      messages: messages.map((m) => ({
        id: m.id,
        senderId: m.senderId,
        senderName: m.senderName,
        senderPlatformId: m.senderPlatformId,
        content: m.content,
        timestamp: m.timestamp,
      })),
      total: messages.length,
    }
  }

  async getMessageContext(messageIds: number[], contextSize: number): Promise<RawMessage[]> {
    return coreGetMessageContext(this.db, messageIds, contextSize)