Code review

Author: emmerich

packages/gitbook/src/lib/document.ts

@@ -30,14 +30,17 @@ export function hasFullWidthBlock(document: JSONDocument): boolean {
     return false;
 }
 
+/**
+ * Returns true if the document has more than `limit` blocks and/or inlines that match the `check` predicate.
+ */
 export function hasMoreThan(
     document: JSONDocument | DocumentBlock,
     check: (block: DocumentBlock | DocumentInline) => boolean,
     limit = 1
 ): boolean {
     let count = 0;
 
-    function traverse(node: JSONDocument | DocumentBlock): boolean {
+    function traverse(node: JSONDocument | DocumentBlock | DocumentFragment): boolean {
         for (const child of 'nodes' in node ? node.nodes : []) {
             if (child.object === 'text') continue;
 
@@ -49,6 +52,12 @@ export function hasMoreThan(
             if (child.object === 'block' && 'nodes' in child) {
                 if (traverse(child)) return true;
             }
+
+            if (child.object === 'block' && 'fragments' in child) {
+                for (const fragment of child.fragments) {
+                    if (traverse(fragment)) return true;
+                }
+            }
         }
         return false;
     }