Readur/frontend/src/services/api.ts

import axios from 'axios'

const api = axios.create({
  baseURL: '/api',
  headers: {
    'Content-Type': 'application/json',
  },
})

export { api }
export default api

export interface Document {
  id: string
  filename: string
  original_filename: string
  file_path: string
  file_size: number
  mime_type: string
  tags: string[]
  created_at: string
  updated_at: string
  user_id: string
  username?: string
  file_hash?: string
  original_created_at?: string
  original_modified_at?: string
  source_path?: string
  source_type?: string
  source_id?: string
  file_permissions?: number
  file_owner?: string
  file_group?: string
  source_metadata?: Record<string, any>
  has_ocr_text: boolean
  ocr_confidence?: number
  ocr_word_count?: number
  ocr_processing_time_ms?: number
  ocr_status?: string
}

export interface SearchRequest {
  query: string
  tags?: string[]
  mime_types?: string[]
  limit?: number
  offset?: number
  include_snippets?: boolean
  snippet_length?: number
  search_mode?: 'simple' | 'phrase' | 'fuzzy' | 'boolean'
}

export interface HighlightRange {
  start: number
  end: number
}

export interface SearchSnippet {
  text: string
  start_offset: number
  end_offset: number
  highlight_ranges: HighlightRange[]
}

export interface EnhancedDocument {
  id: string
  filename: string
  original_filename: string
  file_size: number
  mime_type: string
  tags: string[]
  created_at: string
  has_ocr_text: boolean
  ocr_confidence?: number
  ocr_word_count?: number
  ocr_processing_time_ms?: number
  ocr_status?: string
  search_rank?: number
  snippets: SearchSnippet[]
}

export interface SearchResponse {
  documents: EnhancedDocument[]
  total: number
  query_time_ms: number
  suggestions: string[]
}

export interface FacetItem {
  value: string
  count: number
}

export interface SearchFacetsResponse {
  mime_types: FacetItem[]
  tags: FacetItem[]
}

// OCR Retry Types
export interface OcrRetryFilter {
  mime_types?: string[]
  file_extensions?: string[]
  failure_reasons?: string[]
  min_file_size?: number
  max_file_size?: number
  created_after?: string
  created_before?: string
  tags?: string[]
  limit?: number
}

export interface BulkOcrRetryRequest {
  mode: 'all' | 'specific' | 'filter'
  document_ids?: string[]
  filter?: OcrRetryFilter
  priority_override?: number
  preview_only?: boolean
}

export interface OcrRetryDocumentInfo {
  id: string
  filename: string
  file_size: number
  mime_type: string
  ocr_failure_reason?: string
  priority: number
  queue_id?: string
}

export interface BulkOcrRetryResponse {
  success: boolean
  message: string
  queued_count: number
  matched_count: number
  documents: OcrRetryDocumentInfo[]
  estimated_total_time_minutes: number
}

export interface OcrRetryStatsResponse {
  failure_reasons: Array<{
    reason: string
    count: number
    avg_file_size_mb: number
    first_occurrence: string
    last_occurrence: string
  }>
  file_types: Array<{
    mime_type: string
    count: number
    avg_file_size_mb: number
  }>
  total_failed: number
}

export interface OcrRetryRecommendation {
  reason: string
  title: string
  description: string
  estimated_success_rate: number
  document_count: number
  filter: OcrRetryFilter
}

export interface OcrRetryRecommendationsResponse {
  recommendations: OcrRetryRecommendation[]
  total_recommendations: number
}

export interface DocumentRetryHistoryItem {
  id: string
  retry_reason: string
  previous_status?: string
  previous_failure_reason?: string
  previous_error?: string
  priority: number
  queue_id?: string
  created_at: string
}

export interface DocumentRetryHistoryResponse {
  document_id: string
  retry_history: DocumentRetryHistoryItem[]
  total_retries: number
}

export interface PaginatedResponse<T> {
  documents: T[]
  pagination: {
    total: number
    limit: number
    offset: number
    has_more: boolean
  }
}

export interface QueueStats {
  pending_count: number
  processing_count: number
  failed_count: number
  completed_today: number
  avg_wait_time_minutes?: number
  oldest_pending_minutes?: number
}

export interface OcrResponse {
  document_id: string
  filename: string
  has_ocr_text: boolean
  ocr_text?: string
  ocr_confidence?: number
  ocr_word_count?: number
  ocr_processing_time_ms?: number
  ocr_status?: string
  ocr_error?: string
  ocr_completed_at?: string
}

export const documentService = {
  upload: (file: File) => {
    const formData = new FormData()
    formData.append('file', file)
    return api.post('/documents', formData, {
      headers: {
        'Content-Type': 'multipart/form-data',
      },
    })
  },

  list: (limit = 50, offset = 0) => {
    return api.get<Document[]>('/documents', {
      params: { limit, offset },
    })
  },

  listWithPagination: (limit = 20, offset = 0, ocrStatus?: string) => {
    const params: any = { limit, offset };
    if (ocrStatus) {
      params.ocr_status = ocrStatus;
    }
    return api.get<{documents: Document[], pagination: {total: number, limit: number, offset: number, has_more: boolean}}>('/documents', {
      params,
    })
  },

  getById: (id: string) => {
    return api.get<Document>(`/documents/${id}`)
  },

  download: (id: string) => {
    return api.get(`/documents/${id}/download`, {
      responseType: 'blob',
    })
  },

  downloadFile: async (id: string, filename?: string) => {
    try {
      const response = await api.get(`/documents/${id}/download`, {
        responseType: 'blob',
      });
      
      // Create blob URL and trigger download
      const blob = new Blob([response.data], { type: response.headers['content-type'] });
      const url = window.URL.createObjectURL(blob);
      const link = document.createElement('a');
      link.href = url;
      link.download = filename || `document-${id}`;
      document.body.appendChild(link);
      link.click();
      document.body.removeChild(link);
      window.URL.revokeObjectURL(url);
    } catch (error) {
      console.error('Download failed:', error);
      throw error;
    }
  },

  getOcrText: (id: string) => {
    return api.get<OcrResponse>(`/documents/${id}/ocr`)
  },

  view: (id: string) => {
    return api.get(`/documents/${id}/view`, {
      responseType: 'blob',
    })
  },

  getThumbnail: (id: string) => {
    return api.get(`/documents/${id}/thumbnail`, {
      responseType: 'blob',
    })
  },

  getProcessedImage: (id: string) => {
    return api.get(`/documents/${id}/processed-image`, {
      responseType: 'blob',
    })
  },

  retryOcr: (id: string) => {
    return api.post(`/documents/${id}/retry-ocr`)
  },

  // Advanced OCR retry functionality
  bulkRetryOcr: (request: BulkOcrRetryRequest) => {
    return api.post<BulkOcrRetryResponse>('/documents/ocr/bulk-retry', request)
  },

  getRetryStats: () => {
    return api.get<OcrRetryStatsResponse>('/documents/ocr/retry-stats')
  },

  getRetryRecommendations: () => {
    return api.get<OcrRetryRecommendationsResponse>('/documents/ocr/retry-recommendations')
  },

  getDocumentRetryHistory: (id: string) => {
    return api.get<DocumentRetryHistoryResponse>(`/documents/${id}/ocr/retry-history`)
  },

  getFailedOcrDocuments: (limit = 50, offset = 0) => {
    return api.get(`/documents/failed`, {
      params: { stage: 'ocr', limit, offset },
    })
  },

  getDuplicates: (limit = 25, offset = 0) => {
    return api.get(`/documents/duplicates`, {
      params: { limit, offset },
    })
  },

  search: (searchRequest: SearchRequest) => {
    return api.get<SearchResponse>('/search', {
      params: searchRequest,
    })
  },

  enhancedSearch: (searchRequest: SearchRequest) => {
    return api.get<SearchResponse>('/search/enhanced', {
      params: {
        ...searchRequest,
        include_snippets: searchRequest.include_snippets ?? true,
        snippet_length: searchRequest.snippet_length ?? 200,
        search_mode: searchRequest.search_mode ?? 'simple',
      },
    })
  },

  getFacets: () => {
    return api.get<SearchFacetsResponse>('/search/facets')
  },

  delete: (id: string) => {
    return api.delete(`/documents/${id}`)
  },

  bulkDelete: (documentIds: string[]) => {
    return api.delete('/documents', {
      data: { document_ids: documentIds }
    })
  },

  deleteLowConfidence: (maxConfidence: number, previewOnly: boolean = false) => {
    return api.post('/documents/delete-low-confidence', {
      max_confidence: maxConfidence,
      preview_only: previewOnly
    })
  },
  deleteFailedOcr: (previewOnly: boolean = false) => {
    return api.post('/documents/delete-failed-ocr', {
      preview_only: previewOnly
    })
  },

  getFailedDocuments: (limit = 25, offset = 0, stage?: string, reason?: string) => {
    const params: any = { limit, offset };
    if (stage) params.stage = stage;
    if (reason) params.reason = reason;
    return api.get('/documents/failed', { params })
  },
}

export interface OcrStatusResponse {
  is_paused: boolean
  status: 'paused' | 'running'
}

export interface OcrActionResponse {
  status: 'paused' | 'resumed'
  message: string
}

export const queueService = {
  getStats: () => {
    return api.get<QueueStats>('/queue/stats')
  },

  requeueFailed: () => {
    return api.post('/queue/requeue-failed')
  },

  getOcrStatus: () => {
    return api.get<OcrStatusResponse>('/queue/status')
  },

  pauseOcr: () => {
    return api.post<OcrActionResponse>('/queue/pause')
  },

  resumeOcr: () => {
    return api.post<OcrActionResponse>('/queue/resume')
  },
}