1 files changed, 85 insertions, 0 deletions
diff --git a/app/RemovePath.hs b/app/RemovePath.hs
new file mode 100644
index 0000000..6f2174b
--- /dev/null
+++ b/app/RemovePath.hs
@@ -0,0 +1,85 @@
+-- |
+-- Copyright   : (c) 2015 Egor Tensin <Egor.Tensin@gmail.com>
+-- License     : MIT
+-- Maintainer  : Egor.Tensin@gmail.com
+-- Stability   : experimental
+-- Portability : Windows-only
+
+module Main (main) where
+
+import Control.Monad   (void, when)
+import Control.Monad.Trans.Except (catchE, runExceptT, throwE)
+import Data.List       ((\\))
+import System.IO.Error (ioError, isDoesNotExistError)
+
+import Options.Applicative
+
+import qualified WindowsEnv
+
+import Utils.Prompt
+import Utils.PromptMessage
+
+data Options = Options
+    { optName   :: WindowsEnv.VarName
+    , optYes    :: Bool
+    , optGlobal :: Bool
+    , optPaths  :: [WindowsEnv.VarValue]
+    } deriving (Eq, Show)
+
+optionParser :: Parser Options
+optionParser = Options
+    <$> optNameDesc
+    <*> optYesDesc
+    <*> optGlobalDesc
+    <*> optPathsDesc
+  where
+    optNameDesc = strOption
+         $ long "name" <> short 'n'
+        <> metavar "NAME" <> value "PATH"
+        <> help "Variable name ('PATH' by default)"
+    optYesDesc = switch
+         $ long "yes" <> short 'y'
+        <> help "Skip confirmation prompt"
+    optGlobalDesc = switch
+         $ long "global" <> short 'g'
+        <> help "Remove for all users"
+    optPathsDesc = many $ argument str
+         $ metavar "PATH"
+        <> help "Directories to remove"
+
+main :: IO ()
+main = execParser parser >>= removePath
+  where
+    parser = info (helper <*> optionParser) $
+        fullDesc <> progDesc "Remove directories from your PATH"
+
+removePath :: Options -> IO ()
+removePath options = runExceptT doRemovePath >>= either ioError return
+  where
+    varName = optName options
+    pathsToRemove = optPaths options
+
+    forAllUsers = optGlobal options
+
+    skipPrompt = optYes options
+
+    emptyIfMissing e
+        | isDoesNotExistError e = return ""
+        | otherwise = throwE e
+
+    doRemovePath = do
+        removePathFrom WindowsEnv.CurrentUser
+        when forAllUsers $
+            removePathFrom WindowsEnv.AllUsers
+
+    removePathFrom profile = do
+        oldValue <- WindowsEnv.query profile varName `catchE` emptyIfMissing
+        let oldPaths = WindowsEnv.pathSplit oldValue
+        let newPaths = oldPaths \\ pathsToRemove
+        when (length oldPaths /= length newPaths) $ do
+            let newValue = WindowsEnv.pathJoin newPaths
+            let promptAnd = if skipPrompt
+                then withoutPrompt
+                else withPrompt $ oldNewMessage profile varName oldValue newValue
+            let engrave = WindowsEnv.engrave profile varName newValue
+            void $ promptAnd engrave