☰

AHC Visualizer → AtCoder CustomTest Runner (fixed responseText guards)

textarea#input の内容を AtCoder custom_test に送信。responseText を安全に扱う修正版。

// ==UserScript==
// @name         AHC Visualizer → AtCoder CustomTest Runner (fixed responseText guards)
// @namespace    idk
// @version      1.2
// @description  textarea#input の内容を AtCoder custom_test に送信。responseText を安全に扱う修正版。
// @match        https://img.atcoder.jp/ahc*
// @match        https://atcoder.jp/contests/*/custom_test
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_xmlhttpRequest
// @grant        GM_addStyle
// @grant        GM_openInTab
// @grant        unsafeWindow
// @connect      atcoder.jp
// @run-at       document-idle
// ==/UserScript==

(function () {
  'use strict';

  const sleep = ms => new Promise(r => setTimeout(r, ms));
  const ENC = encodeURIComponent;
  const TOKEN_KEY = 'ahc_csrf_token_v1';
  const CONTEST_KEY = 'ahc_contest_slug_v1';

  // save token/contest when on atcoder custom_test page
  if (location.host === 'atcoder.jp' && /\/contests\/[^/]+\/custom_test/.test(location.pathname)) {
    try {
      let token = (typeof unsafeWindow !== 'undefined' && unsafeWindow.csrfToken) ? unsafeWindow.csrfToken : null;
      if (!token) {
        const el = document.querySelector('input[name="csrf_token"]');
        if (el) token = el.value;
      }
      const contest = location.pathname.split('/')[2] || null;
      if (token) GM_setValue(TOKEN_KEY, token);
      if (contest) GM_setValue(CONTEST_KEY, contest);
      console.log('AHC Runner: saved token/contest', { token: !!token, contest });
    } catch (e) {
      console.warn('AHC Runner: failed to save token', e);
    }
    return;
  }

  GM_addStyle(`
    #ahc-mini { position: fixed; right:12px; top:12px; width:420px; max-width:86vw; z-index:2147483647;
      background:#fff; border:1px solid #ddd; padding:10px; border-radius:8px; box-shadow:0 6px 18px rgba(0,0,0,0.12);}
    #ahc-mini textarea{ width:100%; box-sizing:border-box; font-family:monospace; font-size:12px; resize:vertical;}
    #ahc-mini .status{ font-size:12px; margin-top:6px; color:#333; word-break:break-word;}
  `);

  const root = document.createElement('div');
  root.id = 'ahc-mini';
  // 1. root.innerHTML は一旦言語selectの中身だけ空にしておく
  root.innerHTML = `
  <div style="font-weight:700;margin-bottom:6px">AHC Visualizer → AtCoder Runner</div>
  <div>
    <label style="font-size:12px">Language:</label>
    <select id="ahc-lang" style="margin-left:6px;">
      <option>Loading...</option>
    </select>
  </div>
  <textarea id="ahc-code" placeholder="Paste your source code here (required)" rows="8"></textarea>
  <div style="margin-top:6px;">
    <button id="ahc-run" style="font-size:110%">Run</button>
    <button id="ahc-open" style="margin-left:6px">Open Code Test</button>
  </div>
  <div class="status" id="ahc-status">Initializing...</div>
`;

  // 2. 言語リストを取得してセレクトを更新する関数
  async function updateLanguageSelect(contest) {
    const langSel = document.getElementById('ahc-lang');
    langSel.innerHTML = ''; // 一旦クリア

    const langs = await fetchAllowedLanguages(contest);
    if (!langs || Object.keys(langs).length === 0) {
      langSel.innerHTML = '<option>Failed to get languages</option>';
      return;
    }

    for (const [id, name] of Object.entries(langs)) {
      const option = document.createElement('option');
      option.value = id;
      option.textContent = name + ' — ' + id;
      langSel.appendChild(option);
    }
  }

  // 3. スクリプト初期化時に contest を推測 or 保存値から取り出し言語を読み込む
  (async () => {
    const t = await GM_getValue(TOKEN_KEY);
    let c = await GM_getValue(CONTEST_KEY) || guessContestFromImg();
    if (!c) {
      statusEl.textContent = 'Cannot guess contest. Please open custom_test page once.';
      return;
    }
    await updateLanguageSelect(c);

    if (t && c) statusEl.textContent = `Token acquired (contest=${c}). Contents of #input will be sent as stdin.`;
    else statusEl.textContent = 'Token not acquired: Please open custom_test page once (auto open possible).';
  })();

  document.body.appendChild(root);

  const statusEl = document.getElementById('ahc-status');
  const runBtn = document.getElementById('ahc-run');
  const openBtn = document.getElementById('ahc-open');
  const codeTa = document.getElementById('ahc-code');
  const langSel = document.getElementById('ahc-lang');

  function guessContestFromImg() {
    const m = location.href.match(/img\.atcoder\.jp\/([^\/?#/]+)/);
    return m ? m[1] : null;
  }

  async function ensureTokenAndContest(timeout = 30000) {
    let token = await GM_getValue(TOKEN_KEY);
    let contest = await GM_getValue(CONTEST_KEY);
    if (!contest) {
      const g = guessContestFromImg();
      if (g) { contest = g; await GM_setValue(CONTEST_KEY, contest); }
    }
    if (token && contest) return { token, contest };

    const guessed = contest || guessContestFromImg();
    if (!guessed) {
      statusEl.textContent = 'Cannot guess contest. Check URL.';
      throw new Error('contest slug not found');
    }

    statusEl.textContent = 'Token not acquired: Opening custom_test in new tab (manual ok)';
    try { GM_openInTab(`https://atcoder.jp/contests/${guessed}/custom_test`, { active:false, insert:true }); }
    catch (e) { statusEl.textContent = 'Auto open blocked. Please open manually.'; }

    const start = Date.now();
    while (Date.now() - start < timeout) {
      token = await GM_getValue(TOKEN_KEY);
      contest = await GM_getValue(CONTEST_KEY);
      if (token && contest) { statusEl.textContent = 'Token acquired. Ready to run.'; return { token, contest }; }
      await sleep(400);
    }
    statusEl.textContent = 'Token acquisition timed out. Please open custom_test page once.';
    throw new Error('token timeout');
  }

  async function fetchAllowedLanguages(contest) {
    const url = `https://atcoder.jp/contests/${contest}/custom_test`;
    return await new Promise((resolve) => {
      GM_xmlhttpRequest({
        method: 'GET',
        url,
        onload: res => {
          try {
            const respText = String(res.responseText ?? "");
            const doc = new DOMParser().parseFromString(respText, 'text/html');
            const sel = doc.querySelector('select[name="data.LanguageId"], select#language, select[name="LanguageId"]');
            if (!sel) return resolve({});
            const opts = Array.from(sel.options).map(o => ({ id: o.value, text: o.textContent.trim() }));
            const map = {};
            opts.forEach(o => { if (o.id) map[o.id] = o.text; });
            resolve(map);
          } catch (e) { resolve({}); }
        },
        onerror: () => resolve({})
      });
    });
  }

  function findOutputElem() {
    return document.getElementById('output') || document.querySelector('textarea[name="output"], textarea#output, textarea.output') ||
      document.querySelector('[contenteditable="true"], pre, code');
  }

  async function submitAndPoll({contest, token, languageId, sourceCode, inputText}) {
    const allowed = await fetchAllowedLanguages(contest);
    console.log('allowed languages:', allowed);
    if (allowed && Object.keys(allowed).length > 0 && !allowed[languageId]) {
      const firstId = Object.keys(allowed)[0];
      statusEl.textContent = `Selected language not allowed. Automatically switching to ${allowed[firstId]} (${firstId}).`;
      languageId = firstId;
      langSel.value = languageId;
    }

    const submitUrl = `https://atcoder.jp/contests/${contest}/custom_test/submit/json`;
    const resultUrl = `https://atcoder.jp/contests/${contest}/custom_test/json?reload=true`;
    const body = `data.LanguageId=${ENC(languageId)}&sourceCode=${ENC(sourceCode)}&input=${ENC(inputText)}&csrf_token=${ENC(token)}`;

    statusEl.textContent = 'Sending...';
    const postPreview = String(body ?? "").slice(0, 1200);
    console.log('[AHC Runner] POST body preview', postPreview);
    await new Promise((resolve, reject) => {
      GM_xmlhttpRequest({
        method: 'POST',
        url: submitUrl,
        headers: { "Content-Type": "application/x-www-form-urlencoded; charset=UTF-8" },
        data: body,
        onload: r => {
          const resp = String(r.responseText ?? "");
          console.log('[AHC Runner] POST response', r.status, resp.slice(0, 400));
          if (r.status >= 400) return reject(new Error('submit HTTP ' + r.status + ': ' + resp));
          if (resp && /言語.*許可/.test(resp)) return reject(new Error(resp));
          resolve(r);
        },
        onerror: e => reject(new Error('submit failed: ' + e))
      });
    });

    statusEl.textContent = 'Running (polling)...';
    const start = Date.now();
    const TIMEOUT = 2 * 60 * 1000;
    while (true) {
      if (Date.now() - start > TIMEOUT) throw new Error('Result retrieval timeout');
      const data = await new Promise((resolve, reject) => {
        GM_xmlhttpRequest({
          method: 'GET',
          url: resultUrl,
          onload: r => {
            const resp = String(r.responseText ?? "");
            try { resolve(JSON.parse(resp)); } catch (e) { resolve({}); }
          },
          onerror: e => reject(e)
        });
      });
      console.log('[AHC Runner] poll:', data);
      if (data && ('Interval' in data)) {
        await sleep(Math.max(200, Number(data.Interval) || 500));
        continue;
      }
      if (!data || !('Result' in data)) { await sleep(300); continue; }
      return data;
    }
  }

  runBtn.addEventListener('click', async () => {
    runBtn.disabled = true;
    statusEl.textContent = 'Preparing...';
    try {
      const { token, contest } = await ensureTokenAndContest(30000);
      let languageId = String(langSel.value || '4003');
      const sourceCode = codeTa.value || '';
      if (!sourceCode.trim()) { alert('Please input source code'); runBtn.disabled = false; return; }

      const inputElem = document.getElementById('input');
      if (!inputElem) { alert('textarea#input not found'); runBtn.disabled = false; return; }
      await sleep(80);
      const inputText = inputElem.value || '';
      console.log('[AHC Runner] inputText preview:', String(inputText).slice(0, 800));
      if (!inputText && !confirm('textarea#input is empty. Run with empty input?')) { runBtn.disabled = false; return; }

      const result = await submitAndPoll({ contest, token, languageId, sourceCode, inputText });

      const stdout = (typeof result.Stdout !== 'undefined') ? String(result.Stdout) : '';
      const stderr = (typeof result.Stderr !== 'undefined') ? String(result.Stderr) : '';
      const exit = result.Result && result.Result.ExitCode;
      const time = result.Result && result.Result.TimeConsumption;

      const outEl = findOutputElem();
      if (outEl) {
        if ('value' in outEl) {
          outEl.value = stdout;
          outEl.dispatchEvent(new Event('input', { bubbles: true }));
        } else {
          outEl.textContent = stdout;
        }
      } else {
        console.log('[AHC Runner] stdout:', stdout.slice(0, 800));
      }

      statusEl.textContent = `Finished: exit=${exit} time=${time}ms stderr=${stderr ? 'present' : 'none'}`;
    } catch (err) {
      console.error('[AHC Runner] error', err);
      statusEl.textContent = 'Error: ' + (err && err.message ? err.message : err);
      alert('An error occurred during execution. Check the console.');
    } finally {
      runBtn.disabled = false;
    }
  });

  openBtn.addEventListener('click', () => {
    const g = guessContestFromImg();
    if (!g) { alert('Cannot guess contest'); return; }
    window.open(`https://atcoder.jp/contests/${g}/custom_test`, '_blank');
  });

  (async () => {
    const t = await GM_getValue(TOKEN_KEY);
    const c = await GM_getValue(CONTEST_KEY) || guessContestFromImg();
    if (t && c) statusEl.textContent = `Token acquired (contest=${c}). Contents of #input will be sent as stdin.`;
    else statusEl.textContent = 'Token not acquired: Please open custom_test page once (auto open possible).';
  })();

  function addCopyButton(textarea) {
    if (!textarea) return;

    const label = document.querySelector(`label[for="${textarea.id}"]`);

    // ラベルとボタンを同じ行に配置
    const header = document.createElement('div');
    header.style.display = 'flex';
    header.style.alignItems = 'center';
    header.style.gap = '6px'; // ラベルとボタンの間隔

    if (label) {
      textarea.parentNode.insertBefore(header, label);
      header.appendChild(label);
    } else {
      textarea.parentNode.insertBefore(header, textarea);
    }

    const btn = document.createElement('button');
    btn.textContent = 'Copy';
    btn.type = 'button';
    btn.style.padding = '1px 6px 1px 6px';
    btn.style.margin = '1px';

    btn.addEventListener('click', () => {
      navigator.clipboard.writeText(textarea.value).then(() => {
        btn.textContent = 'Copied';
        setTimeout(() => { btn.textContent = 'Copy'; }, 1000);
      }).catch(() => {
        alert('failed to copy');
      });
    });

    header.appendChild(btn);

    // textarea をラベル・ボタンの下に配置
    header.insertAdjacentElement('afterend', textarea);
  }

  window.addEventListener('load', () => {
    addCopyButton(document.getElementById('input'));
    addCopyButton(document.getElementById('output'));
  });

})();