PLT: completion support.

2 files changed, 66 insertions, 1 deletions

diff --git a/scheme/plt/geiser.ss b/scheme/plt/geiser.ss
index 4f4e665..71a5f90 100644
--- a/scheme/plt/geiser.ss
+++ b/scheme/plt/geiser.ss
@@ -29,17 +29,20 @@
            geiser/compile
            geiser/load-file
            geiser/compile-file
+           geiser/completions
            geiser/autodoc
            geiser/make-repl-reader)
 
   (compile-enforce-module-constants #f)
-  (require (lib "geiser/eval"))
+  (require (lib "geiser/eval")
+           (lib "geiser/completions"))
 
   (define geiser/eval eval-in)
   (define geiser/compile compile-in)
   (define geiser/load-file load-file)
   (define geiser/compile-file compile-file)
   (define (geiser/autodoc . x) #f)
+  (define geiser/completions completions)
 
   (define prompt (make-parameter "mzscheme@(geiser)"))
   (define (geiser/make-repl-reader builtin-reader)
diff --git a/scheme/plt/geiser/completions.ss b/scheme/plt/geiser/completions.ss
new file mode 100644
index 0000000..3cd62fd
--- /dev/null
+++ b/scheme/plt/geiser/completions.ss
@@ -0,0 +1,62 @@
+;; completions.ss -- completion support
+
+;; Copyright (C) 2009 Jose Antonio Ortega Ruiz
+
+;; Author: Jose Antonio Ortega Ruiz <jao@gnu.org>
+;; Start date: Sun Apr 26, 2009 19:02
+
+;; 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:
+
+;; Finding completions of given prefixes.
+
+;;; Code:
+
+#lang scheme
+
+(provide completions)
+
+(require srfi/13)
+
+(define (completions prefix . context)
+  (let ((context (and (not (null? context)) (car context)))
+        (prefix? (lambda (s) (string-prefix? prefix s))))
+    (append (filter prefix? (map symbol->string (local-bindings context)))
+            (sort (filter prefix? (map symbol->string (namespace-mapped-symbols)))
+                  string<?))))
+
+(define (pair->list pair)
+  (let loop ((d pair) (s '()))
+    (cond ((null? d) (reverse s))
+          ((symbol? d) (reverse (cons d s)))
+          (else (loop (cdr d) (cons (car d) s))))))
+
+(define (local-bindings form)
+  (define (body f) (if (> (length f) 2) (cddr f) '()))
+  (let loop ((form form) (bindings '()))
+    (cond ((not (pair? form)) bindings)
+          ((list? (car form))
+           (loop (cdr form) (append (local-bindings (car form)) bindings)))
+          ((and (list? form) (< (length form) 2)) bindings)
+          ((memq (car form) '(define define* lambda))
+           (loop (body form) (append (pair->list (cadr form)) bindings)))
+          ((and (memq (car form) '(let let* letrec letrec*))
+                (list? (cadr form)))
+           (loop (body form) (append (map car (cadr form)) bindings)))
+          ((and (eq? 'let (car form)) (symbol? (cadr form)))
+           (loop (cons 'let (body form)) (cons (cadr form) bindings)))
+          (else (loop (cdr form) bindings)))))
+
+;;; completions.ss ends here