ルートとテンプレートをYAMLファイルで設定できるようにしました。

Author: javecs

build.gradle

@@ -40,6 +40,7 @@ dependencies {
     compile "com.google.guava:guava:21.0"
     compile "com.atilika.kuromoji:kuromoji-ipadic:0.9.0"
     compile "org.antlr:ST4:4.0.8"
+    compile "org.yaml:snakeyaml:1.18"
     compile "org.reflections:reflections:0.9.11"
     compile "org.jetbrains.kotlin:kotlin-stdlib-jre8:$kotlin_version"
     testCompile 'junit:junit:4.12'

src/main/kotlin/xyz/javecs/tools/text2expr/utils/TemplateConfig.kt

@@ -0,0 +1,18 @@
+package xyz.javecs.tools.text2expr.utils
+
+import org.yaml.snakeyaml.Yaml
+
+data class Templates(var templates: List<RuleTemplate> = ArrayList<RuleTemplate>())
+data class RuleTemplate(var rule:String = "", var template:String = "")
+
+class TemplateConfig(path: String = "rule-template.yml", val defaultTemplate: String = "") {
+    private val config = Yaml().loadAs(read(path), Templates::class.java)
+    fun templateOf(rule:String): String {
+        val item = config.templates.find { it.rule == rule }
+        return if (item == null) {
+            defaultTemplate
+        } else {
+            read(item.template)
+        }
+    }
+}

src/main/resources/rule-template.yml

@@ -0,0 +1,5 @@
+templates:
+  - rule: default
+    template: templates/reply.st
+  - rule: rules/length/mileToKilometer.txt
+    template: templates/simple-expr.st

src/main/resources/templates/simple-expr.st

@@ -0,0 +1,2 @@
+答えは、
+<value>

src/test/kotlin/xyz/javecs/tools/text2expr/test/kotlin/TemplateConfigTest.kt

@@ -0,0 +1,51 @@
+package xyz.javecs.tools.text2expr.test.kotlin
+
+import org.junit.Test
+import xyz.javecs.tools.text2expr.utils.TemplateConfig
+import kotlin.test.assertEquals
+
+class TemplateConfigTest {
+    @Test fun template1() {
+        val config = TemplateConfig("config/rule-template-test1.yml")
+        assertEquals("HELLO", config.templateOf("hello"))
+        assertEquals("", config.templateOf("world"))
+
+    }
+
+    @Test fun template2() {
+        val config = TemplateConfig("config/rule-template-test2.yml")
+        assertEquals("HELLO", config.templateOf("hello"))
+        assertEquals("WORLD", config.templateOf("world"))
+    }
+
+    @Test fun template3() {
+        val config = TemplateConfig("config/rule-template-test2.yml")
+        assertEquals("HELLO", config.templateOf("hello"))
+        assertEquals("WORLD", config.templateOf("world"))
+    }
+
+    @Test fun template4() {
+        val config = TemplateConfig("config/rule-template-test1.yml", defaultTemplate = "Ok")
+        assertEquals("Ok", config.templateOf("xyz"))
+    }
+
+    @Test fun template5() {
+        val expected = """
+            |<if(variables)>
+            |<variables:{v|<v.key> = <v.value>
+            |}>
+            |<endif>
+            |<if(expr)>
+            |こうだから、
+            |<expr:{e|<e>
+            |}>
+            |<endif>
+            |答えは、
+            |<value>
+            |
+            """.trimMargin("|")
+        val config = TemplateConfig()
+        assertEquals(expected, config.templateOf("default"))
+    }
+
+}

src/test/resources/config/rule-template-test1.yml

@@ -0,0 +1,3 @@
+templates:
+  - rule: hello
+    template: config/templates/template1.st

src/test/resources/config/rule-template-test2.yml

@@ -0,0 +1,5 @@
+templates:
+  - rule: hello
+    template: config/templates/template1.st
+  - rule: world
+    template: config/templates/template2.st

src/test/resources/config/templates/default-template.st

@@ -0,0 +1 @@
+Hello, world

src/test/resources/config/templates/template1.st

@@ -0,0 +1 @@
+HELLO

src/test/resources/config/templates/template2.st

@@ -0,0 +1 @@
+WORLD