fix: issue rendering attrs & return after a get

src/__tests__/fixtures/attrs/__snapshots__/attrs-after-get/node.render.expected.html

@@ -0,0 +1,3 @@
+<div>
+  Hello world
+</div>

src/__tests__/fixtures/attrs/__snapshots__/attrs-after-get/web.render.expected.html

@@ -0,0 +1,3 @@
+<div>
+  Hello world
+</div>

src/__tests__/fixtures/attrs/index.test.ts

@@ -5,6 +5,11 @@ describe(
   fixture("./templates/basic.marko", { value: "World" })
 );
 
+describe(
+  "<attrs> after get",
+  fixture("./templates/after-get/index.marko", { value: "World" })
+);
+
 describe(
   "<attrs> error args",
   fixture("./templates/error-args.marko", { value: "World" })

src/__tests__/fixtures/attrs/templates/after-get/components/provider.marko

@@ -0,0 +1,3 @@
+<set="Hello">
+  <${input.renderBody}/>
+</set>

src/__tests__/fixtures/attrs/templates/after-get/components/receiver.marko

@@ -0,0 +1,3 @@
+<get/data="provider"/>
+<attrs/{ value }/>
+<div>${data} ${value}</div>

src/__tests__/fixtures/attrs/templates/after-get/index.marko

@@ -0,0 +1,3 @@
+<provider>
+  <receiver value="world"/>
+</provider>

src/__tests__/fixtures/return/__snapshots__/return-after-get/node.render.expected.html

@@ -0,0 +1 @@
+ Hello world

src/__tests__/fixtures/return/__snapshots__/return-after-get/web.render.expected.html

@@ -0,0 +1 @@
+ Hello world

src/__tests__/fixtures/return/index.test.ts

@@ -2,6 +2,8 @@ import fixture, { FixtureHelpers } from "../../fixture";
 
 describe("<return> basic", fixture("./templates/basic.marko"));
 
+describe("<return> after get", fixture("./templates/after-get/index.marko"));
+
 describe("<return> spread", fixture("./templates/spread.marko"));
 
 describe("<return> read multiple", fixture("./templates/read-multiple.marko"));

src/__tests__/fixtures/return/templates/after-get/components/provider.marko

@@ -0,0 +1,3 @@
+<set="world">
+  <${input.renderBody}/>
+</set>

src/__tests__/fixtures/return/templates/after-get/components/receiver.marko

@@ -0,0 +1,3 @@
+<get/data="provider"/>
+<return=data/>
+

src/__tests__/fixtures/return/templates/after-get/index.marko

@@ -0,0 +1,4 @@
+<provider>
+  <receiver/value/>
+  Hello ${value}
+</provider>

src/components/attrs/translate.ts

@@ -1,5 +1,6 @@
 import { types as t } from "@marko/compiler";
 import deepFreeze from "../../util/deep-freeze/transform";
+import isAtRoot from "../../util/is-at-root";
 const usedTag = new WeakSet<t.Hub>();
 
 export = (tag: t.NodePath<t.MarkoTag>) => {
@@ -8,7 +9,7 @@ export = (tag: t.NodePath<t.MarkoTag>) => {
     ? "can only be used once within a template"
     : !tagVar
     ? "requires a tag variable to be assigned to"
-    : !tag.parentPath.parentPath!.isProgram()
+    : !isAtRoot(tag)
     ? "can only used at the root of the template"
     : tag.node.attributes.length > 0
     ? "does not support attributes"

src/components/return/translate.ts

@@ -1,12 +1,13 @@
 import { types as t } from "@marko/compiler";
+import isAtRoot from "../../util/is-at-root";
 const usedTag = new WeakSet<t.Hub>();
 
 export = (tag: t.NodePath<t.MarkoTag>) => {
   const errorMessage = usedTag.has(tag.hub)
     ? "can only be used once within a template"
     : tag.node.var
     ? "does not support a tag variable"
-    : !tag.parentPath.parentPath!.isProgram()
+    : !isAtRoot(tag)
     ? "can only used at the root of the template"
     : !tag.node.attributes.length
     ? "requires a default attribute"

src/util/is-at-root.ts

@@ -0,0 +1,9 @@
+import { types as t } from "@marko/compiler";
+import isCoreTag from "./is-core-tag";
+
+export default function isAtRoot(tag: t.NodePath<t.MarkoTag>) {
+  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);
+}