content.js

function getLeetClipIcon() {
  const iconUrl = chrome.runtime.getURL("assets/copy.svg");

  return `<img src="${iconUrl}" style="width: 18px; height: 18px;" alt="LeetClip">`;
}

function getCheckIcon(color = "#2cbb5d") {
  return `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" width="18" height="18" fill="${color}">
            <path d="M9 16.17L4.83 12l-1.42 1.41L9 19 21 7l-1.41-1.41z"/>
          </svg>`;
}

function extractProblemDescription() {
  const descElement = document.querySelector(
    '[data-track-load="description_content"]',
  );
  return descElement ? descElement.innerText : "Problem description not found.";
}

// Listen for the Monaco code from inject.js
window.addEventListener("message", (event) => {
  if (
    event.source !== window ||
    !event.data ||
    event.data.type !== "LEETCLIP_CODE"
  ) {
    return;
  }

  const code = event.data.text;
  const description = extractProblemDescription();

  chrome.storage.local.get(["activePrompt"], (data) => {
    let activePromptText = "Is this code correct? Give me the time and space complexity analysis and suggest improvements.";
    
    if (data.activePrompt) {
      activePromptText = data.activePrompt.text;
    }
    
    const finalPayload = `### Problem Context & Constraints\n${description}\n\n### My Current Code\n\`\`\`\n${code}\n\`\`\`\n\n### Task\n${activePromptText}`;
    navigator.clipboard.writeText(finalPayload).then(() => {
      const btn = document.getElementById("leetclip-btn");
      if (btn) {
        btn.dataset.state = "success";

        btn.style.backgroundColor = "rgba(44, 187, 93, 0.2)";
        btn.innerHTML = getCheckIcon();

        setTimeout(() => {
          btn.dataset.state = "";

          btn.style.backgroundColor = btn.matches(":hover")
            ? "rgba(0,0,0,0.1)"
            : "rgba(0,0,0,0.05)";
          btn.innerHTML = getLeetClipIcon();
        }, 2000);
      }
    });
  });
});

function requestCopyToLLM() {
  const script = document.createElement("script");
  script.src = chrome.runtime.getURL("inject.js");
  script.onload = function () {
    this.remove();
  };
  (document.head || document.documentElement).appendChild(script);
}

// --- DOM INJECTION ---

function injectLeetClipButton() {
  if (document.getElementById("leetclip-btn")) return;

  const toolbar = document.querySelector(".h-full.py-2");
  if (!toolbar) return;

  const actionBtnGrp = toolbar.children[0];

  const copyBtn = document.createElement("button");
  copyBtn.id = "leetclip-btn";
  copyBtn.title = "Copy to LeetClip";

  Object.assign(copyBtn.style, {
    width: "32px",
    height: "32px",
    borderRadius: "6px",
    border: "none",
    cursor: "pointer",
    backgroundColor: "rgba(0,0,0,0.05)",
    display: "flex",
    alignItems: "center",
    justifyContent: "center",
    transition: "background-color 0.2s ease",
  });

  // Hover effects
  copyBtn.addEventListener("mouseenter", () => {
    if (copyBtn.dataset.state !== "success") {
      copyBtn.style.backgroundColor = "rgba(0,0,0,0.1)";
    }
  });
  
  copyBtn.addEventListener("mouseleave", () => {
    if (copyBtn.dataset.state !== "success") {
      copyBtn.style.backgroundColor = "rgba(0,0,0,0.05)";
    }
  });

  copyBtn.innerHTML = getLeetClipIcon();

  copyBtn.addEventListener("click", () => {
    requestCopyToLLM();
  });

  actionBtnGrp.appendChild(copyBtn);
}

setInterval(injectLeetClipButton, 2000);