module Text.ParserCombinators.Poly.Text
  ( -- * The Parser datatype
    Parser(P)
  , Result(..)
  , runParser
    -- ** Basic parsers
  , next
  , eof
  , satisfy
  , onFail
    -- ** Derived parsers (but implemented more efficiently)
  , manySatisfy
  , many1Satisfy
    -- ** Re-parsing
  , reparse
    -- * Re-export all more general combinators
  , module Text.ParserCombinators.Poly.Base
  , module Control.Applicative
  ) where


import Text.ParserCombinators.Poly.Base
import Text.ParserCombinators.Poly.Result
import qualified Data.Text.Lazy as T
import Data.Text.Lazy (Text)
import Control.Applicative
import qualified Control.Monad.Fail as Fail

-- | This @Parser@ datatype is a specialised parsing monad with error
--   reporting.  Whereas the standard version can be used for arbitrary
--   token types, this version is specialised to Text input only.
newtype Parser a = P (Text -> Result Text a)

-- | Apply a parser to an input token sequence.
runParser :: Parser a -> Text -> (Either String a, Text)
runParser :: Parser a -> Text -> (Either String a, Text)
runParser (P p :: Text -> Result Text a
p) = Result Text a -> (Either String a, Text)
forall z a. Result z a -> (Either String a, z)
resultToEither (Result Text a -> (Either String a, Text))
-> (Text -> Result Text a) -> Text -> (Either String a, Text)
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Result Text a
p

instance Functor Parser where
    fmap :: (a -> b) -> Parser a -> Parser b
fmap f :: a -> b
f (P p :: Text -> Result Text a
p) = (Text -> Result Text b) -> Parser b
forall a. (Text -> Result Text a) -> Parser a
P ((a -> b) -> Result Text a -> Result Text b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap a -> b
f (Result Text a -> Result Text b)
-> (Text -> Result Text a) -> Text -> Result Text b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Result Text a
p)

instance Monad Parser where
    return :: a -> Parser a
return       = a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure
    (P f :: Text -> Result Text a
f) >>= :: Parser a -> (a -> Parser b) -> Parser b
>>= g :: a -> Parser b
g  = (Text -> Result Text b) -> Parser b
forall a. (Text -> Result Text a) -> Parser a
P (Result Text a -> Result Text b
continue (Result Text a -> Result Text b)
-> (Text -> Result Text a) -> Text -> Result Text b
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Result Text a
f)
      where
        continue :: Result Text a -> Result Text b
continue (Success ts :: Text
ts x :: a
x)             = let (P g' :: Text -> Result Text b
g') = a -> Parser b
g a
x in Text -> Result Text b
g' Text
ts
        continue (Committed r :: Result Text a
r)              = Result Text b -> Result Text b
forall z a. Result z a -> Result z a
Committed (Result Text a -> Result Text b
continue Result Text a
r)
        continue (Failure ts :: Text
ts e :: String
e)             = Text -> String -> Result Text b
forall z a. z -> String -> Result z a
Failure Text
ts String
e

#if !MIN_VERSION_base(4,13,0)
    fail         = Fail.fail
#endif

instance Fail.MonadFail Parser where
    fail :: String -> Parser a
fail e :: String
e       = (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (\ts :: Text
ts-> Text -> String -> Result Text a
forall z a. z -> String -> Result z a
Failure Text
ts String
e)

instance Commitment Parser where
    commit :: Parser a -> Parser a
commit (P p :: Text -> Result Text a
p)         = (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (Result Text a -> Result Text a
forall z a. Result z a -> Result z a
Committed (Result Text a -> Result Text a)
-> (Text -> Result Text a) -> Text -> Result Text a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Result Text a -> Result Text a
forall z a. Result z a -> Result z a
squash (Result Text a -> Result Text a)
-> (Text -> Result Text a) -> Text -> Result Text a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Result Text a
p)
      where
        squash :: Result z a -> Result z a
squash (Committed r :: Result z a
r) = Result z a -> Result z a
squash Result z a
r
        squash r :: Result z a
r             = Result z a
r
    (P p :: Text -> Result Text a
p) adjustErr :: Parser a -> (String -> String) -> Parser a
`adjustErr` f :: String -> String
f  = (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (Result Text a -> Result Text a
forall z a. Result z a -> Result z a
adjust (Result Text a -> Result Text a)
-> (Text -> Result Text a) -> Text -> Result Text a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Result Text a
p)
      where
        adjust :: Result z a -> Result z a
adjust (Failure z :: z
z e :: String
e) = z -> String -> Result z a
forall z a. z -> String -> Result z a
Failure z
z (String -> String
f String
e)
        adjust (Committed r :: Result z a
r) = Result z a -> Result z a
forall z a. Result z a -> Result z a
Committed (Result z a -> Result z a
adjust Result z a
r)
        adjust  good :: Result z a
good         = Result z a
good

    oneOf' :: [(String, Parser a)] -> Parser a
oneOf' = [(String, String)] -> [(String, Parser a)] -> Parser a
forall a. [(String, String)] -> [(String, Parser a)] -> Parser a
accum []
      where accum :: [(String, String)] -> [(String, Parser a)] -> Parser a
accum errs :: [(String, String)]
errs [] =
                String -> Parser a
forall (m :: * -> *) a. MonadFail m => String -> m a
fail ("failed to parse any of the possible choices:\n"
                            String -> String -> String
forall a. [a] -> [a] -> [a]
++Int -> String -> String
indent 2 (((String, String) -> String) -> [(String, String)] -> String
forall (t :: * -> *) a b. Foldable t => (a -> [b]) -> t a -> [b]
concatMap (String, String) -> String
showErr ([(String, String)] -> [(String, String)]
forall a. [a] -> [a]
reverse [(String, String)]
errs)))
            accum errs :: [(String, String)]
errs ((e :: String
e,P p :: Text -> Result Text a
p):ps :: [(String, Parser a)]
ps) =
                (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (\ts :: Text
ts-> case Text -> Result Text a
p Text
ts of
                           Failure _ err :: String
err ->
                                       let (P p' :: Text -> Result Text a
p') = [(String, String)] -> [(String, Parser a)] -> Parser a
accum ((String
e,String
err)(String, String) -> [(String, String)] -> [(String, String)]
forall a. a -> [a] -> [a]
:[(String, String)]
errs) [(String, Parser a)]
ps
                                       in Text -> Result Text a
p' Text
ts
                           r :: Result Text a
r@(Success _ _)    -> Result Text a
r
                           r :: Result Text a
r@(Committed _)    -> Result Text a
r )
            showErr :: (String, String) -> String
showErr (name :: String
name,err :: String
err) = String
nameString -> String -> String
forall a. [a] -> [a] -> [a]
++":\n"String -> String -> String
forall a. [a] -> [a] -> [a]
++Int -> String -> String
indent 2 String
err

instance Applicative Parser where
    pure :: a -> Parser a
pure x :: a
x    = (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (\ts :: Text
ts-> Text -> a -> Result Text a
forall z a. z -> a -> Result z a
Success Text
ts a
x)
    pf :: Parser (a -> b)
pf <*> :: Parser (a -> b) -> Parser a -> Parser b
<*> px :: Parser a
px = do { a -> b
f <- Parser (a -> b)
pf; a
x <- Parser a
px; b -> Parser b
forall (m :: * -> *) a. Monad m => a -> m a
return (a -> b
f a
x) }
#if defined(GLASGOW_HASKELL) && GLASGOW_HASKELL > 610
    p  <*  q  = p `discard` q
#endif

instance Alternative Parser where
    empty :: Parser a
empty     = String -> Parser a
forall (m :: * -> *) a. MonadFail m => String -> m a
fail "no parse"
    p :: Parser a
p <|> :: Parser a -> Parser a -> Parser a
<|> q :: Parser a
q   = Parser a
p Parser a -> Parser a -> Parser a
forall a. Parser a -> Parser a -> Parser a
`onFail` Parser a
q

instance PolyParse Parser

------------------------------------------------------------------------

-- | Simply return the next token in the input tokenstream.
next :: Parser Char
next :: Parser Char
next = (Text -> Result Text Char) -> Parser Char
forall a. (Text -> Result Text a) -> Parser a
P (\bs :: Text
bs-> case Text -> Maybe (Char, Text)
T.uncons Text
bs of
                Nothing       -> Text -> String -> Result Text Char
forall z a. z -> String -> Result z a
Failure Text
bs "Ran out of input (EOF)"
                Just (c :: Char
c, bs' :: Text
bs') -> Text -> Char -> Result Text Char
forall z a. z -> a -> Result z a
Success Text
bs' Char
c )

-- | Succeed if the end of file/input has been reached, fail otherwise.
eof :: Parser ()
eof :: Parser ()
eof = (Text -> Result Text ()) -> Parser ()
forall a. (Text -> Result Text a) -> Parser a
P (\bs :: Text
bs -> if Text -> Bool
T.null Text
bs
                then Text -> () -> Result Text ()
forall z a. z -> a -> Result z a
Success Text
bs ()
                else Text -> String -> Result Text ()
forall z a. z -> String -> Result z a
Failure Text
bs "Expected end of input (EOF)" )

-- | Return the next token if it satisfies the given predicate.
satisfy :: (Char -> Bool) -> Parser Char
satisfy :: (Char -> Bool) -> Parser Char
satisfy f :: Char -> Bool
f = do { Char
x <- Parser Char
next
               ; if Char -> Bool
f Char
x then Char -> Parser Char
forall (m :: * -> *) a. Monad m => a -> m a
return Char
x else String -> Parser Char
forall (m :: * -> *) a. MonadFail m => String -> m a
fail "Parse.satisfy: failed"
               }

-- | @p `onFail` q@ means parse p, unless p fails, in which case
--   parse q instead.
--   Can be chained together to give multiple attempts to parse something.
--   (Note that q could itself be a failing parser, e.g. to change the error
--   message from that defined in p to something different.)
--   However, a severe failure in p cannot be ignored.
onFail :: Parser a -> Parser a -> Parser a
(P p :: Text -> Result Text a
p) onFail :: Parser a -> Parser a -> Parser a
`onFail` (P q :: Text -> Result Text a
q) = (Text -> Result Text a) -> Parser a
forall a. (Text -> Result Text a) -> Parser a
P (\ts :: Text
ts-> Text -> Result Text a -> Result Text a
continue Text
ts (Result Text a -> Result Text a) -> Result Text a -> Result Text a
forall a b. (a -> b) -> a -> b
$ Text -> Result Text a
p Text
ts)
  where continue :: Text -> Result Text a -> Result Text a
continue ts :: Text
ts (Failure _ _) = Text -> Result Text a
q Text
ts
    --  continue _  (Committed r) = r	-- no, remain Committed
        continue _  r :: Result Text a
r             = Result Text a
r

------------------------------------------------------------------------

-- | @manySatisfy p@ is a more efficient fused version of @many (satisfy p)@
manySatisfy :: (Char->Bool) -> Parser Text
manySatisfy :: (Char -> Bool) -> Parser Text
manySatisfy f :: Char -> Bool
f = (Text -> Result Text Text) -> Parser Text
forall a. (Text -> Result Text a) -> Parser a
P (\bs :: Text
bs-> let (pre :: Text
pre,suf :: Text
suf) = (Char -> Bool) -> Text -> (Text, Text)
T.span Char -> Bool
f Text
bs in Text -> Text -> Result Text Text
forall z a. z -> a -> Result z a
Success Text
suf Text
pre)

-- | @many1Satisfy p@ is a more efficient fused version of @many1 (satisfy p)@
many1Satisfy :: (Char->Bool) -> Parser Text
many1Satisfy :: (Char -> Bool) -> Parser Text
many1Satisfy f :: Char -> Bool
f = do Text
x <- (Char -> Bool) -> Parser Text
manySatisfy Char -> Bool
f
                    if Text -> Bool
T.null Text
x then String -> Parser Text
forall (m :: * -> *) a. MonadFail m => String -> m a
fail "Parse.many1Satisfy: failed"
                                else Text -> Parser Text
forall (m :: * -> *) a. Monad m => a -> m a
return Text
x

------------------------------------------------------------------------

-- | Push some tokens back onto the front of the input stream and reparse.
--   This is useful e.g. for recursively expanding macros.  When the
--   user-parser recognises a macro use, it can lookup the macro
--   expansion from the parse state, lex it, and then stuff the
--   lexed expansion back down into the parser.
reparse    :: Text -> Parser ()
reparse :: Text -> Parser ()
reparse ts :: Text
ts  = (Text -> Result Text ()) -> Parser ()
forall a. (Text -> Result Text a) -> Parser a
P (\inp :: Text
inp-> Text -> () -> Result Text ()
forall z a. z -> a -> Result z a
Success (Text
ts Text -> Text -> Text
`T.append` Text
inp) ())

------------------------------------------------------------------------