The fetchFile API can provide the extension of the downloaded file

coder-hxl · Feb 16, 2023 · d69df02 · d69df02
1 parent 1457f7e
commit d69df02
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -146,11 +146,11 @@ export default class XCrawl {
       const { id, headers, data } = requestResItem
 
       const mimeType = headers['content-type'] ?? ''
-      const suffix = mimeType.split('/').pop()
+      const fileExtension = fileConfig.extension ?? mimeType.split('/').pop()
       const fileName = new Date().getTime().toString()
       const filePath = path.resolve(
         fileConfig.storeDir,
-        `${fileName}.${suffix}`
+        `${fileName}.${fileExtension}`
       )
 
       try {

diff --git a/src/types.ts b/src/types.ts
@@ -77,6 +77,7 @@ export interface IFetchDataConfig extends IFetchBaseConifg {}
 export interface IFetchFileConfig extends IFetchBaseConifg {
   fileConfig: {
     storeDir: string
+    extension?: string
   }
 }