suzanne.soy/beautiful-racket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tweaks

Browse Source
This commit is contained in:
Matthew Butterick 2016-04-29 16:05:58 -07:00
parent a7112554e3
commit fb0522957a
1 changed files with 25 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
beautiful-racket-lib/br/define.rkt
View File

@ -1,5 +1,5 @@
#lang racket/base #lang racket/base
(require (for-syntax racket/list racket/base syntax/parse racket/syntax syntax/datum syntax/strip-context) sugar/define) (require (for-syntax racket/list racket/base syntax/parse racket/syntax syntax/datum syntax/strip-context racket/string) sugar/define)
(provide (all-defined-out)) (provide (all-defined-out))
;; everything is prefixed br: whether it needs it or not so it can be stripped by #lang br ;; everything is prefixed br: whether it needs it or not so it can be stripped by #lang br
@ -14,13 +14,13 @@
(define-for-syntax (generate-literals pats) (define-for-syntax (generate-literals pats)
;; generate literals for any symbols that are not ... or _ or _underscore-prefixed ;; generate literals for any symbols that are not ... or _ or _underscore-prefixed
(for*/list ([pat-arg (in-list (syntax-flatten pats))] (define pattern-arg-prefixer "_")
[pat-datum (in-value (syntax->datum pat-arg))] (for/list ([pat-arg (in-list (syntax-flatten pats))]
#:when (and (symbol? pat-datum) #:when (let ([pat-datum (syntax->datum pat-arg)])
(not (eq? pat-datum '...)) (not (eq? pat-datum '_)) (not (eq? pat-datum 'else)) (and (symbol? pat-datum)
(not (let ([str (symbol->string pat-datum)]) (not (member pat-datum '(... _ else))) ; exempted from literality
(regexp-match #rx"^_" str))))) (not (string-prefix? (symbol->string pat-datum) pattern-arg-prefixer)))))
pat-arg)) pat-arg))
;; expose the caller context within br:define macros with syntax parameter ;; expose the caller context within br:define macros with syntax parameter
(begin-for-syntax (begin-for-syntax
@ -65,28 +65,26 @@
#'((pat result-expr) ... else-result-expr)] #'((pat result-expr) ... else-result-expr)]
[(((syntax pat) result-expr) ...) [(((syntax pat) result-expr) ...)
#'((pat result-expr) ... (raise-syntax-error 'define-cases (format "no matching case for syntax pattern ~v" (syntax->datum stx)) (syntax->datum #'top-id.name)))])] #'((pat result-expr) ... (raise-syntax-error 'define-cases (format "no matching case for syntax pattern ~v" (syntax->datum stx)) (syntax->datum #'top-id.name)))])]
[(LITERAL ...) (generate-literals #'(pat ...))]) [LITERALS (generate-literals #'(pat ...))])
#'(define-syntax top-id.name (λ (stx) #'(define-syntax top-id.name (λ (stx)
(define result (define result
(syntax-case stx (LITERAL ...) (syntax-case stx LITERALS
[pat (syntax-parameterize ([caller-stx (make-rename-transformer #'stx)]) [pat (syntax-parameterize ([caller-stx (make-rename-transformer #'stx)])
(syntax-parameterize ([shared-syntax (make-shared-syntax-macro caller-stx)]) (syntax-parameterize ([shared-syntax (make-shared-syntax-macro caller-stx)])
result-expr))] ... result-expr))] ...
[else else-result-expr])) [else else-result-expr]))
(if (not (syntax? result)) (if (syntax? result)
(datum->syntax #'top-id.name result) result
result))))] (datum->syntax #'top-id.name result)))))]
;; function matcher ;; function matcher
[(_ top-id:id [(_ pat-arg ... . rest-arg) body ...] ...) [(_ top-id:id [(_ . pat-args) . body] ...)
#'(define top-id #'(define top-id
(case-lambda (case-lambda
[(pat-arg ... . rest-arg) body ...] ... [pat-args . body] ...
[else (raise-syntax-error 'define-cases "no matching case for argument pattern" (object-name top-id))]))])) [else (raise-syntax-error 'define-cases "no matching case for argument pattern" (object-name top-id))]))]))
(module+ test (module+ test
(require rackunit) (require rackunit)
(define foo-val 'got-foo-val) (define foo-val 'got-foo-val)
@ -147,8 +145,8 @@
#:literals (syntax) #:literals (syntax)
;; syntax ;; syntax
[(_ (syntax (id pat-arg ... . rest-arg)) body ...) ; (define #'(foo arg) #'(+ arg arg)) [(_ (syntax (id . pat-args)) . body) ; (define #'(foo arg) #'(+ arg arg))
#'(br:define-cases (syntax id) [(syntax (_ pat-arg ... . rest-arg)) body ...])] #'(br:define-cases (syntax id) [(syntax (_ . pat-args)) . body])]
[(_ sid:syntaxed-id sid2:syntaxed-id) ; (define #'f1 #'f2) [(_ sid:syntaxed-id sid2:syntaxed-id) ; (define #'f1 #'f2)
#'(define-syntax sid.name (make-rename-transformer sid2))] #'(define-syntax sid.name (make-rename-transformer sid2))]
@ -162,9 +160,10 @@
[(_ sid:syntaxed-id (λ (stx-arg ...) expr ...)) ; (define #'f1 (λ(stx) expr ...) [(_ sid:syntaxed-id (λ (stx-arg ...) expr ...)) ; (define #'f1 (λ(stx) expr ...)
#:fail-when (not (= (length (syntax->datum #'(stx-arg ...))) 1)) #:fail-when (not (= (length (syntax->datum #'(stx-arg ...))) 1))
(raise-syntax-error 'define "did not get exactly one argument for macro" (syntax->datum #'(stx-arg ...))) (raise-syntax-error 'define "did not get exactly one argument for macro" (syntax->datum #'(stx-arg ...)))
#'(define-syntax (sid.name stx-arg ...) expr ...)] (with-syntax ([(first-stx-arg other ...) #'(stx-arg ...)])
#'(define-syntax (sid.name first-stx-arg) expr ...))]
[(_ args ...) #'(define args ...)])) [(_ arg ...) #'(define arg ...)]))
(module+ test (module+ test
(require rackunit) (require rackunit)
@ -254,7 +253,7 @@
(define-syntax (br:define-cases-inverting stx) (define-syntax (br:define-cases-inverting stx)
(syntax-case stx (syntax) (syntax-case stx (syntax)
[(_ (syntax _id) [(syntax _pat) . _bodyexprs] ...) [(_ (syntax _id) [(syntax _pat) . _bodyexprs] ...)
(with-syntax ([(LITERAL ...) (generate-literals #'(_pat ...))]) (with-syntax ([LITERALS (generate-literals #'(_pat ...))])
#'(define-syntax (_id stx) #'(define-syntax (_id stx)
(syntax-case stx () (syntax-case stx ()
[(_id . rest) [(_id . rest)
@ -262,14 +261,14 @@
[fused-stx (with-syntax ([(expanded-macro (... ...)) expanded-macros]) [fused-stx (with-syntax ([(expanded-macro (... ...)) expanded-macros])
#`(_id expanded-macro (... ...)))]) #`(_id expanded-macro (... ...)))])
(define result (define result
(syntax-case fused-stx (LITERAL ...) ;; put id back together with args to make whole pattern (syntax-case fused-stx LITERALS ;; put id back together with args to make whole pattern
[_pat (syntax-parameterize ([caller-stx (make-rename-transformer #'stx)]) [_pat (syntax-parameterize ([caller-stx (make-rename-transformer #'stx)])
(syntax-parameterize ([shared-syntax (make-shared-syntax-macro caller-stx)]) (syntax-parameterize ([shared-syntax (make-shared-syntax-macro caller-stx)])
. _bodyexprs))] ... . _bodyexprs))] ...
[else (raise-syntax-error 'define-cases-inverting (format "no matching case for syntax pattern ~v" (syntax->datum stx)) (syntax->datum #'_id))])) [else (raise-syntax-error 'define-cases-inverting (format "no matching case for syntax pattern ~v" (syntax->datum stx)) (syntax->datum #'_id))]))
(if (not (syntax? result)) (if (syntax? result)
(datum->syntax #'_id result) result
result))])))])) (datum->syntax #'_id result)))])))]))
(module+ test (module+ test