// ==UserScript==
// @name         修复B站评论区楼层
// @namespace    MotooriKashin
// @version      0.0.3
// @description  修复评论区评论的楼层号，包括视频、动态、专栏、话题、活动……
// @author       MotooriKashin
// @match        *://*.bilibili.com/*
// @grant        none
// @license      MIT License
// @downloadURL none
// ==/UserScript==

(function() {
    'use strict';

    let src, mode, type;

    const url = {
        reply : "https://api.bilibili.com/x/v2/reply",
        replymain : "https://api.bilibili.com/x/v2/reply/main",
        replycursor : "https://api.bilibili.com/x/v2/reply/reply/cursor"
    }
    const xhr = (url) => {
        return new Promise((resolve, reject) => {
            let xhr = new XMLHttpRequest();
            xhr.open('get', url, true);
            xhr.withCredentials = true;
            xhr.onload = () => {
                if (xhr.status >= 200 && xhr.status < 300) resolve(xhr.response)
                else reject({status: xhr.status, statusText: xhr.statusText})
            };
            xhr.onerror = () => reject({status: xhr.status, statusText: xhr.statusText})
            xhr.send();
        });
    }
    const obj2search = (url, obj) =>{
        if (obj) {
            let arr = [],i = 0;
            for (let key in obj) {
                if(obj[key] !== "" && obj[key] !== "undefined" && obj[key] !== null) {
                    arr[i] = key + "=" + obj[key];
                    i++;
                }
            }
            url = url + "?" + arr.join("&");
        }
        return url;
    }
    const setReplyFloor = async (link) => {
        src = "";
        try {
            let oid, sort, pn, data;
            link = link.split('?')[1].split('&');
            for (let i = 0; i < link.length; i++) {
                let key = link[i].split('=');
                if (key[0] == "oid") oid = key[1];
                if (key[0] == "sort") sort = key[1];
                if (key[0] == "pn") pn = key[1];
                if (key[0] == "type") type = key[1];
            }
            if (sort == 0) mode = 1;
            if (sort == 1) return;
            if (sort == 2) mode = 3;
            if (sort == 2) data = await xhr(obj2search(url.replymain, {"oid": oid,"next": pn,"type": type,"mode": mode}));
            else {
                if (pn == 1) data = await xhr(obj2search(url.replymain, {"oid": oid,"type": type,"mode": mode}));
                else{
                    pn = pn - 1;
                    data = await xhr(obj2search(url.reply, {"type": type,"sort": sort,"oid": oid,"pn": pn}));
                    data = JSON.parse(data).data;
                    let i = data.replies.length - 1;
                    oid = data.replies[0].oid;
                    let root = data.replies[i].rpid;
                    data = await xhr(obj2search(url.replycursor, {"oid": oid,"root": root,"type": type}));
                    data = JSON.parse(data).data;
                    oid = data.root.oid;
                    let next = data.root.floor;
                    data = await xhr(obj2search(url.replymain, {"oid": oid,"next": next,"type": type,"mode": mode}));
                }
            }
            data = JSON.parse(data).data;
            let floor = {}, top = data.top, hots = data.hots, replies = data.replies;
            let list_item = document.getElementsByClassName("list-item");
            let main_floor = document.getElementsByClassName("main-floor");
            if (hots && hots[0]) for (let i = 0; i < hots.length; i++) floor[hots[i].rpid] = hots[i].floor;
            if (replies && replies[0]) for (let i = 0;i < replies.length; i++) floor[replies[i].rpid] = replies[i].floor;
            if (top && top.admin) floor[top.admin.rpid] = top.admin.floor;
            if (top && top.upper) floor[top.upper.rpid] = top.upper.floor;
            if (top && top.vote) floor[top.vote.rpid] = top.vote.floor;
            if (main_floor[0]) {
                for (let i = 0; i < main_floor.length; i++) {
                    let rpid = main_floor[i].getAttribute("id").split('_')[2];
                    if (rpid in floor) main_floor[i].getElementsByClassName("floor-num")[0].innerText = "#" + floor[rpid];
                }
            }
            if (list_item[0]) {
                for (let i = 0; i<list_item.length; i++) {
                    let rpid = list_item[i].getAttribute("data-id");
                    if (rpid in floor) {
                        let node = list_item[i].getElementsByClassName("info")[0];
                        let span = document.createElement("span");
                        span.setAttribute("class", "floor");
                        span.innerText = "#" + floor[rpid];
                        node.insertBefore(span,node.firstChild);
                    }
                }
            }
        } catch(e) {console.error(e)}
    }
    document.addEventListener("DOMNodeInserted",(msg) => {
        if (msg.target.src && msg.target.src.startsWith('https://api.bilibili.com/x/v2/reply?')) src = msg.target.src;
        if (src && msg.target.className && (msg.target.className == "main-floor" || msg.target.className == "list-item reply-wrap ")) setReplyFloor(src);
    });
})();