feat: add ast node compile_hints and make it so that AstSourceMap can…

…'t end up inlined into execute

examples/basic_source_map/src/data/demo/functions/foo.mcfunction

@@ -2,3 +2,10 @@ say abc
 
 function demo:bar:
     say def
+
+execute as @p:
+    say something
+
+execute as @p:
+    say foo
+    say bar

mecha/ast.py

@@ -99,7 +99,7 @@
 )
 from uuid import UUID
 
-from beet.core.utils import extra_field, required_field
+from beet.core.utils import JsonDict, extra_field, required_field
 from nbtlib import Byte, ByteArray, Compound, CompoundMatch, Double, Int, IntArray
 from nbtlib import List as ListTag
 from nbtlib import ListIndex, LongArray, NamedKey, Numeric, Path, String
@@ -140,6 +140,7 @@ class AstNode:
     end_location: SourceLocation = extra_field(default=UNKNOWN_LOCATION)
 
     parser: ClassVar[Optional[str]] = None
+    compile_hints: ClassVar[JsonDict] = {}
 
     def __iter__(self) -> Iterator["AstNode"]:
         for f in fields(self):

mecha/contrib/nesting.py

@@ -152,8 +152,18 @@ def nesting_execute_function(self, node: AstCommand):
     def nesting_execute_commands(self, node: AstCommand):
         root = cast(AstRoot, node.arguments[0])
 
-        if len(root.commands) == 1:
-            subcommand = root.commands[0]
+        single_command = None
+        for command in root.commands:
+            if command.compile_hints.get("skip_execute_inline_single_command"):
+                continue
+            if single_command is None:
+                single_command = command
+            else:
+                single_command = None
+                break
+
+        if single_command:
+            subcommand = single_command
 
             if subcommand.identifier == "execute:subcommand":
                 return subcommand.arguments[0]

mecha/contrib/source_map.py

@@ -59,6 +59,8 @@ class AstSourceMap(AstCommandSentinel):
 
     header: str = required_field()
 
+    compile_hints = {"skip_execute_inline_single_command": True}
+
 
 @dataclass
 class SourceMapTransformer(MutatingReducer):

tests/snapshots/examples__build_basic_source_map__0.pack.md

@@ -20,6 +20,8 @@
 ```mcfunction
 # [source_map] src/data/demo/functions/foo.mcfunction
 say abc
+execute as @p run say something
+execute as @p run function demo:foo/nested_execute_0
 ```
 
 `@function demo:thing`
@@ -29,6 +31,14 @@ say abc
 say 456
 ```
 
+`@function demo:foo/nested_execute_0`
+
+```mcfunction
+# [source_map] src/data/demo/functions/foo.mcfunction
+say foo
+say bar
+```
+
 `@function demo:bar`
 
 ```mcfunction