@satorijs/adapter-lark 3.7.1 → 3.7.3

package/src/types/internal.ts

@@ -10,12 +10,15 @@ export interface BaseResponse {
   msg: string
 }
 
-type Method = 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH'
+export interface InternalConfig {
+  multipart?: boolean
+  noExtractData?: boolean
+}
 
 export class Internal {
   constructor(private bot: LarkBot) {}
 
-  private assertResponse(response: HTTP.Response<BaseResponse>) {
+  private _assertResponse(response: HTTP.Response<BaseResponse>) {
     if (!response.data.code) return
     this.bot.logger.debug('response: %o', response.data)
     const error = new HTTP.Error(`request failed`)
@@ -23,10 +26,23 @@ export class Internal {
     throw error
   }
 
-  static define(routes: Dict<Partial<Record<Method, string | string[]>>>, extractData = true) {
+  private _buildData(arg: object, options: InternalConfig) {
+    if (options.multipart) {
+      const form = new FormData()
+      for (const key in arg) {
+        const value = arg[key]
+        form.append(key, value, value instanceof File ? value.name : undefined)
+      }
+      return form
+    } else {
+      return arg
+    }
+  }
+
+  static define(routes: Dict<Partial<Record<HTTP.Method, string | string[]>>>, options: InternalConfig = {}) {
     for (const path in routes) {
       for (const key in routes[path]) {
-        const method = key as Method
+        const method = key as HTTP.Method
         for (const name of makeArray(routes[path][method])) {
           Internal.prototype[name] = async function (this: Internal, ...args: any[]) {
             const raw = args.join(', ')
@@ -39,17 +55,17 @@ export class Internal {
               if (method === 'GET' || method === 'DELETE') {
                 config.params = args[0]
               } else {
-                config.data = args[0]
+                config.data = this._buildData(args[0], options)
               }
             } else if (args.length === 2 && method !== 'GET' && method !== 'DELETE') {
-              config.data = args[0]
+              config.data = this._buildData(args[0], options)
               config.params = args[1]
             } else if (args.length > 1) {
               throw new Error(`too many arguments for ${path}, received ${raw}`)
             }
             const response = await this.bot.http(method, url, config)
-            this.assertResponse(response)
-            return extractData ? response.data.data : response.data
+            this._assertResponse(response)
+            return options.noExtractData ? response.data : response.data.data
           }
         }
       }