From 07f134b10ecc374b58ead2afbf702a2a8b2ceeb8 Mon Sep 17 00:00:00 2001
From: Henrik Lissner <henrik@lissner.net>
Date: Sun, 31 Jan 2021 04:27:41 -0500
Subject: [PATCH] doom purge: delete old ELN directories

---
 core/cli/packages.el | 29 ++++++++++++++++++++++++++---
 core/cli/sync.el     |  2 +-
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/core/cli/packages.el b/core/cli/packages.el
index 4c3f617e1..53258ed06 100644
--- a/core/cli/packages.el
+++ b/core/cli/packages.el
@@ -24,6 +24,7 @@ Emacs (as byte-code is generally not forward-compatible)."
     ((nobuilds-p ["-b" "--no-builds"] "Don't purge unneeded (built) packages")
      (noelpa-p   ["-p" "--no-elpa"]   "Don't purge ELPA packages")
      (norepos-p  ["-r" "--no-repos"]  "Don't purge unused straight repos")
+     (noeln-p    ["-e" "--no-eln"]    "Don't purge old ELN bytecode")
      (regraft-p  ["-g" "--regraft"]   "Regraft git repos (ie. compact them)"))
   "Deletes orphaned packages & repos, and compacts them.
 
@@ -39,7 +40,8 @@ list remains lean."
          (not noelpa-p)
          (not norepos-p)
          (not nobuilds-p)
-         regraft-p)
+         regraft-p
+         (not noeln-p))
     (doom-autoloads-reload))
   t)
 
@@ -532,7 +534,23 @@ declaration) or dependency thereof that hasn't already been."
                    (filename path)
                    e)))))))
 
-(defun doom-cli-packages-purge (&optional elpa-p builds-p repos-p regraft-repos-p)
+(defun doom--cli-packages-purge-eln ()
+  (if-let (dirs
+           (cl-delete (expand-file-name comp-native-version-dir doom--eln-output-path)
+                      (directory-files doom--eln-output-path t "^[^.]" t)
+                      :test #'file-equal-p))
+      (progn
+        (print! (start "Purging old native bytecode..."))
+        (print-group!
+         (dolist (dir dirs)
+           (print! (info "Deleting %S") (relpath dir doom--eln-output-path))
+           (delete-directory dir 'recursive))
+         (print! (success "Purged %d directory(ies)" (length dirs))))
+        (length dirs))
+    (print! (info "No ELN directories to purge"))
+    0))
+
+(defun doom-cli-packages-purge (&optional elpa-p builds-p repos-p regraft-repos-p eln-p)
   "Auto-removes orphaned packages and repos.
 
 An orphaned package is a package that isn't a primary package (i.e. doesn't have
@@ -574,4 +592,9 @@ If ELPA-P, include packages installed with package.el (M-x package-install)."
              (/= 0 (doom--cli-packages-purge-repos repos-to-purge)))
            (if (not regraft-repos-p)
                (ignore (print! (info "Skipping regrafting")))
-             (doom--cli-packages-regraft-repos repos-to-regraft)))))))
+             (doom--cli-packages-regraft-repos repos-to-regraft))
+           (when (require 'comp nil t)
+             (if (not eln-p)
+                 (ignore (print! (info "Skipping native bytecode")))
+               (doom--cli-packages-purge-eln)))))
+     (print! (success "Finished purging")))))
diff --git a/core/cli/sync.el b/core/cli/sync.el
index efc6055c2..8af3ca4c2 100644
--- a/core/cli/sync.el
+++ b/core/cli/sync.el
@@ -31,7 +31,7 @@ stale."
        (doom-cli-packages-build)
        (when update-p
          (doom-cli-packages-update))
-       (doom-cli-packages-purge purge-p 'builds-p purge-p purge-p)
+       (doom-cli-packages-purge purge-p 'builds-p purge-p purge-p purge-p)
        (run-hooks 'doom-sync-post-hook)
        (when (doom-autoloads-reload)
          (print! (info "Restart Emacs or use 'M-x doom/reload' for changes to take effect")))