feat: Add document download link component, refactor download URL resolution with expiration checks

frontend/src/documents/components/DocumentDownloadLink.tsx

@@ -0,0 +1,40 @@
+import React from 'react';
+import { DownloadIcon } from '../../components/icons';
+import { resolveDocumentDownloadHref } from '../documentActions';
+import type { Document } from '../../types/documents';
+
+interface DocumentDownloadLinkProps extends React.AnchorHTMLAttributes<HTMLAnchorElement> {
+    document?: Document | null;
+    children?: React.ReactNode;
+}
+
+const DocumentDownloadLink: React.FC<DocumentDownloadLinkProps> = ({
+    document,
+    children,
+    className = 'icon-button',
+    title = 'Download document',
+    'aria-label': ariaLabel = 'Download document',
+    ...rest
+}) => {
+    const downloadUrl = resolveDocumentDownloadHref(document);
+
+    if (!downloadUrl) {
+        return null;
+    }
+
+    return (
+        <a
+            href={downloadUrl}
+            target="_blank"
+            rel="noopener noreferrer"
+            className={className}
+            title={title}
+            aria-label={ariaLabel}
+            {...rest}
+        >
+            {children || <DownloadIcon />}
+        </a>
+    );
+};
+
+export default DocumentDownloadLink;

frontend/src/documents/documentActions.ts

@@ -8,15 +8,18 @@ import type { Document } from '../types/documents';
 
 const asyncFalse = async () => false;
 
-const resolveDocumentDownloadHref = (document?: Document | null): string | null => {
+export const resolveDocumentDownloadHref = (document?: Document | null): string | null => {
   if (!document) {
     return null;
   }
-  const downloadUrl = (document.current_version as { download?: { url: string } | null } | null)?.download?.url;
-  if (!downloadUrl) {
+  const download = document.current_version?.download;
+  if (!download?.url) {
     return null;
   }
-  return downloadUrl;
+  if (download.expires_at && download.expires_at <= Date.now()) {
+    return null;
+  }
+  return download.url;
 };
 
 const hasDocumentTextContentAsset = (document?: Document | null, getDocumentAsset?: GetDocumentAsset | null): boolean => {

frontend/src/styles/index.css

@@ -2,7 +2,7 @@
 @import './base/controls.css';
 @import './layout/panel-header-controls.css';
 @import './components/resize-handle.css';
-@import './viewer/preview-zoom.css';
+@import '../viewer/preview-zoom.css';
 @import './base/iconography.css';
 @import '../documents/styles/controls.css';
 @import './layout/structure.css';
@@ -21,4 +21,4 @@
 @import '../login/login.css';
 @import '../documents/styles/shared-snippets.css';
 @import './uploads/overlay.css';
-@import './base/text-button.css';
+@import './base/text-button.css';

frontend/src/viewer/DocumentViewerLayout.tsx

@@ -2,6 +2,7 @@ import { useCallback, useMemo, useRef } from 'react';
 import type { JSX } from 'react';
 import DocumentInfoPanel from './components/DocumentInfoPanel';
 import UnifiedDocumentViewer from './UnifiedDocumentViewer';
+import { resolveDocumentDownloadHref } from '../documents/documentActions';
 
 import type { Document } from '../types/documents';
 
@@ -56,13 +57,13 @@ const DocumentViewerLayout = ({
 
   const renderViewportPane = useCallback(() => (
     <div className="document-viewer__viewport" ref={viewportRef}>
-      {!document?.current_version?.download?.url ? (
+      {!resolveDocumentDownloadHref(document) ? (
         <div className="document-viewer__message">{previewLoadingMessage}</div>
       ) : (
         previewContent
       )}
     </div>
-  ), [previewContent, document?.current_version?.download?.url, previewLoadingMessage, viewportRef]);
+  ), [previewContent, document, previewLoadingMessage, viewportRef]);
 
   const viewportPane = renderViewportPane();
 

frontend/src/viewer/DocumentViewerPanel.tsx

@@ -1,12 +1,12 @@
 import React, {
   useCallback,
+  useEffect,
   useMemo,
   useRef,
 } from 'react';
 import type { ReactNode } from 'react';
 import { useNavigate } from 'react-router-dom';
 import {
-  DownloadIcon,
   CloseIcon,
   IconZoomInArea,
   WindowMaximizeIcon,
@@ -15,10 +15,10 @@ import {
   buildCorrespondentOptions,
   sortCorrespondents,
 } from './components/DocumentSummarySection';
+import DocumentDownloadLink from '../documents/components/DocumentDownloadLink';
 import { usePreviewContext } from './PreviewContext';
 import type { DocumentSummarySectionProps } from './components/DocumentSummarySection';
 import { extractDocumentMetadataPayload } from './logic/documentSummary';
-import { createDocumentActionState } from '../documents/documentActions';
 import { resolveDocumentAssetUrl } from '../lib/assets/AssetManager';
 import PanelHeader from '../components/PanelHeader';
 import BreadcrumbTrail from '../components/BreadcrumbTrail';
@@ -48,33 +48,14 @@ interface DocumentViewerPanelProps extends DocumentSummarySectionProps {
 
 const createDocumentViewerHeaderActions = ({
   document,
-  actionState,
-  onZoom,
-  canZoom = false,
 }) => {
   if (!document) {
     return null;
   }
 
-  const downloadHref = actionState?.downloadHref;
-  if (!downloadHref && !(canZoom && onZoom)) {
-    return null;
-  }
-
   return (
     <>
-      {downloadHref ? (
-        <a
-          className="icon-button"
-          href={downloadHref}
-          target="_blank"
-          rel="noopener noreferrer"
-          aria-label="Download document"
-          title="Download document"
-        >
-          <DownloadIcon />
-        </a>
-      ) : null}
+      <DocumentDownloadLink document={document} />
     </>
   );
 };
@@ -93,7 +74,6 @@ const DocumentViewerPanel: React.FC<DocumentViewerPanelProps> = ({
   ensureAssetUrl,
   getDocumentAsset,
   ensurePreviewData,
-  notifyApiError,
   sidebarToggle = null,
   onClose,
   resolveFolderPath,
@@ -125,6 +105,18 @@ const DocumentViewerPanel: React.FC<DocumentViewerPanelProps> = ({
     return Boolean(getDocumentAsset(document, 'text-content'));
   }, [document, getDocumentAsset]);
 
+  useEffect(() => {
+    if (!document?.id || !ensurePreviewData) {
+      return;
+    }
+    const download = document.current_version?.download;
+    if (download?.expires_at && download.expires_at <= Date.now()) {
+      ensurePreviewData(document.id).catch((error) => {
+        console.warn('Failed to refresh expired document', error);
+      });
+    }
+  }, [document, ensurePreviewData]);
+
   const navigateToFolder = useCallback(
     (folderId: FolderId | null) => {
       const target = folderId == null
@@ -242,27 +234,6 @@ const DocumentViewerPanel: React.FC<DocumentViewerPanelProps> = ({
     ? 'document-viewer document-viewer--stacked'
     : 'document-viewer';
 
-  const actionState = useMemo(
-    () =>
-      document
-        ? createDocumentActionState({
-          document,
-          ensurePreviewData,
-          ensureAssetUrl,
-          getDocumentAsset,
-          notifyApiError,
-          ocrErrorMessage: 'Unable to open text content.',
-        })
-        : null,
-    [
-      document,
-      ensurePreviewData,
-      ensureAssetUrl,
-      getDocumentAsset,
-      notifyApiError,
-    ],
-  );
-
   const breadcrumbs = useMemo(() => {
     if (!document || !resolveFolderPath) {
       return [];
@@ -294,9 +265,6 @@ const DocumentViewerPanel: React.FC<DocumentViewerPanelProps> = ({
 
   const headerActions = createDocumentViewerHeaderActions({
     document,
-    actionState,
-    onZoom: handleZoomOpen,
-    canZoom: Boolean(document),
   });
 
   const maximizeButton = isSidebarVariant && onMaximize

frontend/src/viewer/UnifiedDocumentViewer.tsx

@@ -1,6 +1,7 @@
 import React, { useMemo } from 'react';
 import PdfViewer from './PdfViewer';
 import MediaViewer from './MediaViewer';
+import { resolveDocumentDownloadHref } from '../documents/documentActions';
 import type { Document } from '../types/documents';
 
 interface UnifiedDocumentViewerProps {
@@ -17,7 +18,7 @@ const UnifiedDocumentViewer: React.FC<UnifiedDocumentViewerProps> = ({
             return null;
         }
 
-        const downloadUrl = document.current_version?.download?.url;
+        const downloadUrl = resolveDocumentDownloadHref(document);
         if (!downloadUrl) {
             return null;
         }

frontend/src/viewer/components/PreviewZoomOverlay.tsx

@@ -1,8 +1,9 @@
 import React, { useEffect, useRef, useState } from 'react';
 import { createPortal } from 'react-dom';
 import UnifiedDocumentViewer from '../UnifiedDocumentViewer';
+import DocumentDownloadLink from '../../documents/components/DocumentDownloadLink';
 import PanelHeader from '../../components/PanelHeader';
-import { IconX, DownloadIcon, FileInfoIcon } from '../../components/icons';
+import { IconX, FileInfoIcon } from '../../components/icons';
 
 import type { Document } from '../../types/documents';
 
@@ -29,7 +30,6 @@ const PreviewZoomOverlay: React.FC<PreviewZoomOverlayProps> = ({
     lastDocumentRef.current = inputDocument;
   }
 
-  const downloadUrl = lastDocumentRef.current?.current_version?.download?.url;
   const documentTitle = lastDocumentRef.current?.title || undefined;
 
   useEffect(() => {
@@ -132,18 +132,7 @@ const PreviewZoomOverlay: React.FC<PreviewZoomOverlayProps> = ({
           }
           actions={
             <>
-              {downloadUrl && (
-                <a
-                  href={downloadUrl}
-                  target="_blank"
-                  rel="noopener noreferrer"
-                  className="icon-button"
-                  aria-label="Download document"
-                  title="Download document"
-                >
-                  <DownloadIcon />
-                </a>
-              )}
+              <DocumentDownloadLink document={lastDocumentRef.current} />
             </>
           }
         />