Better log-level option

* Allow using strings with `--loglevel`
* `-v` now raises log level relative to `--loglevel` or `--silent`
* Use GmLogLevel instead of Int for parser base
This commit is contained in:
Nikolay Yakimov 2015-12-25 08:19:10 +03:00
parent 5e4026b946
commit 41b9c0bbf2
2 changed files with 47 additions and 15 deletions

View File

@ -26,6 +26,9 @@ import Options.Applicative
import Options.Applicative.Types import Options.Applicative.Types
import Language.Haskell.GhcMod.Types import Language.Haskell.GhcMod.Types
import Control.Arrow import Control.Arrow
import Data.Char (toUpper, toLower)
import Data.List (intercalate)
import Language.Haskell.GhcMod.Read
import GHCMod.Options.Commands import GHCMod.Options.Commands
import GHCMod.Version import GHCMod.Version
import GHCMod.Options.DocUtils import GHCMod.Options.DocUtils
@ -74,31 +77,48 @@ argAndCmdSpec = (,) <$> globalArgSpec <*> commandsSpec
splitOn :: Eq a => a -> [a] -> ([a], [a]) splitOn :: Eq a => a -> [a] -> ([a], [a])
splitOn c = second (drop 1) . break (==c) splitOn c = second (drop 1) . break (==c)
getLogLevel :: Int -> GmLogLevel
getLogLevel = toEnum . min 7
logLevelParser :: Parser GmLogLevel logLevelParser :: Parser GmLogLevel
logLevelParser = logLevelParser =
getLogLevel logLevelSwitch <*>
<$> silentSwitch logLevelOption
<||> logLevelSwitch <||> silentSwitch
<||> logLevelOption
where where
logLevelOption = logLevelOption =
option int option parseLL
$$ long "verbose" $$ long "verbose"
<=> metavar "LEVEL" <=> metavar "LEVEL"
<=> value 4 <=> value GmWarning
<=> showDefault <=> showDefaultWith showLL
<=> help "Set log level. (0-7)" <=> help' $$$ do
"Set log level ("
<> int' (fromEnum (minBound :: GmLogLevel))
<> "-"
<> int' (fromEnum (maxBound :: GmLogLevel))
<> ")"
"You can also use strings (case-insensitive):"
para'
$ intercalate ", "
$ map showLL ([minBound..maxBound] :: [GmLogLevel])
logLevelSwitch = logLevelSwitch =
(4+) . length <$> many $$ flag' () repeatAp succ' . length <$> many $$ flag' ()
$$ short 'v' $$ short 'v'
<=> help "Increase log level" <=> help "Increase log level"
silentSwitch = flag' 0 silentSwitch = flag' GmSilent
$$ long "silent" $$ long "silent"
<=> short 's' <=> short 's'
<=> help "Be silent, set log level to 0" <=> help "Be silent, set log level to 'silent'"
showLL = drop 2 . map toLower . show
repeatAp f n = foldr (.) id (replicate n f)
succ' x | x == maxBound = x
| otherwise = succ x
parseLL = do
v <- readerAsk
let
il'= toEnum . min maxBound <$> readMaybe v
ll' = readMaybe ("Gm" ++ capFirst v)
maybe (readerError $ "Not a log level \"" ++ v ++ "\"") return $ ll' <|> il'
capFirst (h:t) = toUpper h : map toLower t
capFirst [] = []
outputOptsSpec :: Parser OutputOpts outputOptsSpec :: Parser OutputOpts
outputOptsSpec = OutputOpts outputOptsSpec = OutputOpts

View File

@ -23,6 +23,8 @@ import qualified Options.Applicative.Help.Pretty as PP
import Control.Monad.State import Control.Monad.State
import GHC.Exts( IsString(..) ) import GHC.Exts( IsString(..) )
import Data.Maybe import Data.Maybe
import Data.Monoid
import Prelude
newtype MyDocM s a = MyDoc {unwrapState :: State s a} newtype MyDocM s a = MyDoc {unwrapState :: State s a}
deriving (Monad, Functor, Applicative, MonadState s) deriving (Monad, Functor, Applicative, MonadState s)
@ -31,6 +33,10 @@ type MyDoc = MyDocM (Maybe Doc) ()
instance IsString (MyDocM (Maybe Doc) a) where instance IsString (MyDocM (Maybe Doc) a) where
fromString = append . para fromString = append . para
instance Monoid (MyDocM (Maybe Doc) ()) where
mappend a b = append $ doc a <> doc b
mempty = append PP.empty
para :: String -> Doc para :: String -> Doc
para = PP.fillSep . map PP.text . words para = PP.fillSep . map PP.text . words
@ -65,3 +71,9 @@ progDesc' = progDescDoc . Just . doc
indent :: Int -> MyDoc -> MyDoc indent :: Int -> MyDoc -> MyDoc
indent n = append . PP.indent n . doc indent n = append . PP.indent n . doc
int' :: Int -> MyDoc
int' = append . PP.int
para' :: String -> MyDoc
para' = append . para