open-brainy/src/mcp/brainyMCPAdapter.ts

/**
 * BrainyMCPAdapter
 * 
 * This class provides an adapter for accessing Brainy data through the Model Control Protocol (MCP).
 * It wraps a Brainy instance and exposes methods for getting vectors, searching similar items,
 * and getting relationships.
 */

import { v4 as uuidv4 } from '../universal/uuid.js'
import { BrainyInterface } from '../types/brainyDataInterface.js'
import { NounType } from '../types/graphTypes.js'
import {
  MCPRequest,
  MCPResponse,
  MCPDataAccessRequest,
  MCPRequestType,
  MCP_VERSION
} from '../types/mcpTypes.js'

export class BrainyMCPAdapter {
  private brainyData: BrainyInterface

  /**
   * Creates a new BrainyMCPAdapter
   * @param brainyData The Brainy instance to wrap
   */
  constructor(brainyData: BrainyInterface) {
    this.brainyData = brainyData
  }

  /**
   * Handles an MCP data access request
   * @param request The MCP request
   * @returns An MCP response
   */
  async handleRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
    try {
      switch (request.operation) {
        case 'get':
          return await this.handleGetRequest(request)
        case 'search':
          return await this.handleSearchRequest(request)
        case 'add':
          return await this.handleAddRequest(request)
        case 'getRelationships':
          return await this.handleGetRelationshipsRequest(request)
        default:
          return this.createErrorResponse(
            request.requestId,
            'UNSUPPORTED_OPERATION',
            `Operation ${request.operation} is not supported`
          )
      }
    } catch (error) {
      return this.createErrorResponse(
        request.requestId,
        'INTERNAL_ERROR',
        error instanceof Error ? error.message : String(error)
      )
    }
  }

  /**
   * Handles a get request
   * @param request The MCP request
   * @returns An MCP response
   */
  private async handleGetRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
    const { id } = request.parameters

    if (!id) {
      return this.createErrorResponse(
        request.requestId,
        'MISSING_PARAMETER',
        'Parameter "id" is required'
      )
    }

    const noun = await this.brainyData.get(id)
    
    if (!noun) {
      return this.createErrorResponse(
        request.requestId,
        'NOT_FOUND',
        `No noun found with id ${id}`
      )
    }

    return this.createSuccessResponse(request.requestId, noun)
  }

  /**
   * Handles a search request
   * @param request The MCP request
   * @returns An MCP response
   */
  private async handleSearchRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
    const { query, k = 10 } = request.parameters

    if (!query) {
      return this.createErrorResponse(
        request.requestId,
        'MISSING_PARAMETER',
        'Parameter "query" is required'
      )
    }

    const results = await this.brainyData.find({ query, limit: k })
    return this.createSuccessResponse(request.requestId, results)
  }

  /**
   * Handles an add request
   * @param request The MCP request
   * @returns An MCP response
   */
  private async handleAddRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
    const { text, metadata } = request.parameters

    if (!text) {
      return this.createErrorResponse(
        request.requestId,
        'MISSING_PARAMETER',
        'Parameter "text" is required'
      )
    }

    // Add data using modern API (interface only supports modern methods)
    const id = await this.brainyData.add({ data: text, type: NounType.Document, metadata })
    return this.createSuccessResponse(request.requestId, { id })
  }

  /**
   * Handles a getRelationships request
   * @param request The MCP request
   * @returns An MCP response
   */
  private async handleGetRelationshipsRequest(request: MCPDataAccessRequest): Promise<MCPResponse> {
    const { id } = request.parameters

    if (!id) {
      return this.createErrorResponse(
        request.requestId,
        'MISSING_PARAMETER',
        'Parameter "id" is required'
      )
    }

    // This is a simplified implementation - in a real implementation, we would
    // need to check if these methods exist on the BrainyInterface
    const outgoing = await (this.brainyData as any).getVerbsBySource?.(id) || []
    const incoming = await (this.brainyData as any).getVerbsByTarget?.(id) || []

    return this.createSuccessResponse(request.requestId, { outgoing, incoming })
  }

  /**
   * Creates a success response
   * @param requestId The request ID
   * @param data The response data
   * @returns An MCP response
   */
  private createSuccessResponse(requestId: string, data: any): MCPResponse {
    return {
      success: true,
      requestId,
      version: MCP_VERSION,
      data
    }
  }

  /**
   * Creates an error response
   * @param requestId The request ID
   * @param code The error code
   * @param message The error message
   * @param details Optional error details
   * @returns An MCP response
   */
  private createErrorResponse(
    requestId: string,
    code: string,
    message: string,
    details?: any
  ): MCPResponse {
    return {
      success: false,
      requestId,
      version: MCP_VERSION,
      error: {
        code,
        message,
        details
      }
    }
  }

  /**
   * Creates a new request ID
   * @returns A new UUID
   */
  generateRequestId(): string {
    return uuidv4()
  }
}