Skip to content

Commit 45ecbad

Browse files
committed
feat: doesn't have to be buffer
1 parent 77139bd commit 45ecbad

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

packages/crawl/api.ts

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ interface InfoFileConfig extends InfoCommonConfig {
9494
fileName: string
9595
filePath: string
9696
data: Buffer
97-
}) => Promise<Buffer> | Buffer | void)
97+
}) => Promise<Buffer | void> | Buffer | void)
9898
| undefined
9999
}
100100

@@ -199,7 +199,7 @@ interface CrawlFileConfig {
199199
fileName: string
200200
filePath: string
201201
data: Buffer
202-
}) => Promise<Buffer> | Buffer | void)
202+
}) => Promise<Buffer | void> | Buffer | void)
203203
| undefined
204204
onCrawlItemComplete:
205205
| ((crawlDataSingleResult: CrawlDataSingleResult<any>) => void)
@@ -1074,7 +1074,9 @@ function fileSingleResultHandle(
10741074

10751075
let saveFileItemPending
10761076
if (isPromise(onBeforeSaveItemFileResult)) {
1077-
saveFileItemPending = onBeforeSaveItemFileResult!.then(saveFile)
1077+
saveFileItemPending = onBeforeSaveItemFileResult!
1078+
.then((res) => (isBuffer(res) ? res : data))
1079+
.then(saveFile)
10781080
} else if (isBuffer(onBeforeSaveItemFileResult)) {
10791081
saveFileItemPending = saveFile(onBeforeSaveItemFileResult)
10801082
} else {

packages/crawl/types/api.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ export interface CrawlFileAdvancedConfig extends CrawlCommonConfig {
178178
fileName: string
179179
filePath: string
180180
data: Buffer
181-
}) => Promise<Buffer> | Buffer | void
181+
}) => Promise<Buffer | void> | Buffer | void
182182
}
183183

184184
/* API Result */

0 commit comments

Comments
 (0)