{-# LANGUAGE DeriveLift #-}
{-# LANGUAGE DerivingStrategies #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TupleSections #-}
{-# LANGUAGE CPP #-}

module Database.Persist.Quasi.Internal.ModelParser
    ( SourceLoc (..)
    , Token (..)
    , tokenContent
    , anyToken
    , ParsedEntityDef
    , parsedEntityDefComments
    , parsedEntityDefEntityName
    , parsedEntityDefIsSum
    , parsedEntityDefEntityAttributes
    , parsedEntityDefFieldAttributes
    , parsedEntityDefExtras
    , parsedEntityDefSpan
    , parseSource
    , memberBlockAttrs
    , ParserWarning
    , parserWarningMessage
    , ParseResult
    , CumulativeParseResult
    , toCumulativeParseResult
    , renderErrors
    , runConfiguredParser
    , ParserErrorLevel (..)
    , initialExtraState
    ) where

import Control.Applicative (Alternative)
import Control.Monad (MonadPlus, mzero, void)
import Control.Monad.Reader (MonadReader, ReaderT, asks, runReaderT)
import Control.Monad.State
import Control.Monad.Writer
import Data.Char (isSpace)
import Data.Either (partitionEithers)
import Data.Foldable (fold)
import Data.Functor.Identity
import Data.List (find, intercalate)
import Data.List.NonEmpty (NonEmpty (..))
import qualified Data.List.NonEmpty as NEL
import qualified Data.Map as M
import Data.Maybe
import Data.Set (Set)
import qualified Data.Set as Set
import Data.Text (Text)
import qualified Data.Text as Text
import Data.Void
import Database.Persist.Quasi.PersistSettings.Internal
import Database.Persist.Types
import Database.Persist.Types.SourceSpan
import Language.Haskell.TH.Syntax (Lift)
import Text.Megaparsec hiding (Token)
import Text.Megaparsec.Char
import qualified Text.Megaparsec.Char.Lexer as L
import qualified Text.Megaparsec.Stream as TMS

-- We'll augment the parser with extra state to accumulate comments seen during parsing.
-- Comments are lexed as whitespace, but will be used to generate documentation later.
data ExtraState = ExtraState
    { ExtraState -> [(SourcePos, CommentToken)]
esPositionedCommentTokens :: [(SourcePos, CommentToken)]
    , ExtraState -> Maybe SourcePos
esLastDocumentablePosition :: Maybe SourcePos
    }

-- @since 2.16.0.0
initialExtraState :: ExtraState
initialExtraState :: ExtraState
initialExtraState =
    ExtraState
        { esPositionedCommentTokens :: [(SourcePos, CommentToken)]
esPositionedCommentTokens = []
        , esLastDocumentablePosition :: Maybe SourcePos
esLastDocumentablePosition = Maybe SourcePos
forall a. Maybe a
Nothing
        }

newtype Parser a = Parser
    { forall a.
Parser a
-> ReaderT
     PersistSettings
     (StateT
        ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
     a
unParser
        :: ReaderT
            PersistSettings
            ( StateT
                ExtraState
                ( ParsecT
                    Void
                    String
                    ( Writer
                        (Set ParserWarning)
                    )
                )
            )
            a
    }
    deriving newtype
        ( (forall a b. (a -> b) -> Parser a -> Parser b)
-> (forall a b. a -> Parser b -> Parser a) -> Functor Parser
forall a b. a -> Parser b -> Parser a
forall a b. (a -> b) -> Parser a -> Parser b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
$cfmap :: forall a b. (a -> b) -> Parser a -> Parser b
fmap :: forall a b. (a -> b) -> Parser a -> Parser b
$c<$ :: forall a b. a -> Parser b -> Parser a
<$ :: forall a b. a -> Parser b -> Parser a
Functor
        , Functor Parser
Functor Parser =>
(forall a. a -> Parser a)
-> (forall a b. Parser (a -> b) -> Parser a -> Parser b)
-> (forall a b c.
    (a -> b -> c) -> Parser a -> Parser b -> Parser c)
-> (forall a b. Parser a -> Parser b -> Parser b)
-> (forall a b. Parser a -> Parser b -> Parser a)
-> Applicative Parser
forall a. a -> Parser a
forall a b. Parser a -> Parser b -> Parser a
forall a b. Parser a -> Parser b -> Parser b
forall a b. Parser (a -> b) -> Parser a -> Parser b
forall a b c. (a -> b -> c) -> Parser a -> Parser b -> Parser c
forall (f :: * -> *).
Functor f =>
(forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
$cpure :: forall a. a -> Parser a
pure :: forall a. a -> Parser a
$c<*> :: forall a b. Parser (a -> b) -> Parser a -> Parser b
<*> :: forall a b. Parser (a -> b) -> Parser a -> Parser b
$cliftA2 :: forall a b c. (a -> b -> c) -> Parser a -> Parser b -> Parser c
liftA2 :: forall a b c. (a -> b -> c) -> Parser a -> Parser b -> Parser c
$c*> :: forall a b. Parser a -> Parser b -> Parser b
*> :: forall a b. Parser a -> Parser b -> Parser b
$c<* :: forall a b. Parser a -> Parser b -> Parser a
<* :: forall a b. Parser a -> Parser b -> Parser a
Applicative
        , Applicative Parser
Applicative Parser =>
(forall a b. Parser a -> (a -> Parser b) -> Parser b)
-> (forall a b. Parser a -> Parser b -> Parser b)
-> (forall a. a -> Parser a)
-> Monad Parser
forall a. a -> Parser a
forall a b. Parser a -> Parser b -> Parser b
forall a b. Parser a -> (a -> Parser b) -> Parser b
forall (m :: * -> *).
Applicative m =>
(forall a b. m a -> (a -> m b) -> m b)
-> (forall a b. m a -> m b -> m b)
-> (forall a. a -> m a)
-> Monad m
$c>>= :: forall a b. Parser a -> (a -> Parser b) -> Parser b
>>= :: forall a b. Parser a -> (a -> Parser b) -> Parser b
$c>> :: forall a b. Parser a -> Parser b -> Parser b
>> :: forall a b. Parser a -> Parser b -> Parser b
$creturn :: forall a. a -> Parser a
return :: forall a. a -> Parser a
Monad
        , Applicative Parser
Applicative Parser =>
(forall a. Parser a)
-> (forall a. Parser a -> Parser a -> Parser a)
-> (forall a. Parser a -> Parser [a])
-> (forall a. Parser a -> Parser [a])
-> Alternative Parser
forall a. Parser a
forall a. Parser a -> Parser [a]
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *).
Applicative f =>
(forall a. f a)
-> (forall a. f a -> f a -> f a)
-> (forall a. f a -> f [a])
-> (forall a. f a -> f [a])
-> Alternative f
$cempty :: forall a. Parser a
empty :: forall a. Parser a
$c<|> :: forall a. Parser a -> Parser a -> Parser a
<|> :: forall a. Parser a -> Parser a -> Parser a
$csome :: forall a. Parser a -> Parser [a]
some :: forall a. Parser a -> Parser [a]
$cmany :: forall a. Parser a -> Parser [a]
many :: forall a. Parser a -> Parser [a]
Alternative
        , Monad Parser
Alternative Parser
(Alternative Parser, Monad Parser) =>
(forall a. Parser a)
-> (forall a. Parser a -> Parser a -> Parser a) -> MonadPlus Parser
forall a. Parser a
forall a. Parser a -> Parser a -> Parser a
forall (m :: * -> *).
(Alternative m, Monad m) =>
(forall a. m a) -> (forall a. m a -> m a -> m a) -> MonadPlus m
$cmzero :: forall a. Parser a
mzero :: forall a. Parser a
$cmplus :: forall a. Parser a -> Parser a -> Parser a
mplus :: forall a. Parser a -> Parser a -> Parser a
MonadPlus
        , MonadState ExtraState
        , MonadReader PersistSettings
        , MonadParsec Void String
        )

type EntityParseError = ParseErrorBundle String Void

-- | Result of parsing a single source text.
--
-- @since 2.16.0.0
type ParseResult a =
    (Set ParserWarning, Either (ParseErrorBundle String Void) a)

type InternalParseResult a = ParseResult (a, ExtraState)

-- | Cumulative result of parsing multiple source texts.
--
-- @since 2.16.0.0
type CumulativeParseResult a = (Set ParserWarning, Either [EntityParseError] a)

toCumulativeParseResult
    :: (Monoid a) => [ParseResult a] -> CumulativeParseResult a
toCumulativeParseResult :: forall a. Monoid a => [ParseResult a] -> CumulativeParseResult a
toCumulativeParseResult [ParseResult a]
prs = do
    let
        (Set ParserWarning
warnings, [Either (ParseErrorBundle String Void) a]
eithers) = [ParseResult a]
-> (Set ParserWarning, [Either (ParseErrorBundle String Void) a])
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
forall (m :: * -> *) a. Monad m => [m a] -> m [a]
sequence [ParseResult a]
prs
    case [Either (ParseErrorBundle String Void) a]
-> ([ParseErrorBundle String Void], [a])
forall a b. [Either a b] -> ([a], [b])
partitionEithers [Either (ParseErrorBundle String Void) a]
eithers of
        ([], [a]
results) -> (Set ParserWarning
warnings, a -> Either [ParseErrorBundle String Void] a
forall a b. b -> Either a b
Right (a -> Either [ParseErrorBundle String Void] a)
-> a -> Either [ParseErrorBundle String Void] a
forall a b. (a -> b) -> a -> b
$ [a] -> a
forall m. Monoid m => [m] -> m
forall (t :: * -> *) m. (Foldable t, Monoid m) => t m -> m
fold [a]
results)
        ([ParseErrorBundle String Void]
errs, [a]
_) -> (Set ParserWarning
warnings, [ParseErrorBundle String Void]
-> Either [ParseErrorBundle String Void] a
forall a b. a -> Either a b
Left [ParseErrorBundle String Void]
errs)

-- | Run a parser using provided PersistSettings and ExtraState
-- @since 2.16.0.0
runConfiguredParser
    :: PersistSettings
    -> ExtraState
    -> Parser a
    -> String
    -> String
    -> InternalParseResult a
runConfiguredParser :: forall a.
PersistSettings
-> ExtraState
-> Parser a
-> String
-> String
-> InternalParseResult a
runConfiguredParser PersistSettings
ps ExtraState
acc Parser a
parser String
fp String
s = (Set ParserWarning
warnings, Either (ParseErrorBundle String Void) (a, ExtraState)
either)
  where
    sm :: StateT
  ExtraState (ParsecT Void String (Writer (Set ParserWarning))) a
sm = ReaderT
  PersistSettings
  (StateT
     ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
  a
-> PersistSettings
-> StateT
     ExtraState (ParsecT Void String (Writer (Set ParserWarning))) a
forall r (m :: * -> *) a. ReaderT r m a -> r -> m a
runReaderT (Parser a
-> ReaderT
     PersistSettings
     (StateT
        ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
     a
forall a.
Parser a
-> ReaderT
     PersistSettings
     (StateT
        ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
     a
unParser Parser a
parser) PersistSettings
ps
    pm :: ParsecT Void String (Writer (Set ParserWarning)) (a, ExtraState)
pm = StateT
  ExtraState (ParsecT Void String (Writer (Set ParserWarning))) a
-> ExtraState
-> ParsecT Void String (Writer (Set ParserWarning)) (a, ExtraState)
forall s (m :: * -> *) a. StateT s m a -> s -> m (a, s)
runStateT StateT
  ExtraState (ParsecT Void String (Writer (Set ParserWarning))) a
sm ExtraState
acc
    wm :: Writer
  (Set ParserWarning)
  (State String Void,
   Either (ParseErrorBundle String Void) (a, ExtraState))
wm = ParsecT Void String (Writer (Set ParserWarning)) (a, ExtraState)
-> State String Void
-> Writer
     (Set ParserWarning)
     (State String Void,
      Either (ParseErrorBundle String Void) (a, ExtraState))
forall (m :: * -> *) e s a.
Monad m =>
ParsecT e s m a
-> State s e -> m (State s e, Either (ParseErrorBundle s e) a)
runParserT' ParsecT Void String (Writer (Set ParserWarning)) (a, ExtraState)
pm State String Void
initialInternalState
    ((State String Void
_is, Either (ParseErrorBundle String Void) (a, ExtraState)
either), Set ParserWarning
warnings) = Writer
  (Set ParserWarning)
  (State String Void,
   Either (ParseErrorBundle String Void) (a, ExtraState))
-> ((State String Void,
     Either (ParseErrorBundle String Void) (a, ExtraState)),
    Set ParserWarning)
forall w a. Writer w a -> (a, w)
runWriter Writer
  (Set ParserWarning)
  (State String Void,
   Either (ParseErrorBundle String Void) (a, ExtraState))
wm

    initialSourcePos :: SourcePos
initialSourcePos =
        SourcePos
            { sourceName :: String
sourceName = String
fp
            , sourceLine :: Pos
sourceLine = Pos
pos1
            , sourceColumn :: Pos
sourceColumn = Pos
pos1
            }
    initialPosState :: PosState String
initialPosState =
        PosState
            { pstateInput :: String
pstateInput = String
s
            , pstateOffset :: Int
pstateOffset = Int
0
            , pstateSourcePos :: SourcePos
pstateSourcePos = SourcePos
initialSourcePos
            , -- for legacy compatibility, we treat each tab as a single unit of whitespace
              pstateTabWidth :: Pos
pstateTabWidth = Pos
pos1
            , pstateLinePrefix :: String
pstateLinePrefix = String
""
            }
    initialInternalState :: State String Void
initialInternalState =
        State
            { stateInput :: String
stateInput = String
s
            , stateOffset :: Int
stateOffset = Int
0
            , statePosState :: PosState String
statePosState = PosState String
initialPosState
            , stateParseErrors :: [ParseError String Void]
stateParseErrors = []
            }

reportWarnings :: Set ParserWarning -> Parser ()
#if MIN_VERSION_megaparsec(9,5,0)
reportWarnings :: Set ParserWarning -> Parser ()
reportWarnings = ReaderT
  PersistSettings
  (StateT
     ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
  ()
-> Parser ()
forall a.
ReaderT
  PersistSettings
  (StateT
     ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
  a
-> Parser a
Parser (ReaderT
   PersistSettings
   (StateT
      ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
   ()
 -> Parser ())
-> (Set ParserWarning
    -> ReaderT
         PersistSettings
         (StateT
            ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
         ())
-> Set ParserWarning
-> Parser ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Set ParserWarning
-> ReaderT
     PersistSettings
     (StateT
        ExtraState (ParsecT Void String (Writer (Set ParserWarning))))
     ()
forall w (m :: * -> *). MonadWriter w m => w -> m ()
tell
#else
reportWarnings _pw = pure ()
#endif

-- | Renders a list of EntityParseErrors as a String using `errorBundlePretty`,
-- separated by line breaks.
-- @since 2.16.0.0
renderErrors :: [EntityParseError] -> String
renderErrors :: [ParseErrorBundle String Void] -> String
renderErrors [ParseErrorBundle String Void]
errs = String -> [String] -> String
forall a. [a] -> [[a]] -> [a]
intercalate String
"\n" ([String] -> String) -> [String] -> String
forall a b. (a -> b) -> a -> b
$ (ParseErrorBundle String Void -> String)
-> [ParseErrorBundle String Void] -> [String]
forall a b. (a -> b) -> [a] -> [b]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ParseErrorBundle String Void -> String
forall s e.
(VisualStream s, TraversableStream s, ShowErrorComponent e) =>
ParseErrorBundle s e -> String
errorBundlePretty [ParseErrorBundle String Void]
errs

-- | Attempts to parse with a provided parser. If it fails with an error matching
-- the provided predicate, it registers a warning with the provided message and falls
-- back to the second provided parser.
tryOrWarn
    :: String
    -> (ParseError String Void -> Bool)
    -> Parser a
    -> Parser a
    -> Parser a
tryOrWarn :: forall a.
String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrWarn String
msg ParseError String Void -> Bool
p Parser a
l Parser a
r = do
    State String Void
parserState <- Parser (State String Void)
forall e s (m :: * -> *). MonadParsec e s m => m (State s e)
getParserState
    (ParseError String Void -> Parser a) -> Parser a -> Parser a
forall a.
(ParseError String Void -> Parser a) -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
(ParseError s e -> m a) -> m a -> m a
withRecovery (PosState String -> ParseError String Void -> Parser a
warnAndRetry (PosState String -> ParseError String Void -> Parser a)
-> PosState String -> ParseError String Void -> Parser a
forall a b. (a -> b) -> a -> b
$ State String Void -> PosState String
forall s e. State s e -> PosState s
statePosState State String Void
parserState) Parser a
l
  where
    warnAndRetry :: PosState String -> ParseError String Void -> Parser a
warnAndRetry PosState String
posState ParseError String Void
err = do
        if ParseError String Void -> Bool
p ParseError String Void
err
            then do
                let
                    ([(ParseError String Void, SourcePos)]
pairs, PosState String
_) = (ParseError String Void -> Int)
-> [ParseError String Void]
-> PosState String
-> ([(ParseError String Void, SourcePos)], PosState String)
forall (t :: * -> *) s a.
(Traversable t, TraversableStream s) =>
(a -> Int) -> t a -> PosState s -> (t (a, SourcePos), PosState s)
attachSourcePos ParseError String Void -> Int
forall s e. ParseError s e -> Int
errorOffset [ParseError String Void
err] PosState String
posState
                Set ParserWarning -> Parser ()
reportWarnings (Set ParserWarning -> Parser ())
-> ([ParserWarning] -> Set ParserWarning)
-> [ParserWarning]
-> Parser ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. [ParserWarning] -> Set ParserWarning
forall a. Ord a => [a] -> Set a
Set.fromList ([ParserWarning] -> Parser ()) -> [ParserWarning] -> Parser ()
forall a b. (a -> b) -> a -> b
$
                    ((ParseError String Void, SourcePos) -> ParserWarning)
-> [(ParseError String Void, SourcePos)] -> [ParserWarning]
forall a b. (a -> b) -> [a] -> [b]
map
                        ( \(ParseError String Void
e, SourcePos
_pos) ->
                            ParserWarning
                                { parserWarningExtraMessage :: String
parserWarningExtraMessage = String
msg String -> String -> String
forall a. Semigroup a => a -> a -> a
<> String
"\n"
                                , parserWarningUnderlyingError :: ParseError String Void
parserWarningUnderlyingError = ParseError String Void
e
                                , parserWarningPosState :: PosState String
parserWarningPosState = PosState String
posState
                                }
                        )
                        [(ParseError String Void, SourcePos)]
pairs
                Parser a
r
            else ParseError String Void -> Parser a
forall a. ParseError String Void -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
ParseError s e -> m a
parseError ParseError String Void
err

-- | Attempts to parse with a provided parser. If it fails with an error matching
-- the provided predicate, it registers a delayed error with the provided message and falls
-- back to the second provided parser.
--
-- This is useful when registering errors in space consumers and other parsers that are called
-- with `try`, since a non-delayed error in this context will cause backtracking and not
-- get reported to the user.
tryOrRegisterError
    :: String
    -> (ParseError String Void -> Bool)
    -> Parser a
    -> Parser a
    -> Parser a
tryOrRegisterError :: forall a.
String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrRegisterError String
msg ParseError String Void -> Bool
p Parser a
l Parser a
r = do
    State String Void
parserState <- Parser (State String Void)
forall e s (m :: * -> *). MonadParsec e s m => m (State s e)
getParserState
    (ParseError String Void -> Parser a) -> Parser a -> Parser a
forall a.
(ParseError String Void -> Parser a) -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
(ParseError s e -> m a) -> m a -> m a
withRecovery (PosState String -> ParseError String Void -> Parser a
delayedError (PosState String -> ParseError String Void -> Parser a)
-> PosState String -> ParseError String Void -> Parser a
forall a b. (a -> b) -> a -> b
$ State String Void -> PosState String
forall s e. State s e -> PosState s
statePosState State String Void
parserState) Parser a
l
  where
    delayedError :: PosState String -> ParseError String Void -> Parser a
delayedError PosState String
posState ParseError String Void
err = do
        if ParseError String Void -> Bool
p ParseError String Void
err
            then do
                let
                    ([(ParseError String Void, SourcePos)]
pairs, PosState String
_) = (ParseError String Void -> Int)
-> [ParseError String Void]
-> PosState String
-> ([(ParseError String Void, SourcePos)], PosState String)
forall (t :: * -> *) s a.
(Traversable t, TraversableStream s) =>
(a -> Int) -> t a -> PosState s -> (t (a, SourcePos), PosState s)
attachSourcePos ParseError String Void -> Int
forall s e. ParseError s e -> Int
errorOffset [ParseError String Void
err] PosState String
posState
                ParseError String Void -> Parser ()
forall e s (m :: * -> *).
MonadParsec e s m =>
ParseError s e -> m ()
registerParseError ParseError String Void
err
                Parser a
r
            else ParseError String Void -> Parser a
forall a. ParseError String Void -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
ParseError s e -> m a
parseError ParseError String Void
err

tryOrReport
    :: Maybe ParserErrorLevel
    -> String
    -> (ParseError String Void -> Bool)
    -> Parser a
    -> Parser a
    -> Parser a
tryOrReport :: forall a.
Maybe ParserErrorLevel
-> String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrReport Maybe ParserErrorLevel
level String
msg ParseError String Void -> Bool
p Parser a
l Parser a
r = case Maybe ParserErrorLevel
level of
    Just ParserErrorLevel
LevelError -> String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
forall a.
String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrRegisterError String
msg ParseError String Void -> Bool
p Parser a
l Parser a
r
    Just ParserErrorLevel
LevelWarning -> String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
forall a.
String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrWarn String
msg ParseError String Void -> Bool
p Parser a
l Parser a
r
    Maybe ParserErrorLevel
Nothing -> Parser a
r

-- | Source location: file and line/col information. This is half of a 'SourceSpan'.
--
-- @since 2.16.0.0
data SourceLoc = SourceLoc
    { SourceLoc -> Text
locFile :: Text
    , SourceLoc -> Int
locStartLine :: Int
    , SourceLoc -> Int
locStartCol :: Int
    }
    deriving (Int -> SourceLoc -> String -> String
[SourceLoc] -> String -> String
SourceLoc -> String
(Int -> SourceLoc -> String -> String)
-> (SourceLoc -> String)
-> ([SourceLoc] -> String -> String)
-> Show SourceLoc
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> SourceLoc -> String -> String
showsPrec :: Int -> SourceLoc -> String -> String
$cshow :: SourceLoc -> String
show :: SourceLoc -> String
$cshowList :: [SourceLoc] -> String -> String
showList :: [SourceLoc] -> String -> String
Show, (forall (m :: * -> *). Quote m => SourceLoc -> m Exp)
-> (forall (m :: * -> *). Quote m => SourceLoc -> Code m SourceLoc)
-> Lift SourceLoc
forall t.
(forall (m :: * -> *). Quote m => t -> m Exp)
-> (forall (m :: * -> *). Quote m => t -> Code m t) -> Lift t
forall (m :: * -> *). Quote m => SourceLoc -> m Exp
forall (m :: * -> *). Quote m => SourceLoc -> Code m SourceLoc
$clift :: forall (m :: * -> *). Quote m => SourceLoc -> m Exp
lift :: forall (m :: * -> *). Quote m => SourceLoc -> m Exp
$cliftTyped :: forall (m :: * -> *). Quote m => SourceLoc -> Code m SourceLoc
liftTyped :: forall (m :: * -> *). Quote m => SourceLoc -> Code m SourceLoc
Lift)

-- @since 2.16.0.0
data Token
    = Quotation Text
    | Equality Text Text
    | Parenthetical Text
    | BlockKey Text
    | PText Text
    deriving (Token -> Token -> Bool
(Token -> Token -> Bool) -> (Token -> Token -> Bool) -> Eq Token
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: Token -> Token -> Bool
== :: Token -> Token -> Bool
$c/= :: Token -> Token -> Bool
/= :: Token -> Token -> Bool
Eq, Eq Token
Eq Token =>
(Token -> Token -> Ordering)
-> (Token -> Token -> Bool)
-> (Token -> Token -> Bool)
-> (Token -> Token -> Bool)
-> (Token -> Token -> Bool)
-> (Token -> Token -> Token)
-> (Token -> Token -> Token)
-> Ord Token
Token -> Token -> Bool
Token -> Token -> Ordering
Token -> Token -> Token
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
$ccompare :: Token -> Token -> Ordering
compare :: Token -> Token -> Ordering
$c< :: Token -> Token -> Bool
< :: Token -> Token -> Bool
$c<= :: Token -> Token -> Bool
<= :: Token -> Token -> Bool
$c> :: Token -> Token -> Bool
> :: Token -> Token -> Bool
$c>= :: Token -> Token -> Bool
>= :: Token -> Token -> Bool
$cmax :: Token -> Token -> Token
max :: Token -> Token -> Token
$cmin :: Token -> Token -> Token
min :: Token -> Token -> Token
Ord, Int -> Token -> String -> String
[Token] -> String -> String
Token -> String
(Int -> Token -> String -> String)
-> (Token -> String) -> ([Token] -> String -> String) -> Show Token
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> Token -> String -> String
showsPrec :: Int -> Token -> String -> String
$cshow :: Token -> String
show :: Token -> String
$cshowList :: [Token] -> String -> String
showList :: [Token] -> String -> String
Show)

-- @since 2.16.0.0
data CommentToken
    = DocComment Text
    | Comment Text
    deriving (CommentToken -> CommentToken -> Bool
(CommentToken -> CommentToken -> Bool)
-> (CommentToken -> CommentToken -> Bool) -> Eq CommentToken
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: CommentToken -> CommentToken -> Bool
== :: CommentToken -> CommentToken -> Bool
$c/= :: CommentToken -> CommentToken -> Bool
/= :: CommentToken -> CommentToken -> Bool
Eq, Eq CommentToken
Eq CommentToken =>
(CommentToken -> CommentToken -> Ordering)
-> (CommentToken -> CommentToken -> Bool)
-> (CommentToken -> CommentToken -> Bool)
-> (CommentToken -> CommentToken -> Bool)
-> (CommentToken -> CommentToken -> Bool)
-> (CommentToken -> CommentToken -> CommentToken)
-> (CommentToken -> CommentToken -> CommentToken)
-> Ord CommentToken
CommentToken -> CommentToken -> Bool
CommentToken -> CommentToken -> Ordering
CommentToken -> CommentToken -> CommentToken
forall a.
Eq a =>
(a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
$ccompare :: CommentToken -> CommentToken -> Ordering
compare :: CommentToken -> CommentToken -> Ordering
$c< :: CommentToken -> CommentToken -> Bool
< :: CommentToken -> CommentToken -> Bool
$c<= :: CommentToken -> CommentToken -> Bool
<= :: CommentToken -> CommentToken -> Bool
$c> :: CommentToken -> CommentToken -> Bool
> :: CommentToken -> CommentToken -> Bool
$c>= :: CommentToken -> CommentToken -> Bool
>= :: CommentToken -> CommentToken -> Bool
$cmax :: CommentToken -> CommentToken -> CommentToken
max :: CommentToken -> CommentToken -> CommentToken
$cmin :: CommentToken -> CommentToken -> CommentToken
min :: CommentToken -> CommentToken -> CommentToken
Ord, Int -> CommentToken -> String -> String
[CommentToken] -> String -> String
CommentToken -> String
(Int -> CommentToken -> String -> String)
-> (CommentToken -> String)
-> ([CommentToken] -> String -> String)
-> Show CommentToken
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> CommentToken -> String -> String
showsPrec :: Int -> CommentToken -> String -> String
$cshow :: CommentToken -> String
show :: CommentToken -> String
$cshowList :: [CommentToken] -> String -> String
showList :: [CommentToken] -> String -> String
Show)

-- | Converts a token into a Text representation for second-stage parsing or presentation to the user
--
-- @since 2.16.0.0
tokenContent :: Token -> Text
tokenContent :: Token -> Text
tokenContent = \case
    Quotation Text
s -> Text
s
    Equality Text
l Text
r -> [Text] -> Text
forall m. Monoid m => [m] -> m
mconcat [Text
l, Text
"=", Text
r]
    Parenthetical Text
s -> Text
s
    PText Text
s -> Text
s
    BlockKey Text
s -> Text
s

commentContent :: CommentToken -> Text
commentContent :: CommentToken -> Text
commentContent = \case
    Comment Text
s -> Text
s
    DocComment Text
s -> Text
s

docComment :: Parser (SourcePos, CommentToken)
docComment :: Parser (SourcePos, CommentToken)
docComment = do
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    String
content <-
        Tokens String -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens String
"-- |" Parser (Tokens String) -> Parser () -> Parser ()
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser ()
validHSpace Parser () -> Parser String -> Parser String
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhileP (String -> Maybe String
forall a. a -> Maybe a
Just String
"character") (Token String -> Token String -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
Token String
'\n')
    (SourcePos, CommentToken) -> Parser (SourcePos, CommentToken)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (SourcePos
pos, Text -> CommentToken
DocComment (String -> Text
Text.pack String
content))

comment :: Parser (SourcePos, CommentToken)
comment :: Parser (SourcePos, CommentToken)
comment = do
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    String
content <-
        (Tokens String -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens String
"--" Parser (Tokens String)
-> Parser (Tokens String) -> Parser (Tokens String)
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Tokens String -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string Tokens String
"#")
            Parser (Tokens String) -> Parser () -> Parser ()
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser ()
validHSpace
            Parser () -> Parser String -> Parser String
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhileP (String -> Maybe String
forall a. a -> Maybe a
Just String
"character") (Token String -> Token String -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
Token String
'\n')
    (SourcePos, CommentToken) -> Parser (SourcePos, CommentToken)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (SourcePos
pos, Text -> CommentToken
Comment (String -> Text
Text.pack String
content))

skipComment :: Parser ()
skipComment :: Parser ()
skipComment = do
    (SourcePos, CommentToken)
content <- Parser (SourcePos, CommentToken)
docComment Parser (SourcePos, CommentToken)
-> Parser (SourcePos, CommentToken)
-> Parser (SourcePos, CommentToken)
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Parser (SourcePos, CommentToken)
comment
    Parser () -> Parser ()
forall (f :: * -> *) a. Functor f => f a -> f ()
void (Parser () -> Parser ()) -> Parser () -> Parser ()
forall a b. (a -> b) -> a -> b
$ (SourcePos, CommentToken) -> Parser ()
appendCommentToState (SourcePos, CommentToken)
content

isValidHSpace :: Bool -> Char -> Bool
isValidHSpace :: Bool -> Char -> Bool
isValidHSpace Bool
allowTabs Char
c =
    if Bool
allowTabs
        then Char -> Bool
isSpace Char
c Bool -> Bool -> Bool
&& Char
c Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
'\n'
        else Char -> Bool
isSpace Char
c Bool -> Bool -> Bool
&& Char
c Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
'\n' Bool -> Bool -> Bool
&& Char
c Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
'\t'

isValidSpace :: Bool -> Char -> Bool
isValidSpace :: Bool -> Char -> Bool
isValidSpace Bool
allowTabs Char
c =
    if Bool
allowTabs
        then Char -> Bool
isSpace Char
c
        else Char -> Bool
isSpace Char
c Bool -> Bool -> Bool
&& Char
c Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
/= Char
'\t'

validSpaceParser
    :: (Maybe String -> (TMS.Token String -> Bool) -> Parser (Tokens String))
    -> (Bool -> Char -> Bool)
    -> Parser ()
validSpaceParser :: (Maybe String -> (Token String -> Bool) -> Parser (Tokens String))
-> (Bool -> Char -> Bool) -> Parser ()
validSpaceParser Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
taker Bool -> Char -> Bool
validator = do
    Maybe ParserErrorLevel
tabErrorLevel <- (PersistSettings -> Maybe ParserErrorLevel)
-> Parser (Maybe ParserErrorLevel)
forall r (m :: * -> *) a. MonadReader r m => (r -> a) -> m a
asks PersistSettings -> Maybe ParserErrorLevel
psTabErrorLevel
    Parser String -> Parser ()
forall (f :: * -> *) a. Functor f => f a -> f ()
void (Parser String -> Parser ()) -> Parser String -> Parser ()
forall a b. (a -> b) -> a -> b
$
        Maybe ParserErrorLevel
-> String
-> (ParseError String Void -> Bool)
-> Parser String
-> Parser String
-> Parser String
forall a.
Maybe ParserErrorLevel
-> String
-> (ParseError String Void -> Bool)
-> Parser a
-> Parser a
-> Parser a
tryOrReport
            Maybe ParserErrorLevel
tabErrorLevel
            String
"use spaces instead of tabs"
            ParseError String Void -> Bool
isUnexpectedTabError
            (Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
taker (String -> Maybe String
forall a. a -> Maybe a
Just String
"valid whitespace") (Bool -> Char -> Bool
validator Bool
False))
            (Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
taker (String -> Maybe String
forall a. a -> Maybe a
Just String
"valid whitespace") (Bool -> Char -> Bool
validator Bool
True))

isUnexpectedTabError :: ParseError String Void -> Bool
isUnexpectedTabError :: ParseError String Void -> Bool
isUnexpectedTabError (TrivialError Int
_ Maybe (ErrorItem (Token String))
ue Set (ErrorItem (Token String))
l) =
    Maybe (ErrorItem Char)
Maybe (ErrorItem (Token String))
ue Maybe (ErrorItem Char) -> Maybe (ErrorItem Char) -> Bool
forall a. Eq a => a -> a -> Bool
== ErrorItem Char -> Maybe (ErrorItem Char)
forall a. a -> Maybe a
Just (NonEmpty Char -> ErrorItem Char
forall t. NonEmpty t -> ErrorItem t
Tokens (Char
'\t' Char -> String -> NonEmpty Char
forall a. a -> [a] -> NonEmpty a
:| String
""))
        Bool -> Bool -> Bool
&& Set (ErrorItem Char)
Set (ErrorItem (Token String))
l Set (ErrorItem Char) -> Set (ErrorItem Char) -> Bool
forall a. Eq a => a -> a -> Bool
== ErrorItem Char -> Set (ErrorItem Char)
forall a. a -> Set a
Set.singleton (NonEmpty Char -> ErrorItem Char
forall t. NonEmpty Char -> ErrorItem t
Label (Char
'v' Char -> String -> NonEmpty Char
forall a. a -> [a] -> NonEmpty a
:| String
"alid whitespace"))
isUnexpectedTabError ParseError String Void
_ = Bool
False

someValidHSpace :: Parser ()
someValidHSpace :: Parser ()
someValidHSpace = (Maybe String -> (Token String -> Bool) -> Parser (Tokens String))
-> (Bool -> Char -> Bool) -> Parser ()
validSpaceParser Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P Bool -> Char -> Bool
isValidHSpace

someValidSpace :: Parser ()
someValidSpace :: Parser ()
someValidSpace = (Maybe String -> (Token String -> Bool) -> Parser (Tokens String))
-> (Bool -> Char -> Bool) -> Parser ()
validSpaceParser Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhile1P Bool -> Char -> Bool
isValidSpace

validHSpace :: Parser ()
validHSpace :: Parser ()
validHSpace = (Maybe String -> (Token String -> Bool) -> Parser (Tokens String))
-> (Bool -> Char -> Bool) -> Parser ()
validSpaceParser Maybe String -> (Token String -> Bool) -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Maybe String -> (Token s -> Bool) -> m (Tokens s)
takeWhileP Bool -> Char -> Bool
isValidHSpace

spaceConsumer :: Parser ()
spaceConsumer :: Parser ()
spaceConsumer =
    Parser () -> Parser () -> Parser () -> Parser ()
forall e s (m :: * -> *).
MonadParsec e s m =>
m () -> m () -> m () -> m ()
L.space
        Parser ()
someValidHSpace
        Parser ()
skipComment
        Parser ()
forall a. Parser a
forall (f :: * -> *) a. Alternative f => f a
empty

spaceConsumerN :: Parser ()
spaceConsumerN :: Parser ()
spaceConsumerN =
    Parser () -> Parser () -> Parser () -> Parser ()
forall e s (m :: * -> *).
MonadParsec e s m =>
m () -> m () -> m () -> m ()
L.space
        Parser ()
someValidSpace
        Parser ()
skipComment
        Parser ()
forall a. Parser a
forall (f :: * -> *) a. Alternative f => f a
empty

contentChar :: Parser Char
contentChar :: Parser Char
contentChar =
    [Parser Char] -> Parser Char
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice
        [ Parser Char
Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
alphaNumChar
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'.'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'['
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
']'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'_'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'\''
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'"'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'!'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'~'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'-'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
':'
        , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
','
        , do
            Char
backslash <- Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'\\'
            Char
nextChar <- Parser Char -> Parser Char
forall a. Parser a -> Parser a
forall e s (m :: * -> *) a. MonadParsec e s m => m a -> m a
lookAhead Parser Char
Parser (Token String)
forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle
            if Char
nextChar Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
'(' Bool -> Bool -> Bool
|| Char
nextChar Char -> Char -> Bool
forall a. Eq a => a -> a -> Bool
== Char
')'
                then Token String -> Parser (Token String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Token s -> m (Token s)
single Char
Token String
nextChar
                else Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
backslash
        ]

nonLineSpaceChar :: Parser Char
nonLineSpaceChar :: Parser Char
nonLineSpaceChar = [Parser Char] -> Parser Char
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice [Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
' ', Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'\t']

-- This is a replacement for `Text.Megaparsec.Char.Lexer.charLiteral`;
-- it does nearly the same thing but additionally supports escaped parentheses.
charLiteral :: Parser Char
charLiteral :: Parser Char
charLiteral = String -> Parser Char -> Parser Char
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"literal character" (Parser Char -> Parser Char) -> Parser Char -> Parser Char
forall a b. (a -> b) -> a -> b
$ do
    Char
char1 <- Parser Char
Parser (Token String)
forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle
    case Char
char1 of
        Char
'\\' -> do
            Char
char2 <- Parser Char
Parser (Token String)
forall e s (m :: * -> *). MonadParsec e s m => m (Token s)
anySingle
            case Char
char2 of
                Char
'(' -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
'('
                Char
')' -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
')'
                Char
'\\' -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
'\\'
                Char
'\"' -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
'\"'
                Char
'\'' -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
'\''
                Char
_ -> ErrorItem (Token String) -> Parser Char
forall e s (m :: * -> *) a.
MonadParsec e s m =>
ErrorItem (Token s) -> m a
unexpected (NonEmpty (Token String) -> ErrorItem (Token String)
forall t. NonEmpty t -> ErrorItem t
Tokens (NonEmpty (Token String) -> ErrorItem (Token String))
-> NonEmpty (Token String) -> ErrorItem (Token String)
forall a b. (a -> b) -> a -> b
$ Char
char2 Char -> String -> NonEmpty Char
forall a. a -> [a] -> NonEmpty a
:| [])
        Char
_ -> Char -> Parser Char
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure Char
char1

equality :: Parser Token
equality :: Parser Token
equality = String -> Parser Token -> Parser Token
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"equality expression" (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
    Parser () -> Parser Token -> Parser Token
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
        String
lhs <- Parser Char -> Parser String
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some Parser Char
contentChar
        Char
_ <- Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'='
        String
rhs <-
            [Parser String] -> Parser String
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice
                [ Parser String
quotation'
                , Parser String
sqlLiteral
                , Parser String
parentheticalInner
                , Parser Char -> Parser String
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some (Parser Char -> Parser String) -> Parser Char -> Parser String
forall a b. (a -> b) -> a -> b
$ Parser Char
contentChar Parser Char -> Parser Char -> Parser Char
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'(' Parser Char -> Parser Char -> Parser Char
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
')'
                ]
        Token -> Parser Token
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Token -> Parser Token) -> Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ Text -> Text -> Token
Equality (String -> Text
Text.pack String
lhs) (String -> Text
Text.pack String
rhs)
  where
    parentheticalInner :: Parser String
parentheticalInner = do
        String
str <- Parser String
parenthetical'
        String -> Parser String
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (String -> Parser String)
-> (String -> String) -> String -> Parser String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> String
forall a. HasCallStack => [a] -> [a]
init (String -> String) -> (String -> String) -> String -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Int -> String -> String
forall a. Int -> [a] -> [a]
drop Int
1 (String -> Parser String) -> String -> Parser String
forall a b. (a -> b) -> a -> b
$ String
str

sqlTypeName :: Parser String
sqlTypeName :: Parser String
sqlTypeName =
    Parser Char -> Parser String
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some (Parser Char -> Parser String) -> Parser Char -> Parser String
forall a b. (a -> b) -> a -> b
$
        [Parser Char] -> Parser Char
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice
            [ Parser Char
Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
alphaNumChar
            , Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'_'
            ]

sqlLiteral :: Parser String
sqlLiteral :: Parser String
sqlLiteral = String -> Parser String -> Parser String
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"SQL literal" (Parser String -> Parser String) -> Parser String -> Parser String
forall a b. (a -> b) -> a -> b
$ do
    String
quote <- Parser () -> Parser String -> Parser String
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer (Parser String -> Parser String) -> Parser String -> Parser String
forall a b. (a -> b) -> a -> b
$ Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'\'' Parser Char -> Parser String -> Parser String
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser Char -> Parser Char -> Parser String
forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
manyTill Parser Char
charLiteral (Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'\'')
    Maybe String
st <- Parser String -> Parser (Maybe String)
forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional (Parser String -> Parser (Maybe String))
-> Parser String -> Parser (Maybe String)
forall a b. (a -> b) -> a -> b
$ do
        String
colons <- Tokens String -> Parser (Tokens String)
forall e s (m :: * -> *).
MonadParsec e s m =>
Tokens s -> m (Tokens s)
string String
Tokens String
"::"
        String
tn <- Parser String
sqlTypeName
        String -> Parser String
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (String -> Parser String) -> String -> Parser String
forall a b. (a -> b) -> a -> b
$ String
colons String -> String -> String
forall a. Semigroup a => a -> a -> a
<> String
tn
    String -> Parser String
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (String -> Parser String) -> String -> Parser String
forall a b. (a -> b) -> a -> b
$
        [String] -> String
forall m. Monoid m => [m] -> m
mconcat
            [ String
"'"
            , String
quote
            , String
"'"
            , String -> Maybe String -> String
forall a. a -> Maybe a -> a
fromMaybe String
"" Maybe String
st
            ]

quotation :: Parser Token
quotation :: Parser Token
quotation = String -> Parser Token -> Parser Token
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"quotation" (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
    String
str <- Parser () -> Parser String -> Parser String
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer Parser String
quotation'
    Token -> Parser Token
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Token -> Parser Token) -> (Text -> Token) -> Text -> Parser Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Token
Quotation (Text -> Parser Token) -> Text -> Parser Token
forall a b. (a -> b) -> a -> b
$ String -> Text
Text.pack String
str

quotation' :: Parser String
quotation' :: Parser String
quotation' = Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'"' Parser Char -> Parser String -> Parser String
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser Char -> Parser Char -> Parser String
forall (m :: * -> *) a end. MonadPlus m => m a -> m end -> m [a]
manyTill Parser Char
charLiteral (Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'"')

parenthetical :: Parser Token
parenthetical :: Parser Token
parenthetical = String -> Parser Token -> Parser Token
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"parenthetical" (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
    String
str <- Parser () -> Parser String -> Parser String
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer Parser String
parenthetical'
    Token -> Parser Token
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Token -> Parser Token)
-> (String -> Token) -> String -> Parser Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Token
Parenthetical (Text -> Token) -> (String -> Text) -> String -> Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> Text
Text.pack (String -> Text) -> (String -> String) -> String -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> String
forall a. HasCallStack => [a] -> [a]
init (String -> String) -> (String -> String) -> String -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Int -> String -> String
forall a. Int -> [a] -> [a]
drop Int
1 (String -> Parser Token) -> String -> Parser Token
forall a b. (a -> b) -> a -> b
$ String
str

parenthetical' :: Parser String
parenthetical' :: Parser String
parenthetical' = do
    String
str <- Parser Char -> Parser Char -> Parser String -> Parser String
forall (m :: * -> *) open close a.
Applicative m =>
m open -> m close -> m a -> m a
between (Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'(') (Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
')') Parser String
q
    String -> Parser String
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (String -> Parser String) -> String -> Parser String
forall a b. (a -> b) -> a -> b
$ String
"(" String -> String -> String
forall a. [a] -> [a] -> [a]
++ String
str String -> String -> String
forall a. [a] -> [a] -> [a]
++ String
")"
  where
    q :: Parser String
q = [String] -> String
forall m. Monoid m => [m] -> m
mconcat ([String] -> String) -> Parser [String] -> Parser String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Parser String -> Parser [String]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some (Parser String
c Parser String -> Parser String -> Parser String
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Parser String
parenthetical')
    c :: Parser String
c = (Char -> String -> String
forall a. a -> [a] -> [a]
: []) (Char -> String) -> Parser Char -> Parser String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [Parser Char] -> Parser Char
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice [Parser Char
contentChar, Parser Char
nonLineSpaceChar, Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'"']

blockKey :: Parser Token
blockKey :: Parser Token
blockKey = String -> Parser Token -> Parser Token
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"block key" (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
    Char
fl <- Parser Char
Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
upperChar
    String
rl <- Parser Char -> Parser String
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many Parser Char
Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
m (Token s)
alphaNumChar
    Token -> Parser Token
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Token -> Parser Token)
-> (String -> Token) -> String -> Parser Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Token
BlockKey (Text -> Token) -> (String -> Text) -> String -> Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> Text
Text.pack (String -> Parser Token) -> String -> Parser Token
forall a b. (a -> b) -> a -> b
$ Char
fl Char -> String -> String
forall a. a -> [a] -> [a]
: String
rl

ptext :: Parser Token
ptext :: Parser Token
ptext = String -> Parser Token -> Parser Token
forall a. String -> Parser a -> Parser a
forall e s (m :: * -> *) a.
MonadParsec e s m =>
String -> m a -> m a
label String
"plain token" (Parser Token -> Parser Token) -> Parser Token -> Parser Token
forall a b. (a -> b) -> a -> b
$ do
    String
str <- Parser () -> Parser String -> Parser String
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer (Parser String -> Parser String) -> Parser String -> Parser String
forall a b. (a -> b) -> a -> b
$ Parser Char -> Parser String
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some Parser Char
contentChar
    Token -> Parser Token
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Token -> Parser Token)
-> (String -> Token) -> String -> Parser Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Text -> Token
PText (Text -> Token) -> (String -> Text) -> String -> Token
forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> Text
Text.pack (String -> Parser Token) -> String -> Parser Token
forall a b. (a -> b) -> a -> b
$ String
str

-- @since 2.16.0.0
anyToken :: Parser Token
anyToken :: Parser Token
anyToken =
    [Parser Token] -> Parser Token
forall (f :: * -> *) (m :: * -> *) a.
(Foldable f, Alternative m) =>
f (m a) -> m a
choice
        [ Parser Token -> Parser Token
forall a. Parser a -> Parser a
forall e s (m :: * -> *) a. MonadParsec e s m => m a -> m a
try Parser Token
equality
        , Parser Token
quotation
        , Parser Token
parenthetical
        , Parser Token
ptext
        ]

data ParsedEntityDef = ParsedEntityDef
    { ParsedEntityDef -> [Text]
parsedEntityDefComments :: [Text]
    , ParsedEntityDef -> EntityNameHS
parsedEntityDefEntityName :: EntityNameHS
    , ParsedEntityDef -> Bool
parsedEntityDefIsSum :: Bool
    , ParsedEntityDef -> [Text]
parsedEntityDefEntityAttributes :: [Attr]
    , ParsedEntityDef -> [([Token], Maybe Text)]
parsedEntityDefFieldAttributes :: [([Token], Maybe Text)]
    , ParsedEntityDef -> Map Text [[Text]]
parsedEntityDefExtras :: M.Map Text [ExtraLine]
    , ParsedEntityDef -> Maybe SourceSpan
parsedEntityDefSpan :: Maybe SourceSpan
    }
    deriving (Int -> ParsedEntityDef -> String -> String
[ParsedEntityDef] -> String -> String
ParsedEntityDef -> String
(Int -> ParsedEntityDef -> String -> String)
-> (ParsedEntityDef -> String)
-> ([ParsedEntityDef] -> String -> String)
-> Show ParsedEntityDef
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> ParsedEntityDef -> String -> String
showsPrec :: Int -> ParsedEntityDef -> String -> String
$cshow :: ParsedEntityDef -> String
show :: ParsedEntityDef -> String
$cshowList :: [ParsedEntityDef] -> String -> String
showList :: [ParsedEntityDef] -> String -> String
Show)

data DocCommentBlock = DocCommentBlock
    { DocCommentBlock -> [Text]
docCommentBlockLines :: [Text]
    , DocCommentBlock -> SourcePos
docCommentBlockPos :: SourcePos
    }
    deriving (Int -> DocCommentBlock -> String -> String
[DocCommentBlock] -> String -> String
DocCommentBlock -> String
(Int -> DocCommentBlock -> String -> String)
-> (DocCommentBlock -> String)
-> ([DocCommentBlock] -> String -> String)
-> Show DocCommentBlock
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> DocCommentBlock -> String -> String
showsPrec :: Int -> DocCommentBlock -> String -> String
$cshow :: DocCommentBlock -> String
show :: DocCommentBlock -> String
$cshowList :: [DocCommentBlock] -> String -> String
showList :: [DocCommentBlock] -> String -> String
Show)

data EntityHeader = EntityHeader
    { EntityHeader -> Bool
entityHeaderSum :: Bool
    , EntityHeader -> Text
entityHeaderTableName :: Text
    , EntityHeader -> [Token]
entityHeaderRemainingTokens :: [Token]
    , EntityHeader -> SourcePos
entityHeaderPos :: SourcePos
    }
    deriving (Int -> EntityHeader -> String -> String
[EntityHeader] -> String -> String
EntityHeader -> String
(Int -> EntityHeader -> String -> String)
-> (EntityHeader -> String)
-> ([EntityHeader] -> String -> String)
-> Show EntityHeader
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> EntityHeader -> String -> String
showsPrec :: Int -> EntityHeader -> String -> String
$cshow :: EntityHeader -> String
show :: EntityHeader -> String
$cshowList :: [EntityHeader] -> String -> String
showList :: [EntityHeader] -> String -> String
Show)

data EntityBlock = EntityBlock
    { EntityBlock -> Maybe DocCommentBlock
entityBlockDocCommentBlock :: Maybe DocCommentBlock
    , EntityBlock -> EntityHeader
entityBlockEntityHeader :: EntityHeader
    , EntityBlock -> [Member]
entityBlockMembers :: [Member]
    }
    deriving (Int -> EntityBlock -> String -> String
[EntityBlock] -> String -> String
EntityBlock -> String
(Int -> EntityBlock -> String -> String)
-> (EntityBlock -> String)
-> ([EntityBlock] -> String -> String)
-> Show EntityBlock
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> EntityBlock -> String -> String
showsPrec :: Int -> EntityBlock -> String -> String
$cshow :: EntityBlock -> String
show :: EntityBlock -> String
$cshowList :: [EntityBlock] -> String -> String
showList :: [EntityBlock] -> String -> String
Show)

entityBlockFirstPos :: EntityBlock -> SourcePos
entityBlockFirstPos :: EntityBlock -> SourcePos
entityBlockFirstPos = EntityHeader -> SourcePos
entityHeaderPos (EntityHeader -> SourcePos)
-> (EntityBlock -> EntityHeader) -> EntityBlock -> SourcePos
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EntityBlock -> EntityHeader
entityBlockEntityHeader

entityBlockLastPos :: EntityBlock -> SourcePos
entityBlockLastPos :: EntityBlock -> SourcePos
entityBlockLastPos EntityBlock
eb = case EntityBlock -> [Member]
entityBlockMembers EntityBlock
eb of
    [] -> EntityBlock -> SourcePos
entityBlockFirstPos EntityBlock
eb
    [Member]
members -> [SourcePos] -> SourcePos
forall a. Ord a => [a] -> a
forall (t :: * -> *) a. (Foldable t, Ord a) => t a -> a
maximum ([SourcePos] -> SourcePos) -> [SourcePos] -> SourcePos
forall a b. (a -> b) -> a -> b
$ (Member -> SourcePos) -> [Member] -> [SourcePos]
forall a b. (a -> b) -> [a] -> [b]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap Member -> SourcePos
memberEndPos [Member]
members

entityBlockBlockAttrs :: EntityBlock -> [BlockAttr]
entityBlockBlockAttrs :: EntityBlock -> [BlockAttr]
entityBlockBlockAttrs = (Member -> [BlockAttr]) -> [Member] -> [BlockAttr]
forall m a. Monoid m => (a -> m) -> [a] -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap Member -> [BlockAttr]
f ([Member] -> [BlockAttr])
-> (EntityBlock -> [Member]) -> EntityBlock -> [BlockAttr]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> EntityBlock -> [Member]
entityBlockMembers
  where
    f :: Member -> [BlockAttr]
f Member
m = case Member
m of
        MemberExtraBlock ExtraBlock
_ -> []
        MemberBlockAttr BlockAttr
ba -> [BlockAttr
ba]

entityBlockExtraBlocks :: EntityBlock -> [ExtraBlock]
entityBlockExtraBlocks :: EntityBlock -> [ExtraBlock]
entityBlockExtraBlocks = (Member -> [ExtraBlock]) -> [Member] -> [ExtraBlock]
forall m a. Monoid m => (a -> m) -> [a] -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap Member -> [ExtraBlock]
f ([Member] -> [ExtraBlock])
-> (EntityBlock -> [Member]) -> EntityBlock -> [ExtraBlock]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> EntityBlock -> [Member]
entityBlockMembers
  where
    f :: Member -> [ExtraBlock]
f Member
m = case Member
m of
        MemberExtraBlock ExtraBlock
eb -> [ExtraBlock
eb]
        MemberBlockAttr BlockAttr
_ -> []

data ExtraBlockHeader = ExtraBlockHeader
    { ExtraBlockHeader -> Text
extraBlockHeaderKey :: Text
    , ExtraBlockHeader -> [Token]
extraBlockHeaderRemainingTokens :: [Token]
    , ExtraBlockHeader -> SourcePos
extraBlockHeaderPos :: SourcePos
    }
    deriving (Int -> ExtraBlockHeader -> String -> String
[ExtraBlockHeader] -> String -> String
ExtraBlockHeader -> String
(Int -> ExtraBlockHeader -> String -> String)
-> (ExtraBlockHeader -> String)
-> ([ExtraBlockHeader] -> String -> String)
-> Show ExtraBlockHeader
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> ExtraBlockHeader -> String -> String
showsPrec :: Int -> ExtraBlockHeader -> String -> String
$cshow :: ExtraBlockHeader -> String
show :: ExtraBlockHeader -> String
$cshowList :: [ExtraBlockHeader] -> String -> String
showList :: [ExtraBlockHeader] -> String -> String
Show)

data ExtraBlock = ExtraBlock
    { ExtraBlock -> Maybe DocCommentBlock
extraBlockDocCommentBlock :: Maybe DocCommentBlock
    , ExtraBlock -> ExtraBlockHeader
extraBlockExtraBlockHeader :: ExtraBlockHeader
    , ExtraBlock -> NonEmpty Member
extraBlockMembers :: NonEmpty Member
    }
    deriving (Int -> ExtraBlock -> String -> String
[ExtraBlock] -> String -> String
ExtraBlock -> String
(Int -> ExtraBlock -> String -> String)
-> (ExtraBlock -> String)
-> ([ExtraBlock] -> String -> String)
-> Show ExtraBlock
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> ExtraBlock -> String -> String
showsPrec :: Int -> ExtraBlock -> String -> String
$cshow :: ExtraBlock -> String
show :: ExtraBlock -> String
$cshowList :: [ExtraBlock] -> String -> String
showList :: [ExtraBlock] -> String -> String
Show)

data BlockAttr = BlockAttr
    { BlockAttr -> Maybe DocCommentBlock
blockAttrDocCommentBlock :: Maybe DocCommentBlock
    , BlockAttr -> [Token]
blockAttrTokens :: [Token]
    , BlockAttr -> SourcePos
blockAttrPos :: SourcePos
    }
    deriving (Int -> BlockAttr -> String -> String
[BlockAttr] -> String -> String
BlockAttr -> String
(Int -> BlockAttr -> String -> String)
-> (BlockAttr -> String)
-> ([BlockAttr] -> String -> String)
-> Show BlockAttr
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> BlockAttr -> String -> String
showsPrec :: Int -> BlockAttr -> String -> String
$cshow :: BlockAttr -> String
show :: BlockAttr -> String
$cshowList :: [BlockAttr] -> String -> String
showList :: [BlockAttr] -> String -> String
Show)

data Member = MemberExtraBlock ExtraBlock | MemberBlockAttr BlockAttr
    deriving (Int -> Member -> String -> String
[Member] -> String -> String
Member -> String
(Int -> Member -> String -> String)
-> (Member -> String)
-> ([Member] -> String -> String)
-> Show Member
forall a.
(Int -> a -> String -> String)
-> (a -> String) -> ([a] -> String -> String) -> Show a
$cshowsPrec :: Int -> Member -> String -> String
showsPrec :: Int -> Member -> String -> String
$cshow :: Member -> String
show :: Member -> String
$cshowList :: [Member] -> String -> String
showList :: [Member] -> String -> String
Show)

-- | The source position at the beginning of the member's final line.
memberEndPos :: Member -> SourcePos
memberEndPos :: Member -> SourcePos
memberEndPos (MemberBlockAttr BlockAttr
fs) = BlockAttr -> SourcePos
blockAttrPos BlockAttr
fs
memberEndPos (MemberExtraBlock ExtraBlock
ex) = Member -> SourcePos
memberEndPos (Member -> SourcePos)
-> (ExtraBlock -> Member) -> ExtraBlock -> SourcePos
forall b c a. (b -> c) -> (a -> b) -> a -> c
. NonEmpty Member -> Member
forall a. NonEmpty a -> a
NEL.last (NonEmpty Member -> Member)
-> (ExtraBlock -> NonEmpty Member) -> ExtraBlock -> Member
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ExtraBlock -> NonEmpty Member
extraBlockMembers (ExtraBlock -> SourcePos) -> ExtraBlock -> SourcePos
forall a b. (a -> b) -> a -> b
$ ExtraBlock
ex

-- | Represents an entity member as a list of BlockAttrs
--
-- @since 2.16.0.0
memberBlockAttrs :: Member -> [BlockAttr]
memberBlockAttrs :: Member -> [BlockAttr]
memberBlockAttrs (MemberBlockAttr BlockAttr
fs) = [BlockAttr
fs]
memberBlockAttrs (MemberExtraBlock ExtraBlock
ex) = (Member -> [BlockAttr]) -> NonEmpty Member -> [BlockAttr]
forall m a. Monoid m => (a -> m) -> NonEmpty a -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap Member -> [BlockAttr]
memberBlockAttrs (NonEmpty Member -> [BlockAttr])
-> (ExtraBlock -> NonEmpty Member) -> ExtraBlock -> [BlockAttr]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ExtraBlock -> NonEmpty Member
extraBlockMembers (ExtraBlock -> [BlockAttr]) -> ExtraBlock -> [BlockAttr]
forall a b. (a -> b) -> a -> b
$ ExtraBlock
ex

extraBlocksAsMap :: [ExtraBlock] -> M.Map Text [ExtraLine]
extraBlocksAsMap :: [ExtraBlock] -> Map Text [[Text]]
extraBlocksAsMap [ExtraBlock]
exs = [(Text, [[Text]])] -> Map Text [[Text]]
forall k a. Ord k => [(k, a)] -> Map k a
M.fromList ([(Text, [[Text]])] -> Map Text [[Text]])
-> [(Text, [[Text]])] -> Map Text [[Text]]
forall a b. (a -> b) -> a -> b
$ (ExtraBlock -> (Text, [[Text]]))
-> [ExtraBlock] -> [(Text, [[Text]])]
forall a b. (a -> b) -> [a] -> [b]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ExtraBlock -> (Text, [[Text]])
asPair [ExtraBlock]
exs
  where
    asPair :: ExtraBlock -> (Text, [[Text]])
asPair ExtraBlock
ex =
        (ExtraBlockHeader -> Text
extraBlockHeaderKey (ExtraBlockHeader -> Text)
-> (ExtraBlock -> ExtraBlockHeader) -> ExtraBlock -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ExtraBlock -> ExtraBlockHeader
extraBlockExtraBlockHeader (ExtraBlock -> Text) -> ExtraBlock -> Text
forall a b. (a -> b) -> a -> b
$ ExtraBlock
ex, ExtraBlock -> [[Text]]
extraLines ExtraBlock
ex)
    extraLines :: ExtraBlock -> [[Text]]
extraLines ExtraBlock
ex = (Member -> [[Text]]) -> NonEmpty Member -> [[Text]]
forall m a. Monoid m => (a -> m) -> NonEmpty a -> m
forall (t :: * -> *) m a.
(Foldable t, Monoid m) =>
(a -> m) -> t a -> m
foldMap Member -> [[Text]]
asExtraLine (ExtraBlock -> NonEmpty Member
extraBlockMembers ExtraBlock
ex)
    asExtraLine :: Member -> [[Text]]
asExtraLine (MemberBlockAttr BlockAttr
fs) = [Token -> Text
tokenContent (Token -> Text) -> [Token] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> BlockAttr -> [Token]
blockAttrTokens BlockAttr
fs]
    asExtraLine Member
_ = []

entityHeader :: Parser EntityHeader
entityHeader :: Parser EntityHeader
entityHeader = do
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    Maybe Char
plus <- Parser Char -> Parser (Maybe Char)
forall (f :: * -> *) a. Alternative f => f a -> f (Maybe a)
optional (Token String -> Parser (Token String)
forall e s (m :: * -> *).
(MonadParsec e s m, Token s ~ Char) =>
Token s -> m (Token s)
char Char
Token String
'+')
    Token
en <- Parser ()
validHSpace Parser () -> Parser Token -> Parser Token
forall a b. Parser a -> Parser b -> Parser b
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser () -> Parser Token -> Parser Token
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer Parser Token
blockKey
    [Token]
rest <- Parser () -> Parser [Token] -> Parser [Token]
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer (Parser Token -> Parser [Token]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many Parser Token
anyToken)
    ()
_ <- Parser ()
setLastDocumentablePosition
    EntityHeader -> Parser EntityHeader
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure
        EntityHeader
            { entityHeaderSum :: Bool
entityHeaderSum = Maybe Char -> Bool
forall a. Maybe a -> Bool
isJust Maybe Char
plus
            , entityHeaderTableName :: Text
entityHeaderTableName = Token -> Text
tokenContent Token
en
            , entityHeaderRemainingTokens :: [Token]
entityHeaderRemainingTokens = [Token]
rest
            , entityHeaderPos :: SourcePos
entityHeaderPos = SourcePos
pos
            }

appendCommentToState :: (SourcePos, CommentToken) -> Parser ()
appendCommentToState :: (SourcePos, CommentToken) -> Parser ()
appendCommentToState (SourcePos, CommentToken)
ptok =
    (ExtraState -> ExtraState) -> Parser ()
forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify ((ExtraState -> ExtraState) -> Parser ())
-> (ExtraState -> ExtraState) -> Parser ()
forall a b. (a -> b) -> a -> b
$ \ExtraState
es ->
        let
            comments :: [(SourcePos, CommentToken)]
comments = ExtraState -> [(SourcePos, CommentToken)]
esPositionedCommentTokens ExtraState
es
         in
            ExtraState
es{esPositionedCommentTokens = ptok : comments}

setLastDocumentablePosition :: Parser ()
setLastDocumentablePosition :: Parser ()
setLastDocumentablePosition = do
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    (ExtraState -> ExtraState) -> Parser ()
forall s (m :: * -> *). MonadState s m => (s -> s) -> m ()
modify ((ExtraState -> ExtraState) -> Parser ())
-> (ExtraState -> ExtraState) -> Parser ()
forall a b. (a -> b) -> a -> b
$ \ExtraState
es -> ExtraState
es{esLastDocumentablePosition = Just pos}

getDcb :: Parser (Maybe DocCommentBlock)
getDcb :: Parser (Maybe DocCommentBlock)
getDcb = do
    ExtraState
es <- Parser ExtraState
forall s (m :: * -> *). MonadState s m => m s
get
    let
        comments :: [(SourcePos, CommentToken)]
comments = [(SourcePos, CommentToken)] -> [(SourcePos, CommentToken)]
forall a. [a] -> [a]
reverse ([(SourcePos, CommentToken)] -> [(SourcePos, CommentToken)])
-> [(SourcePos, CommentToken)] -> [(SourcePos, CommentToken)]
forall a b. (a -> b) -> a -> b
$ ExtraState -> [(SourcePos, CommentToken)]
esPositionedCommentTokens ExtraState
es
    ()
_ <- ExtraState -> Parser ()
forall s (m :: * -> *). MonadState s m => s -> m ()
put ExtraState
es{esPositionedCommentTokens = []}
    let
        candidates :: [(SourcePos, CommentToken)]
candidates = ((SourcePos, CommentToken) -> Bool)
-> [(SourcePos, CommentToken)] -> [(SourcePos, CommentToken)]
forall a. (a -> Bool) -> [a] -> [a]
dropWhile (\(SourcePos
_sp, CommentToken
ct) -> Bool -> Bool
not (CommentToken -> Bool
isDocComment CommentToken
ct)) [(SourcePos, CommentToken)]
comments
        filteredCandidates :: [(SourcePos, CommentToken)]
filteredCandidates = ((SourcePos, CommentToken) -> Bool)
-> [(SourcePos, CommentToken)] -> [(SourcePos, CommentToken)]
forall a. (a -> Bool) -> [a] -> [a]
dropWhile (ExtraState -> (SourcePos, CommentToken) -> Bool
commentIsIncorrectlyPositioned ExtraState
es) [(SourcePos, CommentToken)]
candidates
    Maybe DocCommentBlock -> Parser (Maybe DocCommentBlock)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe DocCommentBlock -> Parser (Maybe DocCommentBlock))
-> Maybe DocCommentBlock -> Parser (Maybe DocCommentBlock)
forall a b. (a -> b) -> a -> b
$ [(SourcePos, CommentToken)] -> Maybe DocCommentBlock
docCommentBlockFromPositionedTokens [(SourcePos, CommentToken)]
filteredCandidates
  where
    commentIsIncorrectlyPositioned
        :: ExtraState -> (SourcePos, CommentToken) -> Bool
    commentIsIncorrectlyPositioned :: ExtraState -> (SourcePos, CommentToken) -> Bool
commentIsIncorrectlyPositioned ExtraState
es (SourcePos, CommentToken)
ptok = case ExtraState -> Maybe SourcePos
esLastDocumentablePosition ExtraState
es of
        Maybe SourcePos
Nothing -> Bool
False
        Just SourcePos
lastDocumentablePos -> (SourcePos -> Pos
sourceLine (SourcePos -> Pos)
-> ((SourcePos, CommentToken) -> SourcePos)
-> (SourcePos, CommentToken)
-> Pos
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SourcePos, CommentToken) -> SourcePos
forall a b. (a, b) -> a
fst) (SourcePos, CommentToken)
ptok Pos -> Pos -> Bool
forall a. Ord a => a -> a -> Bool
<= SourcePos -> Pos
sourceLine SourcePos
lastDocumentablePos

extraBlock :: Parser Member
extraBlock :: Parser Member
extraBlock = Parser ()
-> Parser (IndentOpt Parser Member Member) -> Parser Member
forall s e (m :: * -> *) a b.
(TraversableStream s, MonadParsec e s m, Token s ~ Char) =>
m () -> m (IndentOpt m a b) -> m a
L.indentBlock Parser ()
spaceConsumerN Parser (IndentOpt Parser Member Member)
innerParser
  where
    mkExtraBlockMember :: Maybe DocCommentBlock -> (ExtraBlockHeader, [Member]) -> Member
mkExtraBlockMember Maybe DocCommentBlock
dcb (ExtraBlockHeader
header, [Member]
blockAttrs) =
        ExtraBlock -> Member
MemberExtraBlock
            ExtraBlock
                { extraBlockExtraBlockHeader :: ExtraBlockHeader
extraBlockExtraBlockHeader = ExtraBlockHeader
header
                , extraBlockMembers :: NonEmpty Member
extraBlockMembers = [Member] -> NonEmpty Member
forall {a}. [a] -> NonEmpty a
ensureNonEmpty [Member]
blockAttrs
                , extraBlockDocCommentBlock :: Maybe DocCommentBlock
extraBlockDocCommentBlock = Maybe DocCommentBlock
dcb
                }
    ensureNonEmpty :: [a] -> NonEmpty a
ensureNonEmpty [a]
members = case [a] -> Maybe (NonEmpty a)
forall a. [a] -> Maybe (NonEmpty a)
NEL.nonEmpty [a]
members of
        Just NonEmpty a
nel -> NonEmpty a
nel
        Maybe (NonEmpty a)
Nothing -> String -> NonEmpty a
forall a. HasCallStack => String -> a
error String
"unreachable" -- members is known to be non-empty
    innerParser :: Parser (IndentOpt Parser Member Member)
innerParser = do
        Maybe DocCommentBlock
dcb <- Parser (Maybe DocCommentBlock)
getDcb
        ExtraBlockHeader
header <- Parser ExtraBlockHeader
extraBlockHeader
        IndentOpt Parser Member Member
-> Parser (IndentOpt Parser Member Member)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (IndentOpt Parser Member Member
 -> Parser (IndentOpt Parser Member Member))
-> IndentOpt Parser Member Member
-> Parser (IndentOpt Parser Member Member)
forall a b. (a -> b) -> a -> b
$
            Maybe Pos
-> ([Member] -> Parser Member)
-> Parser Member
-> IndentOpt Parser Member Member
forall (m :: * -> *) a b.
Maybe Pos -> ([b] -> m a) -> m b -> IndentOpt m a b
L.IndentSome Maybe Pos
forall a. Maybe a
Nothing (Member -> Parser Member
forall a. a -> Parser a
forall (m :: * -> *) a. Monad m => a -> m a
return (Member -> Parser Member)
-> ([Member] -> Member) -> [Member] -> Parser Member
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Maybe DocCommentBlock -> (ExtraBlockHeader, [Member]) -> Member
mkExtraBlockMember Maybe DocCommentBlock
dcb ((ExtraBlockHeader, [Member]) -> Member)
-> ([Member] -> (ExtraBlockHeader, [Member])) -> [Member] -> Member
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (ExtraBlockHeader
header,)) Parser Member
blockAttr

extraBlockHeader :: Parser ExtraBlockHeader
extraBlockHeader :: Parser ExtraBlockHeader
extraBlockHeader = do
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    Token
tn <- Parser () -> Parser Token -> Parser Token
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer Parser Token
blockKey
    [Token]
rest <- Parser () -> Parser [Token] -> Parser [Token]
forall e s (m :: * -> *) a. MonadParsec e s m => m () -> m a -> m a
L.lexeme Parser ()
spaceConsumer (Parser Token -> Parser [Token]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many Parser Token
anyToken)
    ()
_ <- Parser ()
setLastDocumentablePosition
    ExtraBlockHeader -> Parser ExtraBlockHeader
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (ExtraBlockHeader -> Parser ExtraBlockHeader)
-> ExtraBlockHeader -> Parser ExtraBlockHeader
forall a b. (a -> b) -> a -> b
$
        ExtraBlockHeader
            { extraBlockHeaderKey :: Text
extraBlockHeaderKey = Token -> Text
tokenContent Token
tn
            , extraBlockHeaderRemainingTokens :: [Token]
extraBlockHeaderRemainingTokens = [Token]
rest
            , extraBlockHeaderPos :: SourcePos
extraBlockHeaderPos = SourcePos
pos
            }

blockAttr :: Parser Member
blockAttr :: Parser Member
blockAttr = do
    Maybe DocCommentBlock
dcb <- Parser (Maybe DocCommentBlock)
getDcb
    SourcePos
pos <- Parser SourcePos
forall s e (m :: * -> *).
(TraversableStream s, MonadParsec e s m) =>
m SourcePos
getSourcePos
    [Token]
line <- Parser Token -> Parser [Token]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
some Parser Token
anyToken
    ()
_ <- Parser ()
setLastDocumentablePosition
    Member -> Parser Member
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Member -> Parser Member) -> Member -> Parser Member
forall a b. (a -> b) -> a -> b
$
        BlockAttr -> Member
MemberBlockAttr
            BlockAttr
                { blockAttrDocCommentBlock :: Maybe DocCommentBlock
blockAttrDocCommentBlock = Maybe DocCommentBlock
dcb
                , blockAttrTokens :: [Token]
blockAttrTokens = [Token]
line
                , blockAttrPos :: SourcePos
blockAttrPos = SourcePos
pos
                }

member :: Parser Member
member :: Parser Member
member = Parser Member -> Parser Member
forall a. Parser a -> Parser a
forall e s (m :: * -> *) a. MonadParsec e s m => m a -> m a
try Parser Member
extraBlock Parser Member -> Parser Member -> Parser Member
forall a. Parser a -> Parser a -> Parser a
forall (f :: * -> *) a. Alternative f => f a -> f a -> f a
<|> Parser Member
blockAttr

entityBlock :: Parser EntityBlock
entityBlock :: Parser EntityBlock
entityBlock = do
    Parser ()
-> Parser (IndentOpt Parser EntityBlock Member)
-> Parser EntityBlock
forall s e (m :: * -> *) a b.
(TraversableStream s, MonadParsec e s m, Token s ~ Char) =>
m () -> m (IndentOpt m a b) -> m a
L.indentBlock Parser ()
spaceConsumerN Parser (IndentOpt Parser EntityBlock Member)
innerParser
  where
    mkEntityBlock :: Maybe DocCommentBlock -> (EntityHeader, [Member]) -> EntityBlock
mkEntityBlock Maybe DocCommentBlock
dcb (EntityHeader
header, [Member]
members) =
        EntityBlock
            { entityBlockEntityHeader :: EntityHeader
entityBlockEntityHeader = EntityHeader
header
            , entityBlockMembers :: [Member]
entityBlockMembers = [Member]
members
            , entityBlockDocCommentBlock :: Maybe DocCommentBlock
entityBlockDocCommentBlock = Maybe DocCommentBlock
dcb
            }
    innerParser :: Parser (IndentOpt Parser EntityBlock Member)
innerParser = do
        Maybe DocCommentBlock
dcb <- Parser (Maybe DocCommentBlock)
getDcb
        EntityHeader
header <- Parser EntityHeader
entityHeader
        IndentOpt Parser EntityBlock Member
-> Parser (IndentOpt Parser EntityBlock Member)
forall a. a -> Parser a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (IndentOpt Parser EntityBlock Member
 -> Parser (IndentOpt Parser EntityBlock Member))
-> IndentOpt Parser EntityBlock Member
-> Parser (IndentOpt Parser EntityBlock Member)
forall a b. (a -> b) -> a -> b
$ Maybe Pos
-> ([Member] -> Parser EntityBlock)
-> Parser Member
-> IndentOpt Parser EntityBlock Member
forall (m :: * -> *) a b.
Maybe Pos -> ([b] -> m a) -> m b -> IndentOpt m a b
L.IndentMany Maybe Pos
forall a. Maybe a
Nothing (EntityBlock -> Parser EntityBlock
forall a. a -> Parser a
forall (m :: * -> *) a. Monad m => a -> m a
return (EntityBlock -> Parser EntityBlock)
-> ([Member] -> EntityBlock) -> [Member] -> Parser EntityBlock
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Maybe DocCommentBlock -> (EntityHeader, [Member]) -> EntityBlock
mkEntityBlock Maybe DocCommentBlock
dcb ((EntityHeader, [Member]) -> EntityBlock)
-> ([Member] -> (EntityHeader, [Member]))
-> [Member]
-> EntityBlock
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (EntityHeader
header,)) Parser Member
member

entitiesFromDocument :: Parser [EntityBlock]
entitiesFromDocument :: Parser [EntityBlock]
entitiesFromDocument = Parser EntityBlock -> Parser [EntityBlock]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
many Parser EntityBlock
entityBlock

docCommentBlockText :: DocCommentBlock -> Text
docCommentBlockText :: DocCommentBlock -> Text
docCommentBlockText DocCommentBlock
dcb = [Text] -> Text
Text.unlines ([Text] -> Text) -> [Text] -> Text
forall a b. (a -> b) -> a -> b
$ DocCommentBlock -> [Text]
docCommentBlockLines DocCommentBlock
dcb

isDocComment :: CommentToken -> Bool
isDocComment :: CommentToken -> Bool
isDocComment CommentToken
tok = case CommentToken
tok of
    DocComment Text
_ -> Bool
True
    CommentToken
_ -> Bool
False

docCommentBlockFromPositionedTokens
    :: [(SourcePos, CommentToken)] -> Maybe DocCommentBlock
docCommentBlockFromPositionedTokens :: [(SourcePos, CommentToken)] -> Maybe DocCommentBlock
docCommentBlockFromPositionedTokens [(SourcePos, CommentToken)]
ptoks =
    case [(SourcePos, CommentToken)]
-> Maybe (NonEmpty (SourcePos, CommentToken))
forall a. [a] -> Maybe (NonEmpty a)
NEL.nonEmpty [(SourcePos, CommentToken)]
ptoks of
        Maybe (NonEmpty (SourcePos, CommentToken))
Nothing -> Maybe DocCommentBlock
forall a. Maybe a
Nothing
        Just NonEmpty (SourcePos, CommentToken)
nel ->
            DocCommentBlock -> Maybe DocCommentBlock
forall a. a -> Maybe a
Just (DocCommentBlock -> Maybe DocCommentBlock)
-> DocCommentBlock -> Maybe DocCommentBlock
forall a b. (a -> b) -> a -> b
$
                DocCommentBlock
                    { docCommentBlockLines :: [Text]
docCommentBlockLines = NonEmpty Text -> [Text]
forall a. NonEmpty a -> [a]
NEL.toList (NonEmpty Text -> [Text]) -> NonEmpty Text -> [Text]
forall a b. (a -> b) -> a -> b
$ ((SourcePos, CommentToken) -> Text)
-> NonEmpty (SourcePos, CommentToken) -> NonEmpty Text
forall a b. (a -> b) -> NonEmpty a -> NonEmpty b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (CommentToken -> Text
commentContent (CommentToken -> Text)
-> ((SourcePos, CommentToken) -> CommentToken)
-> (SourcePos, CommentToken)
-> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SourcePos, CommentToken) -> CommentToken
forall a b. (a, b) -> b
snd) NonEmpty (SourcePos, CommentToken)
nel
                    , docCommentBlockPos :: SourcePos
docCommentBlockPos = (SourcePos, CommentToken) -> SourcePos
forall a b. (a, b) -> a
fst ((SourcePos, CommentToken) -> SourcePos)
-> (SourcePos, CommentToken) -> SourcePos
forall a b. (a -> b) -> a -> b
$ NonEmpty (SourcePos, CommentToken) -> (SourcePos, CommentToken)
forall a. NonEmpty a -> a
NEL.head NonEmpty (SourcePos, CommentToken)
nel
                    }

parseEntities
    :: PersistSettings
    -> Text
    -> String
    -> ParseResult [EntityBlock]
parseEntities :: PersistSettings -> Text -> String -> ParseResult [EntityBlock]
parseEntities PersistSettings
ps Text
fp String
s = do
    let
        (Set ParserWarning
warnings, Either (ParseErrorBundle String Void) ([EntityBlock], ExtraState)
res) =
            PersistSettings
-> ExtraState
-> Parser [EntityBlock]
-> String
-> String
-> (Set ParserWarning,
    Either (ParseErrorBundle String Void) ([EntityBlock], ExtraState))
forall a.
PersistSettings
-> ExtraState
-> Parser a
-> String
-> String
-> InternalParseResult a
runConfiguredParser PersistSettings
ps ExtraState
initialExtraState Parser [EntityBlock]
entitiesFromDocument (Text -> String
Text.unpack Text
fp) String
s
    case Either (ParseErrorBundle String Void) ([EntityBlock], ExtraState)
res of
        Left ParseErrorBundle String Void
peb ->
            (Set ParserWarning
warnings, ParseErrorBundle String Void
-> Either (ParseErrorBundle String Void) [EntityBlock]
forall a b. a -> Either a b
Left ParseErrorBundle String Void
peb)
        Right ([EntityBlock]
entities, ExtraState
_comments) ->
            (Set ParserWarning
warnings, [EntityBlock]
-> Either (ParseErrorBundle String Void) [EntityBlock]
forall a. a -> Either (ParseErrorBundle String Void) a
forall (f :: * -> *) a. Applicative f => a -> f a
pure [EntityBlock]
entities)

toParsedEntityDef :: Maybe SourceLoc -> EntityBlock -> ParsedEntityDef
toParsedEntityDef :: Maybe SourceLoc -> EntityBlock -> ParsedEntityDef
toParsedEntityDef Maybe SourceLoc
mSourceLoc EntityBlock
eb =
    ParsedEntityDef
        { parsedEntityDefComments :: [Text]
parsedEntityDefComments = [Text]
comments
        , parsedEntityDefEntityName :: EntityNameHS
parsedEntityDefEntityName = EntityNameHS
entityNameHS
        , parsedEntityDefIsSum :: Bool
parsedEntityDefIsSum = Bool
isSum
        , parsedEntityDefEntityAttributes :: [Text]
parsedEntityDefEntityAttributes = [Text]
entityAttributes
        , parsedEntityDefFieldAttributes :: [([Token], Maybe Text)]
parsedEntityDefFieldAttributes = [([Token], Maybe Text)]
parsedFieldAttributes
        , parsedEntityDefExtras :: Map Text [[Text]]
parsedEntityDefExtras = Map Text [[Text]]
extras
        , parsedEntityDefSpan :: Maybe SourceSpan
parsedEntityDefSpan = Maybe SourceSpan
mSpan
        }
  where
    comments :: [Text]
comments =
        [Text]
-> (DocCommentBlock -> [Text]) -> Maybe DocCommentBlock -> [Text]
forall b a. b -> (a -> b) -> Maybe a -> b
maybe
            []
            DocCommentBlock -> [Text]
docCommentBlockLines
            (EntityBlock -> Maybe DocCommentBlock
entityBlockDocCommentBlock EntityBlock
eb)
    entityAttributes :: [Text]
entityAttributes =
        Token -> Text
tokenContent (Token -> Text) -> [Token] -> [Text]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (EntityHeader -> [Token]
entityHeaderRemainingTokens (EntityHeader -> [Token])
-> (EntityBlock -> EntityHeader) -> EntityBlock -> [Token]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EntityBlock -> EntityHeader
entityBlockEntityHeader) EntityBlock
eb
    isSum :: Bool
isSum = EntityHeader -> Bool
entityHeaderSum (EntityHeader -> Bool)
-> (EntityBlock -> EntityHeader) -> EntityBlock -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EntityBlock -> EntityHeader
entityBlockEntityHeader (EntityBlock -> Bool) -> EntityBlock -> Bool
forall a b. (a -> b) -> a -> b
$ EntityBlock
eb
    entityNameHS :: EntityNameHS
entityNameHS = Text -> EntityNameHS
EntityNameHS (Text -> EntityNameHS)
-> (EntityBlock -> Text) -> EntityBlock -> EntityNameHS
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EntityHeader -> Text
entityHeaderTableName (EntityHeader -> Text)
-> (EntityBlock -> EntityHeader) -> EntityBlock -> Text
forall b c a. (b -> c) -> (a -> b) -> a -> c
. EntityBlock -> EntityHeader
entityBlockEntityHeader (EntityBlock -> EntityNameHS) -> EntityBlock -> EntityNameHS
forall a b. (a -> b) -> a -> b
$ EntityBlock
eb

    attributePair :: BlockAttr -> ([Token], Maybe Text)
attributePair BlockAttr
a = (BlockAttr -> [Token]
blockAttrTokens BlockAttr
a, DocCommentBlock -> Text
docCommentBlockText (DocCommentBlock -> Text) -> Maybe DocCommentBlock -> Maybe Text
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> BlockAttr -> Maybe DocCommentBlock
blockAttrDocCommentBlock BlockAttr
a)
    parsedFieldAttributes :: [([Token], Maybe Text)]
parsedFieldAttributes = (BlockAttr -> ([Token], Maybe Text))
-> [BlockAttr] -> [([Token], Maybe Text)]
forall a b. (a -> b) -> [a] -> [b]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap BlockAttr -> ([Token], Maybe Text)
attributePair (EntityBlock -> [BlockAttr]
entityBlockBlockAttrs EntityBlock
eb)

    extras :: Map Text [[Text]]
extras = [ExtraBlock] -> Map Text [[Text]]
extraBlocksAsMap (EntityBlock -> [ExtraBlock]
entityBlockExtraBlocks EntityBlock
eb)
    filepath :: Text
filepath = Text -> (SourceLoc -> Text) -> Maybe SourceLoc -> Text
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Text
"" SourceLoc -> Text
locFile Maybe SourceLoc
mSourceLoc
    relativeStartLine :: Int
relativeStartLine = Int -> (SourceLoc -> Int) -> Maybe SourceLoc -> Int
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Int
0 SourceLoc -> Int
locStartLine Maybe SourceLoc
mSourceLoc
    relativeStartCol :: Int
relativeStartCol = Int -> (SourceLoc -> Int) -> Maybe SourceLoc -> Int
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Int
0 SourceLoc -> Int
locStartCol Maybe SourceLoc
mSourceLoc
    mSpan :: Maybe SourceSpan
mSpan =
        SourceSpan -> Maybe SourceSpan
forall a. a -> Maybe a
Just
            SourceSpan
                { spanFile :: Text
spanFile = Text
filepath
                , spanStartLine :: Int
spanStartLine =
                    Int
relativeStartLine Int -> Int -> Int
forall a. Num a => a -> a -> a
+ (Pos -> Int
unPos (Pos -> Int) -> (SourcePos -> Pos) -> SourcePos -> Int
forall b c a. (b -> c) -> (a -> b) -> a -> c
. SourcePos -> Pos
sourceLine (SourcePos -> Int) -> SourcePos -> Int
forall a b. (a -> b) -> a -> b
$ EntityBlock -> SourcePos
entityBlockFirstPos EntityBlock
eb)
                , spanEndLine :: Int
spanEndLine = Int
relativeStartLine Int -> Int -> Int
forall a. Num a => a -> a -> a
+ (Pos -> Int
unPos (Pos -> Int) -> (SourcePos -> Pos) -> SourcePos -> Int
forall b c a. (b -> c) -> (a -> b) -> a -> c
. SourcePos -> Pos
sourceLine (SourcePos -> Int) -> SourcePos -> Int
forall a b. (a -> b) -> a -> b
$ EntityBlock -> SourcePos
entityBlockLastPos EntityBlock
eb)
                , spanStartCol :: Int
spanStartCol =
                    Int
relativeStartCol Int -> Int -> Int
forall a. Num a => a -> a -> a
+ (Pos -> Int
unPos (Pos -> Int) -> (SourcePos -> Pos) -> SourcePos -> Int
forall b c a. (b -> c) -> (a -> b) -> a -> c
. SourcePos -> Pos
sourceColumn (SourcePos -> Int) -> SourcePos -> Int
forall a b. (a -> b) -> a -> b
$ EntityBlock -> SourcePos
entityBlockFirstPos EntityBlock
eb)
                , spanEndCol :: Int
spanEndCol = Pos -> Int
unPos (Pos -> Int) -> (SourcePos -> Pos) -> SourcePos -> Int
forall b c a. (b -> c) -> (a -> b) -> a -> c
. SourcePos -> Pos
sourceColumn (SourcePos -> Int) -> SourcePos -> Int
forall a b. (a -> b) -> a -> b
$ EntityBlock -> SourcePos
entityBlockLastPos EntityBlock
eb
                }

parseSource
    :: PersistSettings
    -> Maybe SourceLoc
    -> Text
    -> ParseResult [ParsedEntityDef]
parseSource :: PersistSettings
-> Maybe SourceLoc -> Text -> ParseResult [ParsedEntityDef]
parseSource PersistSettings
ps Maybe SourceLoc
mSourceLoc Text
source =
    ([EntityBlock] -> [ParsedEntityDef])
-> Either (ParseErrorBundle String Void) [EntityBlock]
-> Either (ParseErrorBundle String Void) [ParsedEntityDef]
forall a b.
(a -> b)
-> Either (ParseErrorBundle String Void) a
-> Either (ParseErrorBundle String Void) b
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ((EntityBlock -> ParsedEntityDef)
-> [EntityBlock] -> [ParsedEntityDef]
forall a b. (a -> b) -> [a] -> [b]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (Maybe SourceLoc -> EntityBlock -> ParsedEntityDef
toParsedEntityDef Maybe SourceLoc
mSourceLoc))
        (Either (ParseErrorBundle String Void) [EntityBlock]
 -> Either (ParseErrorBundle String Void) [ParsedEntityDef])
-> ParseResult [EntityBlock] -> ParseResult [ParsedEntityDef]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> PersistSettings -> Text -> String -> ParseResult [EntityBlock]
parseEntities PersistSettings
ps Text
filepath (Text -> String
Text.unpack Text
source)
  where
    filepath :: Text
filepath = Text -> (SourceLoc -> Text) -> Maybe SourceLoc -> Text
forall b a. b -> (a -> b) -> Maybe a -> b
maybe Text
"" SourceLoc -> Text
locFile Maybe SourceLoc
mSourceLoc