module DDC.Core.Parser.Base
( Parser
, pModuleName
, pQualName
, pName
, pCon, pConSP
, pLit, pLitSP
, pString, pStringSP
, pIndex, pIndexSP
, pVar, pVarSP, pVarNamedSP
, pTok, pTokSP
, pTokAs, pTokAsSP
, pOpSP
, pOpVarSP
, pPragmaSP)
where
import DDC.Base.Pretty
import DDC.Core.Module
import DDC.Core.Lexer.Tokens
import DDC.Base.Parser ((<?>), SourcePos)
import Data.Text (Text)
import qualified DDC.Base.Parser as P
type Parser n a
= P.Parser (Tok n) a
pModuleName :: Pretty n => Parser n ModuleName
pModuleName
= do ms <- P.sepBy1 pModuleName1 (pTok KDot)
return $ ModuleName
$ concat
$ map (\(ModuleName ss) -> ss) ms
pModuleName1 :: Pretty n => Parser n ModuleName
pModuleName1 = P.pTokMaybe f
where f (KN (KCon n)) = Just $ ModuleName [ renderPlain $ ppr n ]
f (KA (KSoConBuiltin c)) = Just $ ModuleName [ renderPlain $ ppr c ]
f (KA (KKiConBuiltin c)) = Just $ ModuleName [ renderPlain $ ppr c ]
f (KA (KTwConBuiltin c)) = Just $ ModuleName [ renderPlain $ ppr c ]
f (KA (KTcConBuiltin c)) = Just $ ModuleName [ renderPlain $ ppr c ]
f _ = Nothing
pQualName :: Pretty n => Parser n (QualName n)
pQualName
= do mn <- pModuleName
pTok KDot
n <- pName
return $ QualName mn n
pName :: Parser n n
pName = P.choice [pCon, pVar]
pCon :: Parser n n
pCon = P.pTokMaybe f
where f (KN (KCon n)) = Just n
f _ = Nothing
pConSP :: Parser n (n, SourcePos)
pConSP = P.pTokMaybeSP f
where f (KN (KCon n)) = Just n
f _ = Nothing
pLit :: Parser n n
pLit = P.pTokMaybe f
where f (KN (KLit n)) = Just n
f _ = Nothing
pLitSP :: Parser n (n, SourcePos)
pLitSP = P.pTokMaybeSP f
where f (KN (KLit n)) = Just n
f _ = Nothing
pString :: Parser n Text
pString = P.pTokMaybe f
where f (KA (KString tx)) = Just tx
f _ = Nothing
pStringSP :: Parser n (Text, SourcePos)
pStringSP = P.pTokMaybeSP f
where f (KA (KString tx)) = Just tx
f _ = Nothing
pVar :: Parser n n
pVar = P.pTokMaybe f
<?> "a variable"
where f (KN (KVar n)) = Just n
f _ = Nothing
pVarSP :: Parser n (n, SourcePos)
pVarSP = P.pTokMaybeSP f
<?> "a variable"
where f (KN (KVar n)) = Just n
f _ = Nothing
pVarNamedSP :: String -> Parser String SourcePos
pVarNamedSP str
= fmap snd (P.pTokMaybeSP f <?> "a variable")
where f (KN (KVar n)) | n == str = Just n
f _ = Nothing
pIndex :: Parser n Int
pIndex = P.pTokMaybe f
<?> "an index"
where f (KA (KIndex i)) = Just i
f _ = Nothing
pIndexSP :: Parser n (Int, SourcePos)
pIndexSP = P.pTokMaybeSP f
<?> "an index"
where f (KA (KIndex i)) = Just i
f _ = Nothing
pOpSP :: Parser n (String, SourcePos)
pOpSP = P.pTokMaybeSP f
where f (KA (KOp str)) = Just str
f _ = Nothing
pOpVarSP :: Parser n (String, SourcePos)
pOpVarSP = P.pTokMaybeSP f
where f (KA (KOpVar str)) = Just str
f _ = Nothing
pPragmaSP :: Parser n (Text, SourcePos)
pPragmaSP = P.pTokMaybeSP f
where f (KA (KPragma txt)) = Just txt
f _ = Nothing
pTok :: TokAtom -> Parser n ()
pTok k = P.pTok (KA k)
pTokSP :: TokAtom -> Parser n SourcePos
pTokSP k = P.pTokSP (KA k)
pTokAs :: TokAtom -> a -> Parser n a
pTokAs k x = P.pTokAs (KA k) x
pTokAsSP :: TokAtom -> a -> Parser n (a, SourcePos)
pTokAsSP k x = P.pTokAsSP (KA k) x