From 35046566a1fa9cbe5067668ac3722320e76c7026 Mon Sep 17 00:00:00 2001
From: Arsen Musayelyan <arsen@arsenm.dev>
Date: Fri, 30 Sep 2022 15:52:46 -0700
Subject: [PATCH] If build dependencies exist, prompt to remove after build

---
 build.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/build.go b/build.go
index c0c362d..31fe6e8 100644
--- a/build.go
+++ b/build.go
@@ -28,6 +28,7 @@ import (
 	"strconv"
 	"strings"
 
+	"github.com/AlecAivazis/survey/v2"
 	_ "github.com/goreleaser/nfpm/v2/apk"
 	_ "github.com/goreleaser/nfpm/v2/arch"
 	_ "github.com/goreleaser/nfpm/v2/deb"
@@ -359,6 +360,23 @@ func buildPackage(ctx context.Context, script string, mgr manager.Manager) ([]st
 		return nil, nil, err
 	}
 
+	if len(vars.BuildDepends) > 0 {
+		var removeBuildDeps bool
+		err = survey.AskOne(&survey.Confirm{
+			Message: "Would you like to remove build dependencies?",
+		}, &removeBuildDeps)
+		if err != nil {
+			return nil, nil, err
+		}
+
+		if removeBuildDeps {
+			err = mgr.Remove(vars.BuildDepends...)
+			if err != nil {
+				return nil, nil, err
+			}
+		}
+	}
+
 	uniq(&pkgPaths, &pkgNames)
 
 	return pkgPaths, pkgNames, nil