ChatRecorder

Saves all chat history.
Wrap lines
// ==UserScript==
// @name         ChatRecorder
// @namespace    http://tampermonkey.net/
// @version      2.7
// @description  Saves all chat history.
// @author       bot_7420 [2937420]
// @match        https://www.torn.com/*
// @run-at       document-start
// @grant        GM_addStyle
// ==/UserScript==

(function () {
  "use strict";

  let db = null;

  window.onload = function () {
    initCSS();
    initControlPanel();
    initChatBoxObserver();
  };

  // Hook chat Websocket on receive message
  const originalSend = WebSocket.prototype.send;
  window.sockets = [];
  WebSocket.prototype.send = function (...args) {
    if (window.sockets.indexOf(this) === -1 && this.url.indexOf("ws-chat") > -1) {
      console.log("ChatRecorder: found chat websocket");
      window.sockets.push(this);
      this.addEventListener("message", function (event) {
        const obj = JSON.parse(event.data);
        if (obj && obj.status === 200 && obj.data) {
          const messageObjList = obj.data;
          for (const messageObj of messageObjList) {
            handleMessage(messageObj);
          }
        }
      });
    }
    return originalSend.call(this, ...args);
  };

  initIndexDB();

  function handleMessage(message) {
    if (!message || !message.messageId) {
      return;
    }
    dbWrite(message);
  }

  function initIndexDB() {
    const openRequest = indexedDB.open("scriptChatRecorderDB", 1);
    openRequest.onupgradeneeded = function (e) {
      console.log("ChatRecorder: initIndexDB open onupgradeneeded");
      db = e.target.result;
      if (!db.objectStoreNames.contains("messageStore")) {
        console.log("ChatRecorder: initIndexDB open onupgradeneeded create store");
        const objectStore = db.createObjectStore("messageStore", { keyPath: "messageId", autoIncrement: false });
        objectStore.createIndex("targetPlayerId", "targetPlayerId", { unique: false });
      }
    };
    openRequest.onsuccess = function (e) {
      console.log("ChatRecorder: initIndexDB open onsuccess");
      db = e.target.result;
    };
    openRequest.onerror = function (e) {
      console.error("ChatRecorder: initIndexDB open onerror");
      console.dir(e);
    };
  }

  function dbWrite(message) {
    if (!db) {
      console.error("ChatRecorder: dbWrite db is null");
    }

    const targetPlayer = getTargetPlayerFromMessage(message);
    if (!targetPlayer) {
      return;
    }
    message.targetPlayerId = targetPlayer.id;
    message.targetPlayerName = targetPlayer.name;

    const transaction = db.transaction(["messageStore"], "readwrite");
    transaction.oncomplete = (event) => {
      //console.log("ChatRecorder: dbWrite transaction oncomplete [" + message.targetPlayerId + " " + message.senderName + ": " + message.messageText + "]");
    };
    transaction.onerror = (event) => {
      console.error("ChatRecorder: dbWrite transaction onerror [" + message.targetPlayerId + " " + message.senderName + ": " + message.messageText + "]");
    };

    const store = transaction.objectStore("messageStore");
    const request = store.put(message);
    request.onsuccess = (event) => {};
  }

  function dbReadByTargetPlayerId(targetPlayerId) {
    if (!db) {
      console.error("ChatRecorder: dbReadByTargetPlayerId db is null");
    }

    const transaction = db.transaction(["messageStore"], "readonly");
    transaction.oncomplete = (event) => {
      console.log("ChatRecorder: dbReadByTargetPlayerId transaction oncomplete [" + targetPlayerId + "]");
    };
    transaction.onerror = (event) => {
      console.error("ChatRecorder: dbReadByTargetPlayerId transaction onerror [" + targetPlayerId + "]");
    };

    const store = transaction.objectStore("messageStore");
    const index = store.index("targetPlayerId");
    const keyRange = IDBKeyRange.only(targetPlayerId);

    return new Promise((resolve, reject) => {
      const resultList = [];
      index.openCursor(keyRange).onerror = (event) => {
        resolve(resultList);
      };
      index.openCursor(keyRange).onsuccess = (event) => {
        const cursor = event.target.result;
        if (cursor) {
          resultList.push(cursor.value);
          cursor.continue();
        } else {
          resolve(resultList);
        }
      };
    });
  }

  function dbReadAllPlayerId() {
    if (!db) {
      console.error("ChatRecorder: dbReadAllPlayerId db is null");
    }

    const transaction = db.transaction(["messageStore"], "readonly");
    transaction.oncomplete = (event) => {
      console.log("ChatRecorder: dbReadAllPlayerId transaction oncomplete");
    };
    transaction.onerror = (event) => {
      console.error("ChatRecorder: dbReadAllPlayerId transaction onerror");
    };

    const store = transaction.objectStore("messageStore");
    const index = store.index("targetPlayerId");
    const keyRange = null;

    return new Promise((resolve, reject) => {
      const resultList = [];
      index.openCursor(keyRange, "nextunique").onerror = (event) => {
        resolve(resultList);
      };
      index.openCursor(keyRange, "nextunique").onsuccess = (event) => {
        const cursor = event.target.result;
        if (cursor) {
          resultList.push(cursor.value);
          cursor.continue();
        } else {
          resolve(resultList);
        }
      };
    });
  }

  function getTargetPlayerFromMessage(message) {
    if (message.roomId.startsWith("Poker:")) {
      return { id: "Poker", name: "Poker" };
    } else if (message.roomId.startsWith("Faction:")) {
      return { id: "Faction", name: "Faction" };
    } else if (message.roomId.startsWith("Company:")) {
      return { id: "Company", name: "Company" };
    } else if (message.roomId.startsWith("Global:")) {
      return { id: "Global", name: "Global" };
    } else if (!message.roomId.startsWith("Users:")) {
      return null;
    }

    // Private chats
    const selfId = getSelfIdFromSession();
    const selfName = getSelfNameFromSession();
    if (!selfId || !selfName) {
      return { id: "others", name: "Other" };
    }
    const strList = message.roomId.split(";");
    if (strList.length !== 3) {
      return { id: "others", name: "Other" };
    }
    let target = { id: "others", name: "Other" };
    if (parseInt(strList[1]) === selfId) {
      target.id = strList[2];
    } else if (parseInt(strList[2]) === selfId) {
      target.id = strList[1];
    } else {
      return { id: "others", name: "Other" };
    }
    if (strList[0].split(":")[1].split(",")[0] === selfName) {
      target.name = strList[0].split(":")[1].split(",")[1];
    } else if (strList[0].split(":")[1].split(",")[1] === selfName) {
      target.name = strList[0].split(":")[1].split(",")[0];
    } else {
      return { id: "others", name: "Other" };
    }
    return target;
  }

  function getSelfIdFromSession() {
    let index = Object.keys(sessionStorage).findIndex((item) => item.startsWith("sidebarData"));
    if (index >= 0) {
      let sidebarData = JSON.parse(sessionStorage.getItem(sessionStorage.key(index)));
      let userID = sidebarData.user.userID;
      return userID;
    }
    return null;
  }

  function getSelfNameFromSession() {
    let index = Object.keys(sessionStorage).findIndex((item) => item.startsWith("sidebarData"));
    if (index >= 0) {
      let sidebarData = JSON.parse(sessionStorage.getItem(sessionStorage.key(index)));
      let userID = sidebarData.user.name;
      return userID;
    }
    return null;
  }

  function initCSS() {
    const isDarkmode = $("body").hasClass("dark-mode");
    GM_addStyle(`.chat-control-panel-popup {
                    position: fixed;
                    top: 10%;
                    left: 15%;
                    border-radius: 10px;
                    padding: 10px;
                    background: ${isDarkmode ? "#282828" : "#F0F0F0"};
                    z-index: 1000;
                    display: none;
                  }

                  .chat-control-panel-results {
                    padding: 10px;
                  }

                  .chat-control-player {
                    margin: 4px 4px 4px 4px !important;
                    display: inline-block !important;
                  }

                  .chat-control-panel-overlay {
                    position: fixed;
                    top: 0;
                    left: 0;
                    background: ${isDarkmode ? "#404040" : "#B0B0B0"};
                    width: 100%;
                    height: 100%;
                    opacity: 0.7;
                    z-index: 900;
                    display: none;
                  }

                  div#chat-player-list {
                    overflow-y: scroll;
                    height: 50px;
                  }

                  a.chat-history-search:hover {
                    color: #318CE7 !important;
                  }

                  .chat-control-panel-item {
                    display: inline-block;
                    margin: 2px 2px 2px 2px;
                  }`);
  }

  function initControlPanel() {
    const $title = $("div#top-page-links-list");
    if ($title.length === 0) {
      console.log("ChatRecorder: nowhere to put control panel button");
    }
    const $controlBtn = $(`<a id="chatHistoryControl" class="t-clear h c-pointer right last">
                              <span class="icon-wrap svg-icon-wrap">
                                <span class="link-icon-svg">
                                    <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 16 10.33"><defs><style>.cls-1{opacity:0.35;}.cls-2{fill:#fff;}.cls-3{fill:#777;}</style></defs><g id="Слой_2" data-name="Слой 2"><g id="icons"><g class="cls-1"><path class="cls-2" d="M10,5.67a2,2,0,0,1-4,0,1.61,1.61,0,0,1,0-.39A1.24,1.24,0,0,0,7.64,3.7a2.19,2.19,0,0,1,.36,0A2,2,0,0,1,10,5.67ZM8,1C3,1,0,5.37,0,5.37s3.22,5,8,5c5.16,0,8-5,8-5S13.14,1,8,1ZM8,9a3.34,3.34,0,1,1,3.33-3.33A3.33,3.33,0,0,1,8,9Z"></path></g><path class="cls-3" d="M10,4.67a2,2,0,0,1-4,0,1.61,1.61,0,0,1,0-.39A1.24,1.24,0,0,0,7.64,2.7a2.19,2.19,0,0,1,.36,0A2,2,0,0,1,10,4.67ZM8,0C3,0,0,4.37,0,4.37s3.22,5,8,5c5.16,0,8-5,8-5S13.14,0,8,0ZM8,8a3.34,3.34,0,1,1,3.33-3.33A3.33,3.33,0,0,1,8,8Z"></path></g></g></svg>
                                </span>
                              </span>
                              <span>ChatRecorder</span>
                            </a>`);
    $title.append($controlBtn);

    const $controlPanelDiv = $(`<div id="chatControlPanel" class="chat-control-panel-popup">control</div>`);
    const $controlPanelOverlayDiv = $(`<div id="chatControlOverlayPanel" class="chat-control-panel-overlay"></div>`);
    $controlPanelDiv.html(`
    <input type="text" class="chat-control-panel-item" id="chat-target-id-input" placeholder="Player ID" size="10" />
    <button id="chat-search" class="chat-control-panel-item" style="cursor: pointer;">Search</button><br>
    <div id="chat-player-list"></div><br>
    <textarea readonly id="chat-results" cols="120" rows="30"></textarea>
    `);

    // Control panel onClick listeners
    $controlPanelDiv.find("button#chat-search").click(function () {
      const inputId = $controlPanelDiv.find("input#chat-target-id-input").val();
      dbReadByTargetPlayerId(inputId).then((result) => {
        let text = "";
        for (const message of result) {
          const timeStr = formatDateString(new Date(message.time * 1000));
          text += timeStr + " " + message.senderName + ": " + message.messageText + "\n";
        }
        text += "Found " + result.length + " records\n";
        const $textarea = $("textarea#chat-results");
        $textarea.val(text);
        $textarea.scrollTop($textarea[0].scrollHeight);
      });
    });

    $title.append($controlPanelDiv);
    $title.append($controlPanelOverlayDiv);

    $controlBtn.click(function () {
      dbReadAllPlayerId().then((result) => {
        const $playerListDiv = $controlPanelDiv.find("div#chat-player-list");
        $playerListDiv.empty();
        let num = 0;
        for (const message of result) {
          if (num == 8) {
            $playerListDiv.append($(`<br>`));
            num = -1;
          }
          num++;
          let a = $(`<a class="chat-control-player">${message.targetPlayerName}</a>`);
          a.click(() => {
            $controlPanelDiv.find("input#chat-target-id-input").val(message.targetPlayerId);
            $controlPanelDiv.find("button#chat-search").trigger("click");
          });
          $playerListDiv.append(a);
        }

        $controlPanelDiv.fadeToggle(200);
        $controlPanelOverlayDiv.fadeToggle(200);
      });
    });

    $controlPanelOverlayDiv.click(function () {
      $controlPanelDiv.fadeOut(200);
      $controlPanelOverlayDiv.fadeOut(200);
    });
  }

  function formatDateString(date) {
    const pad = (v) => {
      return v < 10 ? "0" + v : v;
    };
    let year = date.getFullYear();
    let month = pad(date.getMonth() + 1);
    let day = pad(date.getDate());
    let hour = pad(date.getHours());
    let min = pad(date.getMinutes());
    let sec = pad(date.getSeconds());
    return year + "/" + month + "/" + day + " " + hour + ":" + min + ":" + sec;
  }

  function initChatBoxObserver() {
    const $chatsList = $("div#chatRoot div._chat-box-wrap_1pskg_111");
    const observerConfig = { attributes: true, childList: true, subtree: false };
    const observer = new MutationObserver(() => {
      handleChatBoxChange();
    });
    observer.observe($chatsList[0], observerConfig);
  }

  function handleChatBoxChange() {
    const $chats = $("div#chatRoot div._chat-box_1pskg_111");
    $chats.each((index, div) => {
      const isChatActive = $(div).hasClass("_chat-active_1pskg_120");
      const isLabelAlreadyAdded = $(div).find("a.chat-history-search").length > 0;
      if (isChatActive && !isLabelAlreadyAdded) {
        addLabelToChat($(div));
      } else if (!isChatActive && isLabelAlreadyAdded) {
        removeLabelFromChat($(div));
      }
    });
  }

  function addLabelToChat($chat) {
    const targetName = $chat.find("div._chat-box-head_1pskg_133 span._name_1pskg_148").text();
    if (targetName === "Faction" || targetName === "Trade" || targetName === "Company" || targetName === "Global" || targetName === "Poker") {
      return;
    }

    dbReadAllPlayerId().then((result) => {
      let targetId = null;
      for (const message of result) {
        if (message.targetPlayerName === targetName) {
          targetId = message.targetPlayerId;
          break;
        }
      }

      if (targetId) {
        dbReadByTargetPlayerId(targetId).then((result) => {
          const isChatActive = $chat.hasClass("_chat-active_1pskg_120");
          const isLabelAlreadyAdded = $chat.find("a.chat-history-search").length > 0;
          if (isChatActive && !isLabelAlreadyAdded) {
            let $insert = $('<a class="chat-history-search"> [' + result.length + " Records]</a>");
            $insert.on("click", function (event) {
              $("a#chatHistoryControl").trigger("click");
              $("div#chatControlPanel input#chat-target-id-input").val(targetId);
              $("div#chatControlPanel button#chat-search").trigger("click");
            });
            $chat.find("div._chat-box-head_1pskg_133 span._name_1pskg_148").append($insert);
          }
        });
      } else {
        const isChatActive = $chat.hasClass("_chat-active_1pskg_120");
        const isLabelAlreadyAdded = $chat.find("a.chat-history-search").length > 0;
        if (isChatActive && !isLabelAlreadyAdded) {
          let $insert = $('<a class="chat-history-search"> [' + "0" + " Records]</a>");
          $chat.find("div._chat-box-head_1pskg_133 span._name_1pskg_148").append($insert);
        }
      }
    });
  }

  function removeLabelFromChat($chat) {
    $chat.find("a.chat-history-search").remove();
  }
})();
欢迎访问Greasy Fork镜像
