weaver-work-cli 0.1.8 → 0.1.21

package/dist/shortcuts/caizhilian/operations/handlers/file-upload.js

@@ -0,0 +1,192 @@
+/**
+ * caizhilian.file-upload — 文件上传（本地文件上传到文件服务）
+ *
+ * 覆盖本地文件上传到 E10 文件服务的完整流程：
+ * - contract: 获取上传接口合约
+ * - prepare: 本地文件校验与元数据准备
+ * - upload: 上传文件并获取 fileId
+ *
+ * 接口：
+ * - 预检查：POST /api/file/preUploadCheck?sf_request_type=ajax
+ * - 上传：POST /api/file/module/upload?sf_request_type=ajax
+ */
+import { CaizhilianCliError, caizhilianValidationError } from '../../errors.js';
+import { caizhilianSuccess } from '../types.js';
+import { requireNonEmpty } from '../shared.js';
+import { createHash } from 'node:crypto';
+import { readFileSync, statSync } from 'node:fs';
+import { basename, extname } from 'node:path';
+const ACTION = 'caizhilian.file-upload';
+// ── 接口路径 ──────────────────────────────────────────────
+const PRE_UPLOAD_CHECK_PATH = '/api/file/preUploadCheck?sf_request_type=ajax';
+const UPLOAD_PATH = '/api/file/module/upload?sf_request_type=ajax';
+// ── 辅助函数 ──────────────────────────────────────────────
+function asRecord(value) {
+    return value && typeof value === 'object' && !Array.isArray(value) ? value : {};
+}
+function isPlainObject(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function md5File(filePath) {
+    const content = readFileSync(filePath);
+    return createHash('md5').update(content).digest('hex');
+}
+function sha256File(filePath) {
+    const content = readFileSync(filePath);
+    return createHash('sha256').update(content).digest('hex');
+}
+// ── 获取上传合约 ──────────────────────────────────────────
+async function getContract(args, env) {
+    return caizhilianSuccess(ACTION, {
+        contract: {
+            version: 'agent_callable_interface_contract.v1',
+            actions: ['contract', 'prepare', 'upload'],
+            required: {
+                prepare: ['file_path', 'module'],
+                upload: ['file_path', 'module', 'user_confirmed'],
+            },
+            optional: {
+                prepare: ['context_file'],
+                upload: ['context_file', 'login_mode'],
+            },
+        },
+    });
+}
+// ── 准备上传（prepare/apply）──────────────────────────────────────
+async function prepareUpload(args, env) {
+    const filePath = requireNonEmpty(args.file_path, 'file_path');
+    const module = requireNonEmpty(args.module, 'module');
+    // 校验文件存在
+    let stat;
+    try {
+        stat = statSync(filePath);
+    }
+    catch (error) {
+        throw caizhilianValidationError('file_not_found', `文件不存在: ${filePath}`);
+    }
+    if (!stat.isFile()) {
+        throw caizhilianValidationError('file_not_regular', `不是普通文件: ${filePath}`);
+    }
+    // 计算元数据
+    const fileName = basename(filePath);
+    const ext = extname(filePath).slice(1).toLowerCase();
+    const size = stat.size;
+    const md5 = md5File(filePath);
+    const sha256 = sha256File(filePath);
+    // 签发 continuation
+    const continuation = await env.continuation.issue({
+        operation: 'caizhilian.file-upload.upload.apply',
+        filePath,
+        module,
+        fileName,
+        ext,
+        size,
+        md5,
+        sha256,
+    });
+    return caizhilianSuccess('caizhilian.file-upload.prepare', {
+        file_path: filePath,
+        file_name: fileName,
+        ext,
+        size,
+        md5,
+        sha256,
+        module,
+        continuation,
+    });
+}
+async function uploadApply(args, env, lease) {
+    const filePath = lease.payload?.filePath;
+    const module = lease.payload?.module;
+    const fileName = lease.payload?.fileName;
+    const ext = lease.payload?.ext;
+    const size = lease.payload?.size;
+    const md5 = lease.payload?.md5;
+    const sha256 = lease.payload?.sha256;
+    // 预检查
+    const preCheckResponse = await env.host.request({
+        method: 'POST',
+        path: PRE_UPLOAD_CHECK_PATH,
+        body: {
+            fileName,
+            fileSize: size,
+            module,
+            md5,
+        },
+        operation: 'caizhilian.file-upload.upload.apply',
+    });
+    const preCheckData = asRecord(preCheckResponse);
+    if (!preCheckData.status && !preCheckData.success) {
+        throw new CaizhilianCliError('api', 'business_error', String(preCheckData.msg || preCheckData.message || '预检查失败'), {
+            code: preCheckData.code ?? preCheckData.status,
+            details: { action: 'caizhilian.file-upload.upload.apply', message: String(preCheckData.msg || preCheckData.message || '预检查失败') },
+            exitCode: 1,
+        });
+    }
+    const preCheckResult = asRecord(preCheckData.data);
+    // 判断上传策略
+    const uploadType = preCheckResult.uploadType || 'common';
+    const preUploadId = preCheckResult.preUploadId;
+    // 如果是 second 策略，直接返回已有 file ID
+    if (uploadType === 'second') {
+        const fileId = preCheckResult.fileId || preCheckResult.fileid || preCheckResult.uploadModuleDto?.fileObj?.fileId;
+        if (!fileId) {
+            throw new CaizhilianCliError('api', 'business_error', 'second 策略未返回 file ID', {
+                details: { action: 'caizhilian.file-upload.upload.apply' },
+                exitCode: 1,
+            });
+        }
+        return caizhilianSuccess('caizhilian.file-upload.upload.apply', {
+            file_id: String(fileId),
+            file_name: fileName,
+            size,
+            ext,
+            upload_type: uploadType,
+        });
+    }
+    // common/chunk 策略：上传文件
+    const fileContent = readFileSync(filePath);
+    const formData = new FormData();
+    formData.append('file', new Blob([fileContent]), fileName);
+    formData.append('preUploadId', preUploadId || '');
+    formData.append('module', module);
+    const uploadResponse = await env.host.request({
+        method: 'POST',
+        path: UPLOAD_PATH,
+        body: formData,
+        operation: 'caizhilian.file-upload.upload.apply',
+    });
+    const uploadData = asRecord(uploadResponse);
+    if (!uploadData.status && !uploadData.success) {
+        throw new CaizhilianCliError('api', 'business_error', String(uploadData.msg || uploadData.message || '上传失败'), {
+            code: uploadData.code ?? uploadData.status,
+            details: { action: 'caizhilian.file-upload.upload.apply', message: String(uploadData.msg || uploadData.message || '上传失败') },
+            exitCode: 1,
+        });
+    }
+    const uploadResult = asRecord(uploadData.data);
+    const fileId = uploadResult.fileId || uploadResult.fileid || '';
+    if (!fileId) {
+        throw new CaizhilianCliError('api', 'business_error', '上传未返回 file ID', {
+            details: { action: 'caizhilian.file-upload.upload.apply' },
+            exitCode: 1,
+        });
+    }
+    return caizhilianSuccess('caizhilian.file-upload.upload.apply', {
+        file_id: String(fileId),
+        file_name: fileName,
+        size,
+        ext,
+        upload_type: uploadType,
+    });
+}
+// ── 处理器汇总 ────────────────────────────────────────────
+export const caizhilianFileUploadHandlers = {
+    'caizhilian.file-upload.contract': { execute: getContract },
+    'caizhilian.file-upload.prepare': { execute: prepareUpload },
+    'caizhilian.file-upload.upload': {
+        prepare: prepareUpload,
+        apply: uploadApply,
+        targetKeys: ['file_path', 'module'],
+    },
+};