173 lines
7.6 KiB
Scheme
173 lines
7.6 KiB
Scheme
(module module-reader scheme/base
|
|
|
|
(provide (rename-out [provide-module-reader #%module-begin]
|
|
[wrap wrap-read-all])
|
|
(except-out (all-from-out scheme/base) #%module-begin))
|
|
|
|
(require (for-syntax scheme/base))
|
|
|
|
(define ar? procedure-arity-includes?)
|
|
|
|
(define-syntax (provide-module-reader stx)
|
|
(define (err str [sub #f])
|
|
(raise-syntax-error 'syntax/module-reader str sub))
|
|
(define-syntax-rule (keywords body [kwd var default] ... [checks ...])
|
|
(begin
|
|
(define var #f) ...
|
|
(set! body
|
|
(let loop ([body body])
|
|
(if (not (and (pair? body)
|
|
(pair? (cdr body))
|
|
(keyword? (syntax-e (car body)))))
|
|
(datum->syntax stx body stx)
|
|
(let* ([k (car body)] [k* (syntax-e k)] [v (cadr body)])
|
|
(case k*
|
|
[(kwd) (if var
|
|
(err (format "got two ~s keywords" k*) k)
|
|
(begin (set! var v) (loop (cddr body))))]
|
|
...
|
|
[else (err "got an unknown keyword" (car body))])))))
|
|
checks ...
|
|
(unless var (set! var default)) ...))
|
|
(define <lang-id> (datum->syntax stx 'language-module stx))
|
|
(define <data-id> (datum->syntax stx 'language-data stx))
|
|
(define (construct-reader lang body)
|
|
(keywords body
|
|
[#:language ~lang lang]
|
|
[#:read ~read #'read]
|
|
[#:read-syntax ~read-syntax #'read-syntax]
|
|
[#:wrapper1 ~wrapper1 #'#f]
|
|
[#:wrapper2 ~wrapper2 #'#f]
|
|
[#:whole-body-readers? ~whole-body-readers? #'#f]
|
|
[#:info ~info #'#f]
|
|
[(when (equal? (and lang #t) (and ~lang #t))
|
|
(err (string-append
|
|
"must specify either a module language, or #:language"
|
|
(if (and lang ~lang) ", not both" ""))))
|
|
(unless (equal? (and ~read #t) (and ~read-syntax #t))
|
|
(err "must specify either both #:read and #:read-syntax, or none"))
|
|
(when (and ~whole-body-readers? (not (and ~read ~read-syntax)))
|
|
(err "got a #:whole-body-readers? without #:read and #:read-syntax"))])
|
|
(quasisyntax/loc stx
|
|
(#%module-begin
|
|
#,@body
|
|
(#%provide (rename lang:read read) (rename lang:read-syntax read-syntax)
|
|
read-properties get-info-getter get-info)
|
|
(define (lang:read in modpath line col pos)
|
|
(wrap-internal/wrapper #f #f in modpath line col pos))
|
|
(define (lang:read-syntax src in modpath line col pos)
|
|
(wrap-internal/wrapper #t src in modpath line col pos))
|
|
(define (wrap-internal/wrapper stx? src in modpath line col pos)
|
|
(let* ([props (read-properties in modpath line col pos)]
|
|
[lang (car props)] [#,<lang-id> lang] ;\ visible in
|
|
[data (cadr props)] [#,<data-id> data] ;/ user-code
|
|
[read (if stx?
|
|
(let ([rd #,~read-syntax]) (lambda (in) (rd src in)))
|
|
#,~read)]
|
|
[w1 #,~wrapper1]
|
|
[w2 #,~wrapper2]
|
|
[whole? #,~whole-body-readers?]
|
|
[rd (lambda (in) (wrap-internal lang in read whole? w1 stx?
|
|
modpath src line col pos))]
|
|
[r (cond [(not w2) (rd in)]
|
|
[(ar? w2 3) (w2 in rd stx?)]
|
|
[else (w2 in rd)])])
|
|
(if stx?
|
|
(syntax-property r 'module-language
|
|
(vector (syntax->datum modpath) 'get-info-getter
|
|
props))
|
|
r)))
|
|
(define lang*
|
|
(let ([lang #,~lang])
|
|
(if (not (procedure? lang))
|
|
(list lang #f)
|
|
(cond [(ar? lang 5) lang]
|
|
[(ar? lang 1) (lambda (in . _) (lang in))]
|
|
[(ar? lang 0) (lambda _ (lang))]
|
|
[else (raise-type-error
|
|
'syntax/module-reader
|
|
"language+reader procedure of 5, 1, or 0 arguments"
|
|
lang)]))))
|
|
(define (read-properties in modpath line col pos)
|
|
(if (not (procedure? lang*))
|
|
lang*
|
|
(call-with-values
|
|
(lambda () (parameterize ([current-input-port in])
|
|
(lang* in modpath line col pos)))
|
|
(lambda xs
|
|
(case (length xs)
|
|
[(2) xs] [(1) (list (car xs) #f)]
|
|
[else (error 'syntax/module-reader
|
|
"wrong number of results from ~a, ~a ~e"
|
|
"the #:language function"
|
|
"expected 1 or 2 values, got"
|
|
(length xs))])))))
|
|
(define (get-info in modpath line col pos)
|
|
(get-info-getter (read-properties in modpath line col pos)))
|
|
(define (get-info-getter props)
|
|
(define lang (car props))
|
|
(define data (cadr props))
|
|
(define (default-info what)
|
|
(case what
|
|
[(module-language) (car props)]
|
|
;; ... more?
|
|
[else #f]))
|
|
(define info
|
|
(let* ([#,<lang-id> lang] ;\ visible in
|
|
[#,<data-id> data] ;/ user-code
|
|
[info #,~info])
|
|
(if (or (not info) (and (procedure? info) (ar? info 2)))
|
|
info
|
|
(raise-type-error 'syntax/module-reader
|
|
"info procedure of 1 or 0 arguments" info))))
|
|
(define (language-info what)
|
|
(if info
|
|
(let ([r (info what default-info)])
|
|
(if (eq? r default-info) (default-info what) r))
|
|
(default-info what)))
|
|
language-info))))
|
|
(syntax-case stx ()
|
|
[(_ lang body ...)
|
|
(not (keyword? (syntax-e #'lang)))
|
|
(construct-reader #''lang (syntax->list #'(body ...)))]
|
|
[(_ body ...) (construct-reader #f (syntax->list #'(body ...)))]))
|
|
|
|
(define (wrap-internal lang port read whole? wrapper stx?
|
|
modpath src line col pos)
|
|
(let* ([lang (if stx? (datum->syntax #f lang modpath modpath) lang)]
|
|
[body (lambda ()
|
|
(if whole?
|
|
(read port)
|
|
(let loop ([a null])
|
|
(let ([v (read port)])
|
|
(if (eof-object? v) (reverse a) (loop (cons v a)))))))]
|
|
[body (cond [(not wrapper) (body)]
|
|
[(ar? wrapper 2) (wrapper body stx?)]
|
|
[else (wrapper body)])]
|
|
[all-loc (vector src line col pos
|
|
(let-values ([(l c p) (port-next-location port)])
|
|
(and p (- p pos))))]
|
|
[body (if (and stx? (not (syntax? body)))
|
|
(datum->syntax #f body all-loc)
|
|
body)]
|
|
[p-name (object-name port)]
|
|
[name (if (path? p-name)
|
|
(let-values ([(base name dir?) (split-path p-name)])
|
|
(string->symbol
|
|
(path->string (path-replace-suffix name #""))))
|
|
'page)]
|
|
[tag-src (lambda (v)
|
|
(if stx?
|
|
(datum->syntax
|
|
#f v (vector src line col pos
|
|
(- (or (syntax-position modpath) (add1 pos))
|
|
pos)))
|
|
v))]
|
|
[r `(,(tag-src 'module) ,(tag-src name) ,lang . ,body)])
|
|
(if stx? (datum->syntax #f r all-loc) r)))
|
|
|
|
(define (wrap lang port read modpath src line col pos)
|
|
(wrap-internal lang port read #f #f #f modpath src line col pos))
|
|
|
|
)
|