ilyakooo0/optparse-applicative
1
0
Fork 0
mirror of https://github.com/ilyakooo0/optparse-applicative.git synced 2024-11-30 14:03:28 +03:00
Code Issues Projects Releases Wiki Activity

Add --bash-completion-script option.

Browse Source
This commit is contained in:
Paolo Capriotti 2012-08-01 20:24:47 +01:00
parent ff7f37c518
commit 5620a636c6
1 changed files with 26 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
Options/Applicative/BashCompletion.hs
View File

@ -4,6 +4,7 @@ module Options.Applicative.BashCompletion
) where
import Control.Applicative
import Data.Foldable (asum)
import Data.List
import System.Exit
@ -19,10 +20,15 @@ bashCompletionParser parser = complParser
{ errMessage = \_ -> unlines opts
, errExitCode = ExitSuccess }
complParser = failure <$>
( bashCompletionQuery parser
<$> (many . strOption) (long "bash-completion-word")
<*> option (long "bash-completion-index") )
complParser = asum
[ failure <$>
( bashCompletionQuery parser
<$> (many . strOption) (long "bash-completion-word")
<*> option (long "bash-completion-index") )
, ParserFailure
<$> (bashCompletionScript <$>
strOption (long "bash-completion-script"))
<*> pure ExitSuccess ]
bashCompletionQuery :: Parser a -> [String] -> Int -> [String]
bashCompletionQuery parser ws i = case runCompletion compl parser of
@ -44,3 +50,19 @@ bashCompletionQuery parser ws i = case runCompletion compl parser of
compl = do
setParser Nothing parser
runParserFully parser (drop 1 ws')
bashCompletionScript :: String -> String -> String
bashCompletionScript prog progn = unlines
[ "_" ++ progn ++ "()"
, "{"
, " local cmdline"
, " CMDLINE=(--bash-completion-index $COMP_CWORD)"
, ""
, " for arg in ${COMP_WORDS[@]}; do"
, " CMDLINE=(${CMDLINE[@]} --bash-completion-word $arg)"
, " done"
, ""
, " COMPREPLY=( $(" ++ prog ++ " \"${CMDLINE[@]}\") )"
, "}"
, ""
, "complete -F _" ++ progn ++ " " ++ progn ]