src/LispScriptEngine.kt


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

package moe.nea.lisp

import java.io.Reader
import javax.script.AbstractScriptEngine
import javax.script.Bindings
import javax.script.ScriptContext
import javax.script.SimpleBindings

class LispScriptEngine(private val factory: LispScriptEngineFactory) : AbstractScriptEngine() {
    val executionContext = LispExecutionContext()

    init {
        executionContext.setupStandardBindings()
    }

    override fun eval(script: String, context: ScriptContext): LispData? {
        val fileName = context.getAttribute("scriptName") as? String ?: "script.lisp"
        val program = LispParser.parse(fileName, script)
        val root = executionContext.genBindings()
        for ((name, value) in context.getBindings(ScriptContext.ENGINE_SCOPE)) {
            when (value) {
                is String -> root.setValueLocal(name, LispData.LispString(value))
                is Number -> root.setValueLocal(name, LispData.LispNumber(value.toDouble()))
                is Boolean -> root.setValueLocal(name, LispData.LispNumber(if (value) 1.0 else 0.0))
                null -> root.setValueLocal(name, LispData.LispNil)
                else -> error("Could not convert $value to lisp value")
            }
        }
        return executionContext.executeProgram(root, program)
    }

    override fun eval(reader: Reader, context: ScriptContext): LispData? {
        return eval(reader.readText(), context)
    }

    override fun createBindings(): Bindings {
        return SimpleBindings()
    }

    override fun getFactory(): LispScriptEngineFactory {
        return factory
    }
}