From 66219f23bbb4293b4ad3377a0086c475f78b7630 Mon Sep 17 00:00:00 2001
From: Robert Helgesson <robert@rycee.net>
Date: Sun, 12 Nov 2017 14:07:41 +0100
Subject: [PATCH] home-manager: add 'remove-generations' command

This command allows the user to immediately remove specific
generations from the profiles directory.
---
 home-manager/home-manager | 49 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/home-manager/home-manager b/home-manager/home-manager
index 2e3f2dd90..0b13a84c5 100644
--- a/home-manager/home-manager
+++ b/home-manager/home-manager
@@ -163,6 +163,39 @@ function doListGens() {
     popd > /dev/null
 }
 
+# Removes linked generations. Takes as arguments identifiers of
+# generations to remove.
+function doRmGenerations() {
+    if [[ -v VERBOSE ]]; then
+        export VERBOSE_ARG="--verbose"
+    else
+        export VERBOSE_ARG=""
+    fi
+
+    if [[ -v DRY_RUN ]] ; then
+        export DRY_RUN_CMD=echo
+    else
+        export DRY_RUN_CMD=""
+    fi
+
+    pushd "/nix/var/nix/profiles/per-user/$USER" > /dev/null
+
+    for generationId in "$@"; do
+        local linkName="home-manager-$generationId-link"
+
+        if [[ ! -e $linkName ]]; then
+            errorEcho "No generation with ID $generationId"
+        elif [[ $linkName == $(readlink home-manager) ]]; then
+            errorEcho "Cannot remove the current generation $generationId"
+        else
+            echo Removing generation $generationId
+            $DRY_RUN_CMD rm $VERBOSE_ARG $linkName
+        fi
+    done
+
+    popd > /dev/null
+}
+
 function doListPackages() {
     local outPath
     outPath="$(nix-env -q --out-path | grep -o '/.*home-manager-path$')"
@@ -242,11 +275,21 @@ function doHelp() {
     echo "  -h           Print this help"
     echo
     echo "Commands"
+    echo
     echo "  help         Print this help"
+    echo
     echo "  build        Build configuration into result directory"
+    echo
     echo "  switch       Build and activate configuration"
+    echo
     echo "  generations  List all home environment generations"
+    echo
+    echo "  remove-generations ID..."
+    echo "      Remove indicated generations. Use 'generations' command to"
+    echo "      find suitable generation numbers."
+    echo
     echo "  packages     List all packages installed in home-manager-path"
+    echo
     echo "  news         Show news entries in a pager"
 }
 
@@ -285,7 +328,8 @@ done
 # Get rid of the options.
 shift "$((OPTIND-1))"
 
-cmd="$*"
+cmd="$1"
+shift 1
 
 case "$cmd" in
     build)
@@ -297,6 +341,9 @@ case "$cmd" in
     generations)
         doListGens
         ;;
+    remove-generations)
+        doRmGenerations "$@"
+        ;;
     packages)
         doListPackages
         ;;