// ==UserScript==
// @name         E-Hentai better viewer
// @namespace    https://github.com/lifegpc/userscript
// @version      0.3.0
// @description  Add a viewer to view original picture on website.
// @author       lifegpc
// @match        https://*.e-hentai.org/s/*/*
// @match        https://*.exhentai.org/s/*/*
// @icon         https://e-hentai.org/favicon.ico
// @grant        GM_getResourceText
// @grant        GM_addElement
// @grant        GM_addStyle
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_registerMenuCommand
// @require      https://github.com/lifegpc/viewerjs/raw/main/dist/viewer.min.js
// @require      https://github.com/emn178/js-sha512/raw/v0.8.0/build/sha512.min.js
// @require      https://openuserjs.org/src/libs/sizzle/GM_config.js
// @resource     viewercss https://github.com/lifegpc/viewerjs/raw/main/dist/viewer.min.css
// @run-at       document-start
// ==/UserScript==
GM_config.init({
    id: 'e-hentai',
    fields: {
        enableRedirectAPI: {
            type: 'checkbox',
            label: 'Enable redirect api. This need a thirdparty server.',
            default: false
        },
        APIEndpoint: {
            type: 'hidden'
        },
        tempAPIEndpoint: {
            label: 'The endpoint of the API:',
            type: 'text',
            save: false
        },
        APISecret: {
            label: 'The API secret:',
            type: 'text'
        },
        LoadOriginalWhenOpen: {
            type: 'checkbox',
            label: 'Load original pictures when opening the viewer modal.',
            default: false
        },
    },
    events: {
        init: () => {
            GM_config.set("tempAPIEndpoint", GM_config.get("APIEndpoint"))
        },
        open: () => {
            GM_config.set("tempAPIEndpoint", GM_config.get("APIEndpoint"))
        },
        save: (values) => {
            console.log(values);
            if (GM_config.get('enableRedirectAPI')) {
                if (values.tempAPIEndpoint !== null) {
                    try {
                        GM_config.set("APIEndpoint", new URL(values.tempAPIEndpoint).toString());
                    } catch (e) {
                        GM_config.set("enableRedirectAPI", false);
                        console.log(e);
                    }
                }
                if (!GM_config.get("APISecret")) {
                    GM_config.set("enableRedirectAPI", false);
                }
            }
        }
    }
})
GM_registerMenuCommand("Edit Settings", () => { GM_config.open() }, "e");
GM_addStyle(GM_getResourceText("viewercss"));
function indirectEval(script) {
    return eval(`"use strict";${script}`);
}
function parse_cookies() {
    let cookies = document.cookie.split(";");
    let o = {};
    for (let c of cookies) {
        c = c.trim().split("=");
        let k = c[0];
        let v = c.slice(1).join("=");
        o[k] = v;
    }
    return o;
}
function find_script() {
    let cols = document.getElementsByTagName("script");
    for (let col of cols) {
        if (col.innerHTML.startsWith("var ")) {
            return col.innerHTML;
        }
    }
}
/**
 * encodeURIComponent as python's quote_plus behavoir
 * @param {string} str
 * @returns
 */
function py_quote(str) {
    let s = encodeURIComponent(str);
    while (s.includes('%20')) {
        s = s.replace('%20', '+');
    }
    while (s.includes('!')) {
        s = s.replace('!', '%21');
    }
    while (s.includes("'")) {
        s = s.replace("'", '%27');
    }
    while (s.includes('(')) {
        s = s.replace('(', '%28');
    }
    while (s.includes(')')) {
        s = s.replace(')', '%29');
    }
    while (s.includes('*')) {
        s = s.replace('*', '%2A');
    }
    return s;
}
class URLParams extends URLSearchParams {
    toString() {
        let r = "";
        for (let p of this.entries()) {
            if (r.length) r += "&";
            r += py_quote(p[0]) + "=" + py_quote(p[1]);
        }
        return r;
    }
}
/**
 * 发送GET请求
 * @param {string} url 网站
 * @param {Object<string, string>|Array<Array<string>|string>} data 字典
 * @param {(content: string)=>void} callback 回调函数
 * @param {()=>void} failedCallback 失败回调函数
 * @param {Object<string, string>} headers HTTP头部
 */
function get(url, data, callback, failedCallback, headers) {
    var xhr = new XMLHttpRequest();
    var uri = new URL(url, window.location.href);
    if (data == undefined);
    else if (Array.isArray(data)) {
        for (let i = 0; i < data.length; i++) {
            var pair = data[i];
            if (Array.isArray(pair)) {
                uri.searchParams.append(pair[0], pair.length > 1 ? pair[1] : "");
            } else if (typeof pair == "string") {
                uri.searchParams.append(pair, "");
            }
        }
    } else {
        Object.getOwnPropertyNames(data).forEach((key) => {
            if (typeof data[key] == "string")
                uri.searchParams.append(key, data[key]);
        })
    }
    xhr.open("GET", uri.href);
    if (callback != undefined) xhr.onload = () => {
        callback(xhr.responseText);
    };
    if (failedCallback != undefined) xhr.onerror = failedCallback;
    if (headers != undefined) {
        Object.getOwnPropertyNames(headers).forEach((key) => {
            if (typeof headers[key] == "string")
                xhr.setRequestHeader(key, headers[key])
        })
    }
    try {
        xhr.send();
    } catch (e) {
        if (failedCallback != undefined) failedCallback();
    }
}
/**
 * Generate sign for data
 * @param {Object<string, string | Array<string>>|FormData} data Data
 * @param {string} secret Secret
 * @param {(data: string) => string|undefined} hash The hash function
 * @returns {string}
 */
function genGetSign(data, secret, hash) {
    if (!hash) hash = sha512;
    /**@type {Array<{k: string, v: string}>} */
    let arr = [];
    if (data.constructor.name == "FormData") {
        for (let pair of data.entries()) {
            if (typeof pair[1] != "string") continue;
            arr.push({ k: pair[0], v: pair[1] });
        }
    } else {
        Object.getOwnPropertyNames(data).forEach((key) => {
            let v = data[key];
            if (typeof v == "string") arr.push({ k: key, v: v });
            else if (Array.isArray(v)) {
                v.forEach((v) => {
                    if (typeof v == "string") arr.push({ k: key, v: v });
                })
            }
        })
    }
    arr.sort((a, b) => {
        return a.k == b.k ? a.v == b.v ? 0 : a.v > b.v ? 1 : -1 : a.k > b.k ? 1 : -1;
    })
    let par = new URLParams();
    arr.forEach((v) => {
        par.append(v.k, v.v);
    })
    return hash(secret + par.toString());
}
function redirect_url(url) {
    return new Promise((resolve, reject) => {
        let api = GM_config.get("APIEndpoint");
        let screct = GM_config.get("APISecret");
        if (!api || !screct) {
            reject("Endpoint or screct not found.");
            return;
        }
        let data = {'t': url, 'ar': '0', 'rraj': '1', 'c': JSON.stringify(parse_cookies()), 'r': 'https://e-hentai.org/' };
        data['sign'] = genGetSign(data, screct);
        get(api, data, (c) => {
            resolve(JSON.parse(c)['location']);
        }, () => {
            reject("Error");
        })
    })
}
let base_url = undefined;
let cur_img = null;
let cur_viewer = null;
let script_text = undefined;
let api_url = undefined;
let gid = undefined;
let showkey = undefined;
let img_keys = {};
let original_url = null;
let redirected_url = null;
let cur_page = null;
let total_page = null;
function get_api_url() {
    if (api_url !== undefined) return api_url;
    if (script_text === undefined) {
        script_text = find_script();
    }
    if (script_text === undefined) return undefined;
    api_url = indirectEval(`${script_text};api_url`);
    return api_url;
}
function get_base_url() {
    if (base_url !== undefined) return base_url;
    base_url = new URL(document.location.href)
    return base_url;
}
function get_gid() {
    if (gid !== undefined) return gid;
    if (script_text === undefined) {
        script_text = find_script();
    }
    if (script_text === undefined) return undefined;
    gid = indirectEval(`${script_text};gid`);
    return gid;
}
function get_showkey() {
    if (showkey !== undefined) return showkey;
    if (script_text === undefined) {
        script_text = find_script();
    }
    if (script_text === undefined) return undefined;
    showkey = indirectEval(`${script_text};showkey`);
    return showkey;
}
function api_call(data) {
    return new Promise((resolve, reject) => {
        let api_url = get_api_url();
        if (!api_url) {
            reject("api_url not found.");
            return;
        }
        let xhr = new XMLHttpRequest();
        xhr.open("POST", api_url);
        xhr.withCredentials = true;
        xhr.onload = (ev) => {
            resolve(xhr);
        };
        xhr.onerror = (ev) => {
            reject(ev);
        }
        xhr.onabort = (ev) => {
            reject(ev);
        }
        xhr.send(JSON.stringify(data));
    })
}
function genUrl(gid, imgkey, page) {
    return new URL(`/s/${imgkey}/${gid}-${page}`, get_base_url())
}
async function loadPage(page) {
    let gid = get_gid();
    if (!gid) throw Error("gid not found.");
    let showkey = get_showkey();
    if (!showkey) throw Error("showkey not found.");
    let imgkey = img_keys[page];
    if (!imgkey) return;
    history.pushState({ page, imgkey }, document.title, genUrl(gid, imgkey, page).toString());
    let xhr = await api_call({ method: "showpage", gid, page, imgkey, showkey });
    let a = JSON.parse(xhr.responseText);
    if (cur_viewer) cur_viewer.destroy();
    cur_img = null;
    cur_viewer = null;
    original_url = null;
    redirected_url = null;
    document.getElementById("i1").style.width = a.x + "px";
    document.getElementById("i2").innerHTML = a.n + a.i;
    document.getElementById("i3").innerHTML = a.i3;
    document.getElementById("i4").innerHTML = a.i + a.n;
    document.getElementById("i5").innerHTML = a.i5;
    document.getElementById("i6").innerHTML = a.i6;
    document.getElementById("i7").innerHTML = a.i7;
    update_window_extents(parseInt(a.x), parseInt(a.y));
    load();
}
function replaceLink(a) {
    let ohtml = a.innerHTML;
    let na = document.createElement('a');
    let matched = a.href.match(/e[x-]hentai\.org\/s\/([^\/]+)\/\d+-(\d+)/);
    if (!matched) return;
    let num = parseInt(matched[2]);
    let img_key = matched[1];
    img_keys[num] = img_key;
    na.innerHTML = ohtml;
    na.addEventListener("click", () => { loadPage(num); });
    a.replaceWith(na);
}
async function loadOriginalImage() {
    let url = original_url;
    if (!url) return false;
    if (GM_config.get("enableRedirectAPI")) {
        if (redirected_url) {
            url = redirected_url;
        } else {
            redirected_url = await redirect_url(url);
            url = redirected_url;
        }
    }
    cur_img.src = url;
    return true;
}
let load = () => {
    let img = document.getElementById("img");
    if (img == null) {
        setTimeout(load, 100);
        return;
    }
    cur_page = parseInt(document.querySelector('#i2 span').innerText);
    total_page = parseInt(document.querySelector('#i2 span:last-child').innerText);
    cur_img = img;
    let parent = img.parentElement;
    parent.replaceWith(img);
    let options = {
        title: () => {
            return document.getElementsByTagName("h1")[0].innerText
        },
        keyboard: false,
        navbar: false,
        loop: false,
        toolbar: { prev: { show: true, click: prevImage }, zoomIn: { show: true }, zoomOut: { show: true }, oneToOne: { show: true }, next: { show: true, click: nextImage } }
    };
    /**@type {HTMLAnchorElement}*/
    let original = document.querySelector("#i7>a");
    if (original != null) {
        let ourl = original.href;
        original_url = ourl;
    }
    console.log(options);
    let viewer = new Viewer(img, options);
    let click = async () => {
        img.removeEventListener("click", click);
        if (GM_config.get("LoadOriginalWhenOpen")) {
            await loadOriginalImage();
        }
        await viewer.init();
        console.log("Inited complete");
        await viewer.show();
    };
    cur_viewer = viewer;
    img.addEventListener("click", click);
    let links = document.querySelectorAll(".sn a");
    for (let i of links) {
        replaceLink(i);
    }
    console.log(img_keys);
};
window.addEventListener("DOMContentLoaded", load);
window.addEventListener("DOMContentLoaded", () => {
    // Remove default keyboard binding.
    // Add custom event listener.
    let scr = document.createElement("script");
    scr.innerHTML = 'document.onkeydown = () => {};document.addEventListener("update_window_extents", (ev) => {xres=ev.xres;yres=ev.yres;update_window_extents();});';
    document.head.append(scr);
});
function zoomIn() {
    if (cur_viewer) cur_viewer.zoom(-0.1);
}
function zoomOut() {
    if (cur_viewer) cur_viewer.zoom(0.1);
}
function moveDown() {
    if (cur_viewer) cur_viewer.move(0, -10);
}
function moveUp() {
    if (cur_viewer) cur_viewer.move(0, 10);
}
function moveLeft() {
    if (cur_viewer) cur_viewer.move(10, 0);
}
function moveRight() {
    if (cur_viewer) cur_viewer.move(-10, 0);
}
function prevImage() {
    if (cur_page === null) return;
    loadPage(cur_page - 1)
}
function nextImage() {
    if (cur_page === null) return;
    loadPage(cur_page + 1)
}
function update_window_extents(xres, yres) {
    let ev = new Event("update_window_extents");
    ev.xres = xres;
    ev.yres = yres;
    document.dispatchEvent(ev);
}
window.addEventListener("keydown", (e) => {
    if (!e.ctrlKey && !e.shiftKey) {
        if (e.key == "ArrowDown") {
            zoomIn();
        } else if (e.key == "ArrowUp") {
            zoomOut();
        } else if (e.key == "ArrowLeft") {
            prevImage();
        } else if (e.key == "ArrowRight") {
            nextImage();
        }
    } else if (e.ctrlKey && !e.shiftKey) {
        if (e.key == "ArrowDown") {
            moveDown();
        } else if (e.key == "ArrowUp") {
            moveUp();
        } else if (e.key == "ArrowLeft") {
            moveLeft();
        } else if (e.key == "ArrowRight") {
            moveRight();
        }
    }
})