1
1
mirror of https://github.com/kanaka/mal.git synced 2024-09-11 21:57:38 +03:00
mal/impls/scala/env.scala

43 lines
1000 B
Scala
Raw Normal View History

import types._list
2014-12-19 08:21:39 +03:00
import scala.collection.mutable
object env {
class Env(outer: Env = null,
binds: Iterator[Any] = null,
exprs: Iterator[Any] = null) {
val data: mutable.Map[Symbol, Any] = mutable.Map()
if (binds != null && exprs != null) {
binds.foreach(b => {
val k = b.asInstanceOf[Symbol]
if (k == '&) {
data(binds.next().asInstanceOf[Symbol]) = _list(exprs.toSeq:_*)
2014-12-19 08:21:39 +03:00
} else {
data(k) = exprs.next()
}
})
}
def find(key: Symbol): Env = {
if (data.contains(key)) {
this
} else if (outer != null) {
outer.find(key)
} else {
null
}
}
def set(key: Symbol, value: Any): Any = {
data(key) = value
value
}
def get(key: Symbol): Any = {
val env = find(key)
if (env == null) throw new Exception("'" + key.name + "' not found")
env.data(key)
}
}
}
// vim:ts=2:sw=2