// ==UserScript==
// @name         GenGen-RMJ-DET
// @namespace    http://tampermonkey.net/
// @version      4.5.6
// @author       GenGen 队
// @match        https://www.luogu.com.cn/record/cf/*
// @match        https://www.luogu.com.cn/record/at/*
// @grant        GM_getValue
// @grant        GM_xmlhttpRequest
// @connect      codeforces.com
// @connect      www.luogu.com.cn
// @run-at       document-start
// @description  GenGen RMJ 提交记录详情
// @license      MIT
// ==/UserScript==
(function() {
  const _gmGet = GM_getValue;
  const _gmReq = GM_xmlhttpRequest;

  const VALID = [
    /^https:\/\/www\.luogu\.com\.cn\/record\/cf\/[^/]+\/[^/?#]+/,
    /^https:\/\/www\.luogu\.com\.cn\/record\/at\/[^/]+\/[^/?#]+/,
  ];
  if (!VALID.some(r => r.test(location.href))) return;

  const m = location.pathname.match(/\/record\/(cf|at)\/([^/]+)\/([^/?#]+)/);
  if (!m) return;
  const [, plat, cid, sid] = m;
  const isCF = plat === 'cf';

  const handle = _gmGet('cf-handle', '') || _gmGet('rmj-cf-handle', '') || '';
  const code = _gmGet(`{${cid},${sid}}`, '');

  document.title = 'GenGen RMJ 加载中';
  const mask = document.createElement('div');
  mask.id = 'rmj-mask';
  mask.style.cssText = 'position:fixed;top:0;left:0;width:100vw;height:100vh;background:rgb(239,239,239);z-index:2147483647;';
  (document.documentElement || document.body || document).appendChild(mask);

  const CF2LG = { OK:12, WRONG_ANSWER:6, TIME_LIMIT_EXCEEDED:5, MEMORY_LIMIT_EXCEEDED:4,
    OUTPUT_LIMIT_EXCEEDED:3, RUNTIME_ERROR:7, COMPILATION_ERROR:2, TESTING:1 };
  const FINAL_CF = new Set(['OK','WRONG_ANSWER','COMPILATION_ERROR','RUNTIME_ERROR',
    'TIME_LIMIT_EXCEEDED','MEMORY_LIMIT_EXCEEDED','OUTPUT_LIMIT_EXCEEDED']);
  const TC = 'Luogu\\DataClass\\Record\\JudgeResult\\TestCaseJudgeResult';
  const VERB_DESC = {
    'OK': 'ok accepted',
    'WRONG_ANSWER': 'wrong answer',
    'TIME_LIMIT_EXCEEDED': 'time limit exceeded',
    'MEMORY_LIMIT_EXCEEDED': 'memory limit exceeded',
    'OUTPUT_LIMIT_EXCEEDED': 'output limit exceeded',
    'RUNTIME_ERROR': 'runtime error',
    'COMPILATION_ERROR': 'compilation error',
    'TESTING': 'judging'
  };

  /* ── 网络请求 ── */
  const gmJson = (url) => new Promise((ok, no) => {
    _gmReq({ method:'GET', url, timeout:15000, anonymous:true,
      onload: r => { try { ok(JSON.parse(r.responseText)); } catch (e) { no(e); } },
      onerror: () => no(new Error('net')), ontimeout: () => no(new Error('timeout')) });
  });

  const fetchTemplate = () => new Promise((ok, no) => {
    _gmReq({ method:'GET', url:'https://www.luogu.com.cn/record/290510067', timeout:15000,
      onload: r => r.status === 200 ? ok(r.responseText) : no(new Error('tpl')),
      onerror: () => no(new Error('tpl net')), ontimeout: () => no(new Error('tpl timeout')) });
  });

  async function findCF() {
    let url = `https://codeforces.com/api/contest.status?contestId=${encodeURIComponent(cid)}&from=1&count=100`;
    if (handle) url += `&handle=${encodeURIComponent(handle)}`;
    const d = await gmJson(url);
    if (d.status !== 'OK') throw new Error('CF ' + d.status);
    return (d.result || []).find(x => String(x.id) === sid) || null;
  }

  /* ── 数据构建（只保留最后一个点，带详细数据，分数全 null） ── */
  function buildCases(sub, time, memKB) {
    const lgCode = CF2LG[sub.verdict] ?? 0;
    const isAC = sub.verdict === 'OK';
    const isCE = sub.verdict === 'COMPILATION_ERROR';
    const passed = sub.passedTestCount ?? 0;

    if (isCE) return [];

    let idNum, status;
    if (isAC) {
      idNum = passed > 0 ? passed : 1;
      status = 12;
    } else {
      idNum = passed + 1; // 这个 ID 要加1
      status = (sub.verdict === 'TESTING') ? 1 : lgCode;
    }

    return [{
      id: `${idNum}`,
      status: status,
      description: VERB_DESC[sub.verdict] || String(sub.verdict).toLowerCase().replace(/_/g, ' '),
      score: null,
      signal: 0,
      exitCode: 0,
      subtaskID: 0,
      __CLASS_NAME: TC,
      time: time ?? null,
      memory: memKB ?? null
    }];
  }

  function buildRecordSubset(sub, recId) {
    const lgCode = CF2LG[sub.verdict] ?? 0;
    const isCE = sub.verdict === 'COMPILATION_ERROR';
    const isFinal = FINAL_CF.has(sub.verdict);
    const time = isFinal ? (sub.timeConsumedMillis ?? 0) : 0;
    const memKB = isFinal ? Math.round((sub.memoryConsumedBytes ?? 0) / 1024) : 0;
    const cases = buildCases(sub, time, memKB);
    const tcMap = {};
    cases.forEach(c => { tcMap[c.id] = c; });

    return JSON.parse(JSON.stringify({
      detail: { __CLASS_NAME:'Luogu\\DataClass\\Record\\RecordDetail',
        compileResult:{ __CLASS_NAME:'Luogu\\DataClass\\Record\\JudgeResult\\CompileResult', message:null, opt2:false, success:!isCE },
        judgeResult:{ __CLASS_NAME:'Luogu\\DataClass\\Record\\JudgeResult\\JudgeResult',
          finishedCaseCount:cases.length, memory:memKB, score: null, status:lgCode,
          subtasks:[{ __CLASS_NAME:'Luogu\\DataClass\\Record\\JudgeResult\\SubtaskJudgeResult',
            id:0, judger:'', memory:memKB, score: null, status:lgCode, testCases:tcMap, time }], time },
        version:400 },
      enableO2:true, id: Number(recId), memory:memKB, status:lgCode, time, score: null,
    }));
  }

  const buildStatusPush = (sub, id) => ({
    _channel:'record.track', _channel_param:String(id),
    _ws_type:'server_broadcast', record:buildRecordSubset(sub, id), type:'status_push'
  });

  /* ── 页面世界钩子（恢复 SharedWorker/WS 劫持） ── */
  const PAGE_HOOK = `
  (function () {
    var PORT = null, latestPush = null;
    class FakePort extends EventTarget {
      constructor() { super(); this._om = null; }
      get onmessage() { return this._om; }
      set onmessage(fn) {
        this._om = fn;
        if (latestPush && typeof this._om === 'function') {
          setTimeout(() => { try { this._om({ type: 'message', data: JSON.stringify(latestPush) + '\\n' }); } catch(e) {} }, 500);
        }
      }
      start() {} close() {}
      postMessage(msg) {
        let p = null;
        try {
          if (typeof msg === 'string') p = JSON.parse(msg);
          else if (msg && msg.type === 'join_channel') p = msg;
          else if (msg && msg.func === 'send' && Array.isArray(msg.parameters)) {
            for (let item of msg.parameters) { if (typeof item === 'string') { p = JSON.parse(item); break; } }
          }
        } catch(e) {}
        if (p && latestPush) {
          var jr = {
            _channel: 'record.track', _channel_param: p.channel_param || latestPush._channel_param,
            _ws_type: 'join_result', client_number: 1, result: 'success', type: 'result',
            welcome_message: { detail: latestPush.record.detail, memory: null, record: latestPush.record, score: latestPush.record.score || null, status: latestPush.record.status, time: null, type: 'status_push' }
          };
          setTimeout(() => this._emit({ type: 'message', data: JSON.stringify(jr) + '\\n' }), 200);
        }
      }
      _emit(data) {
        var ev = new MessageEvent('message', { data: data });
        this.dispatchEvent(ev);
        if (typeof this._om === 'function') try { this._om(ev); } catch (err) {}
      }
    }
    function FakeSW() { this.port = PORT = new FakePort(); }
    if (window.SharedWorker) {
      var OSW = window.SharedWorker;
      window.SharedWorker = new Proxy(OSW, { construct: function () { return new FakeSW(); } });
    }
    window.addEventListener('message', function(e) {
      if (e.data && e.data.__rmj_ws_push__) {
        latestPush = e.data.payload;
        if (PORT && typeof PORT._om === 'function') {
          PORT._emit({ type: 'message', data: JSON.stringify(latestPush) + '\\n' });
        }
      }
    });
  })();
  `;

  /* ── HTML 构建 ── */
  function buildHtml(tpl, sub) {
    tpl = tpl.replace(/<head[^>]*>/i, m => m + '<script>' + PAGE_HOOK + '</' + 'script>');

    const FE_RE = /window\._feInjection\s*=\s*JSON\.parse\(decodeURIComponent\("([^"]+)"\)\)/;
    const fm = tpl.match(FE_RE);
    if (!fm) return tpl;
    let data = JSON.parse(decodeURIComponent(fm[1]));
    const rec = data?.currentData?.record;
    if (!rec) return tpl;

    const cu = data.currentUser;
    if (rec.user && cu) Object.assign(rec.user, { uid: cu.uid, name: cu.name, avatar: cu.avatar, color: cu.color || '', slogan: cu.slogan || '' });

    if (code) {
      rec.sourceCode = code;
      rec.sourceCodeLength = code.length;
    } else {
      rec.sourceCode = '// 未找到该提交的源代码\n// 请前往原站查看';
      rec.sourceCodeLength = 0;
    }

    if (isCF && sub) {
      const lgCode = CF2LG[sub.verdict] ?? 0;
      const isFinal = FINAL_CF.has(sub.verdict);
      const time = isFinal ? (sub.timeConsumedMillis ?? 0) : 0;
      const memKB = isFinal ? Math.round((sub.memoryConsumedBytes ?? 0) / 1024) : 0;

      if (rec.problem && sub.problem) {
        rec.problem.pid = `CF${cid}${sub.problem.index}`;
        rec.problem.title = sub.problem.name;
        rec.problem.fullScore = 0;
      }

      rec.status = lgCode;
      rec.time = time;
      rec.memory = memKB;
      rec.score = null; // 分数直接返回 null
      rec.id = Number(sid) || rec.id;
      if (sub.creationTimeSeconds) rec.submitTime = sub.creationTimeSeconds;
      if (rec.detail?.compileResult) rec.detail.compileResult.success = sub.verdict !== 'COMPILATION_ERROR';

      const cases = buildCases(sub, time, memKB);
      const tcMap = {};
      cases.forEach(c => { tcMap[c.id] = c; });

      if (rec.detail?.judgeResult) {
        rec.detail.judgeResult.subtasks = [{ __CLASS_NAME:'Luogu\\DataClass\\Record\\JudgeResult\\SubtaskJudgeResult', id:0, score: null, status:lgCode, testCases:tcMap, judger:'', time, memory:memKB }];
        rec.detail.judgeResult.finishedCaseCount = cases.length;
        rec.detail.judgeResult.status = lgCode;
        rec.detail.judgeResult.time = time;
        rec.detail.judgeResult.memory = memKB;
        rec.detail.judgeResult.score = null;
      }
      data.currentTitle = `CF${cid}${sub.problem?.index || ''} #${sid}`;
    }

    data = JSON.parse(JSON.stringify(data));
    const newExpr = 'window._feInjection = JSON.parse(decodeURIComponent("' + encodeURIComponent(JSON.stringify(data)) + '"))';
    return tpl.replace(FE_RE, () => newExpr);
  }

  /* ── 主流程 ── */
  (async () => {
    if (window.__RMJ_MAIN_RAN__) return;
    window.__RMJ_MAIN_RAN__ = true;

    const [tpl, sub] = await Promise.all([
      fetchTemplate().catch(() => null),
      isCF ? findCF().catch(() => null) : Promise.resolve(null)
    ]);

    if (isCF && sub === null) {
      location.replace(`https://codeforces.com/contest/${cid}/submission/${sid}`);
      return;
    }
    if (!tpl) return;

    const parser = new DOMParser();
    const doc = parser.parseFromString(buildHtml(tpl, sub), 'text/html');
    document.replaceChild(doc.documentElement, document.documentElement);

    document.querySelectorAll('script').forEach(oldScript => {
      const newScript = document.createElement('script');
      Array.from(oldScript.attributes).forEach(attr => newScript.setAttribute(attr.name, attr.value));
      newScript.textContent = oldScript.textContent;
      oldScript.parentNode.replaceChild(newScript, oldScript);
    });

    const origUrl = `https://codeforces.com/contest/${cid}/submission/${sid}`;

    let currentSub = sub;

    // DOM 补丁与篡改定时器 (500ms)
    setInterval(() => {
      // 1. 移除遮罩
      const m = document.getElementById('rmj-mask');
      if (m && (document.querySelector('#app') || document.querySelector('.lfe-body'))) {
        m.remove();
      }

      // 2. 代码长度显示 -
      if (!code) {
        const vals = document.querySelectorAll('.value.lfe-caption');
        if (vals.length >= 2 && vals[1].textContent !== '-') {
          vals[1].textContent = '-';
        }

        // 3. 无代码时的代码区域替换
        const ce = document.querySelector('pre code');
        if (ce) {
          const pre = ce.closest('pre');
          if (pre) {
            pre.outerHTML = `<div class="card padding-default" style="text-align:center; color:#e74c3c; padding:40px; display:flex; flex-direction:column; align-items:center; justify-content:center;">
              <div>本代码非 GenGen RMJ 提交，无法查看</div>
              <a href="${origUrl}" target="_blank" style="margin-top:8px; color:#3498db; text-decoration:none;">前往原站</a>
            </div>`;
          }
        }
      }

      // 4. 底部原站链接
      const card = document.querySelector('.card.padding-default');
      if (card && !card.querySelector('.rmj-orig-link')) {
        const a = document.createElement('a');
        a.href = origUrl;
        a.target = '_blank';
        a.className = 'rmj-orig-link';
        a.textContent = '前往原站';
        a.style.cssText = 'display:block; text-align:center; margin-top:16px; color:#3498db; text-decoration:none;';
        card.appendChild(a);
      }

      // 5. 补全 class="id" 的 ID 为 #N (只在 Judging、AC 等时候通过 DOM 补全，不动其他地方)
      if (currentSub) {
        const passed = currentSub.passedTestCount ?? 0;
        const isAC = currentSub.verdict === 'OK';
        const isCE = currentSub.verdict === 'COMPILATION_ERROR';
        if (!isCE) {
          let idNum;
          if (isAC) {
            idNum = passed > 0 ? passed : 1;
          } else {
            idNum = passed + 1;
          }

          const tcs = document.querySelectorAll('.test-case');
          tcs.forEach(tc => {
            const idEl = tc.querySelector('.id');
            if (idEl && idEl.textContent.trim() !== `#${idNum}`) {
              idEl.textContent = `#${idNum}`;
            }
          });
        }
      }
    }, 500);

    if (!isCF || !sub) return;

    const pushToPage = (s) => {
      window.postMessage({ __rmj_ws_push__: true, payload: buildStatusPush(s, sid) }, '*');
    };

    // 先广播 Judging 再开始篡改 (如果已经是最终状态，则伪造一个 Judging 状态先发出去触发 UI 渲染)
    if (!FINAL_CF.has(sub.verdict)) {
      pushToPage(sub);
    } else {
      const fakeJudgingSub = { ...sub, verdict: 'TESTING', passedTestCount: Math.max(0, (sub.passedTestCount ?? 1) - 1) };
      pushToPage(fakeJudgingSub);
      setTimeout(() => {
        pushToPage(sub);
      }, 600);
    }

    // 3s 轮询获取最新状态
    if (!FINAL_CF.has(sub.verdict)) {
      let polling = false;
      const intervalId = setInterval(async () => {
        if (polling) return;
        polling = true;
        try {
          const s = await findCF();
          if (s) {
            currentSub = s;
            pushToPage(s);
            if (FINAL_CF.has(s.verdict)) clearInterval(intervalId);
          }
        } catch (e) {}
        finally { polling = false; }
      }, 3000);
    }
  })();
})();