diff options
Diffstat (limited to '')
| -rw-r--r-- | apps/UnsetEnv.hs | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/apps/UnsetEnv.hs b/apps/UnsetEnv.hs index eebef00..767c681 100644 --- a/apps/UnsetEnv.hs +++ b/apps/UnsetEnv.hs @@ -11,8 +11,8 @@ import Control.Monad (void) import Options.Applicative import qualified Windows.Environment as Env -import Banner import Prompt +import PromptMessage data Options = Options { optYes :: Bool @@ -20,13 +20,13 @@ data Options = Options , optName :: Env.VarName } deriving (Eq, Show) -options :: Parser Options -options = Options - <$> optYes +optionParser :: Parser Options +optionParser = Options + <$> optYesDesc <*> optGlobalDesc <*> optNameDesc where - optYes = switch $ + optYesDesc = switch $ long "yes" <> short 'y' <> help "Skip confirmation prompt" optGlobalDesc = switch $ @@ -39,14 +39,12 @@ options = Options main :: IO () main = execParser parser >>= unsetEnv where - parser = info (helper <*> options) $ + parser = info (helper <*> optionParser) $ fullDesc <> progDesc "Unset environment variable" unsetEnv :: Options -> IO () -unsetEnv options = void $ prompt banner $ Env.wipe profile varName +unsetEnv options = void $ promptAnd wipe where - banner = wipeBanner profile varName - varName = optName options forAllUsers = optGlobal options @@ -55,6 +53,8 @@ unsetEnv options = void $ prompt banner $ Env.wipe profile varName | otherwise = Env.CurrentUser skipPrompt = optYes options - prompt - | skipPrompt = const withoutPrompt - | otherwise = withPrompt + promptAnd + | skipPrompt = withoutPrompt + | otherwise = withPrompt $ wipeMessage profile varName + + wipe = Env.wipe profile varName |