ArtPlayer/packages/artplayer-plugin-danmuku/src/danmuku.js

import i18n from './i18n';
import { opacity, size, speed } from './setting';
import { filter, getRect, getDanmuRef } from './utils';
import { bilibiliDanmuParseFromUrl } from './bilibili';
import getDanmuTop from './getDanmuTop';

export default class Danmuku {
    constructor(art, option) {
        art.i18n.update(i18n);
        art.setting.add(opacity);
        art.setting.add(size);
        art.setting.add(speed);
        this.art = art;
        this.queue = [];
        this.refs = [];
        this.option = {};
        this.config(option);
        this.isStop = false;
        this.animationFrameTimer = null;
        this.$danmuku = art.template.$danmuku;
        art.on('video:play', this.start.bind(this));
        art.on('video:playing', this.start.bind(this));
        art.on('video:pause', this.stop.bind(this));
        art.on('video:waiting', this.stop.bind(this));
        art.on('resize', this.resize.bind(this));
        art.on('destroy', this.stop.bind(this));
        if (typeof this.option.danmuku === 'function') {
            this.option.danmuku().then(danmus => {
                danmus.forEach(this.emit.bind(this));
                art.emit('artplayerPluginDanmuku:loaded');
            });
        } else if (typeof this.option.danmuku === 'string') {
            bilibiliDanmuParseFromUrl(this.option.danmuku).then(danmus => {
                danmus.forEach(this.emit.bind(this));
                art.emit('artplayerPluginDanmuku:loaded');
            });
        } else {
            this.option.danmuku.forEach(this.emit.bind(this));
            art.emit('artplayerPluginDanmuku:loaded');
        }
    }

    static get option() {
        return {
            danmuku: [],
            speed: 5,
            maxlength: 50,
            margin: [10, 100],
            opacity: 1,
            fontSize: 25,
        };
    }

    static get scheme() {
        return {
            danmuku: 'array|function|string',
            speed: 'number',
            maxlength: 'number',
            margin: 'array',
            opacity: 'number',
            fontSize: 'number',
        };
    }

    config(option) {
        const {
            utils: { clamp },
            validator,
        } = this.art.constructor;
        this.option = Object.assign({}, Danmuku.option, this.option, option);
        validator(this.option, Danmuku.scheme);
        this.option.speed = clamp(this.option.speed, 1, 10);
        this.option.maxlength = clamp(this.option.maxlength, 10, 100);
        this.option.margin[0] = clamp(this.option.margin[0], 0, 100);
        this.option.margin[1] = clamp(this.option.margin[1], 0, 100);
        this.option.opacity = clamp(this.option.opacity, 0, 1);
        this.option.fontSize = clamp(this.option.fontSize, 12, 30);
        this.art.emit('artplayerPluginDanmuku:config', this.option);
    }

    continue() {
        filter(this.queue, 'stop', danmu => {
            danmu.$state = 'emit';
            danmu.$lastStartTime = Date.now();
            switch (danmu.mode) {
                case 'scroll':
                    danmu.$ref.style.transform = `translateX(${-danmu.$restWidth}px) translateY(0px) translateZ(0px)`;
                    danmu.$ref.style.transition = `transform ${danmu.$restTime}s linear 0s`;
                    break;
                default:
                    break;
            }
        });
    }

    suspend() {
        const { $player } = this.art.template;
        filter(this.queue, 'emit', danmu => {
            danmu.$state = 'stop';
            switch (danmu.mode) {
                case 'scroll': {
                    const { left: playerLeft, width: playerWidth } = getRect($player);
                    const { left: danmuLeft } = getRect(danmu.$ref);
                    const translateX = playerWidth - (danmuLeft - playerLeft) + 5;
                    danmu.$ref.style.transform = `translateX(${-translateX}px) translateY(0px) translateZ(0px)`;
                    danmu.$ref.style.transition = 'transform 0s linear 0s';
                    break;
                }
                default:
                    break;
            }
        });
    }

    resize() {
        const { $player } = this.art.template;
        const danmuLeft = getRect($player, 'width');
        filter(this.queue, 'wait', danmu => {
            if (danmu.$ref) {
                danmu.$ref.style.border = 'none';
                danmu.$ref.style.left = `${danmuLeft}px`;
                danmu.$ref.style.marginLeft = '0px';
                danmu.$ref.style.transform = 'translateX(0px) translateY(0px) translateZ(0px)';
                danmu.$ref.style.transition = 'transform 0s linear 0s';
            }
        });
    }

    getDanmuTop(mode) {
        const { $player } = this.art.template;
        const { left: playerLeft, top: playerTop, height: playerHeight, width: playerWidth } = getRect($player);
        const danmus = this.queue.filter(danmu => danmu.mode === mode && danmu.$state === 'emit');

        if (danmus.length === 0) {
            return this.option.margin[0];
        }

        const danmusBySort = danmus
            .map(danmu => {
                const { left: danmuLeft, top: danmuTop, width: danmuWidth, height: danmuHeight } = getRect(danmu.$ref);
                const top = danmuTop - playerTop;
                const left = danmuLeft - playerLeft;
                const right = playerWidth - left - danmuWidth;
                return {
                    top,
                    left,
                    right,
                    height: danmuHeight,
                    width: danmuWidth,
                };
            })
            .sort((prev, next) => {
                return prev.top - next.top;
            });

        danmusBySort.unshift({
            top: 0,
            left: 0,
            right: 0,
            height: this.option.margin[0],
            width: playerWidth,
        });

        danmusBySort.push({
            top: playerHeight - this.option.margin[1],
            left: 0,
            right: 0,
            height: this.option.margin[1],
            width: playerWidth,
        });

        return getDanmuTop(danmusBySort);
    }

    update() {
        const {
            player,
            template: { $player },
        } = this.art;
        this.animationFrameTimer = window.requestAnimationFrame(() => {
            if (player.playing) {
                const danmuLeft = getRect($player, 'width');

                filter(this.queue, 'emit', danmu => {
                    danmu.$restTime -= (Date.now() - danmu.$lastStartTime) / 1000;
                    danmu.$lastStartTime = Date.now();
                    if (danmu.$restTime <= 0) {
                        danmu.$state = 'wait';
                        danmu.$ref.style.border = 'none';
                        danmu.$ref.style.left = `${danmuLeft}px`;
                        danmu.$ref.style.marginLeft = '0px';
                        danmu.$ref.style.transform = 'translateX(0px) translateY(0px) translateZ(0px)';
                        danmu.$ref.style.transition = 'transform 0s linear 0s';
                    }
                });

                this.queue
                    .filter(danmu => {
                        return (
                            player.currentTime + 0.1 >= danmu.time &&
                            danmu.time >= player.currentTime - 0.1 &&
                            danmu.$state === 'wait'
                        );
                    })
                    .forEach(danmu => {
                        danmu.$state = 'emit';
                        danmu.$ref = getDanmuRef(this.queue);
                        this.$danmuku.appendChild(danmu.$ref);
                        if (!this.refs.includes(danmu.$ref)) {
                            this.refs.push(danmu.$ref);
                        }
                        danmu.$ref.style.opacity = this.option.opacity;
                        danmu.$ref.style.fontSize = `${this.option.fontSize}px`;
                        danmu.$ref.innerText = danmu.text;
                        danmu.$ref.style.color = danmu.color;
                        danmu.$ref.style.border = danmu.border ? `1px solid ${danmu.color}` : 'none';
                        danmu.$restTime = this.option.speed;
                        danmu.$lastStartTime = Date.now();
                        const danmuWidth = getRect(danmu.$ref, 'width');
                        const danmuTop = this.getDanmuTop(danmu.mode);
                        switch (danmu.mode) {
                            case 'scroll': {
                                danmu.$restWidth = danmuLeft + danmuWidth + 5;
                                danmu.$ref.style.left = `${danmuLeft}px`;
                                danmu.$ref.style.top = `${danmuTop}px`;
                                danmu.$ref.style.transform = `translateX(${-danmu.$restWidth}px) translateY(0px) translateZ(0px)`;
                                danmu.$ref.style.transition = `transform ${danmu.$restTime}s linear 0s`;
                                break;
                            }
                            case 'static':
                                danmu.$ref.style.top = `${danmuTop}px`;
                                danmu.$ref.style.left = '50%';
                                danmu.$ref.style.marginLeft = `-${danmuWidth / 2}px`;
                                break;
                            default:
                                break;
                        }
                    });
            }

            if (!this.isStop) {
                this.update();
            }
        });
    }

    stop() {
        this.isStop = true;
        this.suspend();
        window.cancelAnimationFrame(this.animationFrameTimer);
        this.art.emit('artplayerPluginDanmuku:stop');
    }

    start() {
        this.isStop = false;
        this.continue();
        this.update();
        this.art.emit('artplayerPluginDanmuku:start');
    }

    show() {
        this.$danmuku.style = 'block';
        this.art.emit('artplayerPluginDanmuku:show');
    }

    hide() {
        this.$danmuku.style = 'none';
        this.art.emit('artplayerPluginDanmuku:hide');
    }

    emit(danmu) {
        const { notice, player, i18n } = this.art;
        if (!danmu.text.trim()) {
            notice.show(i18n.get('Danmu text cannot be empty'));
            return;
        }

        if (danmu.text.length > this.option.maxlength) {
            notice.show(`${i18n.get('The length of the danmu does not exceed')} ${this.option.maxlength}`);
            return;
        }

        if (typeof danmu.time !== 'number') {
            danmu.time = player.currentTime;
        }

        this.queue.push({
            ...danmu,
            $state: 'wait',
            $ref: null,
            $restTime: 0,
            $lastStartTime: 0,
            $restWidth: 0,
        });
    }
}