fix: issue with multiple gets before attrs/return

marko-js · Oct 28, 2021 · 6f9840c · 6f9840c
1 parent 907ab23
commit 6f9840c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/util/is-at-root.ts b/src/util/is-at-root.ts
@@ -1,9 +1,12 @@
 import { types as t } from "@marko/compiler";
 import isCoreTag from "./is-core-tag";
 
-export default function isAtRoot(tag: t.NodePath<t.MarkoTag>) {
+export default function isAtRoot(tag: t.NodePath<t.MarkoTag>): boolean {
   const parentPath = tag.parentPath.parentPath!;
   // Special case `<get>` since it currently wraps it's children
   // which is an implementation detail.
-  return parentPath!.isProgram() || isCoreTag("get", parentPath);
+  return (
+    parentPath!.isProgram() ||
+    (isCoreTag("get", parentPath) && isAtRoot(parentPath))
+  );
 }