;;; core-spacebind.el --- Spacemacs Core File -*- lexical-binding: t -*-
;;
;; Copyright (c) 2012-2020 Sylvain Benner & Contributors
;;
;; Author: Eugene "JAremko" Yaremenko <w3techplayground@gmail.com>
;; URL: https://github.com/syl20bnr/spacemacs
;;
;; This file is not part of GNU Emacs.
;;
;;; License: GPLv3

(require 'core-keybindings)

(defvar spacebind--eager-bind nil
  "If true bind keys right after `spacmeacs|spacebind' macro-expanse.
Otherwise binding happens at the next event loop.")

;; Binding stacks
(defvar spacebind--bs-add-minor-mode-replacements '()
  "Binding stack for `spacemacs/add-key-based-replacements-for-minor-mode'.")
(defvar spacebind--bs-add-major-mode-replacements '()
  "Binding stack for `which-key-add-major-mode-key-based-replacements'.")
(defvar spacebind--bs-declare-prefix '()
  "Binding stack for `spacemacs/declare-prefix'.")
(defvar spacebind--bs-declare-prefix-for-mode '()
  "Binding stack for `spacemacs/declare-prefix-for-mode'.")
(defvar spacebind--bs-set-leader-keys '()
  "Binding stack for `spacemacs/set-leader-keys'.")
(defvar spacebind--bs-set-leader-keys-for-major-mode '()
  "Binding stack for `spacemacs/set-leader-keys-for-major-mode'.")
(defvar spacebind--bs-set-leader-keys-for-minor-mode '()
  "Binding stack for `spacemacs/set-leader-keys-for-minor-mode'.")
(defvar spacebind--bs-global-replacements '()
  "Binding stack for `which-key-add-key-based-replacements'.")
(defvar spacebind--bs-add-fn-key-seq-override '()
  "Binding stack for `spacemacs/add-which-key-fn-key-seq-override'.")

(defvar spacebind--timer [t]
  "`run-with-idle-timer' return value for `spacebind//process-bind-stack'.")

(defun spacebind//process-bind-stack ()
  "Drains bind stacks and binds keys and prefixes."
  (unwind-protect
      (progn
        ;; `spacemacs/add-key-based-replacements-for-minor-mode'
        (dolist (args spacebind--bs-add-minor-mode-replacements)
          (let ((mode (car args))
                (keys (string-join (append `(,(spacemacs/leader-key))
                                           (cadr args))
                                   " "))
                (label (caddr args)))
            (spacemacs/add-key-based-replacements-for-minor-mode
             mode keys label)))

        ;; `which-key-add-major-mode-key-based-replacements'
        (dolist (args spacebind--bs-add-major-mode-replacements)
          (let ((mode (car args))
                (keys (string-join (append `(,(spacemacs/leader-key))
                                           `(,(spacemacs/major-mode-prefix))
                                           (cadr args))
                                   " "))
                (label (caddr args)))
            (which-key-add-major-mode-key-based-replacements mode keys label)))

        ;; `spacemacs/declare-prefix'
        (dolist (args spacebind--bs-declare-prefix)
          (let ((prefix (string-join (car args) " "))
                (label (cadr args)))
            (spacemacs/declare-prefix prefix label)))

        ;; `spacemacs/declare-prefix-for-mode'
        (dolist (args spacebind--bs-declare-prefix-for-mode)
          (let ((mode (car args))
                (prefix (string-join (cadr args) " "))
                (label (caddr args)))
            (spacemacs/declare-prefix prefix label)))

        ;; `spacemacs/set-leader-keys'
        (dolist (args spacebind--bs-set-leader-keys)
          (let ((keys (string-join (car args) " "))
                (fn-sym (cadr args)))
            (spacemacs/set-leader-keys keys fn-sym)))

        ;; `spacemacs/set-leader-keys-for-major-mode'
        (dolist (args spacebind--bs-set-leader-keys-for-major-mode)
          (let ((mode (car args))
                (keys (string-join (cadr args) " "))
                (fn-sym (caddr args)))
            (spacemacs/set-leader-keys-for-major-mode mode keys fn-sym)))

        ;; `spacemacs/set-leader-keys-for-minor-mode'
        (dolist (args spacebind--bs-set-leader-keys-for-minor-mode)
          (let ((mode (car args))
                (keys (string-join (cadr args) " "))
                (fn-sym (caddr args)))
            (spacemacs/set-leader-keys-for-minor-mode mode keys fn-sym)))

        ;; `which-key-add-key-based-replacements'
        (dolist (args spacebind--bs-global-replacements)
          (let ((keys (string-join (append `(,(spacemacs/leader-key))
                                           (car args))
                                   " "))
                (label (cadr args)))
            (which-key-add-key-based-replacements keys label)))

        ;; `spacemacs/add-which-key-fn-key-seq-override'
        (dolist (args spacebind--bs-add-fn-key-seq-override)
          (let ((sym (car args))
                (rep (cadr args))
                (label (caddr args)))
            (spacemacs/add-which-key-fn-key-seq-override sym rep label))))

    ;; Reset stacks
    (setq spacebind--bs-global-replacements nil
          spacebind--bs-set-leader-keys-for-minor-mode nil
          spacebind--bs-set-leader-keys-for-major-mode nil
          spacebind--bs-set-leader-keys nil
          spacebind--bs-declare-prefix-for-mode nil
          spacebind--bs-declare-prefix nil
          spacebind--bs-add-major-mode-replacements nil
          spacebind--bs-add-minor-mode-replacements nil
          spacebind--bs-add-fn-key-seq-override nil
          ;; Reset timer var
          spacebind--timer [t])))

;; TODO: Make this configurable.
(defun spacemacs/major-mode-prefix ()
  "Get current prefix for major modes.
NOTE: `dotspacemacs-major-mode-leader-key' isn't the same."
  "m")

(defun spacemacs/leader-key ()
  "Returns `dotspacemacs-leader-key'"
  dotspacemacs-leader-key)

(defun spacebind//strip-docstring (binding-form)
  "Remove second element of BINDING-FORM if it is a string."
  (if (stringp (cadr binding-form))
      (cons (car binding-form) (cddr binding-form))
    binding-form))

(defun spacemacs/add-key-based-replacements-for-minor-mode
    (mode key-sequence replacement &rest more)
  "Proxy for `which-key-add-key-based-replacements'
MODE currently ignored.
FIXME: We should disable replacements when the mode is disabled.
The function `which-key-add-major-mode-key-based-replacements' might give
a clue how to do this.
Also there is discussion about the feature:
https://github.com/justbur/emacs-which-key/issues/212"
  (apply #'which-key-add-key-based-replacements key-sequence replacement more))

(defun spacemacs/add-which-key-fn-key-seq-override
    (fn-name key-rep label)
  "Replace FN-NAME function's key sequence with KEY-REP and LABEL."
  (push `((nil . ,fn-name) . (,key-rep . ,label)) which-key-replacement-alist))

(cl-defstruct spacemacs--spacebind-state
  "State object for `spacebind' macro implementation.
CTYPE - current binding type.
RSEXP - accumulator with the macro output.
This structure has one interpreter method for each supported CTYPE.
CTYPE is a type of a currently processed binding."
  ctype rsexp)

(cl-defgeneric spacemacs//spacebind-dispatch (state binding)
  (:documentation "Based on BINDING type modify STATE using BINDING value."))

(cl-defmethod spacemacs//spacebind-dispatch ((state spacemacs--spacebind-state)
                                             (keyword-or-symbol symbol))
  "If KEYWORD is a keyword, set STATE slot ctype(current type) to its value.
If KEYWORD is a non-keyword symbol (without \":\" prefix in the name), wrap
its value into a list and re-apply the function to it."
  (if (not (keywordp keyword-or-symbol))
      (spacemacs//spacebind-dispatch state (list keyword-or-symbol))
    (setf (spacemacs--spacebind-state-ctype state) keyword-or-symbol)
    state))

(cl-defmethod spacemacs//spacebind-dispatch ((state spacemacs--spacebind-state)
                                             (sexp list))
  "Apply STATE method from ctype slot to SEXP and append output to rsexp slot."
  (cl-callf append (spacemacs--spacebind-state-rsexp state)
    (funcall (spacemacs--spacebind-state-ctype state) state sexp))
  state)

(cl-defmethod spacemacs//spacebind-dispatch ((state spacemacs--spacebind-state)
                                             (_ string))
  "Ignore strings - used to implement doc-strings."
  state)

(defun spacemacs//spacebind-form-visitor (form path k-fn p-fn)
  "Applies K-FN to FORM if it is a key binding form. Otherwise applies P-FN.
PATH passed to the applied function.
NOTE: This function strips all newline characters, replaces successive spaces
with a singular in string elements of FORM and trims tails of function labels
delimited by \"|\" character."
  (list
   (cl-labels ((apply-str-fmt
                (el)
                (thread-last el
                  ;; Convert new lines and multiply spaces into singular.
                  ;; This is done to enable better binding forms.
                  (replace-regexp-in-string "[\n[:space:]]+" " ")
                  ;; Discard everything after | symbol in labels.
                  ;; This way we can add extra text into the README.org
                  ;; files while omitting it in labels.
                  (replace-regexp-in-string "[[:punct:][:space:]]*|.+" "")))
               (str-fmt-rec
                (depth el)
                (cond
                 ((stringp el) (apply-str-fmt el))
                 ((and (= depth 0)
                       (listp el))
                  ;; We don't want to go deeper than a single level.
                  (mapcar (apply-partially #'str-fmt-rec (1+ depth)) el))
                 (t el)))
               (str-fmt
                (el)
                (str-fmt-rec 0 el)))
     (cl-destructuring-bind
         (key-or-prefix-form
          leader-label-or-fn-symbol
          leader-label-or-next-form)
         (mapcar #'str-fmt (seq-take form 3))
       (let ((full-key-or-prefix (append
                                  path
                                  ;; ("key" :label "label") or "key".
                                  `(,(or (car-safe key-or-prefix-form)
                                         key-or-prefix-form))))
             (key-or-prefix-label (thread-first key-or-prefix-form
                                    (cdr-safe)
                                    (plist-get :label))))
         (if (symbolp leader-label-or-fn-symbol)
             (funcall k-fn
                      full-key-or-prefix
                      key-or-prefix-label
                      leader-label-or-fn-symbol
                      ;; Either "label" or ("doc label" :label "label").
                      (or (thread-first leader-label-or-next-form
                            (cdr-safe)
                            (plist-get :label))
                          leader-label-or-next-form))
           (funcall p-fn full-key-or-prefix leader-label-or-fn-symbol)))))))

(defun spacemacs//spacebind-form-walker-rec (path k-fn p-fn form)
  "Recursive body of `spacemacs//spacebind-form-walker'."
  (let* ((fn-sym-or-label (car-safe (cdr-safe form)))
         (prefix-form? (stringp fn-sym-or-label))
         (binding-form? (and fn-sym-or-label (symbolp fn-sym-or-label)))
         (list-of-forms? (and form (every #'consp form)))
         (binding-or-prefix-form? (or binding-form?
                                      prefix-form?))
         (head (car form))
         (cur-path (if prefix-form?
                       (append path `(,(or (car-safe head)
                                           head)))
                     path))
         ;; Strip key and label from prefix forms.
         (bindings (if prefix-form?
                       (cddr form)
                     form)))
    (append
     (when binding-or-prefix-form?
       (spacemacs//spacebind-form-visitor form path k-fn p-fn))
     (when (or prefix-form?
               list-of-forms?)
       (seq-mapcat
        (apply-partially
         'spacemacs//spacebind-form-walker-rec cur-path k-fn p-fn)
        bindings)))))

(defun spacemacs//spacebind-form-walker (b-forms k-fn p-fn)
  "Part of `spacemacs--spacebind-state' interpreters implementation.
B-FORMS is a root node of a binding tree without mode (car of the root form).
K-FN called for each key binding node with 3 arguments: full_key_sequence,
function_symbol and label_for_leader_menu.
P-FN called for each prefix binding node with 2 arguments:
full_key_prefix_sequence and label_for_leader_menu.
Both K-FN and P-FN should return binding evaluation forms.
The forms will be concatenated and substituted by `spacebind' macro."
  (spacemacs//spacebind-form-walker-rec nil k-fn p-fn b-forms))

;; Key bindings - keywords handlers

(cl-defmethod :print-debug ((_ spacemacs--spacebind-state) form)
  "`message' logging interpreter for debugging."
  (let* ((form (spacebind//strip-docstring form))
         (mode (pop form)))
    (spacemacs//spacebind-form-walker
     form
     (lambda (key-seq key-label fn-symbol label)
       `(message "Key args: key-seq: %S Key-label: %S fn-symbol: %S label: %S"
                 ,key-seq, key-label ,fn-symbol ,label))
     (lambda (key-prefix label)
       `(message "Prefix args: key-prefix: %S label: %S"
                 ,key-prefix ,label)))))

(cl-defmethod :global ((_ spacemacs--spacebind-state) form)
  "Interpreter for global binding forms."
  (spacemacs//spacebind-form-walker
   ;; Strip optional doc-string.
   (if (char-or-string-p (car form))
       (cdr form)
     form)
   (lambda (key-seq key-label fn-symbol label)
     `(progn
        (push (list ',key-seq ,label) spacebind--bs-global-replacements)
        ,(when key-label
           `(push (list ,(symbol-name fn-symbol)
                        ,key-label
                        ,label)
                  spacebind--bs-add-fn-key-seq-override))
        (push (list ',key-seq ',fn-symbol) spacebind--bs-set-leader-keys)))
   (lambda (key-prefix label)
     `(push (list ',key-prefix ,label) spacebind--bs-declare-prefix))))

(cl-defmethod :major ((_ spacemacs--spacebind-state) form)
  "Interpreter for major mode binding forms."
  (let* ((form (spacebind//strip-docstring form))
         (mode (pop form)))
    (spacemacs//spacebind-form-walker
     form
     (lambda (key-seq key-label fn-symbol label)
       `(progn
          (push (list ',mode ',key-seq ,label)
                spacebind--bs-add-major-mode-replacements)
          ,(when key-label
             `(push (list ,(symbol-name fn-symbol)
                          ,key-label
                          ,label)
                    spacebind--bs-add-fn-key-seq-override))
          (push (list ',mode ',key-seq ',fn-symbol)
                spacebind--bs-set-leader-keys-for-major-mode)))
     (lambda (key-prefix label)
       `(push (list ',mode ',key-prefix ,label)
              spacebind--bs-declare-prefix-for-mode)))))

(cl-defmethod :minor ((_ spacemacs--spacebind-state) form)
  "Interpreter for minor mode binding forms."
  (let* ((form (spacebind//strip-docstring form))
         (mode (pop form)))
    (spacemacs//spacebind-form-walker
     form
     (lambda (key-seq key-label fn-symbol label)
       `(progn
          (push (list ',mode ',key-seq ,label)
                spacebind--bs-add-minor-mode-replacements)
          ,(when key-label
             `(push (list ,(symbol-name fn-symbol)
                          ,key-label
                          ,label)
                    spacebind--bs-add-fn-key-seq-override))
          (push (list ',mode ',key-seq ',fn-symbol)
                spacebind--bs-set-leader-keys-for-minor-mode)))
     (lambda (key-prefix label)
       `(push (list ',mode ',key-prefix ,label)
              spacebind--bs-declare-prefix-for-mode)))))

(defmacro spacemacs|spacebind (&rest bindings)
  "Bind keys and their prefixes declared via BINDINGS tree like structure.
BINDINGS format:
 <DELIMITER_KEYWORD>
  <BINDING_FORM>
  <BINDING_FORM>
  ...
 <DELIMITER_KEYWORD>
  <BINDING_FORM>
  ...
 ...

DELIMITER_KEYWORD - specifies a type of following <BINDING_FORM> (or forms).
Currently supported types: :major, :minor and :global.

:major and :minor bindings have this shape:
(<MODE>
 <OPTIONAL_DOC_STRING>
 <PREFIX_OR_BINDING>
 <PREFIX_OR_BINDING>
 ...)

:global forms have a similar shape:
(<OPTIONAL_DOC_STRING>
 <PREFIX_OR_BINDING>
 <PREFIX_OR_BINDING>
 ...)

<OPTIONAL_DOC_STRING> is a string that will be used to generate a key-bindings
section in the corresponding README.org files.

<PREFIX_OR_BINDING> is a recursive form that can be:
  A prefix form:
  (<PREFIX_KEY> <TEXT> <PREFIX_OR_BINDING>)
  Or a key-binding form:
  (<KEY> <FUNCTION_SYMBOL> <TEXT>)

<TEXT> is what will be displayed in the menu and used for the documentation
generation.

<PREFIX_KEY> and <KEY> are singular keys represented as strings.

<FUNCTION_SYMBOL> is the function that will be bound to the <KEY>.

See core-spacebind-utest.el for examples.

NOTE: This macro also has `use-package' integration via `:spacebind' key

NOTE: <TEXT> strings support formatting:
      - \n and multiply spaces are converted into single spaces in the <TEXT>.
      - Everything after and including | symbol is ignored and punctuation
        before the character trimmed. This is done so you can provide additional
        information for the binding documentation while keeping labels brief.

NOTE: You can override key labels and displayed sequences :label <value>
      Example: ((\"k\" :label \"press k\")
                foo-fn
                (\"for docs\" :label \"displayed\"))

\(fn <<DELIMITER_KEYWORD> <BINDING_FORMS>...>...)"
  (append
   (spacemacs--spacebind-state-rsexp
    (seq-reduce 'spacemacs//spacebind-dispatch
                bindings
                (make-spacemacs--spacebind-state
                 :rsexp `(progn))))
   ;; Schedule stacks processing with `spacebind//process-bind-stack' function.
   `((when (aref spacebind--timer 0)
       (if (not spacebind--eager-bind)
           (setq spacebind--timer
                 (run-with-idle-timer 0 nil #'spacebind//process-bind-stack))
         (setq spacebind--timer [t])
         (spacebind//process-bind-stack))))))

(provide 'core-spacebind)