;;; Standard lib, in scheme

; Lists

(define (foldl f i l)
  (if (null? l)
      i
      (foldl f (f (car l) i) (cdr l))))

(define (foldr f i l)
  (if (null? l)
      i
      (f (car l) (foldr f i (cdr l)))))


(define (map f l)
  (if (null? l)
      '()
      (foldr (lambda (e a) (cons (f e) a)) '() l)))

(define (length l)
  (foldl + 0 (map (lambda (x) 1) l)))

(define (reverse l)
  (foldl cons '() l))

(define (append l1 l2)
  (foldr cons l2 l1))

(define list-tail; Taken from https://people.csail.mit.edu/jaffer/r5rs/Pairs-and-lists.html
       (lambda (x k)
         (if (zero? k)
             x
             (list-tail (cdr x) (- k 1)))))


; Math

(define (= a b)
  (if (<= a b)
      (>= a b)
      #f))

(define (zero?     x) (= x 0))
(define (positive? x) (> x 0))
(define (negative? x) (< x 0))
(define (odd?      x) (= (mod x 2) 1))
(define (even?     x) (= (mod x 2) 0))

; Test string

(display "Hello World") (newline)

; Test functions

(define add-three (lambda (x y z) (+ (+ x z) y)))

; Test factorial

(define fac  (lambda (x) (if (>= x 1) (* x (fac (+ x -1))) 1)))

(define (faca x a)
                (if (>= x 1)
                   (faca (+ x -1) (* x a))
                   a))

(define x 5)
(display x)          (newline)
(display (fac x))    (newline)
(display (faca x 1)) (newline)

; Test stuff?

(define (f . l) (cdr l))
(define y '(1 2 3))

(display y)          (newline)
(display (car y))    (newline)
(display (cdr y))    (newline)
(display (append '(1 2 3) '(4 5 6)))    (newline)

(display "\n* R5RS: Quotation and quasi-quotation *\n")

(display "Testing quotation!\n\tExpect: (1 2 3)\n\tGotten: ")
(define (x) (quote (1 2 3)))
(display (x))
(newline)

(display "Testing quotation!\n\tExpect: (1 2 3)\n\tGotten: ")
(define (x) '(1 2 3))
(display (x))
(newline)

(display "Testing quasi-notation basics!\n\tExpect: (1 2 3)\n\tGotten: ")
(define (x) `(1 2 3))
(display (x))
(newline)

(display "Testing quasi-notation!\n\tExpect: (1 a 2 a 3 a 4)\n\tGotten: ")
(define (x p) `(1 ,p 2 ,p 3 ,p 4))
(display (x 'a))
(newline)

(display "Testing quasi-notation advanced!\n\tExpect: (1 2 3 4 5)\n\tGotten: ")
(define (x p) `(1 ,@p 3))
(display (x '(2 3 4)))
(newline)

(display "Testing quasi-notation advanced!\n\tExpect: (1 `(,a ,x 3) 4 5)\n\tGotten: ")
(define (f x y) `(,x `(,a ,x 3) 4 ,y))
(display (f 1 5))
(newline)

; Check set!

(set! x 10)
(display x) (newline)

(newline)
(display "* R5RS: Testing booleans *")
(newline)

(display "Testing not\n\tExpect: #f #t #f #f\n\tGotten: ")
(display (not #t))   (display " ")
(display (not #f))   (display " ")
(display (not "hi")) (display " ")
(display (not 6))    (newline)

(newline)
(display "* R5RS: Testing numerical operators *")
(newline)

(display "Testing number?\n\tExpect: #t #f #f #f\n\tGotten: ")
(display (number? 5))
(display " ")
(display (number? #f))
(display " ")
(display (number? "hello"))
(display " ")
(display (number? '(1 2)))
(newline)

(newline)
(display "* R5RS: Testing fold and map *")
(newline)

(display "Testing sum over foldl!\n\tExpect: 28\n\tGotten: ")
(display (foldl + 0 '(1 2 3 4 5 6 7)))
(newline)

(display "Testing sum over foldr!\n\tExpect: 28\n\tGotten: ")
(display (foldr + 0 '(1 2 3 4 5 6 7)))
(newline)

(display "Testing map!\n\tExpect: (21 22 23)\n\tGotten: ")
(display (map (lambda (a) (+ 20 a)) '(1 2 3)))
(newline)

(display "Testing length!\n\tExpect: 13\n\tGotten: ")
(display (length '(1 2 3 1 2 3 1 2 3 1 2 3 1)))
(newline)

(display "Testing length!\n\tExpect: (5 4 3 2 1)\n\tGotten: ")
(display (reverse '(1 2 3 4 5)))
(newline)

(newline)
(display "* R5RS: Test string *")
(newline)

(display "Testing string-concat\n\tExpect: Hello World\n\tGotten: ")
(display (string-append "Hello" " " "World"))
(newline)

(define (concat-string s sep)
  (foldr (lambda (e a) (string-append (datum->string e) sep a))
         ""
         s))

(define test
  (lambda (name tests)
    (begin (display name)
           (display "\n\tExpect: ")
           (display (concat-string (map car tests) "  "))
           (display "\n\tGotten: ")
           (display (concat-string (map cdr tests) "  "))
           (newline))))

(define (header name) (display (string-append "\n * " name " *\n")))

(header "R5RS: Test equivalence")

(define (list-tail ls index)
  (if (zero? index)
      ls
      (list-tail (cdr ls) (+ index -1))))

(define (list-ref ls index)
  (car (list-tail ls index)))

(define do-stuffer
  (lambda (name_i arg1_i arg2_i)
    (lambda (x)
      (cons (list-ref x name_i)
            (eqv? (list-ref x arg1_i)
                  (list-ref x arg2_i))))))

(define eqv-test
  (lambda (name tests)
    (test name (append (map (do-stuffer 0 1 2) tests)
                       (map (do-stuffer 0 2 1) tests)
                       ))))

(eqv-test "eqv? booleans"
      '(
        (#t  #t #t)
        (#t  #f #f)
        (#f  #t #f)
        (#f  #f #t)
       ))

(eqv-test "eqv? inter-types"
      '(
        (#f    2    a)
        (#f    4    "Derp")
        (#f    ()   (a b))
        (#f    9    ())
       ))

(eqv-test "eqv? symbols"
      '(
        (#t  hello hello)
        (#t  World wOrld)
        (#f  hello world)
        (#f  hello World)
       ))


(header "R5RS: Test if-expressions")

(test "If expressions exp!"
      `(
        (6 . ,(* 3 (if #t 2 0)))
        (0 . ,(* 3 (if #f 2 0)))
        ("Hello World" . ,(if #t "Hello World"))
        ("???" . ,(if #f "Hello World"))
      ))

(header "R5RS: Testing Eval system")

(display "Eval expression. Expect 6: ")
(display (eval '(* 2 (cdr '(2 . 3))) (scheme-report-environment 5)))
(newline)

(newline)
(display "* R5RS: Testing Macro system *")
(newline)

; Test define-syntax

(display "Test define-syntax:\n    Expect: 4 ()\n    Gotten: ")

(define-syntax const
  (lambda (so)
    `(lambda () ,(car (cdr so)))))

(define x (const 4))
(display (x))
(display " ")
(define x (const '()))
(display (x))
(display "\n")


(display "\n* R5RS: Lexical convention *\n")

(define a-variable 20)
(if #f (begin
  (display "Test case-insensitivity (required by R5RS)\n\tExpect: 20 20 20 20\n\tGotten: ")
  (display a-variable) (display " ")
  (display A-variable) (display " ")
  (display a-VARiable) (display " ")
  (display A-VARIABLE))
    (display "Skipped test!\n"))

; Test environment set

(newline)
(display "* Testing custom library *")
(newline)

(display "Testing set-env! Expect 32: ")
(display ((set-env! (lambda () x) '((x . 32)) )))
(newline)

(display "Testing string->datum! Expect (2): ")
(display (cdr (string->datum "(display 2)")))
(newline)

; Test syntax-rules

(newline)
(display "* Testing syntax-rules (move to macro once implemented) *")
(newline)

(display "Attempting to define true using syntax rules.\n\tExpect: #t #f\n\tGotten: ")

(define-syntax true
 (syntax-rules ()
    ((true) #t)))

(display (true))
(display " ")
(display (not (true)))


(display "Attempting to define and using syntax rules.\n\tExpect: #t #f #f #f\n\tGotten: ")

(define-syntax and
 (syntax-rules ()
    ((and) #t)
    ((and test) test)
    ((and test1 test2 ...)
     (if test1 (and test2 ...) #f))))

(display (and #t #t)) (display " ")
(display (and #f #t)) (display " ")
(display (and #f #f)) (display " ")
(display (and #t #f)) (newline)

; TODO: Use syntax-rules to implement let and or.

(display "")