1 files changed, 20 insertions, 0 deletions

@@ -93,4 +93,24 @@ indicator in the mode-line."
              (> winum--window-count 1))
     (format winum-format (winum-get-number-string))))
 
+(defcustom acdw-modeline/word-count-modes
+  (mapcar (lambda (m) (cons m nil)) simple-modeline-word-count-modes)
+  "Alist of modes to functions that `acdw-modeline/word-count' should dispatch.
+If the cdr of the cons cell is nil, use the default function (`count-words').
+Otherwise, cdr should be a function that takes two points (see `count-words')."
+  :type '(alist :key-type (symbol :tag "Major-Mode")
+                :value-type function)
+  :group 'simple-modeline)
+
+(defun acdw-modeline/word-count ()
+  "Display a buffer word count, depending on the major mode.
+Uses `acdw-modeline/word-count-modes' to determine which function to use."
+  (when-let ((modefun
+              (assoc major-mode acdw-modeline/word-count-modes #'equal)))
+    (let ((fn (or (cdr modefun)
+                  #'count-words))
+          (min (if (region-active-p) (region-beginning) (point-min)))
+          (max (if (region-active-p) (region-end) (point-max))))
+      (format "%dW" (funcall fn min max)))))
+
 (provide 'acdw-modeline)