ねことくまとへび

// ==UserScript==
// @name         soul2naga
// @namespace    lions.blue
// @icon         http://1.gravatar.com/avatar/6fa3836d10d691125749472297cf516a
// @version      1.0.0
// @description  downloadlogsで取得する牌譜をNAGAで解析可能な形式に変換する
// @include      https://mahjongsoul.game.yo-star.com/*
// @include      https://game.mahjongsoul.com/*
// @include      https://majsoul.union-game.com/0/*
// ==/UserScript==

(function() {
    // ダウンロードリンクのhref属性において、牌譜データに先行する部分の文字列。
    const DOWNLOAD_HREF_PREFIX = "data:text/plain;charset=utf-8,";

    // 天鳳牌譜エディタのURLにおいて、牌譜データに先行する部分の文字列。
    const EDITOR_URL_PREFIX = "https://tenhou.net/6/#json=";

    // ダウンロードイベントの前に割り込んで牌譜データを書き換える。
    document.addEventListener("click", function(e) {
        const links = document.body.getElementsByTagName("a");
        for (let i = 0; i < links.length; i++) {
            if (isDownloadLink(links[i])) {
                const soulJson = fetchSoulJson(links[i]);
                const urls = createViewerUrls(soulJson);
                links[i].href = buildDownloadHref(urls);
                links[i].download = buildFileName(links[i].download);
                return;
            }
        }
    }, {capture: true});

    /**
     * オブジェクトをディープコピーする。
     * 
     * @param {Object} src コピー対象のオブジェクトを指定する。
     * @returns {Object} 複製したオブジェクトを返す。
     */
    function deepCopy(src) {
        // JSON文字列化してからオブジェクトに戻すことでディープコピーを実現する。
        return JSON.parse(JSON.stringify(src));
    }

    /**
     * ダウンロードリンクであるか判定する。
     * 
     * @param {HTMLElement} element 判定対象の要素を指定する。
     * @returns {Boolean} ダウンロードリンクの場合はtrue、それ以外の場合はfalseを返す。
     */
    function isDownloadLink(element) {
        // href属性の先頭部分で判定する。
        return element.href.startsWith(DOWNLOAD_HREF_PREFIX);
    }

    /**
     * ダウンロードリンクから雀魂の牌譜データを抽出する。
     * 
     * @param {HTMLElement} element ダウンロードリンクを指定する。
     * @returns {String} 雀魂の牌譜データを返す。
     */
    function fetchSoulJson(element) {
        // href属性から牌譜データを抽出する。
        return decodeURIComponent(element.href.replace(DOWNLOAD_HREF_PREFIX, ""));
    }

    /**
     * ダウンロードリンクのhref属性を組み立てる。
     * 
     * @param {Array<String>} urls 牌譜エディタのURL群を指定する。
     * @returns {String} ダウンロードリンクのhref属性を返す。
     */
    function buildDownloadHref(urls) {
        // ダウンロードリンクのhref属性を組み立てる。
        return DOWNLOAD_HREF_PREFIX + encodeURIComponent(urls.join("\n"));
    }

    /**
     * ダウンロードファイル名を組み立てる。
     * 
     * @param {String} baseFileName もともとのダウンロードファイル名を指定する。
     * @returns {String} 組み立てたダウンロードファイル名を返す。
     */
    function buildFileName(baseFileName) {
        // 卓名を雀魂っぽく変換し、拡張子を.txtに変更する。
        return toSoulTable(baseFileName).replace(".json", ".txt");
    }

    /**
     * 雀魂の牌譜JSONを牌譜エディタのURL群に変換する。
     * 
     * @param {String} soulJson 雀魂の牌譜JSONを指定する。
     * @returns {Array<String>} 牌譜エディタのURL群を返す。
     */
    function createViewerUrls(soulJson) {
        // 雀魂の牌譜JSONをオブジェクトに変換する。
        const soulPaifu = JSON.parse(soulJson);

        // title内の卓名を雀魂っぽく変換する。
        // 
        // 変換前のtitle:
        //     "title": [ "玉の間南喰赤", "2021/10/20 20:48:01" ]
        // 変換後のtitle:
        //     "title": [ "玉の間四人南", "2021/10/20 20:48:01" ]
        const title = deepCopy(soulPaifu.title);
        title[0] = toSoulTable(title[0]);

        const name = deepCopy(soulPaifu.name);
        const encodedName = name.map(function(v) {
            return encodeURIComponent(v);
        });

        // rule内の卓名を雀魂っぽく変換する。
        // 
        // 変換前のrule:
        //     "rule": { "disp": "玉の間南喰赤", "aka53": 1, "aka52": 1, "aka51": 1 }
        // 変換後のrule:
        //     "rule": { "disp": "玉の間四人南", "aka53": 1, "aka52": 1, "aka51": 1 }
        const rule = deepCopy(soulPaifu.rule);
        rule.disp = toSoulTable(rule.disp);

        // logを局ごとのデータに分割し、牌譜エディタのURL群として返す。
        return soulPaifu.log.map(function(v) {
            return EDITOR_URL_PREFIX + JSON.stringify({
                "title": title,
                "name": encodedName,
                "rule": rule,
                "log": [toNagaLog(v)],
            });
        });
    }

    /**
     * 卓名を雀魂っぽく変換する。
     * 
     * @param {String} tenhouTable 天鳳っぽい卓名を指定する。
     * @returns {String} 雀魂っぽい卓名を返す。
     */
    function toSoulTable(tenhouTable) {
        // 表記の好みの問題なので、必ずしも必要となる処理ではない。
        return tenhouTable.replace("南喰赤", "四人南");
    }

    /**
     * logをNAGAが解析可能な形式に変換する。
     * 
     * @param {Array<Array>} soulLog 雀魂形式のlogを指定する。
     * @returns {Array<Array>} NAGAで解析可能な形式のlogを返す。
     */
    function toNagaLog(soulLog) {
        // 流局のデータは変換の必要がない。
        if (soulLog[16].length < 3) {
            return soulLog;
        }
        const nagaLog = deepCopy(soulLog);

        // 当該局の場風を算出する。
        // 
        // 局を表す数字と意味:
        //     0 => 東1局, 1 => 東2局, ...
        const prevalent = ["東", "南", "西", "北"][Math.floor(nagaLog[0][0] / 4)];

        // 役名をNAGAが解析可能な表記に変換する。
        // ダブロン・トリロンに対応するため複数回繰り返す。
        for (let i = 1; i < nagaLog[16].length; i += 2) {
            // 当該局における和了者の自風を設定する。
            // 
            // 算出方法:
            //     (和了者のプレイヤー番号 - 親の位置 + 4) % 4
            const seat = ["東", "南", "西", "北"][
                (nagaLog[16][i].indexOf(Math.max(...nagaLog[16][i])) - (nagaLog[0][0] % 4) + 4) % 4
            ];

            // 役名をNAGAが解析可能な表記に変換する。
            nagaLog[16][i + 1] = nagaLog[16][i + 1].slice(0, 4).concat(
                nagaLog[16][i + 1].slice(4).map(function(v) {
                    return toNagaHand(v, prevalent, seat);
                }
            ));
        }

        // 変換後のlogを返す。
        return nagaLog;
    }

    /**
     * 役名をNAGAが解析可能な表記に変換する。
     * 
     * @param {String} hand 和了役を指定する。
     * @param {String} prevalent 場風を指定する。
     * @param {String} seat 和了者の自風を指定する。
     * @returns {String} NAGAで解析可能な表記の役名を返す。
     */
    function toNagaHand(hand, prevalent, seat) {
        // 対応が必要な役が判明次第、随時追加する。
        switch(hand) {
            case "役牌:場風牌(1飜)":
                return `場風 ${prevalent}(1飜)`;
            case "役牌:自風牌(1飜)":
                return `自風 ${seat}(1飜)`;
            case "ダブル立直(2飜)":
                return "両立直(2飜)";
            default:
                return hand;
        }
    }
})();