(require 'ob)

(require 'ob-ref)

(require 'ob-comint)

(require 'ob-eval)

(require 'prolog)

(add-to-list 'org-babel-tangle-lang-exts '("prolog" . "pl"))

(defvar org-babel-prolog-command (or prolog-system "swipl")

"Name of the prolog executable command.")

(defconst org-babel-header-args:prolog

'((:goal . :any))

"Prolog specific header arguments.")

(defvar org-babel-default-header-args:prolog

`((:goal . nil)))

(defun org-babel-prolog--elisp-to-pl (value)

(cond ((stringp value)

(format "'%s'"

(replace-regexp-in-string

"'" "\\'" value)))

((listp value)

(format "[%s]"

(mapconcat #'org-babel-prolog--elisp-to-pl

value

", ")))

(t (prin1-to-string value))))

(defun org-babel-prolog--variable-assignment (pair)

(format "recorda('%s', %s)"

(car pair)

(org-babel-prolog--elisp-to-pl (cdr pair))))

(defun org-babel-variable-assignments:prolog (params)

(let ((strs (mapcar #'org-babel-prolog--variable-assignment

(mapcar #'cdr (org-babel--get-vars params :var)))))

(when strs

(list (concat ":- " (mapconcat #'identity strs ", ") ".\n")))))

(defun org-babel-prolog--parse-goal (goal)

"Evaluate inline emacs-lisp in prolog goal parameter.

(when goal

(with-temp-buffer

(insert goal)

(while (search-backward "=" nil t)

(delete-char 1 t)

(forward-sexp)

(let ((value (eval (preceding-sexp))))

(kill-sexp -1)

(insert (format "%S" value))))

(buffer-string))))

(defun org-babel-execute:prolog (body params)

"Execute a block of Prolog code with org-babel. This function is

(message "executing Prolog source code block")

(let* ((result-params (cdr (assq :result-params params)))

(session (cdr (assq :session params)))

(goal (org-babel-prolog--parse-goal

(cdr (assq :goal params))))

(vars (org-babel-variable-assignments:prolog params))

(full-body (org-babel-expand-body:generic body params vars))

(results (if (string= "none" session)

(org-babel-prolog-evaluate-external-process

goal full-body)

(org-babel-prolog-evaluate-session

session goal full-body))))

(org-babel-reassemble-table

(org-babel-result-cond result-params

results

(let ((tmp (org-babel-temp-file "prolog-results-")))

(with-temp-file tmp (insert results))

(org-babel-import-elisp-from-file tmp)))

(org-babel-pick-name (cdr (assq :colname-names params))

(cdr (assq :colnames params)))

(org-babel-pick-name (cdr (assq :rowname-names params))

(cdr (assq :rownames params))))))

(defun org-babel-load-session:prolog (session body params)

"Load BODY into SESSION."

(let* ((params (org-babel-process-params params))

(session (org-babel-prolog-initiate-session

(cdr (assq :session session)))))

(org-babel-prolog-initiate-session session)))

(defun org-babel-prolog-evaluate-external-process (goal body)

"Evaluates the GOAL given the BODY in a external Prolog

(let* ((tmp-file (org-babel-temp-file "prolog-"))

(command (format "%s -q -l %s -t \"%s\""

org-babel-prolog-command

tmp-file

(replace-regexp-in-string

"\"" "\\\"" (or goal "halt")))))

(with-temp-file tmp-file

(insert (org-babel-chomp body)))

(org-babel-eval command "")))

(defun org-babel-prolog-evaluate-session (session goal body)

"Evaluates the GOAL in the BODY of the prolog block in the

(let* ((session (org-babel-prolog-initiate-session session))

(body (split-string (org-babel-trim body) "\n")))

(org-babel-trim

(with-temp-buffer

(with-current-buffer session

(setq comint-prompt-regexp "^|: *"))

(org-babel-comint-input-command session "consult(user).\n")

(apply #'insert

(org-babel-comint-with-output (session "\n")

(setq comint-prompt-regexp (prolog-prompt-regexp))

(dolist (line body)

(insert line)

(comint-send-input nil t)

(accept-process-output

(get-buffer-process session)))

(comint-send-eof)))

(ansi-color-apply-on-region (point-min) (point-max))

(goto-char (point-max))

(if (save-excursion

(search-backward "ERROR: " nil t))

(progn

(save-excursion

(while (search-backward "|: " nil t)

(replace-match "" nil t)))

(search-backward "true." nil t)

(kill-whole-line)

(org-babel-eval-error-notify -1 (buffer-string))

(buffer-string))

(when goal

(kill-region (point-min) (point-max))

(apply #'insert

(org-babel-comint-with-output (session "")

(insert (concat goal ", !."))

(comint-send-input nil t))))

(ansi-color-apply-on-region (point-min) (point-max))

(if (not (save-excursion

(search-backward "ERROR: " nil t)))

(let ((delete-trailing-lines t))

(delete-trailing-whitespace (point-min))

(buffer-string))

(search-backward "?-" nil t)

(kill-whole-line)

(org-babel-eval-error-notify -1 (buffer-string))

(buffer-string)))))))

(defun org-babel-prolog--answer-correction (string)

(when (string-match-p "Correct to: \".*\"\\?" string)

(insert "no")

(comint-send-input nil t)))

(defun org-babel-prolog--exit-debug (string)

(when (string-match-p "\\(.\\|\n\\)*Exception.* \\? $" string)

(insert "no debug")

(comint-send-input nil t)))

(defun org-babel-prolog-initiate-session (&optional session)

"If there is not a current inferior-process-buffer in SESSION

(unless (string= session "none")

(let ((session (get-buffer-create (or session "*prolog*"))))

(unless (comint-check-proc session)

(with-current-buffer session

(kill-region (point-min) (point-max))

(prolog-inferior-mode)

(setq prolog-program-name org-babel-prolog-command)

(apply #'make-comint-in-buffer

"prolog"

(current-buffer)

(prolog-program-name)

nil

(cons "-q" (prolog-program-switches)))

(add-hook 'comint-output-filter-functions

#'org-babel-prolog--answer-correction nil t)

(add-hook 'comint-output-filter-functions

#'org-babel-prolog--exit-debug nil t)

(while (progn

(goto-char comint-last-input-end)

(not (save-excursion

(re-search-forward comint-prompt-regexp nil t))))

(accept-process-output

(get-buffer-process session)))))

session)))

(provide 'ob-prolog)