content.js

function applyStyles(settings) {
  const isFontActive = settings.setFont !== false;
  const isExpandedActive = settings.expandedMode || false;
  const isRtlFixActive = settings.fixRtl || false; // Added

  const customUrl = settings.customFontUrl || "";

  const fontStylesId = "gemini-vazirmatn-font-styles";
  let fontStyleElement = document.getElementById(fontStylesId);

  if (isFontActive) {
    if (!fontStyleElement) {
      fontStyleElement = document.createElement("style");
      fontStyleElement.id = fontStylesId;
      document.head.appendChild(fontStyleElement);
    }

    let fontImport = "";
    let fontName = "'Vazirmatn', 'Segoe UI', Tahoma, sans-serif";

    if (customUrl.startsWith("https://fonts.googleapis.com/css2")) {
      try {
        const urlParams = new URLSearchParams(customUrl.split("?")[1]);
        const familiesParam = urlParams.get("family");
        if (familiesParam) {
          const mainFamily = familiesParam.split(":")[0].replace(/\+/g, " ");
          fontName = `'${mainFamily}', 'Segoe UI', Tahoma, sans-serif`;
        }
      } catch (e) {
        console.error("Could not parse custom font URL:", e);
      }
      fontImport = `@import url('${customUrl}');\n`;
    } else {
      fontImport = `
                @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@400;700&display=swap');
                @import url('https://fonts.googleapis.com/css2?family=Rubik:ital,wght@0,300..900;1,300..900&display=swap');
            `;
    }
    fontStyleElement.textContent = `
            ${fontImport}
            body, p, h1, h2, h3, h4, h5, h6, div, span, a, li, td, th, input, textarea, button {
                font-family: ${fontName} !important;
            }
            *:lang(fa) {
                font-family: ${fontName} !important;
            }
        `;
  } else {
    if (fontStyleElement) {
      fontStyleElement.remove();
    }
  }

  const widthStylesId = "gemini-expanded-width-styles";
  let widthStyleElement = document.getElementById(widthStylesId);

  if (isExpandedActive) {
    if (!widthStyleElement) {
      widthStyleElement = document.createElement("style");
      widthStyleElement.id = widthStylesId;
      document.head.appendChild(widthStyleElement);
    }
    widthStyleElement.textContent = `
            .conversation-container {
                max-width: 90% !important;
                width: 90% !important;
            }
            .input-area-container {
                max-width: 90% !important;
                width: 90% !important;
            }
            user-query {
                max-width: 100% !important;
            }
            .user-query-bubble-with-background,
            .response-container .response-text-container {
                max-width: 100% !important;
            }
        `;
  } else {
    if (widthStyleElement) {
      widthStyleElement.remove();
    }
  }

  // --- Added RTL Fix Engine Section ---
  const rtlStylesId = "gemini-rtl-fix-styles";
  let rtlStyleElement = document.getElementById(rtlStylesId);

  if (isRtlFixActive) {
    if (!rtlStyleElement) {
      rtlStyleElement = document.createElement("style");
      rtlStyleElement.id = rtlStylesId;
      document.head.appendChild(rtlStyleElement);
    }
    // Forces the browser to evaluate the textual direction dynamically based on content string characteristics.
    // Targets the main response text contents and the textarea user input block.
    rtlStyleElement.textContent = `
            .user-query-bubble-with-background, 
            .response-container .response-text-container,
            .rich-textarea,
            textarea,
            p, 
            li {
                direction: attr(dir, auto) !important;
                unicode-bidi: plaintext !important;
                text-align: start !important;
            }
        `;
  } else {
    if (rtlStyleElement) {
      rtlStyleElement.remove();
    }
  }
}

// Ensure "fixRtl" is initialized during initialization phase
chrome.storage.sync.get(
  ["setFont", "expandedMode", "fixRtl", "customFontUrl"],
  (settings) => {
    applyStyles(settings);
  },
);

chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
  if (request.action === "update_styles") {
    chrome.storage.sync.get(
      ["setFont", "expandedMode", "fixRtl", "customFontUrl"],
      (settings) => {
        applyStyles(settings);
      },
    );
  }
});