blob: a158d6600b26596472dbb775b00860c574c50374 (
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
|
;; introspection.scm -- name says it all
;; Copyright (C) 2009 Jose Antonio Ortega Ruiz
;; Author: Jose Antonio Ortega Ruiz <jao@gnu.org>
;; Start date: Sun Feb 08, 2009 18:44
;; 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/>.
;;; Comentary:
;; Procedures introspecting on scheme objects and their properties.
;;; Code:
(define-module (geiser introspection)
#:export (arguments completions symbol-location)
#:use-module (system vm program)
#:use-module (ice-9 session)
#:use-module (srfi srfi-1))
(define (arguments sym . syms)
(let loop ((sym sym) (syms syms))
(cond ((obj-args (resolve-symbol sym)) => (lambda (args) (cons sym args)))
((null? syms) #f)
(else (loop (car syms) (cdr syms))))))
(define (resolve-symbol sym)
(module-ref (current-module) sym))
(define (obj-args obj)
(cond ((not obj) #f)
((program? obj) (program-args obj))
((procedure? obj) (procedure-args obj))
((macro? obj) (macro-args obj))
(else #f)))
(define (symbol-module sym)
(call/cc
(lambda (k)
(apropos-fold (lambda (module name var init)
(if (eq? name sym) (k (module-name module)) init))
#f
(symbol->string sym)
(apropos-fold-accessible (current-module))))))
(define (program-args program)
(let* ((arity (program-arity program))
(arg-no (first arity))
(opt (> (second arity) 0))
(args (map first (take (program-bindings program) arg-no))))
(format-args (if opt (drop-right args 1) args)
(and opt (last args))
(program-module program))))
(define (procedure-args proc)
(let ((name (procedure-name proc)))
(cond ((procedure-source proc) => (lambda (src)
(procedure-args-from-source name src)))
(else (let* ((arity (procedure-property proc 'arity))
(req (first arity))
(opt (third arity)))
(format-args (map (lambda (n)
(string->symbol (format "arg~A" (+ 1 n))))
(iota req))
(and opt 'rest)
(and name (symbol-module name))))))))
(define (procedure-args-from-source name src)
(let ((formals (cadr src)))
(cond ((list? formals) (format-args formals #f (symbol-module name)))
((pair? formals) (format-args (car formals)
(cdr formals)
(symbol-module name)))
(else #f))))
(define (macro-args macro)
(let ((prog (macro-transformer macro)))
(if prog
(obj-args prog)
(format-args '(...) #f #f))))
(define (format-args args opt module)
(list (cons 'required args)
(cons 'optional (or opt '()))
(cons 'module (cond ((module? module) (module-name module))
((list? module) module)
(else '())))))
(define (completions prefix)
(sort! (map symbol->string
(apropos-internal (string-append "^" prefix)))
string<?))
(define (symbol-location sym)
(cond ((symbol-module sym) => make-location-from-module-name)
(else '())))
(define (make-location file line)
(list (cons 'file (if (string? file) file '()))
(cons 'line (if (number? line) (+ 1 line) '()))))
(define module-filename (@@ (ice-9 session) module-filename))
(define (make-location-from-module-name name)
(make-location (module-filename name) #f))
;;; introspection.scm ends here
|
