@xmanrui/dsh-im 1.3.0 → 1.5.0

package/src/channels/weixin/weixin-api.mjs

@@ -457,6 +457,123 @@ function validateLoginResponse(value) {
 export function createWeixinApi({ fetchImpl = fetch } = {}) {
   if (typeof fetchImpl !== 'function') throw new TypeError('fetchImpl must be a function');
 
+  async function sendArtifact({
+    baseUrl,
+    token,
+    toUserId,
+    file,
+    contextToken,
+    runId,
+    signal,
+  }, { mediaType, createItem }) {
+    const recipient = nonEmptyString(toUserId);
+    if (!recipient || !file || typeof file !== 'object'
+      || typeof file.fileName !== 'string' || !file.fileName
+      || !Buffer.isBuffer(file.bytes)) {
+      throw new TypeError('toUserId and a file are required');
+    }
+    signal?.throwIfAborted();
+    const fileKey = randomBytes(16).toString('hex');
+    const aesKey = randomBytes(16);
+    const rawMd5 = createHash('md5').update(file.bytes).digest('hex');
+    let upload;
+    try {
+      upload = await requestJson(fetchImpl, {
+        method: 'POST',
+        baseUrl,
+        endpoint: 'ilink/bot/getuploadurl',
+        token,
+        signal,
+        body: {
+          filekey: fileKey,
+          media_type: mediaType,
+          to_user_id: recipient,
+          rawsize: file.bytes.byteLength,
+          rawfilemd5: rawMd5,
+          filesize: aesEcbPaddedSize(file.bytes.byteLength),
+          no_need_thumb: true,
+          aeskey: aesKey.toString('hex'),
+          base_info: baseInfo(),
+        },
+      });
+    } catch (error) {
+      if (signal?.aborted) throw abortError(signal);
+      const status = Number(error?.status);
+      const fallback = error?.code === 'http-error' && status >= 400 && status < 500
+        ? 'artifact-provider-rejected'
+        : 'artifact-provider-failed';
+      throw weixinArtifactError(error, { fallback });
+    }
+    const uploadRejection = rejectedProviderResponse(upload);
+    if (uploadRejection) {
+      throw weixinArtifactError(new WeixinApiError(
+        'upload-url-rejected',
+        '微信服务拒绝了文件上传请求。',
+        { providerCode: uploadRejection },
+      ));
+    }
+    const uploadUrl = weixinCdnUploadUrl(upload, fileKey);
+    const ciphertext = encryptWeixinUpload(file.bytes, aesKey);
+    let downloadParam;
+    try {
+      downloadParam = await uploadWeixinCdn(fetchImpl, uploadUrl, ciphertext, { signal });
+    } catch (error) {
+      if (signal?.aborted) throw abortError(signal);
+      const status = Number(error?.status);
+      const fallback = error?.code === 'upload-rejected' || (status >= 400 && status < 500)
+        ? 'artifact-provider-rejected'
+        : 'artifact-provider-failed';
+      throw weixinArtifactError(error, { fallback });
+    }
+    signal?.throwIfAborted();
+    const deliverySeed = nonEmptyString(file.deliveryKey) ?? nonEmptyString(file.artifactId)
+      ?? randomUUID();
+    const clientIdSeed = mediaType === 3 ? deliverySeed : `${deliverySeed}\u0000${mediaType}`;
+    const clientId = `dsh-weixin-${createHash('sha256')
+      .update(clientIdSeed)
+      .digest('hex')
+      .slice(0, 32)}`;
+    const media = {
+      encrypt_query_param: downloadParam,
+      aes_key: Buffer.from(aesKey.toString('hex')).toString('base64'),
+      encrypt_type: 1,
+    };
+    let response;
+    try {
+      response = await requestJson(fetchImpl, {
+        method: 'POST',
+        baseUrl,
+        endpoint: 'ilink/bot/sendmessage',
+        token,
+        signal,
+        body: {
+          msg: {
+            from_user_id: '',
+            to_user_id: recipient,
+            client_id: clientId,
+            message_type: 2,
+            message_state: 2,
+            item_list: [createItem({ file, media, ciphertextSize: ciphertext.byteLength })],
+            ...(nonEmptyString(contextToken) ? { context_token: contextToken.trim() } : {}),
+            ...(nonEmptyString(runId) ? { run_id: runId.trim() } : {}),
+          },
+          base_info: baseInfo(),
+        },
+      });
+    } catch (error) {
+      throw classifyWeixinFinalDeliveryError(error, signal);
+    }
+    const sendRejection = rejectedProviderResponse(response);
+    if (sendRejection) {
+      throw weixinArtifactError(new WeixinApiError(
+        'send-rejected',
+        '微信服务拒绝了文件消息。',
+        { providerCode: sendRejection },
+      ));
+    }
+    return { messageId: clientId };
+  }
+
   return Object.freeze({
     inboundImages(message) {
       return extractWeixinImages(message, { fetchImpl });
@@ -549,115 +666,31 @@ export function createWeixinApi({ fetchImpl = fetch } = {}) {
       return true;
     },
 
-    async sendFile({ baseUrl, token, toUserId, file, contextToken, runId, signal }) {
-      const recipient = nonEmptyString(toUserId);
-      if (!recipient || !file || typeof file !== 'object'
-        || typeof file.fileName !== 'string' || !file.fileName
-        || !Buffer.isBuffer(file.bytes)) {
-        throw new TypeError('toUserId and a file are required');
-      }
-      signal?.throwIfAborted();
-      const fileKey = randomBytes(16).toString('hex');
-      const aesKey = randomBytes(16);
-      const rawMd5 = createHash('md5').update(file.bytes).digest('hex');
-      let upload;
-      try {
-        upload = await requestJson(fetchImpl, {
-          method: 'POST',
-          baseUrl,
-          endpoint: 'ilink/bot/getuploadurl',
-          token,
-          signal,
-          body: {
-            filekey: fileKey,
-            media_type: 3,
-            to_user_id: recipient,
-            rawsize: file.bytes.byteLength,
-            rawfilemd5: rawMd5,
-            filesize: aesEcbPaddedSize(file.bytes.byteLength),
-            no_need_thumb: true,
-            aeskey: aesKey.toString('hex'),
-            base_info: baseInfo(),
+    async sendFile(request) {
+      return sendArtifact(request, {
+        mediaType: 3,
+        createItem: ({ file, media }) => ({
+          type: 4,
+          file_item: {
+            media,
+            file_name: file.fileName,
+            len: String(file.bytes.byteLength),
           },
-        });
-      } catch (error) {
-        if (signal?.aborted) throw abortError(signal);
-        const status = Number(error?.status);
-        const fallback = error?.code === 'http-error' && status >= 400 && status < 500
-          ? 'artifact-provider-rejected'
-          : 'artifact-provider-failed';
-        throw weixinArtifactError(error, { fallback });
-      }
-      const uploadRejection = rejectedProviderResponse(upload);
-      if (uploadRejection) {
-        throw weixinArtifactError(new WeixinApiError(
-          'upload-url-rejected',
-          '微信服务拒绝了文件上传请求。',
-          { providerCode: uploadRejection },
-        ));
-      }
-      const uploadUrl = weixinCdnUploadUrl(upload, fileKey);
-      const ciphertext = encryptWeixinUpload(file.bytes, aesKey);
-      let downloadParam;
-      try {
-        downloadParam = await uploadWeixinCdn(fetchImpl, uploadUrl, ciphertext, { signal });
-      } catch (error) {
-        if (signal?.aborted) throw abortError(signal);
-        const status = Number(error?.status);
-        const fallback = error?.code === 'upload-rejected' || (status >= 400 && status < 500)
-          ? 'artifact-provider-rejected'
-          : 'artifact-provider-failed';
-        throw weixinArtifactError(error, { fallback });
-      }
-      signal?.throwIfAborted();
-      const deliverySeed = nonEmptyString(file.deliveryKey) ?? nonEmptyString(file.artifactId)
-        ?? randomUUID();
-      const clientId = `dsh-weixin-${createHash('sha256').update(deliverySeed).digest('hex').slice(0, 32)}`;
-      let response;
-      try {
-        response = await requestJson(fetchImpl, {
-          method: 'POST',
-          baseUrl,
-          endpoint: 'ilink/bot/sendmessage',
-          token,
-          signal,
-          body: {
-            msg: {
-              from_user_id: '',
-              to_user_id: recipient,
-              client_id: clientId,
-              message_type: 2,
-              message_state: 2,
-              item_list: [{
-                type: 4,
-                file_item: {
-                  media: {
-                    encrypt_query_param: downloadParam,
-                    aes_key: Buffer.from(aesKey.toString('hex')).toString('base64'),
-                    encrypt_type: 1,
-                  },
-                  file_name: file.fileName,
-                  len: String(file.bytes.byteLength),
-                },
-              }],
-              ...(nonEmptyString(contextToken) ? { context_token: contextToken.trim() } : {}),
-              ...(nonEmptyString(runId) ? { run_id: runId.trim() } : {}),
-            },
-            base_info: baseInfo(),
+        }),
+      });
+    },
+
+    async sendImage(request) {
+      return sendArtifact(request, {
+        mediaType: 1,
+        createItem: ({ media, ciphertextSize }) => ({
+          type: 2,
+          image_item: {
+            media,
+            mid_size: ciphertextSize,
           },
-        });
-      } catch (error) {
-        throw classifyWeixinFinalDeliveryError(error, signal);
-      }
-      const sendRejection = rejectedProviderResponse(response);
-      if (sendRejection) {
-        throw weixinArtifactError(new WeixinApiError(
-          'send-rejected',
-          '微信服务拒绝了文件消息。',
-          { providerCode: sendRejection },
-        ));
-      }
-      return { messageId: clientId };
+        }),
+      });
     },
 
     async notifyStart({ baseUrl, token, signal }) {

package/src/channels/weixin/weixin-bridge.mjs

@@ -38,14 +38,9 @@ import {
   prefetchInboundFiles,
 } from '../shared/inbound-file.mjs';
 import { rememberConnectionTestTarget } from '../shared/connection-test.mjs';
+import { deliverOutboundArtifacts } from '../shared/semantic/artifact-delivery.mjs';
 import {
-  materializeOutboundArtifact,
-  releaseOutboundArtifact,
-} from '../shared/semantic/artifact.mjs';
-import {
-  createArtifactFailureReceipt,
   createDeliveryReceipt,
-  mergeDeliveryReceipts,
   providerMessageIdsFor,
 } from '../shared/semantic/delivery.mjs';
 
@@ -869,73 +864,40 @@ export class WeixinHarnessBridge {
   }
 
   async #deliverArtifacts(toUserId, replyTo, artifacts, contextToken, runId, baseReceipt) {
-    const receipts = baseReceipt ? [baseReceipt] : [];
-    let userVisible = Boolean(baseReceipt);
-    for (const artifact of artifacts) {
-      this.#signal?.throwIfAborted();
-      try {
-        if (typeof this.#api.sendFile !== 'function') {
-          const unavailable = new Error('Weixin file delivery is unavailable');
-          unavailable.code = 'artifact-provider-unavailable';
-          throw unavailable;
-        }
-        const file = await materializeOutboundArtifact(artifact, {
-          signal: this.#signal,
-        });
-        const result = await this.#api.sendFile({
-          baseUrl: this.#baseUrl,
-          token: this.#token,
-          toUserId,
-          file,
-          contextToken,
-          runId,
-          signal: this.#signal,
-        });
-        receipts.push(createDeliveryReceipt({
-          deliveryId: file.deliveryKey,
-          presentation: 'weixin-file',
-          providerMessageIds: providerMessageIdsFor(result),
-          artifacts: [{ artifactId: file.artifactId, outcome: 'sent' }],
-        }));
-        userVisible = true;
-        this.#status.artifactsSent = (this.#status.artifactsSent ?? 0) + 1;
-      } catch (error) {
-        if (this.#signal?.aborted) throw error;
-        this.#status.artifactSendErrors = (this.#status.artifactSendErrors ?? 0) + 1;
-        this.#logger.warn?.(
-          `[dsh-weixin] result file delivery failed (${error?.code ?? 'unknown'})`,
-        );
-        let noticeSent = false;
-        const providerMessageIds = await this.#send(
-          toUserId,
-          artifactFailureText(artifact?.fileName, error),
-          contextToken,
-          runId,
-        ).then((ids) => {
-          noticeSent = true;
-          return ids;
-        }).catch(() => []);
-        const failureReceipt = createArtifactFailureReceipt({
-          artifactId: artifact?.artifactId ?? 'unknown',
-          deliveryId: artifact?.deliveryKey ?? artifact?.artifactId ?? 'unknown',
-          error,
-          providerMessageIds,
-        });
-        receipts.push(failureReceipt);
-        if (noticeSent || failureReceipt.artifacts[0]?.outcome === 'unknown') userVisible = true;
-      } finally {
-        releaseOutboundArtifact(artifact);
-      }
-    }
-    const receipt = receipts.length === 0
-      ? null
-      : receipts.length === 1
-        ? receipts[0]
-        : mergeDeliveryReceipts({
-            deliveryId: replyTo,
-            presentation: baseReceipt ? 'weixin-text-and-files' : 'weixin-files',
-            receipts,
-          });
-    return { receipt, userVisible };
+    const sendArtifact = (method, file) => this.#api[method]({
+      baseUrl: this.#baseUrl,
+      token: this.#token,
+      toUserId,
+      file,
+      contextToken,
+      runId,
+      signal: this.#signal,
+    });
+    const delivery = await deliverOutboundArtifacts({
+      artifacts,
+      baseReceipt,
+      deliveryId: replyTo,
+      aggregatePresentation: baseReceipt ? 'weixin-text-and-files' : 'weixin-files',
+      channelKey: 'weixin',
+      signal: this.#signal,
+      sendImage: typeof this.#api.sendImage === 'function'
+        ? (file) => sendArtifact('sendImage', file)
+        : undefined,
+      sendFile: typeof this.#api.sendFile === 'function'
+        ? (file) => sendArtifact('sendFile', file)
+        : undefined,
+      sendFailureNotice: (artifact, error) => this.#send(
+        toUserId,
+        artifactFailureText(artifact?.fileName, error),
+        contextToken,
+        runId,
+      ),
+      logger: this.#logger,
+    });
+    this.#status.artifactsSent = (this.#status.artifactsSent ?? 0)
+      + delivery.artifactsSent;
+    this.#status.artifactSendErrors = (this.#status.artifactSendErrors ?? 0)
+      + delivery.artifactSendErrors;
+    return { receipt: delivery.receipt, userVisible: delivery.userVisible };
   }
 }

package/src/channels/whatsapp/whatsapp-runtime.mjs

@@ -313,12 +313,31 @@ function waitWithSignal(promise, signal) {
 
 function uncertainArtifactDelivery(error) {
   if (error?.code === 'artifact-delivery-uncertain') return error;
-  const uncertain = new Error('WhatsApp could not confirm result file delivery.');
+  const uncertain = new Error('WhatsApp could not confirm artifact delivery.');
   uncertain.code = 'artifact-delivery-uncertain';
   uncertain.cause = error;
   return uncertain;
 }
 
+function whatsappArtifactError(error) {
+  if (error?.code?.startsWith?.('artifact-')) return error;
+  const status = error?.output?.statusCode
+    ?? error?.data?.statusCode
+    ?? error?.statusCode;
+  let code;
+  if (status === 401 || status === 403) code = 'artifact-permission-required';
+  else if (status === 413) code = 'artifact-too-large';
+  else if (status === 429) code = 'artifact-rate-limited';
+  else if ([400, 404, 405, 406, 410, 415, 422].includes(status)) {
+    code = 'artifact-provider-rejected';
+  }
+  if (!code) return uncertainArtifactDelivery(error);
+  const wrapped = new Error('WhatsApp rejected artifact delivery.');
+  wrapped.code = code;
+  wrapped.cause = error;
+  return wrapped;
+}
+
 export class WhatsappBotClient {
   #socket;
   #outboundIds;
@@ -354,14 +373,32 @@ export class WhatsappBotClient {
   }
 
   async sendFile(target, file) {
+    return this.#sendArtifact(target, file, {
+      document: file.bytes,
+      mimetype: file.mediaType ?? 'application/octet-stream',
+      fileName: file.fileName,
+    }, 'file');
+  }
+
+  async sendImage(target, file) {
+    return this.#sendArtifact(target, file, {
+      image: file.bytes,
+      mimetype: file.mediaType ?? 'image/jpeg',
+    }, 'image');
+  }
+
+  async #sendArtifact(target, file, content, presentation) {
     this.#signal?.throwIfAborted();
     await this.#stopTyping(target.jid);
     this.#signal?.throwIfAborted();
     const deliverySeed = typeof file.deliveryKey === 'string' && file.deliveryKey
       ? file.deliveryKey
       : file.artifactId;
-    const messageId = typeof deliverySeed === 'string' && deliverySeed
-      ? createHash('sha256').update(deliverySeed).digest('hex').slice(0, 20).toUpperCase()
+    const messageIdSeed = presentation === 'image'
+      ? `${deliverySeed}:image`
+      : deliverySeed;
+    const messageId = typeof messageIdSeed === 'string' && messageIdSeed
+      ? createHash('sha256').update(messageIdSeed).digest('hex').slice(0, 20).toUpperCase()
       : undefined;
     const options = {
       ...(target.quoted ? { quoted: target.quoted } : {}),
@@ -375,11 +412,7 @@ export class WhatsappBotClient {
     try {
       const pending = this.#socket.sendMessage(
         target.jid,
-        {
-          document: file.bytes,
-          mimetype: file.mediaType ?? 'application/octet-stream',
-          fileName: file.fileName,
-        },
+        content,
         options,
       );
       trackOutboundArtifactProviderPromise(file, pending);
@@ -390,7 +423,7 @@ export class WhatsappBotClient {
       result = await waitWithSignal(pending, waitSignal);
     } catch (error) {
       if (this.#signal?.aborted) throw abortReason(this.#signal);
-      throw uncertainArtifactDelivery(error);
+      throw whatsappArtifactError(error);
     }
     this.#signal?.throwIfAborted();
     this.#outboundIds.remember(result?.key?.id);