doomemacs/core/autoload/message.el

;;; core/autoload/message.el -*- lexical-binding: t; -*-

(defconst doom-message-fg
  '((black    30 term-color-black)
    (red      31 term-color-red)
    (green    32 term-color-green)
    (yellow   33 term-color-yellow)
    (blue     34 term-color-blue)
    (magenta  35 term-color-magenta)
    (cyan     36 term-color-cyan)
    (white    37 term-color-white))
  "List of text colors.")

(defconst doom-message-bg
  '((on-black   40 term-color-black)
    (on-red     41 term-color-red)
    (on-green   42 term-color-green)
    (on-yellow  43 term-color-yellow)
    (on-blue    44 term-color-blue)
    (on-magenta 45 term-color-magenta)
    (on-cyan    46 term-color-cyan)
    (on-white   47 term-color-white))
  "List of colors to draw text on.")

(defconst doom-message-fx
  '((bold       1 :weight bold)
    (dark       2)
    (italic     3 :slant italic)
    (underscore 4 :underline t)
    (blink      5)
    (rapid      6)
    (contrary   7)
    (concealed  8)
    (strike     9 :strike-through t))
  "List of styles.")

;;;###autoload
(defun doom-ansi-apply (code message &rest args)
  "Apply CODE to formatted MESSAGE with ARGS. CODE is derived from any of
`doom-message-fg', `doom-message-bg' or `doom-message-fx'.

In a noninteractive session, this wraps the result in ansi color codes.
Otherwise, it maps colors to a term-color-* face."
  (let ((text (apply #'format message args)))
    (if noninteractive
        (format "\e[%dm%s\e[%dm"
                (cadr
                 (or (assq code doom-message-fg)
                     (assq code doom-message-bg)
                     (assq code doom-message-fx)))
                text 0)
      (require 'term)  ; piggyback on term's color faces
      (propertize
       text 'face
       (let (spec)
         (cond ((setq spec (caddr (assq code doom-message-fg)))
                `(:foreground ,(face-foreground spec)))
               ((setq spec (caddr (assq code doom-message-bg)))
                `(:background ,(face-background spec)))
               ((cddr (assq code doom-message-fx)))))))))

;;;###autoload
(defmacro format! (message &rest args)
  "An alternative to `format' that understands (color ...) and converts them
into faces or ANSI codes depending on the type of sesssion we're in."
  `(cl-flet*
       (,@(cl-loop for rule
                   in (append doom-message-fg doom-message-bg doom-message-fx)
                   collect
                   `(,(car rule)
                     (lambda (message &rest args)
                       (apply #'doom-ansi-apply ',(car rule) message args))))
        (color
         (lambda (code format &rest args)
           (apply #'doom-ansi-apply code format args))))
     (format ,message ,@args)))

;;;###autoload
(defmacro print! (message &rest args)
  "Uses `message' in interactive sessions and `princ' otherwise (prints to
standard out).

Can be colored using (color ...) blocks:

  (print! \"Hello %s %s\" (bold (blue \"How are you?\")))
  (print! \"Hello %s %s\" (red \"World\"))
  (print! (green \"Great %s!\" \"success\"))

Uses faces in interactive sessions and ANSI codes otherwise."
  `(if (not noninteractive)
       (message (format! ,message ,@args))
     ;; princ prints to stdout, message to stderr
     (princ (format! ,message ,@args))
     (terpri)))