fix: fix image transfer

DreamOfIce · Jul 24, 2023 · c840800 · c840800
1 parent c6bf934
commit c840800
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/messanger.ts b/src/messanger.ts
@@ -186,7 +186,7 @@ export class VillaMessanger extends Messenger<VillaBot> {
       }
       case "image": {
         const url = (element.attrs as Dict<string, "url">)["url"];
-        const newUrl = await this.bot.transferImage(url);
+        const newUrl = await this.bot.transferImage(url, this.villaId);
         const { protocol } = new URL(newUrl);
         if (protocol !== "http:" && protocol !== "https:")
           throw new Error(`Unsupported image protocol: ${protocol}`);

diff --git a/src/utils/transferImage.ts b/src/utils/transferImage.ts
@@ -14,6 +14,7 @@ const images: Record<string, ArrayBuffer> = {};
 export async function transferImage(
   this: VillaBot,
   url: string,
+  villaId: string,
 ): Promise<string> {
   const { hostname, protocol } = new URL(url);
   let hash: string | undefined, sourceUrl: string;
@@ -87,9 +88,13 @@ export async function transferImage(
         await this.axios.axios<API.TransferImage.Response>(
           "/vila/api/bot/platform/transferImage",
           {
+            method: "POST",
             data: defineStruct<API.TransferImage.Request>({
               url: sourceUrl,
             }),
+            headers: {
+              "x-rpc-bot_villa_id": villaId,
+            },
             validateStatus: (status) =>
               (status >= 200 && status < 300) || status === 429,
           },