aeo.js 0.0.3 → 0.0.4

package/dist/astro.js

@@ -1139,7 +1139,7 @@ function escapeAttr(str) {
 }
 
 // src/plugins/astro.ts
-function scanBuiltPages(dir, baseUrl) {
+function scanBuiltPages(dir, _baseUrl) {
   const pages = [];
   function walk(currentDir) {
     try {
@@ -1168,7 +1168,8 @@ function scanBuiltPages(dir, baseUrl) {
               pathname,
               title,
               description,
-              content: textContent
+              content: textContent,
+              filePath: fullPath
             });
           } catch {
           }
@@ -1212,6 +1213,55 @@ function scanDevPages(pagesDir) {
   }
   return pages;
 }
+function escapeAttr2(str) {
+  return str.replace(/&/g, "&amp;").replace(/"/g, "&quot;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+}
+function injectHeadTags(pages, config) {
+  let injectedCount = 0;
+  for (const page of pages) {
+    let html;
+    try {
+      html = fs.readFileSync(page.filePath, "utf-8");
+    } catch {
+      continue;
+    }
+    const tags = [];
+    if (!/name=["']description["']/i.test(html)) {
+      const desc = page.description || config.description;
+      if (desc) {
+        tags.push(`<meta name="description" content="${escapeAttr2(desc)}" />`);
+      }
+    }
+    if (!/rel=["']canonical["']/i.test(html)) {
+      const pageUrl = page.pathname === "/" ? config.url : `${config.url.replace(/\/$/, "")}${page.pathname}`;
+      tags.push(`<link rel="canonical" href="${escapeAttr2(pageUrl)}" />`);
+    }
+    if (config.og.enabled && !/property=["']og:/i.test(html)) {
+      const ogHtml = generateOGTagsHtml(page, config);
+      if (ogHtml) tags.push(ogHtml);
+    }
+    if (config.schema.enabled && !/application\/ld\+json/i.test(html)) {
+      const siteSchemas = generateSiteSchemas(config);
+      const pageSchemas = generatePageSchemas(page, config);
+      const jsonLdHtml = generateJsonLdScript([...siteSchemas, ...pageSchemas]);
+      if (jsonLdHtml) tags.push(jsonLdHtml);
+    }
+    if (!/rel=["']alternate["'][^>]*llms\.txt/i.test(html)) {
+      tags.push(`<link rel="alternate" type="text/plain" href="/llms.txt" title="LLM Summary" />`);
+      tags.push(`<link rel="alternate" type="text/plain" href="/llms-full.txt" title="Full Content for LLMs" />`);
+      tags.push(`<link rel="alternate" type="application/json" href="/docs.json" title="Documentation Manifest" />`);
+      tags.push(`<link rel="alternate" type="application/json" href="/ai-index.json" title="AI-Optimized Index" />`);
+    }
+    if (tags.length === 0) continue;
+    const injection = "\n    " + tags.join("\n    ") + "\n  ";
+    const newHtml = html.replace("</head>", injection + "</head>");
+    if (newHtml !== html) {
+      fs.writeFileSync(page.filePath, newHtml, "utf-8");
+      injectedCount++;
+    }
+  }
+  return injectedCount;
+}
 function aeoAstroIntegration(options = {}) {
   let resolvedConfig = resolveConfig(options);
   let astroConfig;
@@ -1292,6 +1342,14 @@ if (!document.querySelector('meta[name="astro-view-transitions-enabled"]')) {
         } catch (error) {
           buildLogger.error(`Failed to generate AEO files: ${error}`);
         }
+        try {
+          const injected = injectHeadTags(discoveredPages, resolvedConfig);
+          if (injected > 0) {
+            buildLogger.info(`Injected head tags into ${injected} pages`);
+          }
+        } catch (error) {
+          buildLogger.error(`Failed to inject head tags: ${error}`);
+        }
       },
       "astro:server:setup": async ({ server, logger }) => {
         const devLogger = logger.fork("aeo.js");