feat: split pdf

src/pages/tools/pdf/split-pdf/index.tsx

@@ -0,0 +1,180 @@
+import { Box, Typography } from '@mui/material';
+import React, { useEffect, useRef, useState } from 'react';
+import ToolFileInput from '@components/input/ToolFileInput';
+import ToolFileResult from '@components/result/ToolFileResult';
+import TextFieldWithDesc from '@components/options/TextFieldWithDesc';
+import ToolContent from '@components/ToolContent';
+import { ToolComponentProps } from '@tools/defineTool';
+import { parsePageRanges, splitPdf } from './service';
+import { CardExampleType } from '@components/examples/ToolExamples';
+import { PDFDocument } from 'pdf-lib';
+import { FormikProps } from 'formik';
+
+type InitialValuesType = {
+  pageRanges: string;
+};
+
+const initialValues: InitialValuesType = {
+  pageRanges: ''
+};
+
+const exampleCards: CardExampleType<InitialValuesType>[] = [
+  {
+    title: 'Extract Specific Pages',
+    description: 'Extract pages 1, 5, 6, 7, and 8 from a PDF document.',
+    sampleText: '',
+    sampleResult: '',
+    sampleOptions: {
+      pageRanges: '1,5-8'
+    }
+  },
+  {
+    title: 'Extract First and Last Pages',
+    description: 'Extract only the first and last pages from a PDF document.',
+    sampleText: '',
+    sampleResult: '',
+    sampleOptions: {
+      pageRanges: '1,10'
+    }
+  },
+  {
+    title: 'Extract a Range of Pages',
+    description: 'Extract a continuous range of pages from a PDF document.',
+    sampleText: '',
+    sampleResult: '',
+    sampleOptions: {
+      pageRanges: '3-7'
+    }
+  }
+];
+
+export default function SplitPdf({ title }: ToolComponentProps) {
+  const [input, setInput] = useState<File | null>(null);
+  const [result, setResult] = useState<File | null>(null);
+  const [isProcessing, setIsProcessing] = useState<boolean>(false);
+  const [totalPages, setTotalPages] = useState<number>(0);
+  const [pageRangePreview, setPageRangePreview] = useState<string>('');
+
+  // Get the total number of pages when a PDF is uploaded
+  useEffect(() => {
+    const getPdfInfo = async () => {
+      if (!input) {
+        setTotalPages(0);
+        return;
+      }
+
+      try {
+        const arrayBuffer = await input.arrayBuffer();
+        const pdf = await PDFDocument.load(arrayBuffer);
+        setTotalPages(pdf.getPageCount());
+      } catch (error) {
+        console.error('Error getting PDF info:', error);
+        setTotalPages(0);
+      }
+    };
+
+    getPdfInfo();
+  }, [input]);
+
+  const onValuesChange = (values: InitialValuesType) => {
+    const { pageRanges } = values;
+    if (!totalPages || !pageRanges?.trim()) {
+      setPageRangePreview('');
+      return;
+    }
+    try {
+      const count = parsePageRanges(pageRanges, totalPages).length;
+      setPageRangePreview(
+        `${count} page${count !== 1 ? 's' : ''} will be extracted`
+      );
+    } catch (error) {
+      setPageRangePreview('');
+    }
+  };
+
+  const compute = async (values: InitialValuesType, input: File | null) => {
+    if (!input) return;
+
+    try {
+      setIsProcessing(true);
+      const splitResult = await splitPdf(input, values.pageRanges);
+      setResult(splitResult);
+    } catch (error) {
+      throw new Error('Error splitting PDF:' + error);
+    } finally {
+      setIsProcessing(false);
+    }
+  };
+
+  return (
+    <ToolContent
+      title={title}
+      input={input}
+      setInput={setInput}
+      initialValues={initialValues}
+      compute={compute}
+      exampleCards={exampleCards}
+      inputComponent={
+        <ToolFileInput
+          value={input}
+          onChange={setInput}
+          accept={['application/pdf']}
+          title={'Input PDF'}
+        />
+      }
+      resultComponent={
+        <ToolFileResult
+          title={'Output PDF with selected pages'}
+          value={result}
+          extension={'pdf'}
+          loading={isProcessing}
+          loadingText={'Extracting pages'}
+        />
+      }
+      getGroups={({ values, updateField }) => [
+        {
+          title: 'Page Selection',
+          component: (
+            <Box>
+              {totalPages > 0 && (
+                <Typography variant="body2" sx={{ mb: 1 }}>
+                  PDF has {totalPages} page{totalPages !== 1 ? 's' : ''}
+                </Typography>
+              )}
+              <TextFieldWithDesc
+                value={values.pageRanges}
+                onOwnChange={(val) => {
+                  updateField('pageRanges', val);
+                }}
+                description={
+                  'Enter page numbers or ranges separated by commas (e.g., 1,3,5-7)'
+                }
+                placeholder={'e.g., 1,5-8'}
+              />
+              {pageRangePreview && (
+                <Typography
+                  variant="body2"
+                  sx={{ mt: 1, color: 'primary.main' }}
+                >
+                  {pageRangePreview}
+                </Typography>
+              )}
+            </Box>
+          )
+        }
+      ]}
+      onValuesChange={onValuesChange}
+      toolInfo={{
+        title: 'How to Use the Split PDF Tool',
+        description: `This tool allows you to extract specific pages from a PDF document. You can specify individual page numbers (e.g., 1,3,5) or page ranges (e.g., 2-6) or a combination of both (e.g., 1,3-5,8).
+
+Leave the page ranges field empty to include all pages from the PDF.
+
+Examples:
+- "1,5,9" extracts pages 1, 5, and 9
+- "1-5" extracts pages 1 through 5
+- "1,3-5,8-10" extracts pages 1, 3, 4, 5, 8, 9, and 10`
+      }}
+    />
+  );
+}