move path-spliting to PathSegment

WorksApplications · Dec 19, 2023 · df9b29a · df9b29a
1 parent 2b82097
commit df9b29a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/src/main/scala/com/worksap/nlp/uzushio/lib/cleaning/PathSegment.scala b/lib/src/main/scala/com/worksap/nlp/uzushio/lib/cleaning/PathSegment.scala
@@ -5,6 +5,9 @@ import scala.collection.mutable.ArrayBuffer
 case class PathSegment(tag: String, id: String, classes: Seq[String]) {
   override def toString: String = classes
     .mkString(tag + (if (classes.isEmpty) "" else "."), ".", if (id == null) "" else s"#$id")
+
+  def splitIdByCase: Seq[String] = id.split("[_-]")
+  def splitClassesByCase: Seq[Array[String]] = classes.map(_.split("[_-]"))
 }
 
 object PathSegment {