@kenkaiiii/gg-pixel 4.3.92 → 4.3.93

package/dist/index.js

@@ -881,6 +881,95 @@ function injectNextClientComponent(layoutPath, clientInitPath) {
   let spec = relative(fromDir, clientInitPath).split(sep).join("/");
   if (!spec.startsWith(".")) spec = "./" + spec;
   spec = spec.replace(/\.tsx$/, "");
+  const astResult = injectClientComponentViaAst(layoutPath, content, spec);
+  if (astResult) return astResult;
+  return injectClientComponentViaRegex(layoutPath, content, spec);
+}
+function injectClientComponentViaAst(layoutPath, content, importSpec) {
+  let recast;
+  let bp;
+  let bt;
+  try {
+    recast = nodeRequire("recast");
+    bp = nodeRequire("@babel/parser");
+    bt = nodeRequire("@babel/types");
+  } catch {
+    return null;
+  }
+  let ast;
+  try {
+    ast = recast.parse(content, {
+      parser: {
+        parse: (src) => bp.parse(src, {
+          sourceType: "module",
+          plugins: ["jsx", "typescript"],
+          allowImportExportEverywhere: true,
+          tokens: true
+        })
+      }
+    });
+  } catch {
+    return null;
+  }
+  const program = ast.program;
+  if (!program || !Array.isArray(program.body)) return null;
+  const bodyEl = findFirstJsxElementByName(ast, "body");
+  if (!bodyEl) return null;
+  const newComponent = bt.jsxElement(
+    bt.jsxOpeningElement(bt.jsxIdentifier("GGPixelClient"), [], true),
+    null,
+    [],
+    true
+  );
+  const leadingText = bt.jsxText("\n        ");
+  const trailingText = bt.jsxText("\n        ");
+  bodyEl.children = [leadingText, newComponent, trailingText, ...bodyEl.children];
+  const importDecl = bt.importDeclaration(
+    [bt.importDefaultSpecifier(bt.identifier("GGPixelClient"))],
+    bt.stringLiteral(importSpec)
+  );
+  const body = program.body;
+  let insertAt = 0;
+  for (let i = 0; i < body.length; i++) {
+    if (body[i]?.type === "ImportDeclaration") insertAt = i + 1;
+  }
+  body.splice(insertAt, 0, importDecl);
+  let out;
+  try {
+    out = recast.print(ast).code;
+  } catch {
+    return null;
+  }
+  writeFileSync(layoutPath, out, "utf8");
+  return { kind: "injected", entryPath: layoutPath };
+}
+function findFirstJsxElementByName(ast, name) {
+  let found = null;
+  function walk(node) {
+    if (found || !node || typeof node !== "object") return;
+    if (Array.isArray(node)) {
+      for (const c of node) walk(c);
+      return;
+    }
+    const n = node;
+    if (n.type === "JSXElement") {
+      const opening = n.openingElement;
+      if (opening?.type === "JSXOpeningElement" && opening.name) {
+        const namedNode = opening.name;
+        if (namedNode.type === "JSXIdentifier" && namedNode.name === name) {
+          found = n;
+          return;
+        }
+      }
+    }
+    for (const key of Object.keys(n)) {
+      walk(n[key]);
+    }
+  }
+  walk(ast);
+  return found;
+}
+function injectClientComponentViaRegex(layoutPath, content, spec) {
   const importLine = `import GGPixelClient from ${JSON.stringify(spec)};`;
   const lines = content.split("\n");
   let insertImportAt = 0;