Stirling-PDF/frontend/src/hooks/useFileManager.ts

import { useState, useCallback } from 'react';
import { useIndexedDB } from '../contexts/IndexedDBContext';
import { FileWithUrl, FileMetadata } from '../types/file';
import { generateThumbnailForFile } from '../utils/thumbnailUtils';

export const useFileManager = () => {
  const [loading, setLoading] = useState(false);
  const indexedDB = useIndexedDB();

  const convertToFile = useCallback(async (fileMetadata: FileMetadata): Promise<File> => {
    if (!indexedDB) {
      throw new Error('IndexedDB context not available');
    }
    
    // Try ID first (preferred)
    if (fileMetadata.id) {
      const file = await indexedDB.loadFile(fileMetadata.id);
      if (file) {
        return file;
      }
    }
    throw new Error(`File not found in storage: ${fileMetadata.name} (ID: ${fileMetadata.id})`);
  }, [indexedDB]);

  const loadRecentFiles = useCallback(async (): Promise<FileMetadata[]> => {
    setLoading(true);
    try {
      if (!indexedDB) {
        return [];
      }
      
      // Get metadata only (no file data) for performance
      const storedFileMetadata = await indexedDB.loadAllMetadata();
      const sortedFiles = storedFileMetadata.sort((a, b) => (b.lastModified || 0) - (a.lastModified || 0));
      
      // Already in correct FileMetadata format
      return sortedFiles;
    } catch (error) {
      console.error('Failed to load recent files:', error);
      return [];
    } finally {
      setLoading(false);
    }
  }, [indexedDB]);

  const handleRemoveFile = useCallback(async (index: number, files: FileMetadata[], setFiles: (files: FileMetadata[]) => void) => {
    const file = files[index];
    if (!file.id) {
      throw new Error('File ID is required for removal');
    }
    if (!indexedDB) {
      throw new Error('IndexedDB context not available');
    }
    try {
      await indexedDB.deleteFile(file.id);
      setFiles(files.filter((_, i) => i !== index));
    } catch (error) {
      console.error('Failed to remove file:', error);
      throw error;
    }
  }, [indexedDB]);

  const storeFile = useCallback(async (file: File, fileId: string) => {
    if (!indexedDB) {
      throw new Error('IndexedDB context not available');
    }
    try {
      // Store file with provided UUID from FileContext (thumbnail generated internally)
      const metadata = await indexedDB.saveFile(file, fileId);

      // Convert file to ArrayBuffer for StoredFile interface compatibility
      const arrayBuffer = await file.arrayBuffer();
      
      // Return StoredFile format for compatibility with old API
      return {
        id: fileId,
        name: file.name,
        type: file.type,
        size: file.size,
        lastModified: file.lastModified,
        data: arrayBuffer,
        thumbnail: metadata.thumbnail
      };
    } catch (error) {
      console.error('Failed to store file:', error);
      throw error;
    }
  }, [indexedDB]);

  const createFileSelectionHandlers = useCallback((
    selectedFiles: string[],
    setSelectedFiles: (files: string[]) => void
  ) => {
    const toggleSelection = (fileId: string) => {
      setSelectedFiles(
        selectedFiles.includes(fileId)
          ? selectedFiles.filter(id => id !== fileId)
          : [...selectedFiles, fileId]
      );
    };

    const clearSelection = () => {
      setSelectedFiles([]);
    };

    const selectMultipleFiles = async (files: FileMetadata[], onFilesSelect: (files: File[]) => void, onStoredFilesSelect?: (filesWithMetadata: Array<{ file: File; originalId: string; metadata: FileMetadata }>) => void) => {
      if (selectedFiles.length === 0) return;

      try {
        // Filter by UUID and convert to File objects
        const selectedFileObjects = files.filter(f => selectedFiles.includes(f.id));
        
        if (onStoredFilesSelect) {
          // NEW: Use stored files flow that preserves IDs
          const filesWithMetadata = await Promise.all(
            selectedFileObjects.map(async (metadata) => ({
              file: await convertToFile(metadata),
              originalId: metadata.id,
              metadata
            }))
          );
          onStoredFilesSelect(filesWithMetadata);
        } else {
          // LEGACY: Old flow that generates new UUIDs (for backward compatibility)
          const filePromises = selectedFileObjects.map(convertToFile);
          const convertedFiles = await Promise.all(filePromises);
          onFilesSelect(convertedFiles); // FileContext will assign new UUIDs
        }
        
        clearSelection();
      } catch (error) {
        console.error('Failed to load selected files:', error);
        throw error;
      }
    };

    return {
      toggleSelection,
      clearSelection,
      selectMultipleFiles
    };
  }, [convertToFile]);

  const touchFile = useCallback(async (id: string) => {
    if (!indexedDB) {
      console.warn('IndexedDB context not available for touch operation');
      return;
    }
    try {
      // Update access time - this will be handled by the cache in IndexedDBContext
      // when the file is loaded, so we can just load it briefly to "touch" it
      await indexedDB.loadFile(id);
    } catch (error) {
      console.error('Failed to touch file:', error);
    }
  }, [indexedDB]);

  return {
    loading,
    convertToFile,
    loadRecentFiles,
    handleRemoveFile,
    storeFile,
    touchFile,
    createFileSelectionHandlers
  };
};