Merge pull request #217 from dubek/lua-interop

[jackhill/mal.git] / mal / env.mal

;; env 

(def! bind-env (fn* [env b e]
  (if (empty? b)
    env

    (if (= "&" (str (first b)))
      (assoc env (str (nth b 1)) e)

      (bind-env (assoc env (str (first b)) (first e))
                (rest b) (rest e))))))

(def! new-env (fn* [& args]
  (if (<= (count args) 1)
    (atom {"--outer--" (first args)})
    (atom (bind-env {"--outer--" (first args)}
                    (nth args 1) (nth args 2))))))

(def! env-find (fn* [env k]
  (let* [ks (str k)
         data @env]
    (if (contains? data ks)
      env
      (if (get data "--outer--")
        (env-find (get data "--outer--") ks)
        nil)))))

(def! env-get (fn* [env k]
  (let* [ks (str k)
         e (env-find env ks)]
    (if e
      (get @e ks)
      (throw (str "'" ks "' not found"))))))

(def! env-set (fn* [env k v]
  (do
    (swap! env assoc (str k) v)
    v)))

;;(prn "loaded env.mal")