All files / src/components/common/FilePreview/hooks usePages.tsx

78.57% Statements 11/14
100% Branches 0/0
66.66% Functions 2/3
78.57% Lines 11/14

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36      3x 2x 2x 2x   2x           2x   2x 1x 1x 1x     2x                          
import { useState } from "react";
import { OnDocumentLoadSuccess } from "react-pdf/dist/cjs/shared/types";
 
const usePages = () => {
  const [numPages, setNumPages] = useState<number>(0);
  const [pageNumber, setPageNumber] = useState<number>(1);
  const [pagesRendered, setPagesRendered] = useState<number>(0);
 
  const onDocumentLoadSuccess: OnDocumentLoadSuccess = (document) => {
    const { numPages } = document;
    setNumPages(numPages);
    setPagesRendered(0);
  };
 
  const pagesRenderedPlusOne = Math.min(pagesRendered + 1, numPages);
 
  const resetPages = () => {
    setNumPages(0);
    setPageNumber(1);
    setPagesRendered(0);
  };
 
  return {
    numPages,
    pageNumber,
    setPageNumber,
    onDocumentLoadSuccess,
    setPagesRendered,
    pagesRendered,
    pagesRenderedPlusOne,
    resetPages,
  };
};
 
export default usePages;