Add initial defaults to page generation

1 files changed, 20 insertions, 0 deletions

diff --git a/doc-tool/src/dotty/tools/dottydoc/staticsite/MarkdownLinkVisitor.scala b/doc-tool/src/dotty/tools/dottydoc/staticsite/MarkdownLinkVisitor.scala
new file mode 100644
index 000000000..ac2b6fc6d
--- /dev/null
+++ b/doc-tool/src/dotty/tools/dottydoc/staticsite/MarkdownLinkVisitor.scala
@@ -0,0 +1,20 @@
+package dotty.tools
+package dottydoc
+package staticsite
+
+import com.vladsch.flexmark.ast._
+
+object MarkdownLinkVisitor {
+  def apply(node: Node): Unit =
+    (new NodeVisitor(
+      new VisitHandler(classOf[Link], new Visitor[Link] {
+        override def visit(node: Link): Unit = {
+          val url = node.getUrl
+          if (url.endsWith(".md")) node.setUrl {
+            url.subSequence(0, url.lastIndexOf('.')).append(".html")
+          }
+        }
+      })
+    ))
+    .visit(node)
+}