limit chapter depth at 400

jmmk · Feb 2, 2021 · c474980 · c474980
1 parent 0f9b88f
commit c474980
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/modules/study/src/main/ChapterRepo.scala b/modules/study/src/main/ChapterRepo.scala
@@ -147,10 +147,11 @@ final class ChapterRepo(val coll: AsyncColl)(implicit
   }
 
   private def childrenTreeToBsonElements(parentPath: Path, children: Node.Children): Vector[(String, Bdoc)] =
-    children.nodes.flatMap { node =>
-      val path = parentPath + node
-      childrenTreeToBsonElements(path, node.children) appended (path.toDbField -> writeNode(node))
-    }
+    (parentPath.depth < Node.MAX_PLIES) ??
+      children.nodes.flatMap { node =>
+        val path = parentPath + node
+        childrenTreeToBsonElements(path, node.children) appended (path.toDbField -> writeNode(node))
+      }
 
   private def setNodeValue[A: BSONWriter](
       field: String,

diff --git a/modules/study/src/main/Path.scala b/modules/study/src/main/Path.scala
@@ -31,6 +31,8 @@ case class Path(ids: Vector[UciCharPair]) extends AnyVal {
     if (ids.isEmpty) s"root.${Path.rootDbKey}"
     else s"root.${Path encodeDbKey this}"
 
+  def depth = ids.size
+
   override def toString = ids.mkString
 }