shadowfacts
/
ekt
mirror of https://github.com/shadowfacts/ekt.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Add support for imports

This commit is contained in:
Shadowfacts 2017-08-19 16:42:04 -04:00
parent c5f94ef738
commit a444838bf9
Signed by: shadowfacts
GPG Key ID: 94A5AB95422746E5
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/kotlin/net/shadowfacts/ekt/EKT.kt
View File

@ -24,6 +24,8 @@ object EKT {
private val startStringRegex = Regex("([:=#])]") private val startStringRegex = Regex("([:=#])]")
private val endStringRegex = Regex("\\[([:=#])") private val endStringRegex = Regex("\\[([:=#])")
private val lastImportRegex = Regex("(import (.*?)\\n)+")
private val scriptPrefix = """ private val scriptPrefix = """
val _env = bindings["_env"] as net.shadowfacts.ekt.EKT.TemplateEnvironment val _env = bindings["_env"] as net.shadowfacts.ekt.EKT.TemplateEnvironment
val _result = StringBuilder() val _result = StringBuilder()
@ -67,7 +69,11 @@ _result.toString()
} }
}) })
val script = scriptPrefix + template + scriptSuffix val lines = template.split("\n")
val imports = lines.filter { it.trim().startsWith("import") }.joinToString("\n")
template = lines.filterNot { it.trim().startsWith("import") }.joinToString("\n")
val script = imports + scriptPrefix + template + scriptSuffix
if (env.cacheDir != null) { if (env.cacheDir != null) {
env.cacheFile.apply { env.cacheFile.apply {

4
src/test/resources/templates/template.ekt
View File

@ -1,3 +1,7 @@
[: import java.util.ArrayList :]
[= ArrayList::class.simpleName =]
Environment type: [= _env::class.simpleName =] Environment type: [= _env::class.simpleName =]
[= list.joinToString("\n") =] [= list.joinToString("\n") =]