mirror of
https://github.com/iib0011/omni-tools.git
synced 2025-12-29 16:16:02 +00:00
created pdf to png tool
This commit is contained in:
@@ -1,62 +1,70 @@
|
||||
import { Box } from '@mui/material';
|
||||
import React, { useState } from 'react';
|
||||
import { useState } from 'react';
|
||||
import ToolContent from '@components/ToolContent';
|
||||
import ToolPdfInput from '@components/input/ToolPdfInput';
|
||||
import { ToolComponentProps } from '@tools/defineTool';
|
||||
import ToolTextInput from '@components/input/ToolTextInput';
|
||||
import ToolTextResult from '@components/result/ToolTextResult';
|
||||
import { GetGroupsType } from '@components/options/ToolOptions';
|
||||
import { CardExampleType } from '@components/examples/ToolExamples';
|
||||
import { main } from './service';
|
||||
import { InitialValuesType } from './types';
|
||||
import { convertPdfToPngImages } from './service';
|
||||
import ToolMultiFileResult from '@components/result/ToolMultiFileResult';
|
||||
|
||||
const initialValues: InitialValuesType = {
|
||||
// splitSeparator: '\n'
|
||||
type ImagePreview = {
|
||||
blob: Blob;
|
||||
url: string;
|
||||
filename: string;
|
||||
};
|
||||
|
||||
const exampleCards: CardExampleType<InitialValuesType>[] = [
|
||||
{
|
||||
title: 'Split a String',
|
||||
description: 'This example shows how to split a string into multiple lines',
|
||||
sampleText: 'Hello World,Hello World',
|
||||
sampleResult: `Hello World
|
||||
Hello World`,
|
||||
sampleOptions: {
|
||||
// splitSeparator: ','
|
||||
}
|
||||
}
|
||||
];
|
||||
export default function PdfToPng({
|
||||
title,
|
||||
longDescription
|
||||
}: ToolComponentProps) {
|
||||
const [input, setInput] = useState<string>('');
|
||||
const [result, setResult] = useState<string>('');
|
||||
export default function PdfToPng({ title }: ToolComponentProps) {
|
||||
const [input, setInput] = useState<File | null>(null);
|
||||
const [images, setImages] = useState<ImagePreview[]>([]);
|
||||
const [zipBlob, setZipBlob] = useState<File | null>(null);
|
||||
const [loading, setLoading] = useState(false);
|
||||
|
||||
const compute = (values: InitialValuesType, input: string) => {
|
||||
setResult(main(input, values));
|
||||
const compute = async (_: {}, file: File | null) => {
|
||||
if (!file) return;
|
||||
setLoading(true);
|
||||
setImages([]);
|
||||
setZipBlob(null);
|
||||
try {
|
||||
const { images, zipFile } = await convertPdfToPngImages(file);
|
||||
setImages(images);
|
||||
setZipBlob(zipFile);
|
||||
} catch (err) {
|
||||
console.error('Conversion failed:', err);
|
||||
} finally {
|
||||
setLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const getGroups: GetGroupsType<InitialValuesType> | null = ({
|
||||
values,
|
||||
updateField
|
||||
}) => [
|
||||
{
|
||||
title: 'Example Settings',
|
||||
component: <Box></Box>
|
||||
}
|
||||
];
|
||||
return (
|
||||
<ToolContent
|
||||
title={title}
|
||||
input={input}
|
||||
inputComponent={<ToolTextInput value={input} onChange={setInput} />}
|
||||
resultComponent={<ToolTextResult value={result} />}
|
||||
initialValues={initialValues}
|
||||
exampleCards={exampleCards}
|
||||
getGroups={getGroups}
|
||||
setInput={setInput}
|
||||
initialValues={{}}
|
||||
compute={compute}
|
||||
toolInfo={{ title: `What is a ${title}?`, description: longDescription }}
|
||||
inputComponent={
|
||||
<ToolPdfInput
|
||||
value={input}
|
||||
onChange={setInput}
|
||||
accept={['application/pdf']}
|
||||
title="Upload a PDF"
|
||||
/>
|
||||
}
|
||||
resultComponent={
|
||||
<ToolMultiFileResult
|
||||
title="Converted PNG Pages"
|
||||
value={images.map((img) => {
|
||||
return new File([img.blob], img.filename, { type: 'image/png' });
|
||||
})}
|
||||
zipFile={zipBlob}
|
||||
loading={loading}
|
||||
loadingText="Converting PDF pages"
|
||||
/>
|
||||
}
|
||||
getGroups={null}
|
||||
toolInfo={{
|
||||
title: 'Convert PDF pages into PNG images',
|
||||
description:
|
||||
'Upload your PDF and get each page rendered as a high-quality PNG. You can preview, download individually, or get all images in a ZIP.'
|
||||
}}
|
||||
/>
|
||||
);
|
||||
}
|
||||
|
||||
@@ -1,6 +0,0 @@
|
||||
import { expect, describe, it } from 'vitest';
|
||||
// import { main } from './service';
|
||||
//
|
||||
// describe('pdf-to-png', () => {
|
||||
//
|
||||
// })
|
||||
@@ -1,5 +1,51 @@
|
||||
import { InitialValuesType } from './types';
|
||||
import * as pdfjsLib from 'pdfjs-dist';
|
||||
import pdfjsWorker from 'pdfjs-dist/build/pdf.worker.min?url';
|
||||
import JSZip from 'jszip';
|
||||
|
||||
export function main(input: string, options: InitialValuesType): string {
|
||||
return input;
|
||||
pdfjsLib.GlobalWorkerOptions.workerSrc = pdfjsWorker;
|
||||
|
||||
type ImagePreview = {
|
||||
blob: Blob;
|
||||
url: string;
|
||||
filename: string;
|
||||
};
|
||||
|
||||
export async function convertPdfToPngImages(pdfFile: File): Promise<{
|
||||
images: ImagePreview[];
|
||||
zipFile: File;
|
||||
}> {
|
||||
const arrayBuffer = await pdfFile.arrayBuffer();
|
||||
const pdf = await pdfjsLib.getDocument({ data: arrayBuffer }).promise;
|
||||
const zip = new JSZip();
|
||||
const images: ImagePreview[] = [];
|
||||
|
||||
for (let i = 1; i <= pdf.numPages; i++) {
|
||||
const page = await pdf.getPage(i);
|
||||
const viewport = page.getViewport({ scale: 2 });
|
||||
|
||||
const canvas = document.createElement('canvas');
|
||||
const context = canvas.getContext('2d')!;
|
||||
canvas.width = viewport.width;
|
||||
canvas.height = viewport.height;
|
||||
|
||||
await page.render({ canvasContext: context, viewport }).promise;
|
||||
|
||||
const blob = await new Promise<Blob>((resolve) =>
|
||||
canvas.toBlob((b) => b && resolve(b), 'image/png')
|
||||
);
|
||||
|
||||
const filename = `page-${i}.png`;
|
||||
const url = URL.createObjectURL(blob);
|
||||
images.push({ blob, url, filename });
|
||||
zip.file(filename, blob);
|
||||
}
|
||||
|
||||
const zipBuffer = await zip.generateAsync({ type: 'arraybuffer' });
|
||||
const zipFile = new File(
|
||||
[zipBuffer],
|
||||
pdfFile.name.replace(/\.pdf$/i, '-pages.zip'),
|
||||
{ type: 'application/zip' }
|
||||
);
|
||||
|
||||
return { images, zipFile };
|
||||
}
|
||||
|
||||
@@ -1,3 +0,0 @@
|
||||
export type InitialValuesType = {
|
||||
// splitSeparator: string;
|
||||
};
|
||||
Reference in New Issue
Block a user