janet/src/compiler/boot.dst

(def defmacro macro
  (fn [name & more] (tuple 'def name 'macro (tuple-prepend (tuple-prepend more name) 'fn))))

(defmacro defn
 [name & more] 
 (tuple 'def name (tuple-prepend (tuple-prepend more name) 'fn)))

(defmacro when
 [cond & body] 
 (tuple 'if cond (tuple-prepend body 'do)))

(def identity (fn [x] x))

(def seq (do
  (defn array-seq [x]
    (def len (length x))
    (var i 0)
    {
      :more (fn [] (< i len))
      :next (fn []
        (def ret (get x i))
        (varset! i (+ i 1))
        ret) 
    })
  (def seqs {
    :array array-seq
    :tuple array-seq
    :struct (fn [x] x)})
  (fn [x]
   (def makeseq (get seqs (type x)))
   (if makeseq (makeseq x) (error "expected sequence")))))

(defn range [top]
  (var i 0)
  {
    :more (fn [] (< i top))
    :next (fn []
      (def ret i)
      (varset! i (+ i 1))
      ret) 
  })

(defn doseq [s] 
 (def s (seq s))
 (def more? (get s :more))
 (def getnext (get s :next))
 (while (more?)
    (getnext)))

(defn map [f s]
 (def s (seq s))
 (def more (get s :more))
 (def getnext (get s :next))
 {
  :more more
  :next (fn [] (f (getnext)))
 })

(defn reduce [f start s]
  (def s (seq s))
  (def more? (get s :more))
  (def getnext (get s :next))
  (var ret start)
  (while (more?)
    (varset! ret (f ret (getnext))))
  ret)

(defmacro for [head & body]
  (def head (ast-unwrap1 head))
  (def sym (get head 0))
  (def start (get head 1))
  (def end (get head 2))
  (def _inc (get head 3))
  (def inc (if _inc _inc 1))
  (tuple 'do
    (tuple 'var sym start)
    (tuple 'while (tuple '< sym end)
     (tuple-prepend body 'do)
     (tuple 'varset! sym (tuple '+ sym 1))
      )))

(def pp (do

(defn pp-seq [pp seen buf a start end]
 (if (get seen a)
  (buffer-push-string buf "<cycle>")
  (do
   (put seen a true)
   (def len (length a))
   (buffer-push-string buf start)
   (for [i 0 len]
    (when (not= i 0) (buffer-push-string buf " "))
    (pp seen buf (get a i)))
   (buffer-push-string buf end)
  ))
 buf)
  
(defn pp-dict [pp seen buf a start end]
 (if (get seen a)
  (buffer-push-string buf "<cycle>")
  (do
   (put seen a true)
   (var k (next a nil))
   (buffer-push-string buf start)
   (while k
    (def v (get a k))
    (pp seen buf k)
    (buffer-push-string buf " ")
    (pp seen buf v)
    (varset! k (next a k))
    (when k (buffer-push-string buf " "))
   )
   (buffer-push-string buf end)
  ))
 buf)

(def _printers {
 :array  (fn [pp seen buf x] (pp-seq pp seen buf x "[" "]"))
 :tuple  (fn [pp seen buf x] (pp-seq pp seen buf x "(" ")"))
 :table  (fn [pp seen buf x] (pp-dict pp seen buf x "@{" "}"))
 :struct (fn [pp seen buf x] (pp-dict pp seen buf x "{" "}"))
})

(defn _default_printer [pp seen buf x] 
 (buffer-push-string buf (string x)) 
 buf)

(defn pp1 [seen buf x]
  (def pmaybe (get _printers (type x)))
  (def p (if pmaybe pmaybe _default_printer))
  (p pp1 seen buf x))

(fn [x] (print (pp1 @{} @"" x)))
))

(defn pairs [x]
 (var lastkey (next x nil))
 {
    :more (fn [] lastkey)
    :next (fn []
        (def ret (tuple lastkey (get x lastkey)))
        (varset! lastkey (next x lastkey))
        ret)
 })

(defn keys [x]
 (var lastkey (next x nil))
 {
    :more (fn [] lastkey)
    :next (fn []
        (def ret lastkey)
        (varset! lastkey (next x lastkey))
        ret)
 })

(defn values [x]
 (var lastkey (next x nil))
 {
    :more (fn [] lastkey)
    :next (fn []
        (def ret (get x lastkey))
        (varset! lastkey (next x lastkey))
        ret)
 })

# Handle errors
(defn onerr [t e]
 (print (string t " error: " e)))

(var *read* nil)
(var *onvalue* identity)

# Cache for require
(def require-loading @{})

# Create a char stream from a function that gets input
(defn char-stream [getchunk ondone]
 (fiber (fn [parent]
   (def buf @"")
   (var len 1)
   (while (< 0 len)
      (buffer-clear buf)
      (getchunk buf) 
      (varset! len (length buf))
      (for [i 0 len]
        (transfer parent (get buf i))))
   (ondone))))

# Convert a charstream into a value
(defn val-stream [chars]
 (fiber (fn [parent]
    (var up parent)
    (def me (fiber-current))
    (def p (parser)) 
    (while true
      (def s (parser-status p)) 
      (if (= s :full)
       (varset! up (transfer up (parser-produce p)))
       (if (= s :error)
        # Parse error
        (onerr "parse" (parser-error p))
        # Normal state, get next char
        (parser-byte p (transfer chars me))))))))

(defn require [path]
 (when (get require-loading path)
  (error (string "circular dependency: module " path "is already loading")))
 (def oldread *read*)
 (def oldonvalue *onvalue*)
 (def f (file-open path))
 (def getter (fn [buf] (file-read f 1024 buf)))
 (def cs (char-stream getter (fn []
    (put require-loading path nil)
    (file-close f)
    (varset! *read* oldread)
    (varset! *onvalue* oldonvalue)
    nil)))
 (def vs (val-stream cs))
 (varset! *onvalue* identity)
 (varset! *read* (fn [] (transfer vs (fiber-current))))
 nil)

(defn dorepl []
 (def oldread *read*)
 (def cs (char-stream (fn [buf]
    (file-write stdout ">> ")
    (file-read stdin :line buf))
  (fn []
    (varset! *read* oldread)
    nil)))
 (def vs (val-stream cs))
 (varset! *onvalue* (fn [ret]
  (put _env '_ @{'value ret})
  (describe ret)))
 (varset! *read* (fn [] (transfer vs (fiber-current)))))

# Main loop
(defn init-loop []
 (while *read*
  (def wrapper (fiber (fn []
    (while *read*
      (def source (*read*))
      (def res (compile source _env))
      (if (= (type res) :function)
        (*onvalue* (res))
        (onerr "compile" (get res :error)))))))
  (def eb (transfer wrapper))
  (if (= (fiber-status wrapper) :error) (onerr "runtime" eb))))

(defn init-repl [] (dorepl) (init-loop))