You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.3 KiB
Racket
46 lines
1.3 KiB
Racket
#lang racket/base
|
|
(require (for-syntax racket/base))
|
|
(require racket/contract)
|
|
|
|
|
|
(provide define/provide define/provide/contract)
|
|
|
|
;; todo: these won't handle nested forms like (define ((foo x) y))
|
|
|
|
(define-syntax (define/provide stx)
|
|
(syntax-case stx ()
|
|
;; order of cases matters, of course
|
|
;; match more complicated shape first,
|
|
;; otherwise second matcher gives false positives
|
|
[(_ (name arg ...) body ...)
|
|
#'(begin
|
|
(provide name)
|
|
(define (name arg ...) body ...))]
|
|
[(_ (name . arg) body ...)
|
|
#'(begin
|
|
(provide name)
|
|
(define (name . arg) body ...))]
|
|
[(_ name body ...)
|
|
#'(begin
|
|
(provide name)
|
|
(define name body ...))]))
|
|
|
|
|
|
(define-syntax (define/provide/contract stx)
|
|
(syntax-case stx ()
|
|
[(_ (name arg ...) contract body ...)
|
|
#'(begin
|
|
(provide (contract-out [name contract]))
|
|
(define (name arg ...) body ...))]
|
|
[(_ (name . arg) contract body ...)
|
|
#'(begin
|
|
(provide (contract-out [name contract]))
|
|
(define (name . arg) body ...))]
|
|
[(_ name contract body ...)
|
|
#'(begin
|
|
(provide (contract-out [name contract]))
|
|
(define name body ...))]))
|
|
|
|
(define/provide/contract (foo #:what x)
|
|
(#:what integer? . -> . integer?)
|
|
(λ(x) x)) |