Bazel DocGen: Fix rewriter.py to correctly rewrite yaml includes

PiperOrigin-RevId: 536336117
Change-Id: I8286f18e5e95c6c60ed03764c4cef3c6743ca488

Author: fweikert

scripts/docs/rewriter.py

@@ -51,8 +51,9 @@ def _fix_md_metadata(content, version):
   return _MD_METADATA_PATTERN.sub(r"\1/versions/{}\2".format(version), content)
 
 
-_YAML_PATH_PATTERN = re.compile(r"((book_|image_)?path: ['\"]?)(/.*?)(['\"]?)$",
-                                re.MULTILINE)
+_YAML_PATH_PATTERN = re.compile(
+    r"(((book_|image_)?path|include): ['\"]?)(/.*?)(['\"]?)$", re.MULTILINE
+)
 
 _YAML_IGNORE_LIST = frozenset(
     ["/", "/_project.yaml", "/versions/", "/versions/_toc.yaml"])
@@ -61,7 +62,7 @@ def _fix_md_metadata(content, version):
 def _fix_yaml_paths(content, version):
 
   def sub(m):
-    prefix, path, suffix = m.group(1, 3, 4)
+    prefix, path, suffix = m.group(1, 4, 5)
     if path in _YAML_IGNORE_LIST:
       return m.group(0)
 

scripts/docs/testdata/expected_output/_book.yaml

@@ -33,3 +33,5 @@ upper_tabs:
         path: /versions/
       - title: Update versions with Bazelisk
         path: /versions/6.6.6/versions/updating-bazel
+      - title: Build encyclopedia
+        include: /versions/6.6.6/reference/be/_toc.yaml

scripts/docs/testdata/input/_book.yaml

@@ -33,3 +33,5 @@ upper_tabs:
         path: /versions/
       - title: Update versions with Bazelisk
         path: /versions/updating-bazel
+      - title: Build encyclopedia
+        include: /reference/be/_toc.yaml