Correct with-eval-after-loads

From this reddit comment:
https://old.reddit.com/r/emacs/comments/pxe9pr/need_help_writing_a_macro/hemygul/

1 files changed, 12 insertions, 4 deletions

@@ -80,12 +80,20 @@ When not specified, ELLIPSIS defaults to '...'."
                 ,@forms))
        (add-function :after after-focus-change-function #',func-name))))
 
-(defmacro with-eval-after-loads (files &rest body)
-  "Execute BODY after FILES are loaded.
+(defmacro with-eval-after-loads (features &rest body)
+  "Execute BODY after FEATURES are loaded.
 This macro simplifies `with-eval-after-load' for multiple nested
 features."
-  (declare (indent 1) (debug (form def-body)))
-  (waterfall-list 'with-eval-after-load files body))
+  (declare (indent 1)
+           (debug (form def-body)))
+  (unless (listp features)
+    (setq features (list features)))
+  (if (null features)
+      (macroexp-progn body)
+    (let* ((this (car features))
+           (rest (cdr features)))
+      `(with-eval-after-load ',this
+         (with-eval-after-loads ,rest ,@body)))))
 
 (defmacro with-message (message &rest body)
   "Execute BODY, messaging 'MESSAGE...' before and 'MESSAGE... Done.' after."