Feature/v2/pageeditor improved (#4289)

# Description of Changes  --- ## Checklist ### General - [ ] I have read the [Contribution Guidelines](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/CONTRIBUTING.md) - [ ] I have read the [Stirling-PDF Developer Guide](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/DeveloperGuide.md) (if applicable) - [ ] I have read the [How to add new languages to Stirling-PDF](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/HowToAddNewLanguage.md) (if applicable) - [ ] I have performed a self-review of my own code - [ ] My changes generate no new warnings ### Documentation - [ ] I have updated relevant docs on [Stirling-PDF's doc repo](https://github.com/Stirling-Tools/Stirling-Tools.github.io/blob/main/docs/) (if functionality has heavily changed) - [ ] I have read the section [Add New Translation Tags](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/HowToAddNewLanguage.md#add-new-translation-tags) (for new translation tags only) ### UI Changes (if applicable) - [ ] Screenshots or videos demonstrating the UI changes are attached (e.g., as comments or direct attachments in the PR) ### Testing (if applicable) - [ ] I have tested my changes locally. Refer to the [Testing Guide](https://github.com/Stirling-Tools/Stirling-PDF/blob/main/devGuide/DeveloperGuide.md#6-testing) for more details. --------- Co-authored-by: Anthony Stirling <[email protected]>
2026-07-16 11:23:10 +02:00 · 2025-08-26 15:30:58 +01:00
parent 9b8091a630
commit ca423f9646
28 changed files with 3463 additions and 2327 deletions
@@ -0,0 +1,176 @@
+import { useMemo } from 'react';
+import { useFileState } from '../../../contexts/FileContext';
+import { PDFDocument, PDFPage } from '../../../types/pageEditor';
+
+export interface PageDocumentHook {
+  document: PDFDocument | null;
+  isVeryLargeDocument: boolean;
+  isLoading: boolean;
+}
+
+/**
+ * Hook for managing PDF document state and metadata in PageEditor
+ * Handles document merging, large document detection, and loading states
+ */
+export function usePageDocument(): PageDocumentHook {
+  const { state, selectors } = useFileState();
+  
+  // Prefer IDs + selectors to avoid array identity churn
+  const activeFileIds = state.files.ids;
+  const primaryFileId = activeFileIds[0] ?? null;
+  
+  // Stable signature for effects (prevents loops)
+  const filesSignature = selectors.getFilesSignature();
+  
+  // UI state
+  const globalProcessing = state.ui.isProcessing;
+  
+  // Get primary file record outside useMemo to track processedFile changes
+  const primaryFileRecord = primaryFileId ? selectors.getFileRecord(primaryFileId) : null;
+  const processedFilePages = primaryFileRecord?.processedFile?.pages;
+  const processedFileTotalPages = primaryFileRecord?.processedFile?.totalPages;
+
+  // Compute merged document with stable signature (prevents infinite loops)
+  const mergedPdfDocument = useMemo((): PDFDocument | null => {
+    if (activeFileIds.length === 0) return null;
+
+    const primaryFile = primaryFileId ? selectors.getFile(primaryFileId) : null;
+    
+    // If we have file IDs but no file record, something is wrong - return null to show loading
+    if (!primaryFileRecord) {
+      console.log('🎬 PageEditor: No primary file record found, showing loading');
+      return null;
+    }
+
+    const name =
+      activeFileIds.length === 1
+        ? (primaryFileRecord.name ?? 'document.pdf')
+        : activeFileIds
+            .map(id => (selectors.getFileRecord(id)?.name ?? 'file').replace(/\.pdf$/i, ''))
+            .join(' + ');
+
+    // Build page insertion map from files with insertion positions
+    const insertionMap = new Map<string, string[]>(); // insertAfterPageId -> fileIds
+    const originalFileIds: string[] = [];
+    
+    activeFileIds.forEach(fileId => {
+      const record = selectors.getFileRecord(fileId);
+      if (record?.insertAfterPageId !== undefined) {
+        if (!insertionMap.has(record.insertAfterPageId)) {
+          insertionMap.set(record.insertAfterPageId, []);
+        }
+        insertionMap.get(record.insertAfterPageId)!.push(fileId);
+      } else {
+        originalFileIds.push(fileId);
+      }
+    });
+    
+    // Build pages by interleaving original pages with insertions
+    let pages: PDFPage[] = [];
+    let totalPageCount = 0;
+    
+    // Helper function to create pages from a file
+    const createPagesFromFile = (fileId: string, startPageNumber: number): PDFPage[] => {
+      const fileRecord = selectors.getFileRecord(fileId);
+      if (!fileRecord) {
+        return [];
+      }
+      
+      const processedFile = fileRecord.processedFile;
+      let filePages: PDFPage[] = [];
+      
+      if (processedFile?.pages && processedFile.pages.length > 0) {
+        // Use fully processed pages with thumbnails
+        filePages = processedFile.pages.map((page, pageIndex) => ({
+          id: `${fileId}-${page.pageNumber}`,
+          pageNumber: startPageNumber + pageIndex,
+          thumbnail: page.thumbnail || null,
+          rotation: page.rotation || 0,
+          selected: false,
+          splitAfter: page.splitAfter || false,
+          originalPageNumber: page.originalPageNumber || page.pageNumber || pageIndex + 1,
+          originalFileId: fileId,
+        }));
+      } else if (processedFile?.totalPages) {
+        // Fallback: create pages without thumbnails but with correct count
+        filePages = Array.from({ length: processedFile.totalPages }, (_, pageIndex) => ({
+          id: `${fileId}-${pageIndex + 1}`,
+          pageNumber: startPageNumber + pageIndex,
+          originalPageNumber: pageIndex + 1,
+          originalFileId: fileId,
+          rotation: 0,
+          thumbnail: null,
+          selected: false,
+          splitAfter: false,
+        }));
+      }
+      
+      return filePages;
+    };
+
+    // Collect all pages from original files (without renumbering yet)
+    const originalFilePages: PDFPage[] = [];
+    originalFileIds.forEach(fileId => {
+      const filePages = createPagesFromFile(fileId, 1); // Temporary numbering
+      originalFilePages.push(...filePages);
+    });
+    
+    // Start with all original pages numbered sequentially  
+    pages = originalFilePages.map((page, index) => ({
+      ...page,
+      pageNumber: index + 1
+    }));
+    
+    // Process each insertion by finding the page ID and inserting after it
+    for (const [insertAfterPageId, fileIds] of insertionMap.entries()) {
+      const targetPageIndex = pages.findIndex(p => p.id === insertAfterPageId);
+      
+      if (targetPageIndex === -1) continue;
+      
+      // Collect all pages to insert
+      const allNewPages: PDFPage[] = [];
+      fileIds.forEach(fileId => {
+        const insertedPages = createPagesFromFile(fileId, 1);
+        allNewPages.push(...insertedPages);
+      });
+      
+      // Insert all new pages after the target page
+      pages.splice(targetPageIndex + 1, 0, ...allNewPages);
+      
+      // Renumber all pages after insertion
+      pages.forEach((page, index) => {
+        page.pageNumber = index + 1;
+      });
+    }
+    
+    totalPageCount = pages.length;
+
+    if (pages.length === 0) {
+      return null;
+    }
+
+    const mergedDoc: PDFDocument = {
+      id: activeFileIds.join('-'),
+      name,
+      file: primaryFile!,
+      pages,
+      totalPages: pages.length,
+    };
+
+    return mergedDoc;
+  }, [activeFileIds, primaryFileId, primaryFileRecord, processedFilePages, processedFileTotalPages, selectors, filesSignature]);
+
+  // Large document detection for smart loading
+  const isVeryLargeDocument = useMemo(() => {
+    return mergedPdfDocument ? mergedPdfDocument.totalPages > 2000 : false;
+  }, [mergedPdfDocument?.totalPages]);
+
+  // Loading state
+  const isLoading = globalProcessing && !mergedPdfDocument;
+
+  return {
+    document: mergedPdfDocument,
+    isVeryLargeDocument,
+    isLoading
+  };
+}