mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-01 05:49:54 +00:00
31 lines
1.0 KiB
TypeScript
31 lines
1.0 KiB
TypeScript
export async function getOggMetadata(file: File) {
|
|
const reader = new FileReader()
|
|
const read_process = new Promise(
|
|
(r) => (reader.onload = (event) => r(event?.target?.result))
|
|
)
|
|
reader.readAsArrayBuffer(file)
|
|
const arrayBuffer = (await read_process) as ArrayBuffer
|
|
const signature = String.fromCharCode(...new Uint8Array(arrayBuffer, 0, 4))
|
|
if (signature !== 'OggS') console.error('Invalid file signature.')
|
|
let oggs = 0
|
|
let header = ''
|
|
while (header.length < arrayBuffer.byteLength) {
|
|
const page = String.fromCharCode(
|
|
...new Uint8Array(arrayBuffer, header.length, header.length + 4096)
|
|
)
|
|
if (page.match('OggS\u0000')) oggs++
|
|
header += page
|
|
if (oggs > 1) break
|
|
}
|
|
let workflow, prompt
|
|
let prompt_s = header
|
|
.match(/prompt=(\{.*?(\}.*?\u0000))/s)?.[1]
|
|
?.match(/\{.*\}/)?.[0]
|
|
if (prompt_s) prompt = JSON.parse(prompt_s)
|
|
let workflow_s = header
|
|
.match(/workflow=(\{.*?(\}.*?\u0000))/s)?.[1]
|
|
?.match(/\{.*\}/)?.[0]
|
|
if (workflow_s) workflow = JSON.parse(workflow_s)
|
|
return { prompt, workflow }
|
|
}
|