plot-plan-designer
Version:
Design Editor Tools with React.js + ant.design + fabric.js
82 lines (81 loc) • 4.93 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
import React, { useCallback, useEffect, useState } from 'react';
import { Document, Page } from 'react-pdf';
import { Button, Modal, InputNumber } from 'antd';
import '../../styles/core/common/pdf-processor.less';
import { getFileName } from './utils/utils';
const PDFProcessor = ({ file, onChange }) => {
const [numPages, setNumPages] = useState(undefined);
const [pageNumber, setPageNumber] = useState(1);
const [visible, setVisible] = useState(false);
const [currentPDF, setCurrentPDF] = useState(null);
const [scale, setScale] = useState(1);
const [dimension, setDimension] = useState({ w: 0, h: 0 });
useEffect(() => {
setVisible(true);
return () => {
setPageNumber(1);
setScale(1);
setDimension({ w: 0, h: 0 });
setCurrentPDF(null);
};
}, [file]);
const onDocumentLoadSuccess = useCallback((pdf) => {
setNumPages(pdf.numPages);
setCurrentPDF(pdf);
}, []);
const onSubmit = useCallback((pageNumber) => {
const exportPageAsImage = (pageNumber) => __awaiter(void 0, void 0, void 0, function* () {
const page = yield currentPDF.getPage(pageNumber);
const viewport = page.getViewport({ scale });
const canvas = document.createElement('canvas');
canvas.height = viewport.height;
canvas.width = viewport.width;
const context = canvas.getContext('2d');
yield page.render({ canvasContext: context, viewport }).promise;
// Set DPI to 300 for better image quality
canvas.toBlob((blob) => {
const { name } = file;
const fileName = getFileName(name);
// @ts-ignore
onChange(new File([blob], `${fileName}.png`, { type: 'image/png', lastModified: new Date() }));
}, 'image/png', 300);
});
exportPageAsImage(pageNumber).then(() => setVisible(false));
}, [currentPDF, scale]);
const onPageLoadSuccess = useCallback(({ originalWidth, originalHeight }) => {
setDimension({ w: originalWidth, h: originalHeight });
}, []);
// @ts-ignore
return (React.createElement("div", null, file && (React.createElement(Modal, { visible: visible, className: "pdf-processor", closable: false, footer: [
React.createElement(React.Fragment, null,
React.createElement("div", { className: "pdf-page-control" },
React.createElement(Button, { shape: "circle", icon: "caret-left", onClick: () => setPageNumber(pageNumber - 1), disabled: pageNumber === 1 }),
React.createElement(Button, { shape: "circle", icon: "caret-right", onClick: () => setPageNumber(pageNumber + 1), disabled: pageNumber === numPages }),
React.createElement("p", { key: "pdf-page", className: "pdf-page-number" },
"Page ",
pageNumber,
" of ",
numPages),
React.createElement("p", { key: "pdf-scale", className: "pdf-page-number", style: { marginLeft: '30px' } }, "Scale:"),
React.createElement(InputNumber, { min: 1, max: 10, defaultValue: scale, onChange: setScale, style: { paddingLeft: '10px', marginLeft: '5px' } }),
React.createElement("p", { key: "pdf-dimension", className: "pdf-page-number" },
"W: ",
Math.ceil(dimension.w) * scale,
" x H: ",
Math.ceil(dimension.h) * scale)),
React.createElement(Button, { key: "back", onClick: () => setVisible(false) }, "Cancel"),
React.createElement(Button, { key: "submit", onClick: () => onSubmit(pageNumber), style: { backgroundColor: '#08979c', color: '#fff' } }, "Use this page")),
] },
React.createElement(Document, { file: file, onLoadSuccess: onDocumentLoadSuccess, className: "pdf-doc-renderer" },
React.createElement(Page, { pageNumber: pageNumber, key: `pdf-${pageNumber}`, onLoadSuccess: onPageLoadSuccess, width: 1024, className: "pdf-page-renderer" }))))));
};
export default PDFProcessor;