Add note blocks and an integration test

1 files changed, 57 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/blog/md/ext/NoteBlocks.kt b/src/main/kotlin/moe/nea/blog/md/ext/NoteBlocks.kt
new file mode 100644
index 0000000..a1f977f
--- /dev/null
+++ b/src/main/kotlin/moe/nea/blog/md/ext/NoteBlocks.kt
@@ -0,0 +1,57 @@
+package moe.nea.blog.md.ext
+
+import moe.nea.blog.gen.DefHtmlFragmentGenerator
+import moe.nea.blog.gen.HtmlDsl
+import moe.nea.blog.gen.MD2HtmlGenerator
+import moe.nea.blog.md.BlockParser
+import moe.nea.blog.md.MarkdownBlock
+import moe.nea.blog.md.MarkdownParser
+import moe.nea.blog.util.indent
+import moe.nea.blog.util.indentSize
+import java.io.PrintStream
+
+data class NoteBlock(val noteType: String, val child: MarkdownBlock) : MarkdownBlock {
+	override fun debugFormat(indent: Int, printStream: PrintStream) {
+		printStream.indent(indent)
+		printStream.println("<note type=\"$noteType\">")
+
+		child.debugFormat(indent + 2, printStream)
+
+		printStream.indent(indent)
+		printStream.println("</note>")
+	}
+}
+
+object NoteBlockGenerator : DefHtmlFragmentGenerator<NoteBlock>() {
+	override fun HtmlDsl.makeHtml(generator: MD2HtmlGenerator, node: NoteBlock) {
+		element("div", mapOf("class" to "note-${node.noteType}"), generator.generateHtml(node.child))
+	}
+}
+
+object NoteBlockParser : BlockParser {
+	override fun detect(line: String): Boolean {
+		return line.startsWith("!!!")
+	}
+
+	override fun parse(parser: MarkdownParser): MarkdownBlock {
+		val line = parser.consumeLine()!!
+		val noteType = line.substring(3).trim()
+		while ((parser.peekLine() ?: error("Unterminated !!! note")).isBlank()) {
+			parser.consumeLine()
+		}
+		val indent = parser.peekLine()!!.indentSize()!!
+		require(indent > 0) { "!!! note not indented" }
+		parser.pushIndent(indent)
+		val list = mutableListOf<MarkdownBlock>()
+		while (true) {
+			val block = parser.readChildBlock() ?: break
+			list.add(block)
+		}
+		parser.popIndent()
+		return NoteBlock(noteType, parser.mergeBlocks(list))
+	}
+
+	override val prio: Int
+		get() = 10
+
+}