document-management/backend/dms/services/file_read_service.py

"""统一文档预览、下载授权及下载计数审计。"""

from __future__ import annotations

import logging
from io import BytesIO
from pathlib import Path
from urllib.parse import quote

from flask import current_app, send_file
from sqlalchemy import select
from werkzeug.utils import secure_filename

from dms.common.enums import AuditAction, AuditTarget, DocumentType
from dms.common.errors import (
    DocumentDownloadForbiddenError,
    FileNotFoundError as DmsFileNotFoundError,
    InternalError,
    PreviewUnavailableError,
    ResourceNotFoundError,
    SecurityLevelForbiddenError,
)
from dms.database.transaction import transaction
from dms.extensions import db
from dms.models import LOCK_DEFER_LOB, Document
from dms.security.auth_context import get_auth_context
from dms.services.audit_service import business_audit
from dms.services.authorization_service import evaluate_plan_access
from dms.services.authorization_service import require_attachment_access
from dms.services.document_password_service import (
    DocumentPasswordAction,
    get_request_password,
    require_document_password,
)
from dms.services.document_query_service import _require_plan_access
from dms.services.preview_service import PreviewService
from dms.services.recycle_bin_service import _verify_file
from dms.storage.envelope import is_envelope, read_plaintext_bytes
from dms.storage.paths import UnsafeStoragePathError, resolve_storage_path
from dms.storage.uploads import IMAGE_EXTENSIONS, MIME_TYPES as UPLOAD_MIME_TYPES

logger = logging.getLogger(__name__)


def _document(document_id: int) -> Document:
    document = db.session.scalar(
        select(Document).where(
            Document.id == document_id,
            Document.is_deleted.is_(False),
            Document.document_type.in_(
                [
                    DocumentType.MAIN.value,
                    DocumentType.SUB_PLAN.value,
                    DocumentType.ATTACHMENT.value,
                ]
            ),
        )
    )
    if document is None:
        raise ResourceNotFoundError("文档不存在")
    return document


def _authorize_view(document: Document) -> None:
    context = get_auth_context()
    if document.document_type == DocumentType.ATTACHMENT.value:
        require_attachment_access(document, context, download=False)
    else:
        _require_plan_access(document, context)


def _authorize_download(document: Document) -> None:
    if document.document_type == DocumentType.ATTACHMENT.value:
        require_attachment_access(
            document, get_auth_context(), download=True
        )
        return
    access = evaluate_plan_access(document, get_auth_context())
    if access.reason == "SECURITY":
        raise SecurityLevelForbiddenError()
    if not access.allowed or not access.can_download:
        raise DocumentDownloadForbiddenError()


def _path(document: Document) -> Path:
    try:
        path = resolve_storage_path(
            document.file_relative_path,
            current_app.config["DMS_STORAGE_ROOT"],
        )
    except UnsafeStoragePathError as exc:
        logger.error("检测到越界文档存储路径：document_id=%s", document.id)
        raise InternalError("文档存储路径异常") from exc
    if not path.is_file():
        raise DmsFileNotFoundError()
    try:
        with path.open("rb") as stream:
            stream.read(1)
    except OSError as exc:
        logger.exception("文档文件读取失败：document_id=%s", document.id)
        raise InternalError("文档文件读取失败") from exc
    return path


def _download_name(value: str) -> str:
    safe = value.replace("\r", "").replace("\n", "").replace("\x00", "")
    return Path(safe).name or "download"


def _inline_content_disposition(value: str) -> str:
    original = _download_name(value)
    fallback = secure_filename(original) or "preview"
    return (
        f'inline; filename="{fallback}"; '
        f"filename*=UTF-8''{quote(original, safe='')}"
    )


def preview_document(document_id: int):
    document = _document(document_id)
    _authorize_view(document)
    require_document_password(
        document,
        DocumentPasswordAction.VIEW.value,
        get_request_password(),
        get_auth_context(),
    )
    verified = _verify_file(document)
    try:
        path = verified.path
        extension = document.file_extension.lower().lstrip(".")
        # 图片原图直出（MIME 必须准确，响应带 nosniff）；其余统一以 PDF 预览
        if extension in IMAGE_EXTENSIONS:
            preview_name = _download_name(document.original_file_name)
            preview_mime = UPLOAD_MIME_TYPES.get(
                extension, "application/octet-stream"
            )
        else:
            preview_name = _download_name(
                f"{Path(document.original_file_name).stem}.pdf"
            )
            preview_mime = "application/pdf"
        preview_service = PreviewService.current()
        preview_path = preview_service.preview_path_for(document)
        verified.close()
        # 契约1.21：信封加密的 original 直出文件解密后经内存流发送
        send_source: Path | BytesIO = preview_path
        if is_envelope(preview_path):
            send_source = BytesIO(read_plaintext_bytes(preview_path))
        response = send_file(
            send_source,
            mimetype=preview_mime,
            as_attachment=False,
            download_name=preview_name,
            conditional=True,
        )
    except Exception:
        verified.close()
        raise
    response.headers["X-Content-Type-Options"] = "nosniff"
    response.headers["Cache-Control"] = "private, no-store"
    response.headers["Content-Disposition"] = _inline_content_disposition(
        preview_name
    )
    return response


def _record_download(document: Document) -> None:
    document_id = document.id
    document_name = document.document_name
    target = (
        AuditTarget.ATTACHMENT
        if document.document_type == DocumentType.ATTACHMENT.value
        else AuditTarget.DOCUMENT
    )
    db.session.rollback()
    try:
        with transaction() as session:
            current = session.scalar(
                select(Document)
                .where(
                    Document.id == document_id,
                    Document.is_deleted.is_(False),
                )
                .options(*LOCK_DEFER_LOB)
                .with_for_update()
            )
            if current is None:
                raise ResourceNotFoundError("文档不存在")
            current.download_count += 1
            session.add(
                business_audit(
                    action=AuditAction.DOWNLOAD_DOCUMENT,
                    target=target,
                    target_id=document_id,
                    target_name=document_name,
                    detail={"documentType": current.document_type},
                )
            )
    except Exception:
        db.session.rollback()
        logger.exception("文档下载计数或审计写入失败：document_id=%s", document_id)


def download_document(document_id: int):
    document = _document(document_id)
    _authorize_download(document)
    require_document_password(
        document,
        DocumentPasswordAction.DOWNLOAD.value,
        get_request_password(),
        get_auth_context(),
    )
    path = _path(document)
    mime_type = document.mime_type or "application/octet-stream"
    download_name = _download_name(document.original_file_name)
    _record_download(document)
    # 契约1.21：信封加密文件解密后经内存流发送
    send_source: Path | BytesIO = path
    if is_envelope(path):
        send_source = BytesIO(read_plaintext_bytes(path))
    response = send_file(
        send_source,
        mimetype=mime_type,
        as_attachment=True,
        download_name=download_name,
        conditional=True,
    )
    response.headers["X-Content-Type-Options"] = "nosniff"
    return response