racket/collects/unstable/lazy-require.rkt

115 lines
4.5 KiB
Racket

#lang racket/base
(require (for-syntax racket/base
compiler/cm-accomplice)
racket/runtime-path
racket/promise)
(provide lazy-require
begin-on-demand)
(define-syntax (lazy-require stx)
(syntax-case stx ()
[(lazy-require [modpath (thing ...)] ...)
#`(begin (define-namespace-anchor anchor)
(lazy-require1 modpath (thing ...) anchor #,stx)
...)]))
(define-syntax (lazy-require1 stx)
(syntax-case stx ()
[(lazy-require1 modpath (name ...) anchor orig-stx)
(with-syntax ([(defn ...)
(for/list ([name (in-list (syntax->list #'(name ...)))])
(unless (identifier? name)
(raise-syntax-error #f "expected identifier" #'orig-stx name))
(with-syntax ([name name]
[(aux) (generate-temporaries (list name))])
#`(begin (define aux (make-lazy-function 'name get-sym))
(define-syntax name
(make-rename-transformer
(syntax-property (quote-syntax aux)
'not-provide-all-defined #t))))))])
;; implicit quasiquote, so can use normal module-path syntax
;; or escape to compute a the module-path via expression
#'(begin (define-runtime-module-path-index mpi-var (quasiquote modpath))
(define-values ()
(let-syntax ([_ (do-registration (#%variable-reference) (quasiquote modpath))])
(values)))
(define (get-sym sym)
(parameterize ((current-namespace (namespace-anchor->namespace anchor)))
(dynamic-require mpi-var sym)))
defn ...))]))
(define (make-lazy-function name get-sym)
;; Use 'delay/sync' because 'delay' promise is not reentrant.
;; FIXME: OTOH, 'delay/sync' promise is not kill-safe.
(let ([fun-p (delay/sync (get-sym name))])
(procedure-rename
(make-keyword-procedure
(lambda (kws kwargs . args)
(keyword-apply (force fun-p) kws kwargs args)))
name)))
(begin-for-syntax
(define (do-registration vr modpath)
(let ([path (resolved-module-path-name
(module-path-index-resolve
(module-path-index-join
modpath
(variable-reference->resolved-module-path vr))))])
(when (path? path)
(register-external-file path)))))
#|
begin-on-demand notes/todo
Currently like lazy-require: only supports functions.
The #:export kw is clunky... one might think it would be nice to just
re-use 'provide' syntax. Would get rename, prefix, etc for free. OTOH,
might be misleading: might seem like provide should apply to
*apparent* enclosing module, not module implicitly created by
begin-on-demand.
Another nice idea would be to implement lazy-require in terms of
begin-on-demand and real require. Unfortunately, that would mean we
couldn't have cyclic lazy-requires, which is very useful.
|#
(define-syntax (begin-on-demand stx)
(syntax-case stx ()
[(begin-on-demand #:export (exp ...) body ...)
(with-syntax ([fresh-name (gensym 'on-demand-submodule)])
#'(begin
(module* fresh-name #f
(no-provide body ...)
(check-procedure exp 'exp) ...
(provide exp ...))
(define-namespace-anchor anchor)
(define get-sym
(let ([sub-mpi
(module-path-index-join '(submod "." fresh-name)
(variable-reference->resolved-module-path
(#%variable-reference)))])
(lambda (sym)
(parameterize ((current-namespace (namespace-anchor->namespace anchor)))
(dynamic-require sub-mpi sym)))))
(define exp (make-lazy-function 'exp get-sym)) ...))]))
(define-syntax (no-provide stx)
(syntax-case stx ()
[(_ form)
(let ([eform (local-expand #'form
(syntax-local-context)
#f)])
(syntax-case eform (begin #%provide)
[(begin e ...)
#'(begin (no-provide e) ...)]
[(#%provide . _)
(raise-syntax-error #f "provide not allowed" eform)]
[_ eform]))]
[(_ form ...)
#'(begin (no-provide form) ...)]))
(define (check-procedure x name)
(unless (procedure? x)
(error 'begin-on-demand "name bound on-demand is not a procedure: ~a" name)))