blob: 33667cf616cdc2a1d366cd7b17a3ad70fdc4d0b2 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
package dotty.tools
package dottydoc
package staticsite
import dotc.core.Contexts.Context
import dotty.tools.dottydoc.util.syntax._
import dotc.config.Printers.dottydoc
import com.vladsch.flexmark.html.HtmlRenderer
import com.vladsch.flexmark.parser.Parser
import com.vladsch.flexmark.ext.front.matter.AbstractYamlFrontMatterVisitor
import liqp.{ Template => LiquidTemplate }
case class IllegalFrontMatter(message: String) extends Exception(message)
trait Page {
import scala.collection.JavaConverters._
def pageContent: String
def params: Map[String, AnyRef]
def yaml(implicit ctx: Context): Map[String, String] = {
if (_yaml eq null) initFields()
_yaml
}
def html(implicit ctx: Context): String = {
if (_html eq null) initFields()
_html
}
protected[this] var _yaml: Map[String, String] = _
protected[this] var _html: String = _
protected[this] def initFields()(implicit ctx: Context) = {
val md = Parser.builder(ctx.docbase.markdownOptions).build.parse(pageContent)
val yamlCollector = new AbstractYamlFrontMatterVisitor()
yamlCollector.visit(md)
_yaml = yamlCollector
.getData().asScala
.mapValues(_.asScala.headOption.getOrElse(""))
.toMap
// YAML must start with "---" and end in either "---" or "..."
val withoutYaml =
if (pageContent.startsWith("---\n")) {
val str =
pageContent.lines
.drop(1)
.dropWhile(line => line != "---" && line != "...")
.drop(1).mkString("\n")
if (str.isEmpty) throw IllegalFrontMatter(pageContent)
else str
}
else pageContent
// make accessible via "{{ page.title }}" in templates
val page = Map("page" -> _yaml.asJava)
_html = LiquidTemplate
.parse(withoutYaml)
.render((page ++ params).asJava)
}
}
class HtmlPage(fileContents: => String, val params: Map[String, AnyRef]) extends Page {
lazy val pageContent = fileContents
}
class MarkdownPage(fileContents: => String, val params: Map[String, AnyRef]) extends Page {
lazy val pageContent = fileContents
override protected[this] def initFields()(implicit ctx: Context) = {
super.initFields()
val md = Parser.builder(ctx.docbase.markdownOptions).build.parse(_html)
_html = HtmlRenderer
.builder(ctx.docbase.markdownOptions)
.escapeHtml(false)
.build()
.render(md)
}
}
|
