;;; lang/org/contrib/jupyter.el -*- lexical-binding: t; -*-
;;;###if (modulep! +jupyter)
;; TODO: Move to :tools jupyter

(use-package! jupyter-repl
  :defer t
  :config
  ;; HACK: If the user is anywhere but the last prompt, typing should move them
  ;;   there instead of unhelpfully spewing read-only errors at them.
  ;; REVIEW: Upstream this (maybe?)
  (defun +jupyter--move-cursor-to-prompt-h ()
    (and (eq this-command 'self-insert-command)
         (> (save-excursion
              (goto-char (point-max))
              (jupyter-repl-cell-code-beginning-position))
            (point))
         (goto-char (point-max))))

  (add-hook! 'jupyter-repl-mode-hook
    (defun +jupyter--init-move-cursor-to-prompt-h ()
      (add-hook 'pre-command-hook #'+jupyter--move-cursor-to-prompt-h
                nil t))))


(use-package! jupyter-org-client
  :defer t
  :config
  ;; Already handled by `+org--ob-jupyter-initiate-session-a'
  (remove-hook 'org-mode-hook #'jupyter-org-interaction-mode))


(use-package! ob-jupyter
  :defer t
  :init
  ;; HACK: ob-juypter don't support ob-async and handles async itself, so
  ;;   piggyback off of `org-babel-jupyter-make-language-alias' to disable it
  ;;   for every current and future kernel language.
  (defadvice! +org-jupyter--suppress-ob-async-a (fn _kernel lang)
    :before #'org-babel-jupyter-make-language-alias
    (with-eval-after-load 'ob-async
      (add-to-list 'ob-async-no-async-languages-alist (concat "jupyter-" lang))))

  (add-hook! '+org-babel-load-functions
    (defun +org-babel-load-jupyter-h (lang)
      (and (string-prefix-p "jupyter-" (symbol-name lang))
           (require 'ob-jupyter nil t)
           (org-babel-jupyter-make-local-aliases))))

  :config
  (defadvice! +org--ob-jupyter-initiate-session-a (&rest _)
    :after #'org-babel-jupyter-initiate-session
    (unless (bound-and-true-p jupyter-org-interaction-mode)
      (jupyter-org-interaction-mode)))

  (require 'tramp))