// ==UserScript==
// @name         ADMS复制
// @namespace    anker-dms-export
// @version      2.6
// @description  在结果集工具栏内嵌入复制按钮，可选含表头、纯文本或CSV格式；不含表头时自动排除序号列；重新查询后按钮自动补回
// @match        https://dms.anker-in.com/*
// @grant        GM_setClipboard
// ==/UserScript==

(function () {
  'use strict';

  const sleep = ms => new Promise(r => setTimeout(r, ms));

  function csvEscape(s) {
    s = String(s ?? '');
    if (/[",\n]/.test(s)) return '"' + s.replace(/"/g, '""') + '"';
    return s;
  }

  function splitGridsByHeader(scope) {
    const grids = Array.from(scope.querySelectorAll('[role="grid"]'));
    const blocks = grids.map(g => {
      const inner = g.querySelector('.ReactVirtualized__Grid__innerScrollContainer') || g;
      const cells = inner.children.length ? Array.from(inner.children) : [g];
      const tops = new Set(cells.map(c => c.style.top || '0px'));
      return { el: g, isHeader: tops.size <= 1, cells };
    });
    return {
      headerBlocks: blocks.filter(b => b.isHeader),
      dataBlocks: blocks.filter(b => !b.isHeader)
    };
  }

  function subHeadersOf(block) {
    return block.cells
      .slice()
      .sort((a, b) => parseFloat(a.style.left || 0) - parseFloat(b.style.left || 0))
      .map(c => c.textContent.trim());
  }

  function blockToColumns(cellMap) {
    const rowsMap = {};
    Object.entries(cellMap).forEach(([key, text]) => {
      const idx = key.lastIndexOf('_');
      const top = key.slice(0, idx), left = key.slice(idx + 1);
      (rowsMap[top] ||= {})[left] = text;
    });
    const sortedTops = Object.keys(rowsMap).sort((a, b) => parseFloat(a) - parseFloat(b));
    const leftsSet = new Set();
    Object.values(rowsMap).forEach(r => Object.keys(r).forEach(l => leftsSet.add(l)));
    const sortedLefts = Array.from(leftsSet).sort((a, b) => parseFloat(a) - parseFloat(b));
    return {
      colCount: sortedLefts.length,
      matrix: sortedTops.map(top => sortedLefts.map(left => rowsMap[top][left] ?? ''))
    };
  }

  function formatOutput(headers, rows, includeHeader, format) {
    if (format === 'csv') {
      const lines = [];
      if (includeHeader) lines.push(headers.map(csvEscape).join(','));
      rows.forEach(r => lines.push(r.map(csvEscape).join(',')));
      return lines.join('\n');
    }
    const lines = [];
    if (includeHeader) lines.push(headers.join('\t'));
    rows.forEach(r => lines.push(r.join('\t')));
    return lines.join('\n');
  }

  function toast(msg) {
    const t = document.createElement('div');
    t.textContent = msg;
    Object.assign(t.style, {
      position: 'fixed', bottom: '20px', right: '20px', zIndex: 999999,
      background: '#333', color: '#fff', padding: '8px 14px',
      borderRadius: '4px', fontSize: '13px'
    });
    document.body.appendChild(t);
    setTimeout(() => t.remove(), 2500);
  }

  async function runExport(scope, includeHeader, format) {
    let { headerBlocks, dataBlocks } = splitGridsByHeader(scope);
    if (!dataBlocks.length) { alert('当前结果集没有找到虚拟表格'); return; }

    // 序号列跟随"包含表头"开关：不含表头时才剔除序号列
    if (!includeHeader) {
      const rowNumIdx = headerBlocks.findIndex(hb => {
        const texts = subHeadersOf(hb);
        return texts.length === 1 && texts[0] === '序号';
      });
      if (rowNumIdx !== -1) {
        headerBlocks = headerBlocks.filter((_, i) => i !== rowNumIdx);
        dataBlocks = dataBlocks.filter((_, i) => i !== rowNumIdx);
      }
    }

    const scrollers = dataBlocks.map(b => b.el.closest('.ReactVirtualized__Grid') || b.el);
    const maxScroll = Math.max(...scrollers.map(s => s.scrollHeight - s.clientHeight));
    const step = Math.max(50, Math.min(...scrollers.map(s => s.clientHeight)) - 20);

    const cellMaps = dataBlocks.map(() => ({}));
    function captureAll() {
      dataBlocks.forEach((b, i) => {
        const inner = b.el.querySelector('.ReactVirtualized__Grid__innerScrollContainer') || b.el;
        Array.from(inner.children).forEach(c => {
          cellMaps[i][`${c.style.top}_${c.style.left}`] = c.textContent;
        });
      });
    }
    captureAll();
    for (let t = step; t <= maxScroll + step; t += step) {
      scrollers.forEach(s => { s.scrollTop = t; s.dispatchEvent(new Event('scroll', { bubbles: true })); });
      await sleep(60);
      captureAll();
    }

    const blockResults = cellMaps.map(blockToColumns);
    const rowCount = Math.max(0, ...blockResults.map(r => r.matrix.length));
    const rows = [];
    for (let r = 0; r < rowCount; r++) {
      let row = [];
      blockResults.forEach(br => { row = row.concat(br.matrix[r] || new Array(br.colCount).fill('')); });
      rows.push(row);
    }

    const headers = [];
    headerBlocks.forEach(hb => headers.push(...subHeadersOf(hb)));

    const output = formatOutput(headers, rows, includeHeader, format);
    GM_setClipboard(output);
    toast(`已复制 ${rows.length} 行到剪贴板`);
  }

  function openConfigPopup(anchorBtn, scope) {
    document.querySelectorAll('.vg-export-popup').forEach(p => p.remove());
    const rect = anchorBtn.getBoundingClientRect();
    const panel = document.createElement('div');
    panel.className = 'vg-export-popup';
    Object.assign(panel.style, {
      position: 'fixed', top: `${rect.bottom + 4}px`, left: `${rect.left}px`,
      zIndex: 999999, background: '#fff', border: '1px solid #d9d9d9', borderRadius: '6px',
      padding: '10px', boxShadow: '0 2px 8px rgba(0,0,0,.15)', fontSize: '13px', width: '190px'
    });
    panel.innerHTML = `
      <div style="margin-bottom:8px;"><label><input type="checkbox" id="vg-header" checked> 包含表头</label></div>
      <div style="margin-bottom:8px;">
        <label><input type="radio" name="vg-fmt" value="plain" checked> 纯文本(Tab分隔)</label><br>
        <label><input type="radio" name="vg-fmt" value="csv"> CSV</label>
      </div>
      <button id="vg-run" style="width:100%;padding:5px;cursor:pointer;">复制</button>
    `;
    document.body.appendChild(panel);
    panel.querySelector('#vg-run').onclick = () => {
      const includeHeader = panel.querySelector('#vg-header').checked;
      const format = panel.querySelector('input[name="vg-fmt"]:checked').value;
      panel.remove();
      runExport(scope, includeHeader, format);
    };
    setTimeout(() => {
      document.addEventListener('click', function closeOnce(e) {
        if (!panel.contains(e.target) && e.target !== anchorBtn) {
          panel.remove();
          document.removeEventListener('click', closeOnce);
        }
      });
    }, 0);
  }

  function findResultScope(extraContentEl) {
    return extraContentEl.closest('[class*="resultTabPaneContainer"]')
      || extraContentEl.closest('.ant-tabs-tabpane')
      || document.body;
  }

  const COPY_ICON_SVG = `<svg viewBox="0 0 1024 1024" width="14" height="14" fill="currentColor">
    <path d="M832 64H384c-35.3 0-64 28.7-64 64v64H192c-35.3 0-64 28.7-64 64v640c0 35.3 28.7 64 64 64h448c35.3 0 64-28.7 64-64v-64h128c35.3 0 64-28.7 64-64V128c0-35.3-28.7-64-64-64zM640 896H192V256h192v512c0 35.3 28.7 64 64 64h192v64zm192-128H384V192h448v576z"/>
  </svg>`;

  function injectButtonInto(extraContentEl, uploadBtn) {
    const btn = document.createElement('span');
    btn.className = 'vg-copy-btn';
    btn.title = '复制（含表头/格式可选）';
    Object.assign(btn.style, {
      display: 'inline-flex', alignItems: 'center', justifyContent: 'center',
      width: '14px', height: '14px', marginRight: '8px', cursor: 'pointer',
      color: 'rgba(0,0,0,.65)', verticalAlign: 'middle'
    });
    btn.innerHTML = COPY_ICON_SVG;
    btn.onclick = () => openConfigPopup(btn, findResultScope(extraContentEl));
    uploadBtn.parentElement.insertBefore(btn, uploadBtn);
  }

  function scanAndInject() {
    document.querySelectorAll('.ant-tabs-extra-content').forEach(el => {
      if (el.querySelector('.vg-copy-btn')) return; // 按钮还在，不用重插
      const icon = el.querySelector('svg[data-icon="upload"], .anticon-upload');
      const uploadBtn = icon ? icon.closest('button') : null;
      if (uploadBtn) injectButtonInto(el, uploadBtn);
    });
  }

  scanAndInject();
  new MutationObserver(() => scanAndInject()).observe(document.body, { childList: true, subtree: true });
})();