Process HTML files like markdown files, fixes #199. (#200)

Process HTML files like markdown files, fixes #199.
scalameta · Oct 31, 2019 · d3d4c8e · d3d4c8e
2 parents 8ce1073 + c273908
commit d3d4c8e
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 3 deletions.
diff --git a/mdoc/src/main/scala/mdoc/internal/cli/MainOps.scala b/mdoc/src/main/scala/mdoc/internal/cli/MainOps.scala
@@ -85,9 +85,11 @@ final class MainOps(
     try {
       if (!settings.isIncluded(file.relpath)) Exit.success
       else {
-        PathIO.extension(file.in.toNIO) match {
-          case "md" => handleMarkdown(file)
-          case _ => handleRegularFile(file)
+        val extension = PathIO.extension(file.in.toNIO)
+        if (settings.isMarkdownFileExtension(extension)) {
+          handleMarkdown(file)
+        } else {
+          handleRegularFile(file)
         }
       }
     } catch {

diff --git a/mdoc/src/main/scala/mdoc/internal/cli/Settings.scala b/mdoc/src/main/scala/mdoc/internal/cli/Settings.scala
@@ -96,6 +96,8 @@ case class Settings(
     usage: Boolean = false,
     @Description("Print out the version number and exit")
     version: Boolean = false,
+    @Description("Set of file extensions to treat as markdown files.")
+    markdownExtensions: List[String] = List("md", "html"),
     @Description(
       "Glob to filter which files to process. Defaults to all files. " +
         "Example: --include **/example.md will process only files with the name example.md."
@@ -135,6 +137,8 @@ case class Settings(
     variablePrinter: Variable => String = ReplVariablePrinter
 ) {
 
+  val isMarkdownFileExtension = markdownExtensions.toSet
+
   def withProperties(props: MdocProperties): Settings =
     copy(
       scalacOptions = props.scalacOptions,

diff --git a/tests/unit/src/test/scala/tests/cli/CliSuite.scala b/tests/unit/src/test/scala/tests/cli/CliSuite.scala
@@ -168,4 +168,27 @@ class CliSuite extends BaseCliSuite {
     }
   )
 
+  checkCli(
+    "html",
+    """
+      |/index.html
+      |<h1>My Fantastic Talk!</h1>
+      |<p>
+      |```scala mdoc
+      |println(42)
+      |```
+      |</p>
+      |""".stripMargin,
+    """
+      |/index.html
+      |<h1>My Fantastic Talk!</h1>
+      |<p>
+      |```scala
+      |println(42)
+      |// 42
+      |```
+      |</p>
+      |""".stripMargin
+  )
+
 }