1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import { Config } from "./config.ts";
const ws = new WebSocket(Deno.args[0])
// deno-lint-ignore no-unused-vars
let config: Config = {} as unknown as Config
const supported = [
"youtube-channel-info"
]
function key_to_url(key: string): string {
const [kind, id] = key.split(":", 2)
if (kind == "youtube-channel-info") return `https://youtube.com/channel/${id}`
throw new Error(`unknown kind ${kind}`);
}
async function dfile(path: string, url: string) {
console.log(`thumbnail download ${path} from ${url}`);
const r = await fetch(url)
if (!r.body) return console.error("thumbnail download failed");
await Deno.writeFile(path, new Uint8Array(await r.arrayBuffer()))
}
async function save_infojson(url: string, key: string, data: { [key: string]: unknown }) {
if (!data.output) throw new Error("output dir missing");
const output = data.output
const infojson = `${output}/channel.info.json`;
const o = await new Deno.Command("yt-dlp", {
args: [
"--skip-download",
"--write-info-json",
"-o", `${output}/channel`,
"--playlist-items", "0",
url
],
stdout: "piped",
stderr: "inherit",
}).output()
if (!o.success) throw new Error("info json download failed");
ws.send(JSON.stringify({ t: "metadata", key, data: { progress: 0.5 } }))
const oj = JSON.parse(await Deno.readTextFile(infojson))
const avatar = oj.thumbnails?.find((t: { id: string }) => t.id == "avatar_uncropped") ?? oj?.thumbnails?.[oj.thumbnails?.length - 1]
const banner = oj.thumbnails?.find((t: { id: string }) => t.id == "banner_uncropped")
if (avatar) await dfile(`${output}/poster.jpeg`, avatar.url)
ws.send(JSON.stringify({ t: "metadata", key, data: { progress: 0.75 } }))
if (banner) await dfile(`${output}/backdrop.jpeg`, banner.url)
ws.send(JSON.stringify({ t: "metadata", key, data: { progress: 1 } }))
}
ws.onerror = () => console.error("ws error")
ws.onclose = () => {
console.error("ws closed")
Deno.exit(1)
}
ws.onopen = () => {
console.log("ws open");
ws.send(JSON.stringify({ t: "register", name: "yt-dlp channel info generator", task_kinds: supported }))
ws.send(JSON.stringify({ t: "accept" }))
}
ws.onmessage = async ev => {
if (typeof ev.data != "string") return
const p = JSON.parse(ev.data)
if (p.t == "config") config = p.config
if (p.t == "error") console.error(`error: ${p.message}`);
if (p.t == "work") {
const url = key_to_url(p.key)
if (!p.data.output) throw new Error("no output");
console.log(`got work ${p.key}`);
await save_infojson(url, p.key, p.data)
ws.send(JSON.stringify({ t: "complete", key: p.key }))
ws.send(JSON.stringify({ t: "save" }))
ws.send(JSON.stringify({ t: "accept" }))
}
}
|
