content-script.js

(() => {
  const SELECTORS = {
    mst: '#nbmst',
    loaiHoaDon: '#lhdon',
    kyHieu: '#khhdon',
    soHoaDon: '#shdon',
    tongTienThue: '#tgtthue',
    tongTienThanhToan: '#tgtttbso'
  };

  function query(selector) {
    try {
      return document.querySelector(selector);
    } catch (_) {
      return null;
    }
  }

  function setInputValue(input, value) {
    if (!input || value == null) return;
    const nativeInputValueSetter = Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype, 'value')?.set;
    nativeInputValueSetter?.call(input, String(value));
    input.dispatchEvent(new Event('input', { bubbles: true }));
    input.dispatchEvent(new Event('change', { bubbles: true }));
  }

  function selectLoaiHoaDonByText(selectRoot, text) {
    if (!selectRoot || !text) return;
    // Ant Design v3/v4 custom select; try click to open and choose by innerText
    const trigger = selectRoot.querySelector('.ant-select-selection, .ant-select-selector');
    (trigger || selectRoot).click();
    const tryPick = () => {
      const options = Array.from(document.querySelectorAll('.ant-select-dropdown .ant-select-dropdown-menu-item, .ant-select-item-option'));
      const target = options.find(o => o.textContent?.trim().toLowerCase() === String(text).trim().toLowerCase());
      if (target) {
        target.click();
        return true;
      }
      return false;
    };
    setTimeout(tryPick, 50);
  }

  async function fillFromStoredDefaults() {
    const defaults = await new Promise(resolve => {
      try {
        chrome.storage.sync.get(null, resolve);
      } catch (_) {
        resolve({});
      }
    });
    applyValues(defaults);
  }

  function applyValues(values) {
    setInputValue(query(SELECTORS.mst), values.mst);
    selectLoaiHoaDonByText(query(SELECTORS.loaiHoaDon), values.loaiHoaDonText || values.loaiHoaDon);
    setInputValue(query(SELECTORS.kyHieu), values.kyHieu);
    setInputValue(query(SELECTORS.soHoaDon), values.soHoaDon);
    setInputValue(query(SELECTORS.tongTienThue), values.tongTienThue);
    setInputValue(query(SELECTORS.tongTienThanhToan), values.tongTienThanhToan);
  }

  function waitForAppReady(callback) {
    const maxWaitMs = 10000;
    const start = Date.now();
    const timer = setInterval(() => {
      const hasForm = query(SELECTORS.mst) && query(SELECTORS.kyHieu) && query(SELECTORS.soHoaDon);
      if (hasForm || Date.now() - start > maxWaitMs) {
        clearInterval(timer);
        callback();
      }
    }, 150);
  }

  // Listen for commands from popup to fill now
  try {
    chrome.runtime.onMessage.addListener((msg, _sender, sendResponse) => {
      if (msg?.type === 'HDDT_FILL') {
        try {
          applyValues(msg.payload || {});
          sendResponse({ ok: true, message: 'Đã điền thành công' });
        } catch (error) {
          console.error('Error applying values:', error);
          sendResponse({ ok: false, error: error.message });
        }
        return true;
      }

      if (msg?.type === 'HDDT_FILL_CAPTCHA') {
        try {
          const captchaInput = findCaptchaInput();
          if (captchaInput) {
            setInputValue(captchaInput, msg.payload.captcha);
            captchaInput.focus(); // Focus to make it easier for user to submit
            sendResponse({ ok: true, message: 'Đã điền CAPTCHA' });
          } else {
            sendResponse({ ok: false, error: 'Không tìm thấy ô nhập CAPTCHA' });
          }
        } catch (error) {
          sendResponse({ ok: false, error: error.message });
        }
        return true;
      }

      if (msg?.type === 'HDDT_EXTRACT_CAPTCHA') {
        try {
          const captchaImg = findCaptchaImage();
          if (captchaImg) {
            const src = captchaImg.src;
            if (src.startsWith('data:image')) {
              // Already base64
              const base64 = src.split(',')[1];
              const mimeType = src.match(/data:(.*?);/)?.[1] || 'image/png';
              sendResponse({ ok: true, data: { base64, mimeType } });
            } else {
              sendResponse({ ok: false, error: 'CAPTCHA không ở dạng base64' });
            }
          } else {
            sendResponse({ ok: false, error: 'Không tìm thấy ảnh CAPTCHA' });
          }
        } catch (error) {
          sendResponse({ ok: false, error: error.message });
        }
        return true;
      }
      return false;
    });
  } catch (error) {
    console.error('Error setting up message listener:', error);
  }

  function findCaptchaInput() {
    // Priority 1: Common IDs/Names on tax sites
    const candidates = [
      '#cvalue', '#captcha', '#captchaCode', '#securityCode', '#txtCaptcha',
      'input[name="cvalue"]', 'input[name="captcha"]', 'input[name="securityCode"]'
    ];

    for (const selector of candidates) {
      const el = document.querySelector(selector);
      if (el) return el;
    }

    // Priority 2: Look for input near "captcha" text
    const inputs = Array.from(document.querySelectorAll('input[type="text"]'));
    return inputs.find(input => {
      const id = (input.id || '').toLowerCase();
      const name = (input.name || '').toLowerCase();
      const placeholder = (input.placeholder || '').toLowerCase();
      return id.includes('captcha') || name.includes('captcha') || placeholder.includes('captcha') || placeholder.includes('mã xác thực');
    });
  }

  function findCaptchaImage() {
    // Priority 1: Known class from screenshot
    const candidates = [
      '.Captcha__Image-sc-1uplk1e-1',
      'img[alt="captcha"]',
      'img[alt*="captcha" i]',
      '.captcha-image',
      '#cimage'
    ];

    for (const selector of candidates) {
      const el = document.querySelector(selector);
      if (el) return el;
    }

    return null;
  }


  waitForAppReady(fillFromStoredDefaults);
})();