blob: 120dac6ab080501674f17c3233f11ecca47f531c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
-----------------------------------------------------------------------------
-- |
-- Module : XMobar.Environment
-- Copyright : (c) William Song
-- License : BSD-style (see LICENSE)
--
-- Maintainer : Will Song <incertia@incertia.net>
-- Stability : stable
-- Portability : portable
--
-- A function to expand environment variables in strings
--
-----------------------------------------------------------------------------
module Environment where
import Control.Applicative ((<$>))
import Data.Maybe (fromMaybe)
import System.Environment (lookupEnv)
expandEnv :: String -> IO String
expandEnv "" = return ""
expandEnv (c:s) = case c of
'$' -> do
envVar <- fromMaybe "" <$> lookupEnv e
remainder <- expandEnv s'
return $ envVar ++ remainder
where (e, s') = getVar s
getVar "" = ("", "")
getVar ('{':s'') = (takeUntil "}" s'', drop 1 . dropUntil "}" $ s'')
getVar s'' = (takeUntil filterstr s'', dropUntil filterstr s'')
filterstr = ",./? \t;:\"'~`!@#$%^&*()<>-+=\\|"
takeUntil f = takeWhile (not . flip elem f)
dropUntil f = dropWhile (not . flip elem f)
'\\' -> case s == "" of
True -> return "\\"
False -> do
remainder <- expandEnv $ drop 1 s
return $ escString s ++ remainder
where escString s' = let (cc:_) = s' in
case cc of
't' -> "\t"
'n' -> "\n"
'$' -> "$"
_ -> [cc]
_ -> do
remainder <- expandEnv s
return $ c : remainder
|
