Flyspell: Fix logic in ambrevar/flyspell-and-whitespace-mode

2018-06-23 23:26:23 +02:00 · 2018-06-23 23:26:23 +02:00 · b027b6f2df
parent 7074e26fbe
commit b027b6f2df
1 changed files with 5 additions and 3 deletions
--- a/.emacs.d/lisp/functions.el
+++ b/.emacs.d/lisp/functions.el
@ -80,9 +80,11 @@ selectively."
 (defun ambrevar/flyspell-and-whitespace-mode ()
  "Toggle `flyspell-mode' and `whitespace-mode'."
  (interactive)
-  (if (derived-mode-p 'text-mode)
-      (flyspell-mode)
-    (if flyspell-mode (flyspell-mode 0) (flyspell-prog-mode)))
+  (if (derived-mode-p 'prog-mode)
+      (flyspell-prog-mode)
+    (flyspell-mode)
+    (when flyspell-mode
+      (flyspell-buffer)))
  (whitespace-mode 'toggle))
 (global-set-key (kbd "<f9>") 'flyspell-and-whitespace-mode)