]> code.delx.au - gnu-emacs/blobdiff - lisp/org/ob-sh.el
Merge from trunk after a lot of time.
[gnu-emacs] / lisp / org / ob-sh.el
index 9349a842a603ee84c18e5123deb4d7241c2271bc..c0e6b15feb9560b8b27b97262e0e492ff39b651b 100644 (file)
@@ -1,11 +1,10 @@
 ;;; ob-sh.el --- org-babel functions for shell evaluation
 
-;; Copyright (C) 2009, 2010  Free Software Foundation, Inc.
+;; Copyright (C) 2009-2013 Free Software Foundation, Inc.
 
 ;; Author: Eric Schulte
 ;; Keywords: literate programming, reproducible research
 ;; Homepage: http://orgmode.org
-;; Version: 7.7
 
 ;; This file is part of GNU Emacs.
 
@@ -57,14 +56,13 @@ This will be passed to  `shell-command-on-region'")
 This function is called by `org-babel-execute-src-block'."
   (let* ((session (org-babel-sh-initiate-session
                   (cdr (assoc :session params))))
-         (result-params (cdr (assoc :result-params params)))
         (stdin ((lambda (stdin) (when stdin (org-babel-sh-var-to-string
-                                        (org-babel-ref-resolve stdin))))
+                                             (org-babel-ref-resolve stdin))))
                 (cdr (assoc :stdin params))))
          (full-body (org-babel-expand-body:generic
                     body params (org-babel-variable-assignments:sh params))))
     (org-babel-reassemble-table
-     (org-babel-sh-evaluate session full-body result-params stdin)
+     (org-babel-sh-evaluate session full-body params stdin)
      (org-babel-pick-name
       (cdr (assoc :colname-names params)) (cdr (assoc :colnames params)))
      (org-babel-pick-name
@@ -92,7 +90,7 @@ This function is called by `org-babel-execute-src-block'."
 ;; helper functions
 
 (defun org-babel-variable-assignments:sh (params)
-  "Return list of shell statements assigning the block's variables"
+  "Return list of shell statements assigning the block's variables."
   (let ((sep (cdr (assoc :separator params))))
     (mapcar
      (lambda (pair)
@@ -109,13 +107,13 @@ var of the same value."
 
 (defun org-babel-sh-var-to-string (var &optional sep)
   "Convert an elisp value to a string."
-  (flet ((echo-var (v) (if (stringp v) v (format "%S" v))))
+  (let ((echo-var (lambda (v) (if (stringp v) v (format "%S" v)))))
     (cond
      ((and (listp var) (listp (car var)))
-      (orgtbl-to-generic var  (list :sep (or sep "\t") :fmt #'echo-var)))
+      (orgtbl-to-generic var  (list :sep (or sep "\t") :fmt echo-var)))
      ((listp var)
-      (mapconcat #'echo-var var "\n"))
-     (t (echo-var var)))))
+      (mapconcat echo-var var "\n"))
+     (t (funcall echo-var var)))))
 
 (defun org-babel-sh-table-or-results (results)
   "Convert RESULTS to an appropriate elisp value.
@@ -135,29 +133,38 @@ Emacs-lisp table, otherwise return the results as a string."
 (defvar org-babel-sh-eoe-output "org_babel_sh_eoe"
   "String to indicate that evaluation has completed.")
 
-(defun org-babel-sh-evaluate (session body &optional result-params stdin)
+(defun org-babel-sh-evaluate (session body &optional params stdin)
   "Pass BODY to the Shell process in BUFFER.
 If RESULT-TYPE equals 'output then return a list of the outputs
 of the statements in BODY, if RESULT-TYPE equals 'value then
 return the value of the last statement in BODY."
   ((lambda (results)
      (when results
-       (if (or (member "scalar" result-params)
-              (member "verbatim" result-params)
-              (member "output" result-params))
-          results
-        (let ((tmp-file (org-babel-temp-file "sh-")))
-          (with-temp-file tmp-file (insert results))
-          (org-babel-import-elisp-from-file tmp-file)))))
+       (let ((result-params (cdr (assoc :result-params params))))
+        (if (or (member "scalar" result-params)
+                (member "verbatim" result-params)
+                (member "output" result-params))
+            results
+          (let ((tmp-file (org-babel-temp-file "sh-")))
+            (with-temp-file tmp-file (insert results))
+            (org-babel-import-elisp-from-file tmp-file))))))
    (cond
     (stdin                             ; external shell script w/STDIN
      (let ((script-file (org-babel-temp-file "sh-script-"))
-          (stdin-file (org-babel-temp-file "sh-stdin-")))
-       (with-temp-file script-file (insert body))
+          (stdin-file (org-babel-temp-file "sh-stdin-"))
+          (shebang (cdr (assoc :shebang params)))
+          (padline (not (string= "no" (cdr (assoc :padline params))))))
+       (with-temp-file script-file
+        (when shebang (insert (concat shebang "\n")))
+        (when padline (insert "\n"))
+        (insert body))
+       (set-file-modes script-file #o755)
        (with-temp-file stdin-file (insert stdin))
        (with-temp-buffer
         (call-process-shell-command
-         (format "%s %s" org-babel-sh-command script-file)
+         (if shebang
+             script-file
+           (format "%s %s" org-babel-sh-command script-file))
          stdin-file
          (current-buffer))
         (buffer-string))))
@@ -183,7 +190,18 @@ return the value of the last statement in BODY."
            (list org-babel-sh-eoe-indicator))))
        2)) "\n"))
     ('otherwise                                ; external shell script
-     (org-babel-eval org-babel-sh-command (org-babel-trim body))))))
+     (if (and (cdr (assoc :shebang params))
+             (> (length (cdr (assoc :shebang params))) 0))
+        (let ((script-file (org-babel-temp-file "sh-script-"))
+              (shebang (cdr (assoc :shebang params)))
+              (padline (not (string= "no" (cdr (assoc :padline params))))))
+          (with-temp-file script-file
+            (when shebang (insert (concat shebang "\n")))
+            (when padline (insert "\n"))
+            (insert body))
+          (set-file-modes script-file #o755)
+          (org-babel-eval script-file ""))
+       (org-babel-eval org-babel-sh-command (org-babel-trim body)))))))
 
 (defun org-babel-sh-strip-weird-long-prompt (string)
   "Remove prompt cruft from a string of shell output."
@@ -193,6 +211,6 @@ return the value of the last statement in BODY."
 
 (provide 'ob-sh)
 
-;; arch-tag: 416dd531-c230-4b0a-a5bf-8d948f990f2d
+
 
 ;;; ob-sh.el ends here