(ns section4
  (:use scheme-helpers
        environment))

(declare execute-application
         primitive-procedure-names
         primitive-procedure-objects
         my-eval
         my-apply
         analyze
         no-operands?
         first-operand
         rest-operands
         actual-value)

; Exercise 4.1
(defn list-of-values [exps env]
  (if (no-operands? exps)
    '()
    (let [left (my-eval (first-operand exps) env)
          right (list-of-values (rest-operands exps) env)]
      (cons left right))))
; Above function imposes a left to right ordering. If the
; assignments inside of let where switched it would be right
; to left

(declare if-predicate if-alternative if-consequent)

(defn eval-if [exp env]
  (if (actual-value (if-predicate exp) env)
    (my-eval (if-consequent exp) env)
    (my-eval (if-alternative exp) env)))

(declare last-exp? first-exp rest-exps)

(defn eval-sequence [exps env]
  (cond (last-exp? exps) (my-eval (first-exp exps) env)
        :else (do (my-eval (first-exp exps) env)
                  (eval-sequence (rest-exps exps) env))))
;make above a recur

(declare assignment-variable assignment-value)

(defn eval-assignment [exp env]
  (set-variable-value! (assignment-variable exp)
                       (my-eval (assignment-value exp) env)
                       env)
  'ok)

(declare definition-variable definition-value)

(defn eval-definition [exp env]
  ;  (println "eval-def: " exp)
  (define-variable!
    (definition-variable exp)
    (my-eval (definition-value exp) env)
    env)
  'ok)

(defn self-evaluating? [exp]
  (or (number? exp)
      (string? exp)
      (and (seq? exp) (self-evaluating? (first exp)))))

(defn variable? [exp]
  (or (symbol? exp)
      (= 'true exp)
      (= 'false exp)))

(defn tagged-list? [exp tag]
  (if (seq? exp)
    (= (first exp) tag)
    false))

(defn quoted? [exp]
  (tagged-list? exp 'quote))

(defn text-of-quotation [exp] (cadr exp))

(defn assignment? [exp]
  (tagged-list? exp 'set!))

(defn assignment-variable [exp] (second exp))

(defn assignment-value [exp] (nth exp 2))

(defn definition? [exp]
  (tagged-list? exp 'define))

(defn definition-variable [exp]
  (if (symbol? (second exp))
    (second exp)
    (first (first (rest exp)))))

(declare make-lambda)

(defn definition-value [exp]
  (if (symbol? (second exp))
    (nth exp 2)
    (make-lambda (rest (first (rest exp))) ; formal parameters
                 (rest (rest exp)))))          ; body

(defn lambda? [exp] (tagged-list? exp 'lambda))

(defn lambda-parameters [exp] (second exp))

(defn lambda-body [exp] (rest (rest exp)))

(defn make-lambda [parameters body]
  (cons 'lambda (cons parameters body)))

(defn if? [exp] (tagged-list? exp 'if))

(defn if-predicate [exp] (cadr exp))

(defn if-consequent [exp] (caddr exp))

(defn if-alternative [exp]
  (if (not (nil? (cdddr exp)))
    (cadddr exp)
    'false))

(defn make-if [predicate consequent alternative]
  (list 'if predicate consequent alternative))

(defn begin? [exp] (tagged-list? exp 'begin))

(defn begin-actions [exp] (cdr exp))

(defn last-exp? [xs] (null? (cdr xs)))

(defn first-exp [xs] (car xs))

(defn rest-exps [xs] (cdr xs))

(defn make-begin [xs] (cons 'begin xs))

(defn sequence->exp [xs]
  (cond (null? xs) xs
        (last-exp? xs) (first-exp xs)
        :else (make-begin xs)))

(defn pair? [x] (seq? x))

(defn application? [exp] (pair? exp))

(defn operator [exp] (car exp))

(defn operands [exp] (cdr exp))

(defn no-operands? [ops] (null? ops))

(defn first-operand [ops] (car ops))

(defn rest-operands [ops] (cdr ops))

(declare expand-clauses)
(defn cond? [exp] (tagged-list? exp 'cond))

(defn cond-clauses [exp] (cdr exp))

(defn cond-predicate [clause] (car clause))

(defn cond-else-clause? [clause]
  (= (cond-predicate clause) 'else))

(defn cond-actions [clause] (cdr clause))

(defn cond->if [exp]
  (expand-clauses (cond-clauses exp)))

(defn extended-cond? [clause]
  (and (list? clause)
       (> (count clause) 2)
       (= (second clause) '=>)))

(defn extended-cond-test [clause]
  (first clause))

(defn extended-cond-recipient [clause]
  (nth clause 2))

(defn expand-clauses [clauses]
  (if (null? clauses)
    'false
    (let [first-clause (car clauses)
          rest-clauses (cdr clauses)]
      (cond (cond-else-clause? first-clause)
            (if (null? rest-clauses)
              (sequence->exp (cond-actions first-clause))
              (Error. (str  "ELSE clause isn't last -- COND->IF"
                            clauses)))
            (extended-cond? first-clause)
            (make-if (extended-cond-test first-clause)
                     (list
                      (extended-cond-recipient first-clause)
                      (extended-cond-test first-clause))
                     (expand-clauses rest-clauses))
            :else
            (make-if (cond-predicate first-clause)
                     (sequence->exp (cond-actions first-clause))
                     (expand-clauses rest-clauses))))))

(declare scan-out-defines)
(defn make-procedure [parameters body env]
  (list 'procedure parameters (scan-out-defines body) env))

(defn compound-procedure? [p]
  (tagged-list? p 'procedure))

(defn procedure-parameters [p] (cadr p))

(defn procedure-body [p] (caddr p))

(defn procedure-environment [p] (cadddr p))

(defn let? [exp]
  (tagged-list? exp 'let))

(defn named-let? [exp]
  (symbol? (second exp)))

(defn let-body [exp]
  (if (named-let? exp)
    (nth exp 3)
    (nth exp 2)))

(defn let-variables [exp]
  (if (named-let? exp)
    (map first (nth exp 2))
    (map first (second exp))))

(defn let-values [exp]
  (if (named-let? exp)
    (map second (nth exp 2))
    (map second (second exp))))

(defn let-name [exp]
  (second exp))

(defn make-definition [fn-name parameters body]
  (list 'define (cons fn-name parameters) body))


; define function
; eval function with arguments
(defn let->combination [exp]
  (let [parameters (let-variables exp)
        args (let-values exp)
        body (let-body exp)]
    (if (named-let? exp)
      (sequence->exp
       (list
        (make-definition (let-name exp)
                         parameters
                         body)
        (cons
         (let-name exp)
         args)))
      (cons
       (make-lambda (let-variables exp)
                    (list (let-body exp)))
       (let-values exp)))))

(defn let*? [exp]
  (tagged-list? exp 'let*))

(defn make-let [clauses body]
  (list 'let clauses body))

(defn let*->nested-lets [exp]
  (let [let-clauses (reverse (second exp))
        body (let-body exp)]
    (reduce #(make-let (list %2) %1) body let-clauses)))

(def primitive-procedures
     (list (list 'car car)
           (list 'cdr cdr)
           (list 'cadr cadr)
           (list 'cons cons)
           (list 'null? null?)
           (list '+ +)
           (list '- -)
           (list '* *)
           (list '/ /)
           (list '= =)
           (list '> >)
           (list '< <)
           (list 'and (fn [& xs] (reduce #(and %1 %2) true xs)))
           (list 'or (fn [& xs] (reduce #(or %1 %2) false xs)))))

(defn primitive-procedure-names []
  (map car primitive-procedures))

(defn primitive-procedure-objects []
  (map (fn [proc] (list 'primitive (cadr proc)))
       primitive-procedures))

(defn setup-environment []
  (let [initial-env
        (extend-environment (primitive-procedure-names)
                            (primitive-procedure-objects)
                            the-empty-environment)]
    (define-variable! 'true true initial-env)
    (define-variable! 'false false initial-env)
    (define-variable! 'nil nil initial-env)
    initial-env))

(def the-global-environment (setup-environment))

(defn reset-global-environment []
  (def the-global-environment (setup-environment)))

; Exercise 4.13
(defn unbind? [exp]
  (tagged-list? exp 'make-unbound!))

(defn eval-unbind [exp env]
  (unbind-variable! (second exp) env)
  'ok)

(defn primitive-procedure? [proc]
  (tagged-list? proc 'primitive))

(defn primitive-implementation [proc] (cadr proc))

(defn apply-primitive-procedure [proc args]
  (apply (primitive-implementation proc) args))

(defn execute-application [proc args]
  (cond (primitive-procedure? proc)
          (apply-primitive-procedure proc args)
        (compound-procedure? proc)
          ((procedure-body proc)
           (extend-environment (procedure-parameters proc)
                               args
                               (procedure-environment proc)))
        :else
        (Error. (str
                 "Unknown procedure type -- EXECUTE-APPLICATION"
                 proc))))

(defn is-define? [e]
  (and (seq? e)
       (tagged-list? e 'define)))

(defn find-defines [exp]
  (filter is-define? exp))

(defn defined-variables [defs]
  (map second defs))

(defn defined-values [defs]
  (map #(nth % 2) defs))

(defn non-defines [exp]
  (remove is-define? exp))

(defn scan-out-defines [exp]
  (let [defs (find-defines exp)]
    (if (zero? (count defs))
      exp
      (let [variables (defined-variables defs)
            values (defined-values defs)
            body (nth (non-defines exp) 2)
            vars (second (non-defines exp))]
        (list 'lambda
              vars
              (cons 'let
                    (cons (map #(list % (quote (quote *unassigned*))) variables)
                          (concat (map
                                   #(list 'set! %1 %2)
                                   variables
                                   values)
                                  (list body)))))))))

; Exercise 4.20
(defn letrec? [exp]
  (tagged-list? exp 'letrec))

(defn letrec->let [exp]
  (let [fns (second exp)
        fn-names (map first fns)
        fn-vals  (map second fns)
        body (nth exp 2)]
    (make-let
     (map #(list % ''*unassigned*) fn-names)
     (make-begin
      (concat
       (map #(list 'set! %1 %2) fn-names fn-vals)
       (list body))))))


(defrecord Thunk [exp env])
(defrecord Evaluated-Thunk [value])

(defn atom? [x]
  (= clojure.lang.Atom (class x)))

(defn thunk? [obj]
;  (println "thunk? " obj " " (type obj) (atom? obj))
  (if (atom? obj)
    (= Thunk (class @obj))
    false))

(defn delay-it [exp env]
   (atom (Thunk. exp env)))

(defn evaluated-thunk? [obj]
  (if (atom? obj)
    (= Evaluated-Thunk (class @obj))))

(defn make-evaled-thunk [old-thunk new-value]
  (Evaluated-Thunk. new-value))

(defn force-it [obj]
;  (println "force-it: " obj)
  (cond
   (thunk? obj)
   (let [result (actual-value (:exp @obj)
                              (:env @obj))]
     (swap! obj make-evaled-thunk result)
     result)
   (evaluated-thunk? obj) (:value @obj)
   :else obj))

(defn actual-value [exp env]
;  (println "Actual value: " exp)
  (let [actual (my-eval exp env)
	forced (force-it actual)]
;    (println "forced: " forced)
    forced))

(defn list-of-arg-values [exp env]
;  (println "list-of-arg-values: " exp)
  (if (no-operands? exp)
    '()
    (cons (actual-value (first-operand exp) env)
          (list-of-arg-values (rest-operands exp)
                              env))))

(defn list-of-delayed-args [exps env]
  (if (no-operands? exps)
    '()
    (cons (delay-it (first-operand exps) env)
          (list-of-delayed-args (rest-operands exps)
                                env))))

(defn my-eval [exp env]
  (cond (self-evaluating? exp) exp
        (variable? exp) (lookup-variable-value exp env)
        (quoted? exp) (text-of-quotation exp)
        (assignment? exp) (eval-assignment exp env)
        (unbind? exp) (eval-unbind exp env)
        (definition? exp) (eval-definition exp env)
        (if? exp) (eval-if exp env)
        (lambda? exp)
          (make-procedure (lambda-parameters exp)
                          (lambda-body exp)
                          env)
        (begin? exp)
          (eval-sequence (begin-actions exp) env)
        (cond? exp) (my-eval (cond->if exp) env)
        (let? exp) (my-eval (let->combination exp) env)
        (let*? exp) (my-eval (let*->nested-lets exp) env)
        (letrec? exp) (my-eval (letrec->let exp) env)
        (application? exp)
          (my-apply (actual-value (operator exp) env)
                    (operands exp)
                    env)
        :else (Error. (str "Unknown expression type -- EVAL " exp))))


(defn my-apply [procedure arguments env]
; (println "apply: " procedure " " arguments)
  (cond (primitive-procedure? procedure)
          (apply-primitive-procedure
           procedure
           (list-of-arg-values arguments env))
        (compound-procedure? procedure)
          (eval-sequence
           (procedure-body procedure)
           (extend-environment
            (procedure-parameters procedure)
            (list-of-delayed-args arguments env)
            (procedure-environment procedure)))
        :else (Error. (str "Unknown procedure type -- APPLY " procedure))))

(defn interpret [exp]
  (actual-value exp the-global-environment))