import { type LoaderFunctionArgs } from "@remix-run/node";
import { getUserSession } from "~/api/login/auth.server";
import { DOCUMENT_URL } from "~/api/axios-client";

/**
 * PDF 代理路由
 * 用于在服务器端添加 JWT 认证后获取 PDF 文件
 */
export async function loader({ request }: LoaderFunctionArgs) {
  try {
    // 获取 JWT token
    const { frontendJWT } = await getUserSession(request);

    // 从查询参数获取文件路径和预览标志
    const url = new URL(request.url);
    const filePath = url.searchParams.get("path");
    const isPreview = url.searchParams.get("preview") === "true";

    // console.log("📄 [PDF Proxy] 请求参数:", {
    //   path: filePath,
    //   preview: url.searchParams.get("preview"),
    //   isPreview: isPreview
    // });

    if (!filePath) {
      return new Response("缺少文件路径参数", { status: 400 });
    }

    // 构建完整的文件 URL
    const fileUrl = `${DOCUMENT_URL}${filePath}`;
    // console.log('fileUrl:', fileUrl)

    // 使用 JWT 认证获取文件
    const response = await fetch(fileUrl, {
      headers: {
        'Authorization': `Bearer ${frontendJWT}`,
      },
    });

    if (!response.ok) {
      return new Response(`获取文件失败: ${response.statusText}`, {
        status: response.status
      });
    }

    // 获取文件内容
    const blob = await response.blob();

    // 从路径中提取文件名
    const fileName = filePath.split('/').pop() || 'document.pdf';

    // 判断文件类型
    const fileExtension = fileName.split('.').pop()?.toLowerCase();
    let contentType = response.headers.get('Content-Type') || 'application/octet-stream';

    // 🔑 根据文件扩展名强制设置正确的 Content-Type
    if (fileExtension === 'pdf') {
      contentType = 'application/pdf';
    } else if (fileExtension === 'doc' || fileExtension === 'docx') {
      contentType = 'application/vnd.openxmlformats-officedocument.wordprocessingml.document';
    }

    // 设置响应头
    const headers: HeadersInit = {
      'Content-Type': contentType,
      'Cache-Control': 'public, max-age=3600', // 缓存1小时
    };

    // 根据 preview 参数决定是预览还是下载
    // 默认行为：强制下载（保持向后兼容性）
    if (isPreview) {
      // 在浏览器中预览
      headers['Content-Disposition'] = `inline; filename="${encodeURIComponent(fileName)}"`;
      // console.log("📄 [PDF Proxy] 设置为预览模式 (inline)");
    } else {
      // 强制下载（默认行为）
      headers['Content-Disposition'] = `attachment; filename="${encodeURIComponent(fileName)}"`;
      // console.log("📄 [PDF Proxy] 设置为下载模式 (attachment)");
    }

    // console.log("📄 [PDF Proxy] 响应头:", headers);

    // 返回文件
    return new Response(blob, { headers });

  } catch (error) {
    console.error('PDF 代理错误:', error);
    return new Response(`服务器错误: ${error instanceof Error ? error.message : '未知错误'}`, {
      status: 500
    });
  }
}
