WindyMadman/novelai-bot
1
0
Fork 0
mirror of https://github.com/koishijs/novelai-bot synced 2025-01-07 07:06:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: some resources do not support the HEAD method (#250)

Browse Source
This commit is contained in:
idranme 2024-06-04 03:31:37 +08:00 committed by GitHub
parent 36ce61d9aa
commit 0f1c140209
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/utils.ts
View File

@ -42,15 +42,15 @@ export async function download(ctx: Context, url: string, headers = {}): Promise
const base64 = arrayBufferToBase64(data)
return { buffer: data, base64, dataUrl: `data:${mime};base64,${base64}` }
} else {
const head = await ctx.http.head(url, { headers })
if (+head.get('content-length') > MAX_CONTENT_SIZE) {
const image = await ctx.http(url, { responseType: 'arraybuffer', headers })
if (+image.headers.get('content-length') > MAX_CONTENT_SIZE) {
throw new NetworkError('.file-too-large')
}
const mimetype = head.get('content-type')
const mimetype = image.headers.get('content-type')
if (!ALLOWED_TYPES.includes(mimetype)) {
throw new NetworkError('.unsupported-file-type')
}
const buffer = await ctx.http.get(url, { responseType: 'arraybuffer', headers })
const buffer = image.data
const base64 = arrayBufferToBase64(buffer)
return { buffer, base64, dataUrl: `data:${mimetype};base64,${base64}` }
}