KnowledgeBase/web/services/uploadService.ts

import { apiClient } from './apiClient';
import { IndexingConfig } from '../types';

// web/services/uploadService.ts
export const uploadService = {
  async uploadFile(file: File, authToken: string): Promise<any> {
    const formData = new FormData();
    formData.append('file', file);

    const response = await apiClient.request('/upload', {
      method: 'POST',
      body: formData,
    });

    if (!response.ok) {
      const errorData = await response.json();
      throw new Error(errorData.message || 'fileUploadFailed');
    }

    return response.json();
  },

  async uploadFileWithConfig(file: File, config: IndexingConfig, authToken: string): Promise<any> {
    const formData = new FormData();
    formData.append('file', file);
    formData.append('chunkSize', config.chunkSize.toString());
    formData.append('chunkOverlap', config.chunkOverlap.toString());
    formData.append('embeddingModelId', config.embeddingModelId);

    // 処理モードを追加（指定されている場合）
    if (config.mode) {
      formData.append('mode', config.mode);
    }

    // 分類を追加（指定されている場合）
    if (config.groupIds && config.groupIds.length > 0) {
      formData.append('groupIds', JSON.stringify(config.groupIds));
    }

    const response = await apiClient.request('/upload', {
      method: 'POST',
      body: formData,
    });

    if (!response.ok) {
      const errorData = await response.json();
      throw new Error(errorData.message || 'fileUploadFailed');
    }

    return response.json();
  },

  async uploadText(content: string, title: string, config: IndexingConfig, authToken: string): Promise<any> {
    const { data } = await apiClient.post('/upload/text', {
      content,
      title,
      chunkSize: config.chunkSize.toString(),
      chunkOverlap: config.chunkOverlap.toString(),
      embeddingModelId: config.embeddingModelId,
      mode: config.mode
    });
    return data;
  },

  /**
   * ファイル処理モードの推奨を取得
   * ファイルの種類、サイズなどの要因に基づいて、高速モードまたは高精度モードの使用を推奨します
   */
  async recommendMode(file: File): Promise<any> {
    // セーフティチェック
    if (!file || !file.name) {
      return {
        recommendedMode: 'fast',
        reason: 'invalidFile',
        warnings: ['incompleteFileInfo'],
      };
    }

    // フロントエンドの簡単な判定ロジック
    const ext = file.name.toLowerCase().split('.').pop();
    const sizeMB = file.size / (1024 * 1024);

    const preciseFormats = ['pdf', 'doc', 'docx', 'ppt', 'pptx'];
    const supportedFormats = [...preciseFormats, 'xls', 'xlsx', 'txt', 'md', 'html', 'json', 'csv'];

    if (!supportedFormats.includes(ext || '')) {
      return {
        recommendedMode: 'fast',
        reason: `unsupportedFileFormat`,
        reasonArgs: [ext],
        warnings: ['willUseFastMode'],
      };
    }

    if (!preciseFormats.includes(ext || '')) {
      return {
        recommendedMode: 'fast',
        reason: `formatNoPrecise`,
        reasonArgs: [ext],
        warnings: ['willUseFastMode'],
      };
    }

    // 小規模なファイルには高速モードを推奨
    if (sizeMB < 5) {
      return {
        recommendedMode: 'fast',
        reason: 'smallFileFastOk',
        estimatedCost: 0,
        estimatedTime: sizeMB * 2,
        warnings: [],
      };
    }

    // 中規模なファイルには高精度モードを推奨
    if (sizeMB < 50) {
      return {
        recommendedMode: 'precise',
        reason: 'mixedContentPreciseRecommended',
        estimatedCost: Math.max(0.01, sizeMB * 0.01),
        estimatedTime: sizeMB * 8,
        warnings: ['willIncurApiCost'],
      };
    }

    // 大規模なファイルには高精度モードを推奨するが警告を表示
    return {
      recommendedMode: 'precise',
      reason: 'largeFilePreciseRecommended',
      estimatedCost: sizeMB * 0.015,
      estimatedTime: sizeMB * 12,
      warnings: ['longProcessingTime', 'highApiCost', 'considerFileSplitting'],
    };
  },
};