;; Temporary test file:
(cond-expand 
  (program
    (import (scheme base) 
            (scheme write) 
            (scheme cyclone ast) 
            (scheme cyclone primitives)
            (scheme cyclone util) 
            (scheme cyclone pretty-print))))

(cond-expand
  (program
;; TODO: write code to find the loop. 
;; may be able to use code in wrap-mutables for this.
;; anyway, here is one:
    (define sexp
     '((%closure
         (lambda
           (self$42 loop$8$19)
           ((%closure
              (lambda
                (self$43 loop$8$19)
                (Cyc-seq
                  (set-cell!
                    loop$8$19
                    (%closure
                      (lambda
                        (self$44 k$28 x$9$20)
                        (if (zero?__inline__ x$9$20)
                          ((%closure-ref write 0)
                           write
                           k$28
                           'done)
                          ((%closure-ref
                             (cell-get (%closure-ref self$44 1))
                             0)
                           (cell-get (%closure-ref self$44 1))
                           k$28
                           (Cyc-fast-sub x$9$20 1))))
                      loop$8$19))
                  ((%closure-ref (cell-get loop$8$19) 0)
                   (cell-get loop$8$19)
                   (%closure-ref self$43 1)
                   10)))
              (%closure-ref self$42 1))
            (cell loop$8$19)))
         (%closure-ref self$41 1))
       #f))

    (define ast (ast:sexp->ast sexp))

    ;(pretty-print
    ;  (ast:ast->pp-sexp
    ;    (ast:sexp->ast sexp))
    ;)
  )
     )

(define (clo->lambda-body sexp)
  (car (ast:lambda-body (cadr sexp))))

(let* ((outer-body (clo->lambda-body (car ast))) ; (clo-call cell)
       (inner-body (clo->lambda-body (car outer-body)))
       (set-cell-exp (cadr inner-body))
       (set-clo (caddr set-cell-exp))
       
      )
  (write outer-body)
  (set-car! (cdr inner-body) #f) ;; Don't do the set
  (set-cdr! outer-body `((cell ,set-clo))) ;; Relocate the closure
  ;; TODO: replace self ref in params to set-clo
  ;;       OR, just handle properly in cgen
    (pretty-print
      (ast:ast->pp-sexp
        ast)
    )
)