From 7cbe38b01143c7a3bdb4f55b5b5f23692ceba62d Mon Sep 17 00:00:00 2001
From: Arjun Kathuria <arjun.kathuria8@gmail.com>
Date: Tue, 28 Jun 2022 19:50:22 +0530
Subject: [PATCH] Behavior Enhancement: make user press "S" only once to set,
 asks to install AND set if tool uninstalled

---
 app/ghcup/BrickMain.hs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/app/ghcup/BrickMain.hs b/app/ghcup/BrickMain.hs
index 1c75d1d..cb1b83d 100644
--- a/app/ghcup/BrickMain.hs
+++ b/app/ghcup/BrickMain.hs
@@ -515,7 +515,14 @@ set' bs input@(_, ListResult {..}) = do
             (V (NotInstalled tool _)) -> do
               promptAnswer <- getUserPromptResponse userPrompt
               case promptAnswer of
-                PromptYes -> install' bs input
+                PromptYes -> do
+                  res <- install' bs input
+                  case res of
+                    (Left err) -> pure $ Left (prettyShow err)
+                    (Right _) -> do
+                      logInfo "Setting now..."
+                      set' bs input
+
                 PromptNo -> pure $ Left (prettyShow e)
               where
                 userPrompt = L.toStrict $