feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/arch/pdfjs-shadow/__tests__/generate-assets.test.ts

@@ -0,0 +1,89 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+
+import { generatePdfAssetsUrl } from '../src/generate-assets';
+import pkg from '../package.json';
+
+describe('generatePdfAssetsUrl', () => {
+  const originalRegion = global.REGION;
+
+  beforeEach(() => {
+    // 重置模拟
+    vi.resetAllMocks();
+  });
+
+  afterEach(() => {
+    // 恢复原始 REGION 值
+    global.REGION = originalRegion;
+  });
+
+  it('应该为 cmaps 生成正确的 URL（中国区域）', () => {
+    // 设置区域为中国
+    global.REGION = 'cn';
+
+    const url = generatePdfAssetsUrl('cmaps');
+
+    // 验证 URL 格式
+    expect(url).toContain('//lf-cdn.coze.cn/obj/unpkg');
+    expect(url).toContain(pkg.name.replace(/^@/, ''));
+    expect(url).toContain('lib/cmaps/');
+  });
+
+  it('应该为 pdf.worker 生成正确的 URL（中国区域）', () => {
+    // 设置区域为中国
+    global.REGION = 'cn';
+
+    const url = generatePdfAssetsUrl('pdf.worker');
+
+    // 验证 URL 格式
+    expect(url).toContain('//lf-cdn.coze.cn/obj/unpkg');
+    expect(url).toContain(pkg.name.replace(/^@/, ''));
+    expect(url).toContain('lib/worker.js');
+  });
+
+  it('应该为 cmaps 生成正确的 URL（国际区域）', () => {
+    // 设置区域为国际
+    global.REGION = 'va';
+
+    const url = generatePdfAssetsUrl('cmaps');
+
+    // 验证 URL 格式
+    expect(url).toContain('//sf-cdn.coze.com/obj/unpkg-va');
+    expect(url).toContain(pkg.name.replace(/^@/, ''));
+    expect(url).toContain('lib/cmaps/');
+  });
+
+  it('应该为 pdf.worker 生成正确的 URL（国际区域）', () => {
+    // 设置区域为国际
+    global.REGION = 'va';
+
+    const url = generatePdfAssetsUrl('pdf.worker');
+
+    // 验证 URL 格式
+    expect(url).toContain('//sf-cdn.coze.com/obj/unpkg-va');
+    expect(url).toContain(pkg.name.replace(/^@/, ''));
+    expect(url).toContain('lib/worker.js');
+  });
+
+  it('应该在传入无效资源类型时抛出错误', () => {
+    // 使用类型断言来测试错误情况
+    expect(() => generatePdfAssetsUrl('invalid' as any)).toThrow(
+      '目前只支持引用 cmaps 与 pdf.worker 文件',
+    );
+  });
+});

frontend/packages/arch/pdfjs-shadow/__tests__/index.test.ts

@@ -0,0 +1,49 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import { describe, it, expect, vi } from 'vitest';
+
+// 模拟 pdfjs-dist 模块
+vi.mock('pdfjs-dist', () => ({
+  getDocument: vi.fn(),
+}));
+
+// 模拟 generate-assets 和 init-pdfjs-dist 模块
+vi.mock('../src/generate-assets', () => ({
+  generatePdfAssetsUrl: vi.fn(),
+}));
+
+vi.mock('../src/init-pdfjs-dist', () => ({
+  initPdfJsWorker: vi.fn(),
+}));
+
+// 导入被测试的模块
+import {
+  generatePdfAssetsUrl,
+  initPdfJsWorker,
+  getDocument,
+} from '../src/index';
+
+describe('pdfjs-shadow index', () => {
+  it('应该导出所有必要的函数和类型', () => {
+    // 验证导出的函数
+    expect(typeof generatePdfAssetsUrl).toBe('function');
+    expect(typeof initPdfJsWorker).toBe('function');
+
+    // 验证从 pdfjs-dist 重新导出的函数和类型
+    expect(getDocument).toBeDefined();
+  });
+});

frontend/packages/arch/pdfjs-shadow/__tests__/init-pdfjs-dist.test.ts

@@ -0,0 +1,78 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+
+// 模拟 pdfjs-dist 模块
+vi.mock('pdfjs-dist', () => ({
+  GlobalWorkerOptions: {
+    workerSrc: '',
+  },
+}));
+
+// 模拟 generate-assets 模块
+vi.mock('../src/generate-assets', () => ({
+  generatePdfAssetsUrl: vi.fn().mockReturnValue('mocked-worker-url'),
+}));
+
+// 导入被测试的模块
+import { GlobalWorkerOptions } from 'pdfjs-dist';
+
+import { initPdfJsWorker } from '../src/init-pdfjs-dist';
+import { generatePdfAssetsUrl } from '../src/generate-assets';
+
+describe('initPdfJsWorker', () => {
+  beforeEach(() => {
+    // 每个测试前重置 GlobalWorkerOptions.workerSrc
+    GlobalWorkerOptions.workerSrc = '';
+    // 清除所有模拟函数的调用记录
+    vi.clearAllMocks();
+  });
+
+  afterEach(() => {
+    // 每个测试后重置模拟
+    vi.resetAllMocks();
+  });
+
+  it('应该设置 GlobalWorkerOptions.workerSrc 当它为空时', () => {
+    // 确保 workerSrc 初始为空
+    expect(GlobalWorkerOptions.workerSrc).toBe('');
+
+    // 调用初始化函数
+    initPdfJsWorker();
+
+    // 验证 generatePdfAssetsUrl 被调用，且参数正确
+    expect(generatePdfAssetsUrl).toHaveBeenCalledTimes(1);
+    expect(generatePdfAssetsUrl).toHaveBeenCalledWith('pdf.worker');
+
+    // 验证 workerSrc 被正确设置
+    expect(GlobalWorkerOptions.workerSrc).toBe('mocked-worker-url');
+  });
+
+  it('不应该重新设置 GlobalWorkerOptions.workerSrc 当它已经有值时', () => {
+    // 预先设置 workerSrc
+    GlobalWorkerOptions.workerSrc = 'existing-worker-url';
+
+    // 调用初始化函数
+    initPdfJsWorker();
+
+    // 验证 generatePdfAssetsUrl 没有被调用
+    expect(generatePdfAssetsUrl).not.toHaveBeenCalled();
+
+    // 验证 workerSrc 保持不变
+    expect(GlobalWorkerOptions.workerSrc).toBe('existing-worker-url');
+  });
+});