diff --git a/src/index.ts b/src/index.ts index f343699..cf146c7 100644 --- 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 index 9b3723e..8b264ac 100644 --- 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 } }