summaryrefslogtreecommitdiff
path: root/scheme/plt/geiser/modules.ss
blob: a1094af946ba75a56a0ce22f3adcebb506c3d030 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
;; modules.ss -- module metadata

;; Copyright (C) 2009 Jose Antonio Ortega Ruiz

;; Author: Jose Antonio Ortega Ruiz <jao@gnu.org>
;; Start date: Wed May 06, 2009 02:35

;; This file is free software; you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation; either version 3 of the License, or
;; (at your option) any later version.

;; This file is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU General Public License for more details.

;; You should have received a copy of the GNU General Public License
;; along with this program.  If not, see <http://www.gnu.org/licenses/>.

;;; Code:

#lang scheme

(provide load-module
         module-spec->namespace
         namespace->module-path-name
         module-path-name->name
         module-spec->path-name
         module-list)

(require srfi/13 scheme/enter)

(define nowhere (open-output-nowhere))

(define (ensure-module-spec spec)
  (cond ((symbol? spec) spec)
        ((not (string? spec)) #f)
        ((not (file-exists? spec)) #f)
        ((absolute-path? spec) `(file ,spec))
        (else spec)))

(define (module-spec->namespace mod-spec)
  (letrec ((spec (ensure-module-spec mod-spec))
           (handler (lambda (e)
                      (load-module spec)
                      (enter! #f)
                      (module->namespace spec))))
    (if spec
        (with-handlers ((exn:fail:contract? handler))
          (module->namespace spec))
        (current-namespace))))

(define (load-module spec . port)
  (parameterize ((current-error-port (if (null? port) nowhere (car port))))
    (eval #`(enter! #,(ensure-module-spec spec)))))

(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->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 "<top>")))


(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)
  (and (bytes=? (or (filename-extension path) #"") #"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))
                 ((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 module-cache #f)

(define (module-list)
  (when (not module-cache)
    (set! module-cache
          (sort (foldl find-modules '() (current-library-collection-paths))
                string<?)))
  module-cache)


;;; modules.ss ends here