fix: load worldgen directives in snapshots

examples/worldgen.md

@@ -0,0 +1,27 @@
+# Lectern snapshot
+
+## Data pack
+
+`@data_pack pack.mcmeta`
+
+```json
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  }
+}
+```
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```

examples/worldgen_overlay.md

@@ -0,0 +1,57 @@
+# Lectern snapshot
+
+## Data pack
+
+`@data_pack pack.mcmeta`
+
+```json
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  },
+  "overlays": {
+    "entries": [
+      {
+        "formats": {
+          "min_inclusive": 48,
+          "max_inclusive": 48
+        },
+        "directory": "overlay_48"
+      }
+    ]
+  }
+}
+```
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```
+
+## Overlay `overlay_48`
+
+`@overlay overlay_48`
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-overlay-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```
+
+`@endoverlay`

lectern/directive.py

@@ -16,6 +16,7 @@
 from zipfile import ZipFile
 
 from beet import Container, DataPack, NamespaceFile, ResourcePack
+from beet.contrib.worldgen import worldgen
 from beet.core.utils import snake_case
 
 from .fragment import Fragment
@@ -52,6 +53,8 @@ def __init__(
         @self.add_resolver
         def _(self: DirectiveRegistry):
             for pack in [self.assets, self.data]:
+                if isinstance(pack, DataPack):
+                    worldgen(pack)
                 for file_type in pack.resolve_scope_map().values():
                     name = snake_case(file_type.__name__)
                     self[name] = NamespacedResourceDirective(file_type)

tests/snapshots/examples__markdown_examples_worldgen_md__0.pack.md

@@ -0,0 +1,27 @@
+# Lectern snapshot
+
+## Data pack
+
+`@data_pack pack.mcmeta`
+
+```json
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  }
+}
+```
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```

tests/snapshots/examples__markdown_examples_worldgen_overlay_md__0.pack.md

@@ -0,0 +1,57 @@
+# Lectern snapshot
+
+## Data pack
+
+`@data_pack pack.mcmeta`
+
+```json
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  },
+  "overlays": {
+    "entries": [
+      {
+        "formats": {
+          "min_inclusive": 48,
+          "max_inclusive": 48
+        },
+        "directory": "overlay_48"
+      }
+    ]
+  }
+}
+```
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```
+
+## Overlay `overlay_48`
+
+`@overlay overlay_48`
+
+### demo
+
+`@dimension demo:demo`
+
+```json
+{
+  "type": "demo:in-the-overlay-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+```
+
+`@endoverlay`

tests/snapshots/examples__text_examples_worldgen_md__0.pack.txt

@@ -0,0 +1,15 @@
+@data_pack pack.mcmeta
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  }
+}
+
+@dimension demo:demo
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}

tests/snapshots/examples__text_examples_worldgen_overlay_md__0.pack.txt

@@ -0,0 +1,38 @@
+@data_pack pack.mcmeta
+{
+  "pack": {
+    "pack_format": 48,
+    "description": ""
+  },
+  "overlays": {
+    "entries": [
+      {
+        "formats": {
+          "min_inclusive": 48,
+          "max_inclusive": 48
+        },
+        "directory": "overlay_48"
+      }
+    ]
+  }
+}
+
+@dimension demo:demo
+{
+  "type": "demo:in-the-base-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+
+@overlay overlay_48
+
+@dimension demo:demo
+{
+  "type": "demo:in-the-overlay-pack",
+  "generator": {
+    "type": "minecraft:debug"
+  }
+}
+
+@endoverlay