// ==UserScript==
// @name         mooket
// @namespace    http://tampermonkey.net/
// @version      2025-03-26.2
// @description  银河奶牛历史价格 show history market data for milkywayidle
// @author       IOMisaka
// @match        https://www.milkywayidle.com/*
// @match        https://test.milkywayidle.com/*
// @connect      mooket.qi-e.top
// @icon         data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==
// @grant        none
// @require      https://cdn.jsdelivr.net/npm/chart.js@4.4.8/dist/chart.umd.min.js
// @license MIT
// @downloadURL none
// ==/UserScript==


(function () {
  'use strict';
  let initData_itemDetailMap = null;
  if (localStorage.getItem("initClientData")) {
    const obj = JSON.parse(localStorage.getItem("initClientData"));
    initData_itemDetailMap = obj.itemDetailMap;
  }
  function hookWS() {
    const dataProperty = Object.getOwnPropertyDescriptor(MessageEvent.prototype, "data");
    const oriGet = dataProperty.get;
    dataProperty.get = hookedGet;
    Object.defineProperty(MessageEvent.prototype, "data", dataProperty);

    function hookedGet() {
      const socket = this.currentTarget;
      if (!(socket instanceof WebSocket)) {
        return oriGet.call(this);
      }
      if (socket.url.indexOf("api.milkywayidle.com/ws") <= -1 && socket.url.indexOf("api-test.milkywayidle.com/ws") <= -1) {
        return oriGet.call(this);
      }
      const message = oriGet.call(this);
      Object.defineProperty(this, "data", { value: message }); // Anti-loop
      return handleMessage(message);
    }
  }
  function handleMessage(message) {
    let obj = JSON.parse(message);
    if (obj && obj.type === "market_item_order_books_updated") {
      requestItemPrice(obj.marketItemOrderBooks.itemHrid, cur_day);
    } else if (obj && obj.type === "market_listings_updated") {//挂单变动

    }
    return message;
  }

  hookWS();

  let cur_day = 1;
  let cur_name = null;
  let w = "500px";
  let h = "280px";
  let configStr = localStorage.getItem("mooket_config");
  let config = configStr ? JSON.parse(configStr) : { "dayIndex": 0, "visible": true, "filter": { "bid": true, "ask": true, "mean": true } };
  cur_day = config.day;//读取设置

  window.onresize = function () {
    checkSize();
  };
  function checkSize() {
    if (window.innerWidth < window.innerHeight) {
      w = "280px";
      h = "500px";
    } else {
      w = "500px";
      h = "280px";
    }
  }
  checkSize();
  // 创建容器元素并设置样式和位置
  const container = document.createElement('div');
  container.style.border = "1px solid #ccc"; //边框样式
  container.style.backgroundColor = "#fff";
  container.style.position = "fixed";
  container.style.zIndex = 10000;
  container.style.top = "50px"; //距离顶部位置
  container.style.left = "130px"; //距离左侧位置
  container.style.width = w; //容器宽度
  container.style.height = h; //容器高度
  container.style.resize = "both";
  container.style.overflow = "auto";
  container.style.display = "flex";
  container.style.flexDirection = "column";
  container.style.flex = "1";
  container.style.minHeight = "33px";
  container.style.minWidth = "65px";
  container.style.cursor = "move";
  container.addEventListener("mousedown", function (e) {
    const rect = container.getBoundingClientRect();
    if (e.clientX > rect.right - 10 && e.clientY > rect.bottom - 10) {
      return;
    }
    let disX = e.clientX - container.offsetLeft;
    let disY = e.clientY - container.offsetTop;
    document.onmousemove = function (e) {
      let x = e.clientX - disX;
      let y = e.clientY - disY;
      container.style.left = x + 'px';
      container.style.top = y + 'px';
    };
    document.onmouseup = function () {
      document.onmousemove = document.onmouseup = null;
    };
  });
  document.body.appendChild(container);

  const ctx = document.createElement('canvas');
  ctx.id = "myChart";
  container.appendChild(ctx);

  // 创建按钮组并设置样式和位置
  let wrapper = document.createElement('div');
  wrapper.style.position = 'absolute';
  wrapper.style.bottom = '42px';
  wrapper.style.right = '0px';
  wrapper.style.fontSize = '14px';

  //wrapper.style.backgroundColor = '#fff';
  wrapper.style.flexShrink = 0;
  container.appendChild(wrapper);

  const days = [1, 3, 7, 30, 180]
  const dayTitle = ['1天', '3天', '7天', '30天', '半年']
  cur_day = days[config.dayIndex];

  for (let i = 0; i < 5; i++) {
    let btn = document.createElement('input');
    btn.id = 'chartType' + i;
    btn.type = 'radio';
    btn.name = 'chartType';
    btn.value = days[i];
    btn.style.cursor = 'pointer';
    btn.style.verticalAlign = "middle";
    btn.checked = i == config.dayIndex;
    btn.onclick = function () {
      cur_day = this.value;
      config.dayIndex = i;
      if (cur_name) requestItemPrice(cur_name, cur_day);
      save_config();
    }

    let label = document.createElement('label');
    label.innerText = dayTitle[i];
    label.style.display = 'inline-block';
    label.style.verticalAlign = 'middle';
    label.style.textAlign = 'center';
    label.htmlFor = btn.id;
    label.style.margin = '1px';
    wrapper.appendChild(btn);
    wrapper.appendChild(label);
  }
  //添加一个btn隐藏canvas和wrapper
  let btn_close = document.createElement('input');
  btn_close.type = 'button';
  btn_close.value = '📈隐藏';
  btn_close.style.textAlign = 'center';
  btn_close.style.display = 'inline';
  btn_close.style.margin = 0;
  btn_close.style.top = '2px';
  btn_close.style.left = '2px';
  btn_close.style.cursor = 'pointer';
  btn_close.style.position = 'absolute';
  let lastWidth;
  let lastHeight;
  btn_close.onclick = toggle;

  function toggle() {
    if (wrapper.style.display === 'none') {
      wrapper.style.display = ctx.style.display = 'block';
      btn_close.value = '📈隐藏';
      container.style.width = lastWidth;
      container.style.height = lastHeight;
      config.visible = true;
      save_config();
    } else {
      lastWidth = container.style.width;
      lastHeight = container.style.height;
      wrapper.style.display = ctx.style.display = 'none';
      container.style.width = "auto";
      container.style.height = "auto";
      btn_close.value = '📈显示';
      config.visible = false;
      save_config();
    }
  };

  container.appendChild(btn_close);
  let chart = new Chart(ctx, {
    type: 'line',
    data: {
      labels: [],
      datasets: [{
        label: '市场',
        data: [],
        backgroundColor: 'rgba(255,99,132,0.2)',
        borderColor: 'rgba(255,99,132,1)',
        borderWidth: 1
      }]
    },
    options: {
      onClick: save_config,
      responsive: true,
      maintainAspectRatio: false,
      pointRadius: 0,
      pointHitRadius: 20,
      tension: 0.5,
      scales: {
        y: {
          beginAtZero: false,
          ticks: {
            // 自定义刻度标签格式化
            callback: showNumber
          }
        }
      }
    }
  });

  function requestItemPrice(name, day = 1) {
    if (initData_itemDetailMap && initData_itemDetailMap[name]) {
      name = initData_itemDetailMap[name].name;
    }

    cur_name = name;
    cur_day = day;

    let time = day * 3600 * 24;
    fetch("https://mooket.qi-e.top/market", {
      method: "POST",
      headers: {
        "Content-Type": "application/json",
      },
      body: JSON.stringify({
        name: name,
        time: time
      })
    }).then(res => {
      res.json().then(data => updateChart(data, cur_day));
    })
  }
  function uploadItemPrice(marketItemOrderBooks, day = 1) {
    let name = marketItemOrderBooks.itemHrid;
    if (initData_itemDetailMap && initData_itemDetailMap[name]) {
      name = initData_itemDetailMap[name].name;
    }

    cur_name = name;
    cur_day = day;

    let time = day * 3600 * 24;
    fetch("https://mooket.qi-e.top/market", {
      method: "POST",
      headers: {
        "Content-Type": "application/json",
      },
      body: JSON.stringify({
        name: name,
        time: time
      })
    }).then(res => {
      res.json().then(data => updateChart(data, cur_day));
    })
  }
  function formatTime(timestamp, range) {
    const date = new Date(timestamp * 1000);
    const pad = n => n.toString().padStart(2, '0');

    // 获取各时间组件
    const hours = pad(date.getHours());
    const minutes = pad(date.getMinutes());
    const day = pad(date.getDate());
    const month = pad(date.getMonth() + 1);
    const shortYear = date.getFullYear().toString().slice(-2);

    // 根据时间范围选择格式
    switch (parseInt(range)) {
      case 1: // 1天：只显示时间
        return `${hours}:${minutes}`;

      case 3: // 3天：日+时段
        return `${hours}:${minutes}`;

      case 7: // 7天：月/日 + 时段
        return `${day}.${hours}`;

      case 30: // 30天：月/日
        return `${month}/${day}`;

      default: // 180天：年/月
        return `${shortYear}/${month}`;
    }
  }

  function showNumber(num) {
    const absNum = Math.abs(num);

    return absNum >= 1e10 ? `${Math.floor(num / 1e9)}B` :
      absNum >= 1e7 ? `${Math.floor(num / 1e6)}M` :
        absNum >= 1e4 ? `${Math.floor(num / 1e3)}K` :
          `${Math.floor(num)}`;
  }
  //data={'bid':[{time:1,price:1}],'ask':[{time:1,price:1}]}
  function updateChart(data, day) {
    //过滤异常元素
    for (let i = data.bid.length - 1; i >= 0; i--) {
      if (data.bid[i].price < 0 || data.ask[i].price < 0) {
        data.bid.splice(i, 1);
        data.ask.splice(i, 1);
      }
    }
    //timestamp转日期时间
    //根据day输出不同的时间表示，<3天显示时分，<=7天显示日时，<=30天显示月日，>30天显示年月

    let labels = data.bid.map(x => formatTime(x.time, day));

    chart.data.labels = labels;

    let sma = [];
    let sma_size = 6;
    for (let i = 0; i < data.bid.length; i++) {
      if (i < sma_size) sma.push((data.bid[i].price + data.ask[i].price) / 2);
      else {
        let sum = 0;
        for (let j = 0; j < sma_size; j++) {
          sum += ((data.bid[i - j].price + data.ask[i - j].price) / 2);
        }
        sma.push(sum / sma_size);
      }
    }

    chart.data.datasets = [
      {
        label: '买入',
        data: data.bid.map(x => x.price),
        borderColor: '#ff3300',
        backgroundColor: '#ff3300',
        borderWidth: 1.5
      },
      {
        label: '卖出',
        data: data.ask.map(x => x.price),
        borderColor: '#00cc00',
        backgroundColor: '#00cc00',
        borderWidth: 1.5
      },
      {
        label: '均线',
        data: sma,
        borderColor: '#ff9900',
        borderWidth: 3,
        fill: true

      }
    ];
    chart.setDatasetVisibility(0, config.filter.ask);
    chart.setDatasetVisibility(1, config.filter.bid);
    chart.setDatasetVisibility(2, config.filter.mean);

    chart.update()
  }
  function save_config() {

    if (chart && chart.data && chart.data.datasets && chart.data.datasets.length == 3) {
      config.filter.ask = chart.getDatasetMeta(0).visible;
      config.filter.bid = chart.getDatasetMeta(1).visible;
      config.filter.mean = chart.getDatasetMeta(2).visible;
    }
    localStorage.setItem("mooket_config", JSON.stringify(config));
  }
  //requestMarket('Apple', 1);
  toggle();

})();