MarkdownPart: identify mdoc code fence modifier

scalameta · Nov 5, 2023 · d7b59fb · d7b59fb
1 parent b8b3fd7
commit d7b59fb
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 9 deletions.
diff --git a/mdoc/src/main/scala/mdoc/internal/markdown/FenceInput.scala b/mdoc/src/main/scala/mdoc/internal/markdown/FenceInput.scala
@@ -15,13 +15,10 @@ case class StringFenceInput(block: CodeFence, input: Input, mod: Str)
 case class ScalaFenceInput(block: CodeFence, input: Input, mod: Modifier)
 
 class FenceInput(ctx: Context, baseInput: Input) {
-  def getModifier(info: Text): Option[Modifier] = {
-    val string = info.value.stripLineEnd
-    if (!string.startsWith("scala mdoc")) None
-    else {
-      if (!string.contains(':')) Some(Modifier.Default())
+  def getModifier(block: CodeFence): Option[Modifier] = {
+    block.getMdocMode.flatMap { mode =>
+      if (mode.isEmpty) Some(Modifier.Default())
       else {
-        val mode = string.stripPrefix("scala mdoc:").takeWhile(!_.isWhitespace)
         Modifier(mode)
           .orElse {
             val (name, info) = mode.split(":", 2) match {
@@ -47,15 +44,15 @@ class FenceInput(ctx: Context, baseInput: Input) {
               }
           }
           .orElse {
-            invalid(info, s"Invalid mode '$mode'")
+            invalid(block.info, s"Invalid mode '$mode'")
             None
           }
       }
     }
   }
 
   private def invalid(info: Text, message: String): Unit = {
-    val offset = "scala mdoc:".length
+    val offset = CodeFence.taglen + 1 // includes colon
     val start = info.posBeg + offset
     val end = info.posEnd - 1
     val pos = Position.Range(baseInput, start, end)
@@ -95,7 +92,7 @@ class FenceInput(ctx: Context, baseInput: Input) {
     }
   }
   def unapply(block: CodeFence): Option[ScalaFenceInput] = {
-    getModifier(block.info) match {
+    getModifier(block) match {
       case Some(mod) =>
         if (isValid(block.info, mod)) {
           val input = Input.Slice(baseInput, block.body.posBeg, block.body.posEnd)

diff --git a/parser/src/main/scala/mdoc/parser/MarkdownPart.scala b/parser/src/main/scala/mdoc/parser/MarkdownPart.scala
@@ -142,6 +142,12 @@ final case class Text(value: String) extends MarkdownPart {
     } else this
   }
 }
+
+object CodeFence {
+  val tag = "scala mdoc"
+  val taglen = tag.length
+}
+
 final case class CodeFence(
     openBackticks: Text,
     info: Text,
@@ -154,4 +160,22 @@ final case class CodeFence(
   var newBody = Option.empty[String]
   def indent: Int = tag.value.length
   def hasBlankTag: Boolean = tag.value.forall(_.isWhitespace)
+
+  def getMdocMode: Option[String] = {
+    val infoStr = info.value
+    if (!infoStr.startsWith(CodeFence.tag)) None
+    else if (infoStr.length == CodeFence.taglen) Some("")
+    else {
+      val head = infoStr.charAt(CodeFence.taglen)
+      if (head == ':') {
+        val idxSpace = infoStr.indexWhere(_.isWhitespace, CodeFence.taglen + 1)
+        Some(
+          if (idxSpace < 0) infoStr.substring(CodeFence.taglen + 1)
+          else infoStr.substring(CodeFence.taglen + 1, idxSpace)
+        )
+      } else {
+        if (head.isWhitespace) Some("") else None
+      }
+    }
+  }
 }