diff --git a/frontend/src/components/viewer/EmbedPdfViewer.tsx b/frontend/src/components/viewer/EmbedPdfViewer.tsx index f20589360..4daa0c554 100644 --- a/frontend/src/components/viewer/EmbedPdfViewer.tsx +++ b/frontend/src/components/viewer/EmbedPdfViewer.tsx @@ -13,7 +13,7 @@ import { useSignature } from '../../contexts/SignatureContext'; import { createStirlingFilesAndStubs } from '../../services/fileStubHelpers'; import NavigationWarningModal from '../shared/NavigationWarningModal'; import { isStirlingFile } from '../../types/fileContext'; -import ViewerRightRailButtons from './ViewerRightRailButtons'; +import { useViewerRightRailButtons } from './useViewerRightRailButtons'; export interface EmbedPdfViewerProps { sidebarsVisible: boolean; @@ -37,6 +37,9 @@ const EmbedPdfViewerContent = ({ const { isThumbnailSidebarVisible, toggleThumbnailSidebar, zoomActions, spreadActions, panActions: _panActions, rotationActions: _rotationActions, getScrollState, getZoomState, getSpreadState, getRotationState, isAnnotationMode, isAnnotationsVisible, exportActions } = useViewer(); + // Register viewer right-rail buttons + useViewerRightRailButtons(); + const scrollState = getScrollState(); const zoomState = getZoomState(); const spreadState = getSpreadState(); @@ -253,7 +256,6 @@ const EmbedPdfViewerContent = ({ overflow: 'hidden', contain: 'layout style paint' }}> - {/* Close Button - Only show in preview mode */} {onClose && previewFile && ( { +export function useViewerRightRailButtons() { const { t } = useTranslation(); const viewer = useViewer(); const [isPanning, setIsPanning] = useState(() => viewer.getPanState()?.isPanning ?? false); @@ -121,8 +121,4 @@ const ViewerRightRailButtons = () => { }, [t, viewer, isPanning]); useRightRailButtons(viewerButtons); - - return null; -}; - -export default ViewerRightRailButtons; +}