|
@@ -1,4 +1,4 @@
|
|
|
-;;; org-babel-shell.el --- org-babel functions for shell evaluation
|
|
|
+;;; org-babel-sh.el --- org-babel functions for shell evaluation
|
|
|
|
|
|
;; Copyright (C) 2009 Eric Schulte
|
|
|
|
|
@@ -48,27 +48,44 @@ function is called by `org-babel-execute-src-block'."
|
|
|
(lambda (pair)
|
|
|
(format "%s=%s"
|
|
|
(car pair)
|
|
|
- (org-babel-shell-var-to-shell (cdr pair))))
|
|
|
+ (org-babel-sh-var-to-sh (cdr pair))))
|
|
|
vars "\n") "\n" body "\n\n")) ;; then the source block body
|
|
|
- (session (org-babel-shell-initiate-session (cdr (assoc :session params))))
|
|
|
- (results (org-babel-shell-evaluate (org-babel-shell-session-buffer session) full-body result-type)))
|
|
|
+ (session (org-babel-sh-initiate-session (cdr (assoc :session params))))
|
|
|
+ (results (org-babel-sh-evaluate session full-body result-type)))
|
|
|
(if (member "scalar" result-params)
|
|
|
results
|
|
|
- (setq results (let ((tmp-file (make-temp-file "org-babel-ruby")))
|
|
|
+ (setq results (let ((tmp-file (make-temp-file "org-babel-shell")))
|
|
|
(with-temp-file tmp-file (insert results))
|
|
|
(org-babel-import-elisp-from-file tmp-file)))
|
|
|
(if (and (member "vector" results) (not (listp results)))
|
|
|
(list (list results))
|
|
|
results))))
|
|
|
|
|
|
-(defun org-babel-shell-var-to-shell (var)
|
|
|
+(defun org-babel-prep-session:sh (session params)
|
|
|
+ "Prepare SESSION according to the header arguments specified in PARAMS."
|
|
|
+ (let* ((session (org-babel-sh-initiate-session session))
|
|
|
+ (vars (org-babel-ref-variables params))
|
|
|
+ (var-lines (mapcar ;; define any variables
|
|
|
+ (lambda (pair)
|
|
|
+ (format "%s=%s"
|
|
|
+ (car pair)
|
|
|
+ (org-babel-sh-var-to-sh (cdr pair))))
|
|
|
+ vars)))
|
|
|
+ (org-babel-comint-in-buffer session
|
|
|
+ (mapc (lambda (var)
|
|
|
+ (insert var) (comint-send-input nil t)
|
|
|
+ (org-babel-comint-wait-for-output session)) var-lines))))
|
|
|
+
|
|
|
+;; helper functions
|
|
|
+
|
|
|
+(defun org-babel-sh-var-to-sh (var)
|
|
|
"Convert an elisp var into a string of shell commands
|
|
|
specifying a var of the same value."
|
|
|
(if (listp var)
|
|
|
- (concat "[" (mapconcat #'org-babel-shell-var-to-shell var ", ") "]")
|
|
|
+ (concat "[" (mapconcat #'org-babel-sh-var-to-sh var ", ") "]")
|
|
|
(format "%S" var)))
|
|
|
|
|
|
-(defun org-babel-shell-table-or-results (results)
|
|
|
+(defun org-babel-sh-table-or-results (results)
|
|
|
"If the results look like a table, then convert them into an
|
|
|
Emacs-lisp table, otherwise return the results as a string."
|
|
|
(org-babel-read
|
|
@@ -81,73 +98,57 @@ Emacs-lisp table, otherwise return the results as a string."
|
|
|
"'" "\"" results)))))
|
|
|
results)))
|
|
|
|
|
|
-;; functions for comint evaluation
|
|
|
+(defvar org-babel-sh-buffers '(:default . nil))
|
|
|
|
|
|
-(defvar org-babel-shell-buffers '(:default . nil))
|
|
|
+(defun org-babel-sh-session-buffer (session)
|
|
|
+ (cdr (assoc session org-babel-sh-buffers)))
|
|
|
|
|
|
-(defun org-babel-shell-session-buffer (session)
|
|
|
- (cdr (assoc session org-babel-shell-buffers)))
|
|
|
-
|
|
|
-(defun org-babel-shell-initiate-session (&optional session)
|
|
|
+(defun org-babel-sh-initiate-session-by-key (&optional session)
|
|
|
"If there is not a current inferior-process-buffer in SESSION
|
|
|
then create. Return the initialized session."
|
|
|
(save-window-excursion
|
|
|
(let* ((session (if session (intern session) :default))
|
|
|
- (shell-buffer (org-babel-shell-session-buffer session))
|
|
|
- (newp (not (org-babel-comint-buffer-livep shell-buffer))))
|
|
|
- (shell shell-buffer)
|
|
|
+ (sh-buffer (org-babel-sh-session-buffer session))
|
|
|
+ (newp (not (org-babel-comint-buffer-livep sh-buffer))))
|
|
|
+ (if (and sh-buffer (get-buffer sh-buffer) (not (buffer-live-p sh-buffer)))
|
|
|
+ (setq sh-buffer nil))
|
|
|
+ (shell sh-buffer)
|
|
|
(when newp
|
|
|
- (setq shell-buffer (current-buffer))
|
|
|
- (org-babel-comint-wait-for-output shell-buffer))
|
|
|
- (setq org-babel-shell-buffers (cons (cons session shell-buffer) (assq-delete-all session org-babel-shell-buffers)))
|
|
|
+ (setq sh-buffer (current-buffer))
|
|
|
+ (org-babel-comint-wait-for-output sh-buffer))
|
|
|
+ (setq org-babel-sh-buffers (cons (cons session sh-buffer) (assq-delete-all session org-babel-sh-buffers)))
|
|
|
session)))
|
|
|
|
|
|
-(defvar org-babel-shell-eoe-indicator "echo 'org_babel_shell_eoe'"
|
|
|
+(defun org-babel-sh-initiate-session (&optional session)
|
|
|
+ (org-babel-sh-session-buffer (org-babel-sh-initiate-session-by-key session)))
|
|
|
+
|
|
|
+(defvar org-babel-sh-eoe-indicator "echo 'org_babel_sh_eoe'"
|
|
|
"Used to indicate that evaluation is has completed.")
|
|
|
-(defvar org-babel-shell-eoe-output "org_babel_shell_eoe"
|
|
|
+(defvar org-babel-sh-eoe-output "org_babel_sh_eoe"
|
|
|
"Used to indicate that evaluation is has completed.")
|
|
|
|
|
|
-(defun org-babel-shell-evaluate (buffer body &optional result-type)
|
|
|
+(defun org-babel-sh-evaluate (buffer body &optional result-type)
|
|
|
"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."
|
|
|
- (org-babel-comint-in-buffer buffer
|
|
|
- (let ((string-buffer "")
|
|
|
- (full-body (mapconcat #'org-babel-chomp
|
|
|
- (list body org-babel-shell-eoe-indicator) "\n"))
|
|
|
- results)
|
|
|
- (flet ((my-filt (text) (setq string-buffer (concat string-buffer text))))
|
|
|
- ;; setup filter
|
|
|
- (add-hook 'comint-output-filter-functions 'my-filt)
|
|
|
- ;; pass FULL-BODY to process
|
|
|
- (goto-char (process-mark (get-buffer-process buffer)))
|
|
|
- (insert full-body)
|
|
|
- (comint-send-input)
|
|
|
- ;; wait for end-of-evaluation indicator
|
|
|
- (while (progn
|
|
|
- (goto-char comint-last-input-end)
|
|
|
- (not (save-excursion (and (re-search-forward comint-prompt-regexp nil t)
|
|
|
- (re-search-forward (regexp-quote org-babel-shell-eoe-output) nil t)))))
|
|
|
- (accept-process-output (get-buffer-process buffer)))
|
|
|
- ;; remove filter
|
|
|
- (remove-hook 'comint-output-filter-functions 'my-filt))
|
|
|
- ;; (message (format "raw-results=%S" string-buffer)) ;; debugging
|
|
|
- ;; (message (format "split-results=%S" (mapcar #'org-babel-trim (split-string string-buffer comint-prompt-regexp)))) ;; debugging
|
|
|
- ;; split results with `comint-prompt-regexp'
|
|
|
- (setq results (cdr (member org-babel-shell-eoe-output
|
|
|
- (reverse (mapcar #'org-babel-shell-strip-weird-long-prompt
|
|
|
- (mapcar #'org-babel-trim (split-string string-buffer comint-prompt-regexp)))))))
|
|
|
- ;; (message (replace-regexp-in-string "%" "%%" (format "processed-results=%S" results))) ;; debugging
|
|
|
- (or (case result-type
|
|
|
- (output (org-babel-trim (mapconcat #'org-babel-trim (reverse results) "\n")))
|
|
|
- (value (car results))
|
|
|
- (t (reverse results))) ""))))
|
|
|
-
|
|
|
-(defun org-babel-shell-strip-weird-long-prompt (string)
|
|
|
+ (let* ((full-body (mapconcat #'org-babel-chomp
|
|
|
+ (list body org-babel-sh-eoe-indicator) "\n"))
|
|
|
+ (raw (org-babel-comint-with-output buffer org-babel-sh-eoe-output nil
|
|
|
+ (insert full-body) (comint-send-input nil t)))
|
|
|
+ (results (cdr (member org-babel-sh-eoe-output
|
|
|
+ (reverse (mapcar #'org-babel-sh-strip-weird-long-prompt
|
|
|
+ (mapcar #'org-babel-trim raw)))))))
|
|
|
+ ;; (message (replace-regexp-in-string "%" "%%" (format "processed-results=%S" results))) ;; debugging
|
|
|
+ (or (case result-type
|
|
|
+ (output (org-babel-trim (mapconcat #'org-babel-trim (reverse results) "\n")))
|
|
|
+ (value (car results))
|
|
|
+ (t (reverse results))) "")))
|
|
|
+
|
|
|
+(defun org-babel-sh-strip-weird-long-prompt (string)
|
|
|
(while (string-match "^% +[\r\n$]+ *" string)
|
|
|
(setq string (substring string (match-end 0))))
|
|
|
string)
|
|
|
|
|
|
-(provide 'org-babel-shell)
|
|
|
-;;; org-babel-shell.el ends here
|
|
|
+(provide 'org-babel-sh)
|
|
|
+;;; org-babel-sh.el ends here
|