userscript/BilibiliEmojisExtractor.user.js at main · jc3213/userscript · GitHub

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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
// ==UserScript==
// @name            Bilibili Emojis Extractor
// @name:zh         哔哩哔哩表情提取
// @namespace       https://github.com/jc3213/userscript
// @version         0.5
// @description     Extract official Emojis in users' comments
// @description:zh  提取评论区的官方表情的图片
// @author          jc3213
// @match           https://www.bilibili.com/video/*
// @match           https://www.bilibili.com/v/*
// @match           https://www.bilibili.com/bangumi/*
// @grant           GM_download
// ==/UserScript==

const callMap = new Map();

function sendMessage(aria2c, params) {
    return new Promise((resolve, reject) => {
        let id = crypto.randomUUID();
        let timer = setTimeout(() => {
            callMap.delete(id);
            reject( new Error('"Download With Aria2" is either not installed, disabled, or lower than v4.17.0.3548.') );
        }, 1000);
        callMap.set(id, (result) => {
            clearTimeout(timer);
            callMap.delete(id);
            resolve(result);
        });
        window.postMessage({ aria2c, id, params });
    });
}

function aria2Status() {
    return sendMessage('aria2c_status');
}

function aria2Download(params) {
    return sendMessage('aria2c_download', params);
}

window.addEventListener('message', (event) => {
    let { aria2c, id, result } = event.data;
    if (aria2c === 'aria2c_response') {
        callMap.get(id)?.(result);
    }
});

let keyword;

let result;

function extractImages(root) {
    let content = root.shadowRoot?.querySelector('bili-rich-text')?.shadowRoot;
    let images = content?.querySelectorAll('img');

    if (!images || images.length === 0) return;

    for (let { alt, src } of images) {
        if (!alt.includes(keyword)) return;
        let url = src.split('@')[0];
        let out = `${alt.slice(1, -1)}.${url.split('.').pop()}`;
        result.set(url, out);
    }
}

function extractEmoji() {
    keyword = prompt('输入表情包关键词：');

    result = new Map();

    let comments = document.getElementById('commentapp').children[0].shadowRoot.children[1].children[1];

    for (let com of comments.children) {
        let [comment, replies] = com.shadowRoot.children;
        extractImages(comment);
        for (let reply of replies.children[0].shadowRoot.children[0].children[0].children) {
            extractImages(reply);
        }
    }

    let { size } = result;

    if (size === 0) {
        alert('未找到所输入关键词对应的表情！');
        return;
    } else {
        alert(`找到${size}个表情……`);
    }

    aria2Status().then(() => {
        let session = [];
        for (let [url, out] of result) {
            session.push({ url, options: { out } });
        }
        aria2Download(session);
    }).catch(() => {
        let index = 0;
        for (let [url, name] of result) {
            setTimeout(() => GM_download(url, name), index++ * 200);
        }
    });
}

document.addEventListener('keydown', (event) => {
    if (event.altKey && event.shiftKey && event.code === 'KeyE') {
        extractEmoji();
    }
});