# End-to-end proof of the self-hosting pipeline: a reader form is analyzed by the
# PORTABLE Clojure analyzer (jolt.analyzer, in jolt-core) into host-neutral IR,
# then the Janet back end lowers the IR to a Janet form and evaluates it. No use
# of compiler.janet's analyzer — this is the Clojure-in-Clojure front end.
(import ../../src/jolt/backend :as backend)
(use ../../src/jolt/api)
(use ../../src/jolt/reader)

(defn ce [ctx s] (normalize-pvecs (backend/compile-and-eval ctx (parse-string s))))

(print "self-host pipeline (Clojure analyzer -> IR -> Janet)...")
(let [ctx (init)]
  # primitives + control flow
  (assert (= 3 (ce ctx "(+ 1 2)")) "+")
  (assert (= 6 (ce ctx "(* 2 3)")) "*")
  (assert (= :a (ce ctx "(if true :a :b)")) "if true")
  (assert (= :b (ce ctx "(if false :a :b)")) "if false")
  (assert (= 10 (ce ctx "(let [x 4 y 6] (+ x y))")) "let")
  (assert (= 6 (ce ctx "(do 1 2 6)")) "do")

  # literals
  (assert (= [2 3 4] (ce ctx "(map inc [1 2 3])")) "vector literal + core fn")
  (assert (= 1 (ce ctx "(get {:a 1 :b 2} :a)")) "map literal")
  (assert (= 42 (ce ctx "(quote 42)")) "quote literal")

  # def + global reference (name-based var resolution)
  (ce ctx "(def base 100)")
  (assert (= 142 (ce ctx "(+ base 42)")) "def + later ref")

  # fn / defn (defn is a macro -> expand -> def of fn*)
  (ce ctx "(defn add [a b] (+ a b))")
  (assert (= 7 (ce ctx "(add 3 4)")) "defn")
  (assert (= 49 (ce ctx "((fn [x] (* x x)) 7)")) "anon fn")

  # recursion through the var cell (no recur needed)
  (ce ctx "(defn fib [n] (if (< n 2) n (+ (fib (- n 1)) (fib (- n 2)))))")
  (assert (= 55 (ce ctx "(fib 10)")) "recursive fib via var")

  # multi-arity + variadic
  (ce ctx "(defn arity ([a] a) ([a b] (+ a b)) ([a b & more] (apply + a b more)))")
  (assert (= 5 (ce ctx "(arity 5)")) "multi-arity 1")
  (assert (= 7 (ce ctx "(arity 3 4)")) "multi-arity 2")
  (assert (= 15 (ce ctx "(arity 1 2 3 4 5)")) "multi-arity variadic")

  # higher-order + nesting
  (assert (= 15 (ce ctx "(reduce + (map inc [0 1 2 3 4]))")) "reduce+map"))

(print "self-host pipeline passed!")