Refactor into multiple functions

frontend/src/hooks/tools/sanitize/useSanitizeOperation.ts

@@ -53,6 +53,103 @@ export const useSanitizeOperation = () => {
     return { operation, operationId, fileId };
   }, []);
 
+  const buildFormData = useCallback((parameters: SanitizeParameters, file: File): FormData => {
+    const formData = new FormData();
+    formData.append('fileInput', file);
+    
+    // Add parameters
+    formData.append('removeJavaScript', parameters.removeJavaScript.toString());
+    formData.append('removeEmbeddedFiles', parameters.removeEmbeddedFiles.toString());
+    formData.append('removeXMPMetadata', parameters.removeXMPMetadata.toString());
+    formData.append('removeMetadata', parameters.removeMetadata.toString());
+    formData.append('removeLinks', parameters.removeLinks.toString());
+    formData.append('removeFonts', parameters.removeFonts.toString());
+    
+    return formData;
+  }, []);
+
+  const sanitizeFile = useCallback(async (
+    file: File,
+    parameters: SanitizeParameters,
+    generateSanitizedFileName: (originalFileName?: string) => string,
+    operationId: string,
+    fileId: string
+  ): Promise<File | null> => {
+    try {
+      const formData = buildFormData(parameters, file);
+
+      const response = await fetch('/api/v1/security/sanitize-pdf', {
+        method: 'POST',
+        body: formData,
+      });
+
+      if (!response.ok) {
+        const errorText = await response.text();
+        markOperationFailed(fileId, operationId, errorText);
+        console.error(`Error sanitizing file ${file.name}:`, errorText);
+        return null;
+      }
+
+      const blob = await response.blob();
+      const sanitizedFileName = generateSanitizedFileName(file.name);
+      const sanitizedFile = new File([blob], sanitizedFileName, { type: blob.type });
+      
+      markOperationApplied(fileId, operationId);
+      return sanitizedFile;
+    } catch (error) {
+      console.error(`Error sanitizing file ${file.name}:`, error);
+      markOperationFailed(fileId, operationId, error instanceof Error ? error.message : 'Unknown error');
+      return null;
+    }
+  }, [buildFormData, markOperationApplied, markOperationFailed]);
+
+  const createDownloadInfo = useCallback(async (results: File[]): Promise<void> => {
+    if (results.length === 1) {
+      const url = window.URL.createObjectURL(results[0]);
+      setDownloadUrl(url);
+    } else {
+      const { zipFile } = await zipFileService.createZipFromFiles(results, 'sanitized_files.zip');
+      const url = window.URL.createObjectURL(zipFile);
+      setDownloadUrl(url);
+    }
+  }, []);
+
+  const generateThumbnailsForResults = useCallback(async (results: File[]): Promise<void> => {
+    const thumbnails = await Promise.all(
+      results.map(async (file) => {
+        try {
+          const thumbnail = await generateThumbnailForFile(file);
+          return thumbnail || '';
+        } catch (error) {
+          console.warn(`Failed to generate thumbnail for ${file.name}:`, error);
+          return '';
+        }
+      })
+    );
+
+    setThumbnails(thumbnails);
+  }, []);
+
+  const processResults = useCallback(async (results: File[]): Promise<void> => {
+    setFiles(results);
+    setIsGeneratingThumbnails(true);
+
+    // Add sanitized files to FileContext for future use
+    await addFiles(results);
+
+    // Create download info - single file or ZIP
+    await createDownloadInfo(results);
+
+    // Generate thumbnails
+    await generateThumbnailsForResults(results);
+
+    setIsGeneratingThumbnails(false);
+    setStatus(results.length === 1 
+      ? t('sanitize.completed', 'Sanitization completed successfully') 
+      : t('sanitize.completedMultiple', 'Sanitized {{count}} files successfully', { count: results.length })
+    );
+  }, [addFiles, createDownloadInfo, generateThumbnailsForResults, t]);
+
   const executeOperation = useCallback(async (
     parameters: SanitizeParameters,
     selectedFiles: File[],
@@ -88,40 +185,11 @@ export const useSanitizeOperation = () => {
             })
         );
 
-        try {
+        const sanitizedFile = await sanitizeFile(file, parameters, generateSanitizedFileName, operationId, fileId);
-          const formData = new FormData();
-          formData.append('fileInput', file);
-          
-          // Add parameters
-          formData.append('removeJavaScript', parameters.removeJavaScript.toString());
-          formData.append('removeEmbeddedFiles', parameters.removeEmbeddedFiles.toString());
-          formData.append('removeXMPMetadata', parameters.removeXMPMetadata.toString());
-          formData.append('removeMetadata', parameters.removeMetadata.toString());
-          formData.append('removeLinks', parameters.removeLinks.toString());
-          formData.append('removeFonts', parameters.removeFonts.toString());
-
-          const response = await fetch('/api/v1/security/sanitize-pdf', {
-            method: 'POST',
-            body: formData,
-          });
-
-          if (!response.ok) {
-            const errorText = await response.text();
-            markOperationFailed(fileId, operationId, errorText);
-            console.error(`Error sanitizing file ${file.name}:`, errorText);
-            failedFiles.push(file.name);
-            continue;
-          }
-
-          const blob = await response.blob();
-          const sanitizedFileName = generateSanitizedFileName(file.name);
-          const sanitizedFile = new File([blob], sanitizedFileName, { type: blob.type });
         
+        if (sanitizedFile) {
           results.push(sanitizedFile);
-          markOperationApplied(fileId, operationId);
+        } else {
-        } catch (error) {
-          console.error(`Error sanitizing file ${file.name}:`, error);
-          markOperationFailed(fileId, operationId, error instanceof Error ? error.message : 'Unknown error');
           failedFiles.push(file.name);
         }
       }
@@ -135,41 +203,7 @@ export const useSanitizeOperation = () => {
       }
 
       if (results.length > 0) {
-        setFiles(results);
+        await processResults(results);
-        setIsGeneratingThumbnails(true);
-
-        // Add sanitized files to FileContext for future use
-        await addFiles(results);
-
-        // Create download info - single file or ZIP
-        if (results.length === 1) {
-          const url = window.URL.createObjectURL(results[0]);
-          setDownloadUrl(url);
-        } else {
-          const { zipFile } = await zipFileService.createZipFromFiles(results, 'sanitized_files.zip');
-          const url = window.URL.createObjectURL(zipFile);
-          setDownloadUrl(url);
-        }
-
-        // Generate thumbnails
-        const thumbnails = await Promise.all(
-          results.map(async (file) => {
-            try {
-              const thumbnail = await generateThumbnailForFile(file);
-              return thumbnail || '';
-            } catch (error) {
-              console.warn(`Failed to generate thumbnail for ${file.name}:`, error);
-              return '';
-            }
-          })
-        );
-
-        setThumbnails(thumbnails);
-        setIsGeneratingThumbnails(false);
-        setStatus(results.length === 1 
-          ? t('sanitize.completed', 'Sanitization completed successfully') 
-          : t('sanitize.completedMultiple', 'Sanitized {{count}} files successfully', { count: results.length })
-        );
       } else {
         setErrorMessage(t('sanitize.errorAllFilesFailed', 'All files failed to sanitize'));
       }
@@ -182,7 +216,7 @@ export const useSanitizeOperation = () => {
     } finally {
       setIsLoading(false);
     }
-  }, [t, createOperation, recordOperation, markOperationApplied, markOperationFailed, addFiles]);
+  }, [t, createOperation, recordOperation, sanitizeFile, processResults]);
 
   const resetResults = useCallback(() => {
     if (downloadUrl) {

frontend/src/types/fileContext.ts

@@ -7,7 +7,7 @@ import { PDFDocument, PDFPage, PageOperation } from './pageEditor';
 
 export type ModeType = 'viewer' | 'pageEditor' | 'fileEditor' | 'merge' | 'split' | 'compress' | 'ocr';
 
-export type OperationType = 'merge' | 'split' | 'compress' | 'add' | 'remove' | 'replace' | 'convert' | 'upload' | 'ocr';
+export type OperationType = 'merge' | 'split' | 'compress' | 'add' | 'remove' | 'replace' | 'convert' | 'upload' | 'ocr' | 'sanitize';
 
 export interface FileOperation {
   id: string;