The contract-repl from takikawa

contract-repl's People

Contributors

Stargazers

Watchers

contract-repl's Issues

Separating out non-contract-specific parts of `transform`

The non-contract-specific parts of transform could be separated out into a useful library function. I'm imagining something like this. It's just your code with the contract-specific bits taken out.

;; local-expand/wrap-return-expression :
;; [Stx -> Stx] Stx StxLocalContext [Maybe [Listof Id]] -> Stx

(define ((local-expand/wrap-return-expression wrap) stx ctx stop-ids)
  (define expanded-stx (local-expand stx ctx stop-ids))
  
  (define (transform stx)
    (syntax-parse stx
      [((~literal begin) e:expr ... e1:expr) #`(begin e ... #,(transform #'e1))]
      [x:id (transform #'(#%expression x))]
      [(~and form (kw:id . _))
       #:when (for/or ([k (list #'#%plain-lambda #'case-lambda #'if #'begin #'begin0
                                #'let-values #'letrec-values #'set! #'quote #'quote-syntax
                                #'values
                                #'with-continuation-mark #'#%plain-app #'#%top
                                #'#%variable-reference #'#%expression)])
                (free-identifier=? k #'kw))
       (wrap #'form)]
      [_ stx]))
  (transform expanded-stx))

This contract-repl library would use the a wrap function like this to deal with the contract-specific parts:

(define (wrap form)
  #`(let ([vals (call-with-values (lambda () #,form) list)])
      (define n (length vals))
      (for ([val vals]
            [i (in-naturals 1)])
        (cond [(has-contract? val)
               (display-val-contract val i n)
               val]
              [else val]))
      (apply values vals))

Recommend Projects