@orca-fe/pdf-viewer
Version:
PDF Viewer
870 lines (852 loc) • 44.9 kB
JavaScript
var _excluded = ["className", "pageGap", "maxZoom", "minZoom", "renderPageCover", "onZoomChange", "onPageScroll", "children", "emptyTips", "loadingTips", "title", "hideToolbar", "defaultTitle", "defaultZoom", "dropFile", "pdfJsParams", "onPageChange", "locale", "outputScale", "renderAllPages", "workerSrc"];
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == typeof i ? i : String(i); }
function _toPrimitive(t, r) { if ("object" != typeof t || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != typeof i) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == typeof h && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(typeof e + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }
function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
/* eslint-disable max-lines */
/* eslint-disable @typescript-eslint/no-explicit-any */
import { clamp, roundBy } from '@orca-fe/tools';
import { useCounter, useDebounceEffect, useDebounceFn, useEventListener, useMemoizedFn, useSetState } from 'ahooks';
import React, { useEffect, useImperativeHandle, useMemo, useRef, useState } from 'react';
import { useGetState, useSizeListener } from '@orca-fe/hooks';
import cn from 'classnames';
import { getDocument } from '@orca-fe/pdfjs-dist-browserify';
import * as pdfjsLib from '@orca-fe/pdfjs-dist-browserify';
// import * as pdfjsWorker from '@orca-fe/pdfjs-dist-browserify/build/pdf.worker';
import { ContextMenu } from '@orca-fe/pocket';
import { saveAs } from 'file-saver';
import { DownloadOutlined } from '@ant-design/icons';
import PDFViewerContext, { PDFToolbarContext } from "./context";
import PDFPage from "./PDFPage";
import { findSortedArr, PixelsPerInch } from "./utils";
import ZoomAndPageController from "./ZoomAndPageController";
import PDFToolbar from "./PDFToolbar";
import useStyle from "./PDFViewer.style";
import { LocaleContext, useLocale } from "./locale/context";
import zhCN from "./locale/zh_CN";
import useCollector from "./useCollector";
import { jsx as _jsx } from "react/jsx-runtime";
import { Fragment as _Fragment } from "react/jsx-runtime";
import { jsxs as _jsxs } from "react/jsx-runtime";
var ef = () => undefined;
var round001 = roundBy(0.001);
var DefaultLoadingTips = () => {
var _useLocale = useLocale(zhCN),
_useLocale2 = _slicedToArray(_useLocale, 1),
l = _useLocale2[0];
return /*#__PURE__*/_jsx("div", {
className: "pdf-viewer-default-loading-tips",
children: l.loadingTips
});
};
var DefaultEmptyTips = () => {
var _useLocale3 = useLocale(zhCN),
_useLocale4 = _slicedToArray(_useLocale3, 1),
l = _useLocale4[0];
return /*#__PURE__*/_jsx("div", {
className: "pdf-viewer-default-empty-tips",
children: l.loadTips
});
};
// 使用全局变量
var isWorkerLoaded = false;
var PDFViewer = /*#__PURE__*/React.forwardRef((props, pRef) => {
var _props$className = props.className,
className = _props$className === void 0 ? '' : _props$className,
_props$pageGap = props.pageGap,
pageGap = _props$pageGap === void 0 ? 24 : _props$pageGap,
_props$maxZoom = props.maxZoom,
maxZoom = _props$maxZoom === void 0 ? 3 : _props$maxZoom,
_props$minZoom = props.minZoom,
minZoom = _props$minZoom === void 0 ? -4 : _props$minZoom,
_props$renderPageCove = props.renderPageCover,
renderPageCover = _props$renderPageCove === void 0 ? ef : _props$renderPageCove,
_props$onZoomChange = props.onZoomChange,
onZoomChange = _props$onZoomChange === void 0 ? ef : _props$onZoomChange,
onPageScroll = props.onPageScroll,
children = props.children,
_props$emptyTips = props.emptyTips,
emptyTips = _props$emptyTips === void 0 ? /*#__PURE__*/_jsx(DefaultEmptyTips, {}) : _props$emptyTips,
_props$loadingTips = props.loadingTips,
loadingTips = _props$loadingTips === void 0 ? /*#__PURE__*/_jsx(DefaultLoadingTips, {}) : _props$loadingTips,
_title = props.title,
hideToolbar = props.hideToolbar,
defaultTitle = props.defaultTitle,
_props$defaultZoom = props.defaultZoom,
defaultZoom = _props$defaultZoom === void 0 ? 'autoWidth' : _props$defaultZoom,
dropFile = props.dropFile,
pdfJsParams = props.pdfJsParams,
_props$onPageChange = props.onPageChange,
onPageChange = _props$onPageChange === void 0 ? ef : _props$onPageChange,
locale = props.locale,
outputScale = props.outputScale,
_props$renderAllPages = props.renderAllPages,
renderAllPages = _props$renderAllPages === void 0 ? false : _props$renderAllPages,
workerSrc = props.workerSrc,
otherProps = _objectWithoutProperties(props, _excluded);
useEffect(() => {
if (!isWorkerLoaded) {
pdfjsLib.GlobalWorkerOptions.workerSrc = workerSrc !== null && workerSrc !== void 0 ? workerSrc : '/pdf.worker.min.js';
isWorkerLoaded = true;
}
// if (!window['pdfjsWorker']) {
// window['pdfjsWorker'] = pdfjsWorker;
// }
}, []);
var styles = useStyle();
// 最外层的根节点
var rootRef = useRef(null);
// 包裹页面部分的 body（带滚动条）
var _useState = useState(null),
_useState2 = _slicedToArray(_useState, 2),
bodyRef = _useState2[0],
setBodyRef = _useState2[1];
// 页面容器（随页面长度）
var pageContainerRef = useRef(null);
// 页面缓存（已经渲染过的页面进行缓存）
var pageCacheRef = useRef({});
var _useGetState = useGetState(0),
_useGetState2 = _slicedToArray(_useGetState, 3),
current = _useGetState2[0],
setCurrent = _useGetState2[1],
getCurrent = _useGetState2[2];
var _useGetState3 = useGetState(typeof defaultZoom === 'number' ? defaultZoom : 0),
_useGetState4 = _slicedToArray(_useGetState3, 3),
zoom = _useGetState4[0],
setZoom = _useGetState4[1],
getZoom = _useGetState4[2];
var _useState3 = useState(defaultTitle),
_useState4 = _slicedToArray(_useState3, 2),
__title = _useState4[0],
setTitle = _useState4[1];
var title = _title !== null && _title !== void 0 ? _title : __title;
var scale = Math.pow(2, zoom);
var _useState5 = useState({
zooming: false,
contextMenu: []
}),
_useState6 = _slicedToArray(_useState5, 1),
_this = _useState6[0];
useEffect(() => {
onPageChange(current);
}, [current]);
var _useState7 = useState(0),
_useState8 = _slicedToArray(_useState7, 2),
setForceUpdateCount = _useState8[1];
var forceUpdate = useMemoizedFn(() => {
setForceUpdateCount(count => count + 1);
});
useDebounceEffect(() => {
_this.zooming = false;
}, [zoom], {
wait: 500
});
var _useState9 = useState([0, 0]),
_useState10 = _slicedToArray(_useState9, 2),
renderRange = _useState10[0],
setRenderRange = _useState10[1];
var _useGetState5 = useGetState([]),
_useGetState6 = _slicedToArray(_useGetState5, 3),
pages = _useGetState6[0],
setPages = _useGetState6[1],
getPages = _useGetState6[2];
var _useState11 = useState(false),
_useState12 = _slicedToArray(_useState11, 2),
loading = _useState12[0],
setLoading = _useState12[1];
var _useCounter = useCounter(0),
_useCounter2 = _slicedToArray(_useCounter, 2),
pluginLoading = _useCounter2[0],
_useCounter2$ = _useCounter2[1],
pluginLoad = _useCounter2$.inc,
pluginLoaded = _useCounter2$.dec;
var setZoomWithScrollLock = useMemoizedFn(newZoom => {
var zoom = getZoom();
if (newZoom === zoom) return;
var dom = pageContainerRef.current;
if (dom) {
var scrollTop = dom.scrollTop,
scrollLeft = dom.scrollLeft;
var zoomDiff = newZoom - zoom;
var newScrollLeft = scrollLeft * Math.pow(2, zoomDiff);
var newScrollTop = scrollTop * Math.pow(2, zoomDiff);
setZoom(newZoom);
var pageDom = pageContainerRef.current;
if (pageDom) {
pageDom.style.setProperty('--scale-factor', `${Math.pow(2, newZoom) * PixelsPerInch.PDF_TO_CSS_UNITS}`);
}
dom.scrollTop = newScrollTop;
dom.scrollLeft = newScrollLeft;
onZoomChange(newZoom);
}
});
// 获取每一页的 viewport 信息
var viewports = useMemo(() => pages.map(page => {
var viewport = page.getViewport({
scale: 1
});
return viewport;
}), [pages]);
// 根据 viewport 信息生成每一页的实际位置信息
var _useMemo = useMemo(() => {
var top = 0;
var maxWidth = 0;
var pageMaxWidth = 0;
var pageMaxHeight = 0;
var topArr = viewports.map(({
height: _height,
width: _width
}) => {
var width = _width * PixelsPerInch.PDF_TO_CSS_UNITS;
var height = _height * PixelsPerInch.PDF_TO_CSS_UNITS;
var _top = top;
top += Math.floor(height + pageGap) * scale;
maxWidth = Math.max(width * scale, maxWidth);
pageMaxWidth = Math.max(width, pageMaxWidth);
pageMaxHeight = Math.max(height, pageMaxHeight);
return _top;
});
return {
topArr,
maxWidth,
pageMaxWidth,
pageMaxHeight
};
}, [viewports, pageGap, scale]),
pageTopArr = _useMemo.topArr,
maxWidth = _useMemo.maxWidth,
pageMaxWidth = _useMemo.pageMaxWidth,
pageMaxHeight = _useMemo.pageMaxHeight;
var _useState13 = useState(typeof defaultZoom === 'number' ? false : defaultZoom),
_useState14 = _slicedToArray(_useState13, 2),
zoomMode = _useState14[0],
setZoomMode = _useState14[1];
var autoZoomDebounce = useDebounceFn(() => {
var newZoom = zoom;
if (zoomMode && _this.size && maxWidth && pageMaxHeight) {
if (zoomMode === 'autoWidth') {
// 调整缩放级别，使其与容器宽度匹配
newZoom = Math.log2((_this.size.width - 32) / pageMaxWidth);
} else if (zoomMode === 'autoHeight') {
newZoom = Math.log2((_this.size.height - 32) / pageMaxHeight);
}
}
newZoom = round001(clamp(newZoom, -4, 3));
if (newZoom !== zoom) {
setZoomWithScrollLock(newZoom);
}
}, {
wait: 250
});
useEffect(() => {
autoZoomDebounce.run();
}, [zoomMode, pageMaxHeight, pageMaxWidth]);
// 自动调整缩放级别
useSizeListener(size => {
if (size.width === 0 || size.height === 0) {
autoZoomDebounce.cancel();
return;
}
_this.size = size;
autoZoomDebounce.run();
}, rootRef);
// 翻頁
var changePage = useMemoizedFn((page, anim = false) => {
var dom = pageContainerRef.current;
if (dom) {
var _viewports$page;
var top = pageTopArr[page];
var height = ((_viewports$page = viewports[page]) === null || _viewports$page === void 0 ? void 0 : _viewports$page.height) * scale * PixelsPerInch.PDF_TO_CSS_UNITS;
var containerHeight = dom.getBoundingClientRect().height;
dom.scrollTo({
top: Math.min(top, top + 0.5 * (height - containerHeight)),
behavior: anim ? 'smooth' : 'auto'
});
}
});
var getPDFInstance = useMemoizedFn(() => _this.pdfDoc);
var getFileSource = useMemoizedFn(() => _this.file);
var close = useMemoizedFn( /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee() {
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
if (!_this.pdfDoc) {
_context.next = 12;
break;
}
_context.prev = 1;
_this.pdfLoadingKey = undefined;
_context.next = 5;
return _this.pdfDoc.destroy();
case 5:
setLoading(false);
setTitle(undefined);
_context.next = 12;
break;
case 9:
_context.prev = 9;
_context.t0 = _context["catch"](1);
console.error('pdfDoc destory failed');
case 12:
_this.file = undefined;
_this.pdfDoc = undefined;
setPages([]);
case 15:
case "end":
return _context.stop();
}
}, _callee, null, [[1, 9]]);
})));
var load = useMemoizedFn( /*#__PURE__*/function () {
var _ref2 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(file, _options = {}) {
var options, _ref3, title, _ref3$resetScrollTop, resetScrollTop, key, pdfContent, pdfJsGetDocumentParams, pdfDoc, pageLength, allPages, dom;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
options = typeof _options === 'string' ? {
title: _options
} : _options;
_ref3 = options || {}, title = _ref3.title, _ref3$resetScrollTop = _ref3.resetScrollTop, resetScrollTop = _ref3$resetScrollTop === void 0 ? true : _ref3$resetScrollTop;
key = `${Date.now()}_${Math.random()}`;
_this.pdfLoadingKey = key;
setLoading(true);
_context3.prev = 5;
pdfContent = file;
if (!(pdfContent instanceof Promise)) {
_context3.next = 11;
break;
}
_context3.next = 10;
return pdfContent;
case 10:
pdfContent = _context3.sent;
case 11:
if (!(pdfContent instanceof Blob)) {
_context3.next = 15;
break;
}
_context3.next = 14;
return pdfContent.arrayBuffer();
case 14:
pdfContent = _context3.sent;
case 15:
if (!(key !== _this.pdfLoadingKey)) {
_context3.next = 17;
break;
}
return _context3.abrupt("return");
case 17:
pdfJsGetDocumentParams = _objectSpread({
enableXfa: true
}, pdfJsParams);
if (typeof pdfContent === 'string' || pdfContent instanceof URL) {
pdfJsGetDocumentParams.url = pdfContent;
} else {
pdfJsGetDocumentParams.data = pdfContent;
}
_context3.next = 21;
return getDocument(pdfJsGetDocumentParams).promise;
case 21:
pdfDoc = _context3.sent;
if (!pdfDoc) {
_context3.next = 35;
break;
}
_this.pdfDoc = pdfDoc;
pageLength = pdfDoc.numPages;
_context3.next = 27;
return Promise.all(new Array(pageLength).fill(0).map( /*#__PURE__*/function () {
var _ref4 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2(_, index) {
var pageNum, page;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
pageNum = index + 1;
_context2.next = 3;
return pdfDoc.getPage(pageNum);
case 3:
page = _context2.sent;
return _context2.abrupt("return", page);
case 5:
case "end":
return _context2.stop();
}
}, _callee2);
}));
return function (_x2, _x3) {
return _ref4.apply(this, arguments);
};
}()));
case 27:
allPages = _context3.sent;
if (!(key !== _this.pdfLoadingKey)) {
_context3.next = 30;
break;
}
return _context3.abrupt("return");
case 30:
setPages(allPages);
_this.file = pdfContent;
if (title != null) {
setTitle(title);
}
dom = pageContainerRef.current;
if (dom && resetScrollTop) {
dom.scrollTop = 0;
}
case 35:
_context3.prev = 35;
if (_this.pdfLoadingKey === key) setLoading(false);
return _context3.finish(35);
case 38:
case "end":
return _context3.stop();
}
}, _callee3, null, [[5,, 35, 38]]);
}));
return function (_x) {
return _ref2.apply(this, arguments);
};
}());
var scrollTo = useMemoizedFn((...args) => {
var dom = pageContainerRef.current;
if (dom) dom.scrollTo(...args);
});
var getCurrentPage = useMemoizedFn(() => getCurrent());
var getPageCount = useMemoizedFn(() => getPages().length);
var getRoot = useMemoizedFn(() => rootRef.current);
// 獲取頁面的圖片
var getPageBlob = useMemoizedFn( /*#__PURE__*/function () {
var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee4(index, options = {}) {
var _options$scale, scale, _options$outputScale, outputScale, pages, canvas, _page, viewport, context, task, blob;
return _regeneratorRuntime().wrap(function _callee4$(_context4) {
while (1) switch (_context4.prev = _context4.next) {
case 0:
_options$scale = options.scale, scale = _options$scale === void 0 ? 1 : _options$scale, _options$outputScale = options.outputScale, outputScale = _options$outputScale === void 0 ? 2 : _options$outputScale;
pages = getPages();
if (!(index < 0 || index >= pages.length)) {
_context4.next = 4;
break;
}
return _context4.abrupt("return", null);
case 4:
canvas = document.createElement('canvas');
document.body.append(canvas);
_context4.prev = 6;
canvas.style.display = 'none';
_page = pages[index];
viewport = _page.getViewport({
scale
});
context = canvas.getContext('2d');
if (!context) {
_context4.next = 23;
break;
}
canvas.width = Math.floor(viewport.width * outputScale);
canvas.height = Math.floor(viewport.height * outputScale);
canvas.style.width = `${Math.floor(viewport.width)}px`;
canvas.style.height = `${Math.floor(viewport.height)}px`;
task = _page.render({
canvasContext: context,
viewport,
transform: [outputScale, 0, 0, outputScale, 0, 0]
});
_context4.next = 19;
return task.promise;
case 19:
_context4.next = 21;
return new Promise(resolve => {
canvas.toBlob(blob => {
resolve(blob);
});
});
case 21:
blob = _context4.sent;
return _context4.abrupt("return", blob);
case 23:
_context4.next = 28;
break;
case 25:
_context4.prev = 25;
_context4.t0 = _context4["catch"](6);
console.error(_context4.t0);
case 28:
_context4.prev = 28;
document.body.removeChild(canvas);
return _context4.finish(28);
case 31:
return _context4.abrupt("return", null);
case 32:
case "end":
return _context4.stop();
}
}, _callee4, null, [[6, 25, 28, 31]]);
}));
return function (_x4) {
return _ref5.apply(this, arguments);
};
}());
var updateRenderRange = useMemoizedFn(() => {
var dom = pageContainerRef.current;
if (dom) {
var top = dom.scrollTop;
var height = dom.clientHeight;
var currentPage = Math.max(findSortedArr(pageTopArr, top + 0.5 * height) - 1, 0);
var startPage = findSortedArr(pageTopArr, top - 50);
startPage = Math.max(0, startPage - 1);
var pageBottom = pageTopArr[startPage];
var endPage = startPage;
while (viewports[endPage] && pageBottom + viewports[endPage].height * scale <= top + height + 50) {
pageBottom += (viewports[endPage].height + pageGap) * scale;
endPage += 1;
}
if (current !== currentPage) {
setCurrent(currentPage);
}
if (renderRange[0] !== startPage || renderRange[1] !== endPage) {
setRenderRange([startPage, endPage]);
}
}
});
var updateRenderRangeDebounce = useDebounceFn(() => {
_this.mousePositionBeforeWheel = undefined;
updateRenderRange();
}, {
wait: 280
});
var downloadPdf = useMemoizedFn( /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee5() {
var doc, data, blob, fileTitle, file;
return _regeneratorRuntime().wrap(function _callee5$(_context5) {
while (1) switch (_context5.prev = _context5.next) {
case 0:
doc = _this.pdfDoc;
if (!doc) {
_context5.next = 10;
break;
}
_context5.next = 4;
return doc.getData();
case 4:
data = _context5.sent;
blob = new Blob([data.buffer], {
type: 'application/pdf'
});
fileTitle = typeof title === 'string' ? title : 'download.pdf';
if (!fileTitle.endsWith('.pdf')) fileTitle += '.pdf';
file = new File([blob], fileTitle, {
type: 'application/pdf'
});
saveAs(file, fileTitle);
case 10:
case "end":
return _context5.stop();
}
}, _callee5);
})));
// 渲染完成的回调，缓存 page
var onRenderFinish = pageIndex => {
pageCacheRef.current[pageIndex] = true;
};
// 监听滚动事件，并更新需要展示的页面范围（虚拟列表）
useEventListener('scroll', ev => {
if (_this.zooming) {
// 如果是因为正在缩放导致的滚动，则需要添加防抖
updateRenderRangeDebounce.run();
} else {
updateRenderRangeDebounce.cancel();
updateRenderRange();
}
}, {
target: pageContainerRef,
passive: true
});
useEffect(() => {
updateRenderRangeDebounce.run();
}, [zoom]);
// 監聽滾輪事件，縮放頁面
useEventListener('wheel', ev => {
var zoom = getZoom();
var dom = pageContainerRef.current;
if (ev.ctrlKey && dom) {
var _dom$getBoundingClien = dom.getBoundingClientRect(),
left = _dom$getBoundingClien.left,
top = _dom$getBoundingClien.top,
width = _dom$getBoundingClien.width;
var clientX = ev.clientX,
clientY = ev.clientY;
var x = clientX - left;
var y = clientY - top;
if (!_this.mousePositionBeforeWheel) {
_this.mousePositionBeforeWheel = {
x: x + dom.scrollLeft - 0.5 * (maxWidth < width ? width - maxWidth : 0),
y: y + dom.scrollTop,
zoom
};
}
var newZoom = round001(clamp(zoom - clamp(ev.deltaY, -40, 40) * 0.01, minZoom, maxZoom));
if (newZoom === zoom) return;
_this.zooming = true;
setZoomMode(false);
setZoom(newZoom);
if (_this.mousePositionBeforeWheel) {
// 更新滾動條高度
var _this$mousePositionBe = _this.mousePositionBeforeWheel,
fullScrollLeft = _this$mousePositionBe.x,
fullScrollTop = _this$mousePositionBe.y,
originZoom = _this$mousePositionBe.zoom;
var zoomDiff = newZoom - originZoom;
var newScrollLeft = fullScrollLeft * Math.pow(2, zoomDiff) - x;
var newScrollTop = fullScrollTop * Math.pow(2, zoomDiff) - y;
var pageDom = pageContainerRef.current;
if (pageDom) {
pageDom.style.setProperty('--scale-factor', `${Math.pow(2, newZoom) * PixelsPerInch.PDF_TO_CSS_UNITS}`);
}
dom.scrollTop = newScrollTop;
dom.scrollLeft = newScrollLeft;
}
onZoomChange(newZoom);
} else {
_this.mousePositionBeforeWheel = undefined;
}
}, {
target: pageContainerRef,
passive: true
});
// 鼠标发生移动，则清空 position 信息，避免后续缩放时出现异常
useEventListener('mousemove', () => {
_this.mousePositionBeforeWheel = undefined;
}, {
target: pageContainerRef
});
// 禁用默认的滚轮缩放（缩放页面）
useEventListener('wheel', ev => {
if (ev.ctrlKey) {
ev.preventDefault();
}
}, {
target: pageContainerRef
});
// 拖拽并打开文件
useEventListener('dragenter', ev => {
ev.currentTarget.setAttribute('over', '1');
}, {
target: bodyRef
});
useEventListener('dragleave', ev => {
ev.currentTarget.removeAttribute('over');
}, {
target: bodyRef
});
useEventListener('dragover', ev => {
if (dropFile) {
ev.preventDefault();
ev.stopPropagation();
}
}, {
target: bodyRef
});
useEventListener('drop', ev => {
ev.currentTarget.removeAttribute('over');
if (dropFile) {
ev.preventDefault();
ev.stopPropagation();
if (ev.dataTransfer) {
var files = ev.dataTransfer.files;
if (files.length > 0) {
var _files = _slicedToArray(files, 1),
file = _files[0];
if (file.name.endsWith('.pdf')) load(file);
}
}
}
}, {
target: bodyRef
});
/* 工具栏 */
var _useState15 = useState(null),
_useState16 = _slicedToArray(_useState15, 2),
toolbarLeftDom = _useState16[0],
setToolbarLeftDom = _useState16[1];
var _useState17 = useState(null),
_useState18 = _slicedToArray(_useState17, 2),
toolbarRightDom = _useState18[0],
setToolbarRightDom = _useState18[1];
var _useState19 = useState([]),
_useState20 = _slicedToArray(_useState19, 2),
_centerToolbarIds = _useState20[0],
setCenterToolbarIds = _useState20[1];
var centerToolbarIds = useMemo(() => _centerToolbarIds.map(([id]) => id), [_centerToolbarIds]);
var addCenterToolbarId = useMemoizedFn((id, order = 0) => {
// 插入 toolbar id 并排序
setCenterToolbarIds(ids => ids.filter(([_id]) => _id !== id).concat([[id, order]]).sort((a, b) => a[1] - b[1]));
});
var removeCenterToolbarId = useMemoizedFn(id => {
setCenterToolbarIds(ids => ids.filter(([_id]) => _id !== id));
});
var pdfViewerHandle = useMemo(() => ({
load,
close,
setZoom: setZoomWithScrollLock,
getZoom,
changePage,
getPageBlob,
getCurrentPage,
getPageCount,
scrollTo: scrollTo,
setTitle,
getRoot,
getFileSource,
getPDFInstance,
pluginLoad,
pluginLoaded
}), []);
useImperativeHandle(pRef, () => pdfViewerHandle, []);
var _useState21 = useState([]),
_useState22 = _slicedToArray(_useState21, 2),
pageCoverRefs = _useState22[0],
setPageCoverRefs = _useState22[1];
useEffect(() => {
setPageCoverRefs(l => l.slice());
}, [renderRange[0], renderRange[1]]);
// 内部共享状态
var _useSetState = useSetState(() => ({
drawingPluginName: ''
})),
_useSetState2 = _slicedToArray(_useSetState, 2),
internalState = _useSetState2[0],
setInternalState = _useSetState2[1];
var menuCollector = useCollector();
var mergedLocale = useLocale(zhCN, locale)[0];
return /*#__PURE__*/_jsx(LocaleContext.Provider, {
value: mergedLocale,
children: /*#__PURE__*/_jsx(PDFViewerContext.Provider, {
value: useMemo(() => ({
loading,
pluginLoading,
pages,
viewports,
zoom,
current,
changePage,
forceUpdate,
pageCoverRefs,
pdfViewer: pdfViewerHandle,
internalState,
setInternalState,
bodyElement: bodyRef,
onMenuCollect: menuCollector.on,
offMenuCollect: menuCollector.off
}), [loading, pluginLoading, pages, viewports, zoom, current, pageCoverRefs, internalState, bodyRef]),
children: /*#__PURE__*/_jsx(PDFToolbarContext.Provider, {
value: useMemo(() => ({
toolbarRightDom,
toolbarLeftDom,
removeCenterToolbarId,
centerToolbarIds,
addCenterToolbarId
}), [toolbarRightDom, toolbarLeftDom, centerToolbarIds]),
children: /*#__PURE__*/_jsxs("div", _objectSpread(_objectSpread({
ref: rootRef,
className: `${styles.root} ${className}`
}, otherProps), {}, {
children: [/*#__PURE__*/_jsx(PDFToolbar, {
hide: hideToolbar,
title: loading ? '' : title,
className: styles.toolbar,
leftRef: dom => {
setToolbarLeftDom(dom);
},
centerIds: centerToolbarIds,
rightRef: dom => {
setToolbarRightDom(dom);
}
}), /*#__PURE__*/_jsxs("div", {
ref: setBodyRef,
className: cn(styles.pagesOuter, {
[styles.droppable]: dropFile
}),
children: [/*#__PURE__*/_jsxs(ContextMenu, {
ref: pageContainerRef,
className: styles.pages,
onScroll: onPageScroll,
mainMenuMinWidth: 200,
data: () => [{
key: 'download',
icon: /*#__PURE__*/_jsx(DownloadOutlined, {}),
text: mergedLocale.downloadCurrentFile || '下载当前文件',
disabled: pages.length <= 0,
onClick() {
downloadPdf();
}
}].concat(menuCollector.collect().sort((a, b) => (a.order || 0) - (b.order || 0))),
style: {
'--scale-factor': scale * PixelsPerInch.PDF_TO_CSS_UNITS,
'--scale-factor-origin': scale,
'--pdf-viewer-page-scale': scale * PixelsPerInch.PDF_TO_CSS_UNITS
},
children: [viewports.length === 0 && !loading && !pluginLoading && emptyTips, viewports.map((viewport, pageIndex) => {
var shouldRender = pageIndex >= renderRange[0] && pageIndex <= renderRange[1] || renderAllPages;
var width = `calc(var(--scale-factor) * ${Math.floor(viewport.width)}px)`;
var height = `calc(var(--scale-factor) * ${Math.floor(viewport.height)}px)`;
var gap = `calc(var(--scale-factor) * ${pageGap}px)`;
// 已渲染好的页面，不应该卸载
if (pageCacheRef.current[pageIndex]) {
shouldRender = true;
}
return /*#__PURE__*/_jsx("div", {
className: styles.pageContainer,
style: {
width,
height,
marginBottom: gap
},
children: shouldRender && /*#__PURE__*/_jsxs(_Fragment, {
children: [/*#__PURE__*/_jsx(PDFPage, {
className: styles.page,
outputScale: outputScale,
index: pageIndex,
zoom: zoom,
render: shouldRender,
onRenderFinish: onRenderFinish
}), /*#__PURE__*/_jsx("div", {
ref: node => pageCoverRefs[pageIndex] = node,
className: styles.pageCover
}), /*#__PURE__*/_jsx("div", {
className: styles.pageCover,
children: renderPageCover(pageIndex, {
viewport,
zoom
})
})]
})
}, pageIndex);
})]
}), (loading || !!pluginLoading) && loadingTips, children]
}), pages.length > 0 && /*#__PURE__*/_jsx(ZoomAndPageController, {
className: styles.pageController,
max: Math.pow(2, maxZoom),
min: Math.pow(2, minZoom),
zoomMode: zoomMode,
onZoomModeChange: setZoomMode
})]
}))
})
})
});
});
PDFViewer.displayName = 'PDFViewer';
export var usePdfViewerRef = () => useRef(null);
export default PDFViewer;