mcp-database-server/src/utils/formatUtils.ts

/**
 * Convert data to CSV format
 * @param data Array of objects to convert to CSV
 * @returns CSV formatted string
 */
export function convertToCSV(data: any[]): string {
  if (data.length === 0) return '';
  
  // Get headers
  const headers = Object.keys(data[0]);
  
  // Create CSV header row
  let csv = headers.join(',') + '\n';
  
  // Add data rows
  data.forEach(row => {
    const values = headers.map(header => {
      const val = row[header];
      // Handle strings with commas, quotes, etc.
      if (typeof val === 'string') {
        return `"${val.replace(/"/g, '""')}"`;
      }
      // Use empty string for null/undefined
      return val === null || val === undefined ? '' : val;
    });
    csv += values.join(',') + '\n';
  });
  
  return csv;
}

/**
 * Format error response
 * @param error Error object or message
 * @returns Formatted error response object
 */
export function formatErrorResponse(error: Error | string): { content: Array<{type: string, text: string}>, isError: boolean } {
  const message = error instanceof Error ? error.message : error;
  return {
    content: [{ 
      type: "text", 
      text: JSON.stringify({ error: message }, null, 2) 
    }],
    isError: true
  };
}

/**
 * Format success response
 * @param data Data to format
 * @returns Formatted success response object
 */
export function formatSuccessResponse(data: any): { content: Array<{type: string, text: string}>, isError: boolean } {
  return {
    content: [{ 
      type: "text", 
      text: JSON.stringify(data, null, 2) 
    }],
    isError: false
  };
}