aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/bin/AddPath.hs
diff options
context:
space:
mode:
Diffstat (limited to 'bin/AddPath.hs')
-rw-r--r--bin/AddPath.hs83
1 files changed, 83 insertions, 0 deletions
diff --git a/bin/AddPath.hs b/bin/AddPath.hs
new file mode 100644
index 0000000..73d86d0
--- /dev/null
+++ b/bin/AddPath.hs
@@ -0,0 +1,83 @@
+-- |
+-- 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 (union)
+import System.IO.Error (ioError, isDoesNotExistError)
+
+import Options.Applicative
+
+import qualified WindowsEnv
+
+import Prompt
+import 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 "Add for all users"
+ optPathsDesc = many $ argument str
+ $ metavar "PATH"
+ <> help "Directories to add"
+
+main :: IO ()
+main = execParser parser >>= addPath
+ where
+ parser = info (helper <*> optionParser) $
+ fullDesc <> progDesc "Add directories to your PATH"
+
+addPath :: Options -> IO ()
+addPath options = runExceptT doAddPath >>= either ioError return
+ where
+ varName = optName options
+ pathsToAdd = optPaths options
+
+ forAllUsers = optGlobal options
+ profile
+ | forAllUsers = WindowsEnv.AllUsers
+ | otherwise = WindowsEnv.CurrentUser
+
+ skipPrompt = optYes options
+
+ emptyIfMissing e
+ | isDoesNotExistError e = return ""
+ | otherwise = throwE e
+
+ doAddPath = do
+ oldValue <- WindowsEnv.query profile varName `catchE` emptyIfMissing
+ let oldPaths = WindowsEnv.pathSplit oldValue
+ let newPaths = oldPaths `union` pathsToAdd
+ 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