Add special set function so that customizing display options will take effect immediately

memento-mori.el

@@ -6,7 +6,7 @@
 ;; Maintainer: Ivan Andrus <[email protected]>
 ;; URL: https://github.com/gvol/emacs-memento-mori
 ;; Version: 0.3.1
-;; Package-Requires: ((emacs "24.3"))
+;; Package-Requires: ((emacs "24.4"))
 ;; Keywords: help
 ;; SPDX-License-Identifier: ISC
 
@@ -53,17 +53,21 @@
   :group 'help)
 
 (defcustom memento-mori-display-in-modeline t
   "If non-nil, `memento-mori-mode' will add mementos to the mode line.
 Really, it adds it to the `global-mode-string', which usually
 appears in the mode line, however, if you have customized your
 mode line it may not appear."
   :group 'memento-mori
-  :type 'boolean)
+  :type 'boolean
+  :set #'memento-mori--set-value-and-update-display
+  :initialize #'custom-initialize-default)
 
 (defcustom memento-mori-display-in-frame-title nil
   "If non-nil, `memento-mori-mode' will add mementos to the frame title."
   :group 'memento-mori
-  :type 'boolean)
+  :type 'boolean
+  :set #'memento-mori--set-value-and-update-display
+  :initialize #'custom-initialize-default)
 
 (defcustom memento-mori-initial-scratch-message nil
   "If non-nil, overwrite the value of `initial-scratch-message' with this.
@@ -73,6 +77,7 @@
 
 Ultimately, it will be displayed as the initial documentation in
 *scratch* buffer at startup."
+  :group 'memento-mori
   :type '(choice (text :tag "Scratch message format")
 		         (const :tag "none" nil)))
 
@@ -82,7 +87,9 @@
 To use `memento-mori-mementos' customize it, or erase customization of
 `memento-mori-birth-date'."
   :group 'memento-mori
-  :type 'string)
+  :type 'string
+  :set #'memento-mori--set-value-and-update
+  :initialize #'custom-initialize-default)
 (make-obsolete-variable 'memento-mori-birth-date
                         'memento-mori-mementos "0.2.1")
 
@@ -162,7 +169,9 @@
                   (plist :key-type symbol
                          :options ((:until string :tag "Date")
                                    (:since string) (:formula function))
-                         :value-type string))))
+                         :value-type string)))
+  :set #'memento-mori--set-value-and-update
+  :initialize #'custom-initialize-default)
 
 (define-obsolete-variable-alias 'memento-mori-age-string
   'memento-mori-string "0.2.0")
@@ -247,10 +256,20 @@
 (defvar memento-mori-string ""
   "The memento mori memento.
 The string shown in the mode line and/or frame title when
 `memento-mori-mode' is enabled.  This is not meant to be changed
 by the user, but can be used in other places such as org-mode
 agendas to display the current momento.")
 
+(defun memento-mori--set-value-and-update-display (symbol value)
+  "Function to set display values, so that the display will then be recalculated."
+  (set-default-toplevel-value symbol value)
+  (memento-mori--add-mementos))
+
+(defun memento-mori--set-value-and-update (symbol value)
+  "Function to set memento config, so that a new memento will be chosen."
+  (set-default-toplevel-value symbol value)
+  (memento-mori--update))
+
 (defvar memento-mori--modeline-construct
   `(memento-mori-mode
     ((:propertize
@@ -280,7 +299,7 @@
     (error "Birth date not set.  Try M-x customize-group memento-mori")))
 
 (defun memento-mori--parse-time (value)
   "Return your birth time in `encode-time' format.
 The birth time is parsed from `memento-mori-birth-date' using
 `parse-time-string'.  An error is signaled if it is not valid."
   (when (symbolp value)
@@ -314,7 +333,7 @@
   "Randomly choose a quote from `memento-mori-quotes'."
   (nth (random (length memento-mori-quotes)) memento-mori-quotes))
 
 (defun memento-mori--format-memento (memento)
   "Format MEMENTO based on the current time."
   (let* ((format-string (car memento))
          (args-plist (cdr memento))
@@ -322,7 +341,7 @@
          (since-value (plist-get args-plist :since))
          (_ (when (not (or until-value since-value))
                  (error ":since or :until required.  Parsing %s" memento)))
          (current-time-list nil)
          (until-seconds (when until-value
                           (float-time
                            (time-subtract
@@ -353,7 +372,7 @@
                                      (/ since-seconds (* 60 60 24))
                                      days))
                            (t 0)))))
     (format-spec format-string
                  `((?Y . ,years)
                    (?y . ,(truncate years))
                    (?M . ,months)
@@ -392,13 +411,13 @@
   (when (and memento-mori-initial-scratch-message
              (boundp 'initial-scratch-message))
     (setq initial-scratch-message
           (format-spec memento-mori-initial-scratch-message
                        `((?m . ,memento-mori-string)
                          (?q . ,(memento-mori--random-quote)))
                        'ignore))))
 
 (defun memento-mori--add-mementos ()
   "Adds constructs to mode line and/or frame title to display mementos.
 Where it adds it is controlled by `memento-mori-display-in-modeline' and
 `memento-mori-display-in-frame-title' respectively.