summaryrefslogtreecommitdiff
path: root/scheme/racket/geiser/modules.rkt
diff options
context:
space:
mode:
Diffstat (limited to 'scheme/racket/geiser/modules.rkt')
-rw-r--r--scheme/racket/geiser/modules.rkt150
1 files changed, 150 insertions, 0 deletions
diff --git a/scheme/racket/geiser/modules.rkt b/scheme/racket/geiser/modules.rkt
new file mode 100644
index 0000000..0de316c
--- /dev/null
+++ b/scheme/racket/geiser/modules.rkt
@@ -0,0 +1,150 @@
+;;; modules.rkt -- module metadata
+
+;; Copyright (C) 2009, 2010 Jose Antonio Ortega Ruiz
+
+;; This program is free software; you can redistribute it and/or
+;; modify it under the terms of the Modified BSD License. You should
+;; have received a copy of the license along with this program. If
+;; not, see <http://www.xfree86.org/3.3.6/COPYRIGHT2.html#5>.
+
+;; Start date: Wed May 06, 2009 02:35
+
+#lang racket
+
+(provide load-module
+ ensure-module-spec
+ module-spec->namespace
+ namespace->module-path-name
+ module-path-name->name
+ module-spec->path-name
+ module-list
+ module-exports)
+
+(require srfi/13 syntax/modresolve syntax/modcode geiser/enter)
+
+(define (ensure-module-spec spec)
+ (cond ((symbol? spec) spec)
+ ((not (string? spec)) #f)
+ (else `(file ,spec))))
+
+(define (module-spec->namespace spec (lang #f))
+ (let ((spec (ensure-module-spec spec))
+ (try-lang (lambda (_)
+ (with-handlers ((exn? (const (current-namespace))))
+ (and lang
+ (begin
+ (load-module lang #f (current-namespace))
+ (module->namespace lang)))))))
+ (or (and spec
+ (with-handlers ((exn? try-lang)) (get-namespace spec)))
+ (current-namespace))))
+
+(define nowhere (open-output-nowhere))
+
+(define (load-module spec (port #f) (ns #f))
+ (parameterize ((current-error-port (or port nowhere)))
+ (enter-module (ensure-module-spec spec))
+ (when (namespace? ns)
+ (current-namespace ns))))
+
+(define (namespace->module-path-name ns)
+ (let ((rmp (variable-reference->resolved-module-path
+ (eval '(#%variable-reference) ns))))
+ (and (resolved-module-path? rmp)
+ (resolved-module-path-name rmp))))
+
+(define (module-spec->path-name spec)
+ (with-handlers ((exn? (lambda (_) #f)))
+ (let ((ns (module-spec->namespace (ensure-module-spec spec))))
+ (namespace->module-path-name ns))))
+
+(define (module-path-name->name path)
+ (cond ((path? path)
+ (let* ((path (path->string path))
+ (cpaths (map (compose path->string path->directory-path)
+ (current-library-collection-paths)))
+ (prefix-len (lambda (p)
+ (let ((pl (string-length p)))
+ (if (= pl (string-prefix-length p path))
+ pl
+ 0))))
+ (lens (map prefix-len cpaths))
+ (real-path (substring path (apply max lens))))
+ (if (absolute-path? real-path)
+ (call-with-values (lambda () (split-path path))
+ (lambda (_ basename __) (path->string basename)))
+ (regexp-replace "\\.[^./]*$" real-path ""))))
+ ((eq? path '#%kernel) "(kernel)")
+ ((string? path) path)
+ ((symbol? path) (symbol->string path))
+ (else "")))
+
+(define (skippable-dir? path)
+ (call-with-values (lambda () (split-path path))
+ (lambda (_ basename __)
+ (member (path->string basename) '(".svn" "compiled")))))
+
+(define path->symbol (compose string->symbol path->string))
+
+(define (path->entry path)
+ (let ((ext (filename-extension path)))
+ (and ext
+ (or (bytes=? ext #"rkt") (bytes=? ext #"ss"))
+ (let ((path (path->string path)))
+ (substring path 0 (- (string-length path) 3))))))
+
+(define (visit-module-path path kind acc)
+ (case kind
+ ((file) (let ((entry (path->entry path)))
+ (if entry (cons entry acc) acc)))
+ ((dir) (cond ((skippable-dir? path) (values acc #f))
+ ((or (file-exists? (build-path path "main.rkt"))
+ (file-exists? (build-path path "main.ss")))
+ (cons (path->string path) acc))
+ (else acc)))
+ (else acc)))
+
+(define (find-modules path acc)
+ (if (directory-exists? path)
+ (parameterize ((current-directory path))
+ (fold-files visit-module-path acc))
+ acc))
+
+(define (known-modules)
+ (sort (foldl find-modules '() (current-library-collection-paths)) string<?))
+
+(define module-cache #f)
+(define (update-module-cache)
+ (when (not module-cache) (set! module-cache (known-modules))))
+
+(define (module-list)
+ (update-module-cache)
+ module-cache)
+
+(define (module-exports mod)
+ (define (extract-ids ls)
+ (append-map (lambda (idls)
+ (map car (cdr idls)))
+ ls))
+ (define (classify-ids ids ns)
+ (let loop ((ids ids) (procs '()) (vars '()))
+ (cond ((null? ids)
+ `((procs ,@(reverse procs)) (vars ,@(reverse vars))))
+ ((procedure?
+ (namespace-variable-value (car ids) #t (const #f) ns))
+ (loop (cdr ids) (cons (car ids) procs) vars))
+ (else (loop (cdr ids) procs (cons (car ids) vars))))))
+ (let-values (((reg syn)
+ (module-compiled-exports
+ (get-module-code (resolve-module-path mod #f)))))
+ (let ((syn (extract-ids syn))
+ (reg (extract-ids reg)))
+ `((syntax ,@syn) ,@(classify-ids reg (module-spec->namespace mod))))))
+
+(define (startup)
+ (thread update-module-cache)
+ (void))
+
+(startup)
+
+;;; modules.rkt ends here