feat: add prepend and append to inline_function plugin

beet/contrib/inline_function.py

@@ -10,7 +10,7 @@
 
 from jinja2.ext import Extension
 from jinja2.nodes import DerivedContextReference  # type: ignore
-from jinja2.nodes import CallBlock, Node
+from jinja2.nodes import CallBlock, Node, TemplateData
 
 from beet import Context, Function
 
@@ -28,6 +28,16 @@ def parse(self, parser: Any) -> Node:
         lineno = next(parser.stream).lineno
 
         args: List[Any] = [DerivedContextReference(), parser.parse_expression()]
+
+        if parser.stream.current.test("name:append"):
+            args.append(TemplateData("append"))
+            parser.stream.skip()
+        elif parser.stream.current.test("name:prepend"):
+            args.append(TemplateData("prepend"))
+            parser.stream.skip()
+        else:
+            args.append(TemplateData("replace"))
+
         body = parser.parse_statements(["name:endfunction"], drop_needle=True)
 
         return CallBlock(
@@ -38,11 +48,17 @@ def parse(self, parser: Any) -> Node:
             lineno=lineno,
         )
 
-    def _function_handler(self, context: Any, path: str, caller: Any) -> str:
+    def _function_handler(self, context: Any, path: str, op: str, caller: Any) -> str:
         ctx: Context = context["ctx"]
 
         with ctx.override(render_path=path, render_group="functions"):
             commands = caller()
 
-        ctx.data[path] = Function(commands)
+        if op == "replace":
+            ctx.data[path] = Function(commands)
+        elif op == "append":
+            ctx.data.functions.setdefault(path, Function()).append(Function(commands))
+        elif op == "prepend":
+            ctx.data.functions.setdefault(path, Function()).prepend(Function(commands))
+
         return ""

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

@@ -19,6 +19,12 @@ function demo:inner
             #!tag "demo:nested"
 
             say from nested {{ i }}
+            #!function "demo:forward" append
+                function {{ nested_function }}
+            #!endfunction
+            #!function "demo:reverse" prepend
+                function {{ nested_function }}
+            #!endfunction
         #!endfunction
     #!endfor
 

tests/snapshots/examples__build_load_inline_functions__0.data_pack/data/demo/functions/forward.mcfunction

@@ -0,0 +1,5 @@
+                function demo:nested_0
+                function demo:nested_1
+                function demo:nested_2
+                function demo:nested_3
+                function demo:nested_4

tests/snapshots/examples__build_load_inline_functions__0.data_pack/data/demo/functions/reverse.mcfunction

@@ -0,0 +1,5 @@
+                function demo:nested_4
+                function demo:nested_3
+                function demo:nested_2
+                function demo:nested_1
+                function demo:nested_0