install .go files under $libdir, not $datadir
[bpt/guile.git] / module / system / base / compile.scm
index f83834d..f995d90 100644 (file)
@@ -1,6 +1,6 @@
 ;;; High-level compiler interface
 
-;; Copyright (C) 2001 Free Software Foundation, Inc.
+;; Copyright (C) 2001, 2009 Free Software Foundation, Inc.
 
 ;; This program is free software; you can redistribute it and/or modify
 ;; it under the terms of the GNU General Public License as published by
@@ -29,7 +29,8 @@
   #:export (syntax-error 
             *current-language*
             compiled-file-name compile-file compile-and-load
-            compile compile-time-environment
+            load-ensuring-compiled
+            compile
             decompile)
   #:export-syntax (call-with-compile-error-catch))
 
         ,thunk
         (lambda (key loc msg exp)
           (if (pair? loc)
-              (format (current-error-port)
-                       "~A:~A: ~A: ~A~%" (car loc) (cdr loc) msg exp)
-              (format (current-error-port)
-                       "unknown location: ~A: ~A~%" msg exp)))))
+               (let ((file (or (assq-ref loc 'filename) "unknown file"))
+                     (line (assq-ref loc 'line))
+                     (col (assq-ref loc 'column)))
+                 (format (current-error-port)
+                         "~A:~A:~A: ~A: ~A~%" file line col msg exp))
+               (format (current-error-port)
+                       "unknown location: ~A: ~S~%" msg exp)))))
 
 \f
 ;;;
@@ -77,8 +81,9 @@
      (lambda ()
        (with-throw-handler #t
          (lambda ()
-           (with-output-to-port tmp
-             (lambda () (proc (current-output-port))))
+           (proc tmp)
+           (chmod tmp (logand #o0666 (lognot (umask))))
+           (close-port tmp)
            (rename-file template filename))
          (lambda args
            (delete-file template)))))))
       x
       (lookup-language x)))
 
-(define* (compile-file file #:key (to 'objcode) (opts '()))
-  (let ((comp (compiled-file-name file))
-        (lang (ensure-language (current-language)))
-        (to (ensure-language to)))
-    (catch 'nothing-at-all
-      (lambda ()
-       (call-with-compile-error-catch
-        (lambda ()
-          (call-with-output-file/atomic comp
-            (lambda (port)
-               (let ((print (language-printer to)))
-                 (print (compile (read-file-in file lang)
-                                 #:from lang #:to to #:opts opts)
-                        port))))
-          (format #t "wrote `~A'\n" comp))))
-      (lambda (key . args)
-       (format #t "ERROR: during compilation of ~A:\n" file)
-       (display "ERROR: ")
-       (apply format #t (cadr args) (caddr args))
-       (newline)
-       (format #t "ERROR: ~A ~A ~A\n" key (car args) (cadddr args))
-       (delete-file comp)))))
-
-(define* (compile-and-load file #:key (to 'value) (opts '()))
-  (let ((lang (ensure-language (current-language))))
-    (compile (read-file-in file lang) #:to 'value #:opts opts)))
-
-(define (compiled-file-name file)
+(define (ensure-directory dir)
+  (or (file-exists? dir)
+      (begin
+        (ensure-directory (dirname dir))
+        (mkdir dir))))
+
+(define* (compile-file file #:key
+                       (output-file #f)
+                       (env #f)
+                       (from (current-language))
+                       (to 'objcode)
+                       (opts '()))
+  (let ((comp (or output-file (compiled-file-name file)))
+        (in (open-input-file file)))
+    (ensure-directory (dirname comp))
+    (call-with-output-file/atomic comp
+      (lambda (port)
+        ((language-printer (ensure-language to))
+         (read-and-compile in #:env env #:from from #:to to #:opts opts)
+         port)))
+    comp))
+
+(define* (compile-and-load file #:key (from 'scheme) (to 'value) (opts '()))
+  (read-and-compile (open-input-file file)
+                    #:from from #:to to #:opts opts))
+
+(define* (load-ensuring-compiled source #:key (from 'scheme)
+                                           (to 'value) (opts '()))
+  (let ((compiled (compiled-file-name source #:readable #t)))
+    (load-compiled
+     (if (and compiled
+              (>= (stat:mtime (stat compiled)) (stat:mtime (stat source))))
+         compiled
+         (let ((to-compile (compiled-file-name source #:writable #t)))
+           (if compiled
+               (warn "source file" source "newer than" compiled))
+           (if (and compiled
+                    (not (string-equal? compiled to-compile))
+                    (file-exists? to-compile)
+                    (>= (stat:mtime (stat to-compile))
+                        (stat:mtime (stat compiled))))
+               (warn "using local compiled copy" to-compile)
+               (begin
+                 (format (current-error-port) ";;; Compiling ~s\n" source)
+                 (compile-file source #:output-file to-compile)
+                 (format (current-error-port) ";;; Success: ~s\n" to-compile)))
+           to-compile)))))
+
+(define (ensure-fallback-path)
+  (let ((home (or (getenv "HOME")
+                          (false-if-exception
+                           (passwd:dir (getpwuid (getuid)))))))
+    (and home
+         (let ((cache (in-vicinity home ".guile-ccache")))
+           (cond
+            ((and (access? cache (logior W_OK X_OK))
+                  (file-is-directory? cache))
+             cache)
+            ((not (file-exists? cache))
+             (and (false-if-exception (mkdir cache))
+                  cache))
+            (else #f))))))
+
+(define load-compiled-path
+  (let ((fallback-path #f))
+    (lambda ()
+      (if (not fallback-path)
+          (let ((cache-path (ensure-fallback-path)))
+            (set! fallback-path
+                  (if cache-path
+                      (list cache-path)
+                      '()))))
+      (append %load-path fallback-path))))
+
+(define* (compiled-file-name file #:key (writable #f) (readable #f))
   (let ((base (basename file))
         (cext (cond ((or (null? %load-compiled-extensions)
                          (string-null? (car %load-compiled-extensions)))
                            %load-compiled-extensions)
                      ".go")
                     (else (car %load-compiled-extensions)))))
-    (let lp ((exts %load-extensions))
-      (cond ((null? exts) (string-append base cext))
-            ((string-null? (car exts)) (lp (cdr exts)))
-            ((string-suffix? (car exts) base)
-             (string-append
-              (substring base 0
-                         (- (string-length base) (string-length (car exts))))
-              cext))
-            (else (lp (cdr exts)))))))
+    (define (strip-source-extension base)
+      (let lp ((exts %load-extensions))
+        (cond ((null? exts) (string-append file cext))
+              ((string-null? (car exts)) (lp (cdr exts)))
+              ((string-suffix? (car exts) base)
+               (substring source 0
+                          (- (string-length source)
+                             (string-length (car exts)))))
+              (else (lp (cdr exts))))))
+    (define (strip-path file paths)
+      (let lp ((paths paths))
+        (cond ((null? paths) file)
+              ((string-prefix? (car paths) file)
+               (substring file (1+ (string-length (car paths)))))
+              (else (lp (cdr paths))))))
+    (let ((sibling (string-append (strip-source-extension file) cext)))
+      (cond
+       (writable
+        ;; either put it right beside the original file, or in our
+        ;; ccache. other things wind up not making sense.
+        (cond
+         ((or (not (file-exists? sibling)) (access? sibling W_OK))
+          sibling)
+         ((ensure-fallback-path)
+          => (lambda (p)
+               (string-append p "/" (strip-path sibling))))
+         (else #f)))
+       (readable
+        (if (access? sibling R_OK)
+            sibling
+            (search-path (load-compiled-path)
+                         (strip-path (strip-source-extension file))
+                         %load-compiled-extensions #t)))
+       (else
+        sibling)))))
+
 
 \f
 ;;;
 ;;; Compiler interface
 ;;;
 
-(define (read-file-in file lang)
-  (call-with-input-file file
-    (or (language-read-file lang)
-        (error "language has no #:read-file" lang))))
-
 (define (compile-passes from to opts)
   (map cdr
        (or (lookup-compilation-order from to)
            (error "no way to compile" from "to" to))))
 
 (define (compile-fold passes exp env opts)
-  (if (null? passes)
-      exp
-      (receive (exp env) ((car passes) exp env opts)
-        (compile-fold (cdr passes) exp env opts))))
+  (let lp ((passes passes) (x exp) (e env) (cenv env) (first? #t))
+    (if (null? passes)
+        (values x e cenv)
+        (receive (x e new-cenv) ((car passes) x e opts)
+          (lp (cdr passes) x e (if first? new-cenv cenv) #f)))))
+
+(define (find-language-joint from to)
+  (let lp ((in (reverse (or (lookup-compilation-order from to)
+                            (error "no way to compile" from "to" to))))
+           (lang to))
+    (cond ((null? in)
+           (error "don't know how to join expressions" from to))
+          ((language-joiner lang) lang)
+          (else
+           (lp (cdr in) (caar in))))))
 
-(define (compile-time-environment)
-  "A special function known to the compiler that, when compiled, will
-return a representation of the lexical environment in place at compile
-time. Useful for supporting some forms of dynamic compilation. Returns
-#f if called from the interpreter."
-  #f)
+(define* (read-and-compile port #:key
+                           (env #f)
+                           (from (current-language))
+                           (to 'objcode)
+                           (opts '()))
+  (let ((from (ensure-language from))
+        (to (ensure-language to)))
+    (let ((joint (find-language-joint from to)))
+      (with-fluids ((*current-language* from))
+        (let lp ((exps '()) (env #f) (cenv env))
+          (let ((x ((language-reader (current-language)) port)))
+            (cond
+             ((eof-object? x)
+              (compile ((language-joiner joint) (reverse exps) env)
+                       #:from joint #:to to #:env env #:opts opts))
+             (else
+              ;; compile-fold instead of compile so we get the env too
+              (receive (jexp jenv jcenv)
+                  (compile-fold (compile-passes (current-language) joint opts)
+                                x cenv opts)
+                (lp (cons jexp exps) jenv jcenv))))))))))
 
 (define* (compile x #:key
                   (env #f)
                   (from (current-language))
                   (to 'value)
                   (opts '()))
-  (compile-fold (compile-passes from to opts)
-                x
-                env
-                opts))
+  (receive (exp env cenv)
+      (compile-fold (compile-passes from to opts) x env opts)
+    exp))
 
 \f
 ;;;