// ==UserScript==
// @name         MWI_DamageTracker
// @namespace    ponchain
// @version      0.0.1
// @description  MWI伤害统计
// @author       ponchain
// @match        https://www.milkywayidle.com/*
// @icon         data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==
// @grant        none
// @license MIT
// @downloadURL none
// ==/UserScript==

(function () {
  'use strict';
  // 代理 Websocket
  const OriginalWebSocket = window.WebSocket;
  const handlerQueue = [];
  function MyWebSocket(url, protocols) {
    const ws = new OriginalWebSocket(url, protocols);
    ws.addEventListener('message', function (event) {
      const msgData = JSON.parse(event.data);
      handlerQueue.reduce((prev, handler) => {
        return handler(prev);
      }, msgData);
    });
    return ws;
  }
  window.WebSocket = MyWebSocket;
  let totalDamage = [];
  let monstersHP = [];
  const getStatisticsDom = () => {
    if (!document.querySelector('.statistics-panel')) {
      let panel = document.createElement('div');
      panel.style.position = 'fixed';
      panel.style.top = '50px';
      panel.style.left = '50px';
      panel.style.background = '#f0f0f0';
      panel.style.border = '1px solid #ccc';
      panel.style.zIndex = '9999';
      panel.style.cursor = 'move';
      panel.style.fontSize = '12px';
      panel.style.padding = '4px';
      panel.innerHTML = '<div style="padding: 10px;">还未开始统计</div>';
      panel.className = 'statistics-panel';
      let dragging = false;
      let offsetX, offsetY;

      panel.addEventListener('mousedown', function (e) {
        dragging = true;
        offsetX = e.clientX - panel.offsetLeft;
        offsetY = e.clientY - panel.offsetTop;
      });

      document.addEventListener('mousemove', function (e) {
        if (dragging) {
          var newX = e.clientX - offsetX;
          var newY = e.clientY - offsetY;
          panel.style.left = newX + 'px';
          panel.style.top = newY + 'px';
        }
      });

      document.addEventListener('mouseup', function () {
        dragging = false;
      });

      document.body.appendChild(panel);
    }
    return document.querySelector('.statistics-panel');
  };
  const updateStatisticsPanel = () => {
    const panel = getStatisticsDom();
    const damageDoms = totalDamage.reduce((prev, cur, index) => {
      return prev + `<div>${index + 1}号位:${cur}</div>`;
    }, '');
    panel.innerHTML = damageDoms;
  };
  const calculateDamage = (msgData) => {
    if (msgData.type === 'new_battle') {
      monstersHP = msgData.monsters.map((monster) => monster.maxHitpoints);
      if (!totalDamage.length) {
        totalDamage = new Array(Object.keys(msgData.players).length).fill(0);
      }
    } else if (msgData.type === 'battle_updated' && monstersHP.length) {
      const mMap = msgData.mMap;
      const userIndex = Object.keys(msgData.pMap);
      monstersHP.forEach((mHP, mIndex) => {
        const monster = mMap[mIndex];
        const userIndex = Object.keys(msgData.pMap)[0];
        if (monster) {
          if (userIndex !== void 0) {
            const hpDiff = mHP - monster.cHP;
            hpDiff > 0 && (totalDamage[userIndex] += hpDiff);
          }
          monstersHP[mIndex] = monster.cHP;
        }
      });
      updateStatisticsPanel();
    } else if (msgData.type === 'actions_updated') {
      monstersHP = [];
      totalDamage = [];
    }
  };

  handlerQueue.push(calculateDamage);
})();