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

feat: 支持下载 file 协议的图片 (#210)

Browse Source
This commit is contained in:
idanran 2023-10-09 22:00:49 +08:00 committed by GitHub
parent 5c8a76c44b
commit cd55f2a649
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/utils.ts
View File

@ -34,17 +34,13 @@ const MAX_CONTENT_SIZE = 10485760
const ALLOWED_TYPES = ['image/jpeg', 'image/png']
export async function download(ctx: Context, url: string, headers = {}): Promise<ImageData> {
if (url.startsWith('data:')) {
const [, type, base64] = url.match(/^data:(image\/\w+);base64,(.*)$/)
if (!ALLOWED_TYPES.includes(type)) {
if (url.startsWith('data:') || url.startsWith('file:')) {
const { mime, data } = await ctx.http.file(url)
if (!ALLOWED_TYPES.includes(mime)) {
throw new NetworkError('.unsupported-file-type')
}
const binary = atob(base64)
const result = new Uint8Array(binary.length)
for (let i = 0; i < binary.length; i++) {
result[i] = binary.charCodeAt(i)
}
return { buffer: result, base64, dataUrl: url }
const base64 = arrayBufferToBase64(data)
return { buffer: data, base64, dataUrl: `data:${mime};base64,${base64}` }
} else {
const head = await ctx.http.head(url, { headers })
if (+head['content-length'] > MAX_CONTENT_SIZE) {