pretty
This commit is contained in:
crumbtoo
172
src/Rlp/AltParse.y
Normal file
172
src/Rlp/AltParse.y
Normal file
@@ -0,0 +1,172 @@
|
||||
{
|
||||
module Rlp.AltParse
|
||||
( parseRlpProg
|
||||
, parseRlpProgR
|
||||
, runP'
|
||||
)
|
||||
where
|
||||
import Data.List.Extra
|
||||
import Data.Text (Text)
|
||||
|
||||
import Control.Comonad
|
||||
import Control.Comonad.Cofree
|
||||
import Control.Lens hiding (snoc)
|
||||
|
||||
import Compiler.RlpcError
|
||||
import Compiler.RLPC
|
||||
import Control.Monad.Errorful
|
||||
|
||||
import Rlp.Lex
|
||||
import Rlp.AltSyntax
|
||||
import Rlp.Parse.Types hiding (PsName)
|
||||
import Core.Syntax qualified as Core
|
||||
}
|
||||
|
||||
%name parseRlpProg StandaloneProgram
|
||||
%name parseRlpExpr StandaloneExpr
|
||||
|
||||
%monad { P }
|
||||
%lexer { lexCont } { Located _ TokenEOF }
|
||||
%error { parseError }
|
||||
%errorhandlertype explist
|
||||
%tokentype { Located RlpToken }
|
||||
|
||||
%token
|
||||
varname { Located _ (TokenVarName _) }
|
||||
conname { Located _ (TokenConName _) }
|
||||
consym { Located _ (TokenConSym _) }
|
||||
varsym { Located _ (TokenVarSym _) }
|
||||
data { Located _ TokenData }
|
||||
case { Located _ TokenCase }
|
||||
of { Located _ TokenOf }
|
||||
litint { Located _ (TokenLitInt _) }
|
||||
'=' { Located _ TokenEquals }
|
||||
'|' { Located _ TokenPipe }
|
||||
'::' { Located _ TokenHasType }
|
||||
';' { Located _ TokenSemicolon }
|
||||
'(' { Located _ TokenLParen }
|
||||
')' { Located _ TokenRParen }
|
||||
'->' { Located _ TokenArrow }
|
||||
vsemi { Located _ TokenSemicolonV }
|
||||
'{' { Located _ TokenLBrace }
|
||||
'}' { Located _ TokenRBrace }
|
||||
vlbrace { Located _ TokenLBraceV }
|
||||
vrbrace { Located _ TokenRBraceV }
|
||||
infixl { Located _ TokenInfixL }
|
||||
infixr { Located _ TokenInfixR }
|
||||
infix { Located _ TokenInfix }
|
||||
let { Located _ TokenLet }
|
||||
letrec { Located _ TokenLetrec }
|
||||
in { Located _ TokenIn }
|
||||
|
||||
%nonassoc '='
|
||||
%right '->'
|
||||
%right in
|
||||
|
||||
%%
|
||||
|
||||
StandaloneProgram :: { Program Name (RlpExpr PsName) }
|
||||
: layout0(Decl) { Program $1 }
|
||||
|
||||
|
||||
StandaloneExpr :: { RlpExpr PsName }
|
||||
: litint { undefined }
|
||||
|
||||
VL :: { () }
|
||||
VL : vlbrace { () }
|
||||
|
||||
VR :: { () }
|
||||
VR : vrbrace { () }
|
||||
| error { () }
|
||||
|
||||
VS :: { () }
|
||||
VS : ';' { () }
|
||||
| vsemi { () }
|
||||
|
||||
Decl :: { Decl PsName (RlpExpr PsName) }
|
||||
: FunD { $1 }
|
||||
| DataD { $1 }
|
||||
|
||||
DataD :: { Decl PsName (RlpExpr PsName) }
|
||||
: data Con TyVars { DataD $2 $3 [] }
|
||||
| data Con TyVars '=' DataCons { DataD $2 $3 $5 }
|
||||
|
||||
DataCons :: { [DataCon PsName] }
|
||||
: DataCon '|' DataCons { $1 : $3 }
|
||||
| DataCon { [$1] }
|
||||
|
||||
DataCon :: { DataCon PsName }
|
||||
: Con list0(Type1) { DataCon $1 $2 }
|
||||
|
||||
Type1 :: { Type PsName }
|
||||
: varname { VarT $ extractVarName $1 }
|
||||
| Con { ConT $1 }
|
||||
|
||||
TyVars :: { [PsName] }
|
||||
: list0(varname) { $1 <&> view ( to extract
|
||||
. singular _TokenVarName) }
|
||||
|
||||
FunD :: { Decl PsName (RlpExpr PsName) }
|
||||
: Var Pat1s '=' Expr { FunD $1 $2 $4 }
|
||||
|
||||
Expr :: { RlpExpr PsName }
|
||||
: AppE { $1 }
|
||||
|
||||
AppE :: { RlpExpr PsName }
|
||||
: AppE VarE { Finl $ Core.AppF $1 $2 }
|
||||
| VarE { $1 }
|
||||
|
||||
VarE :: { RlpExpr PsName }
|
||||
: Var { Finl $ Core.VarF $1 }
|
||||
|
||||
Pat1s :: { [Pat PsName] }
|
||||
: list0(Pat1) { $1 }
|
||||
|
||||
Pat1 :: { Pat PsName }
|
||||
: Var { VarP $1 }
|
||||
|
||||
Con :: { PsName }
|
||||
: conname { $1 ^. to extract
|
||||
. singular _TokenConName }
|
||||
| '(' consym ')' { $1 ^. to extract
|
||||
. singular _TokenConSym }
|
||||
|
||||
Var :: { PsName }
|
||||
: varname { $1 ^. to extract
|
||||
. singular _TokenVarName }
|
||||
| '(' varsym ')' { $2 ^. to extract
|
||||
. singular _TokenVarSym }
|
||||
|
||||
-- list0(p : α) : [α]
|
||||
list0(p) : {- epsilon -} { [] }
|
||||
| list0(p) p { $1 `snoc` $2 }
|
||||
|
||||
-- layout0(p : β) :: [β]
|
||||
layout0(p) : '{' layout_list0(';',p) '}' { $2 }
|
||||
| VL layout_list0(VS,p) VR { $2 }
|
||||
|
||||
-- layout_list0(sep : α, p : β) :: [β]
|
||||
layout_list0(sep,p) : p { [$1] }
|
||||
| layout_list1(sep,p) sep p { $1 `snoc` $3 }
|
||||
| {- epsilon -} { [] }
|
||||
|
||||
-- layout1(p : β) :: [β]
|
||||
layout1(p) : '{' layout_list1(';',p) '}' { $2 }
|
||||
| VL layout_list1(VS,p) VR { $2 }
|
||||
|
||||
-- layout_list1(sep : α, p : β) :: [β]
|
||||
layout_list1(sep,p) : p { [$1] }
|
||||
| layout_list1(sep,p) sep p { $1 `snoc` $3 }
|
||||
|
||||
{
|
||||
|
||||
extractVarName = view $ to extract . singular _TokenVarName
|
||||
|
||||
parseRlpProgR :: (Monad m) => Text -> RLPCT m (Program Name (RlpExpr PsName))
|
||||
parseRlpProgR s = liftErrorful $ errorful (ma,es)
|
||||
where
|
||||
(_,es,ma) = runP' parseRlpProg s
|
||||
|
||||
parseError = error "explode"
|
||||
|
||||
}
|
||||
91
src/Rlp/AltSyntax.hs
Normal file
91
src/Rlp/AltSyntax.hs
Normal file
@@ -0,0 +1,91 @@
|
||||
{-# LANGUAGE TemplateHaskell, PatternSynonyms #-}
|
||||
module Rlp.AltSyntax
|
||||
(
|
||||
-- * AST
|
||||
Program(..), Decl(..), ExprF(..), Pat(..)
|
||||
, RlpExprF, RlpExpr
|
||||
, DataCon(..), Type(..)
|
||||
|
||||
, Core.Name, PsName
|
||||
|
||||
-- * Functor-related tools
|
||||
, Fix(..), Cofree(..), Sum(..), pattern Finl, pattern Finr
|
||||
)
|
||||
where
|
||||
--------------------------------------------------------------------------------
|
||||
import Data.Functor.Sum
|
||||
import Control.Comonad.Cofree
|
||||
import Data.Fix
|
||||
|
||||
import Text.Show.Deriving
|
||||
import Data.Text qualified as T
|
||||
import Data.Pretty
|
||||
|
||||
import Compiler.Types
|
||||
import Core.Syntax qualified as Core
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
type PsName = T.Text
|
||||
|
||||
newtype Program b a = Program [Decl b a]
|
||||
deriving Show
|
||||
|
||||
data Decl b a = FunD b [Pat b] a
|
||||
| DataD b [b] [DataCon b]
|
||||
deriving Show
|
||||
|
||||
data DataCon b = DataCon b [Type b]
|
||||
deriving Show
|
||||
|
||||
data Type b = VarT b
|
||||
| ConT b
|
||||
| AppT (Type b) (Type b)
|
||||
| FunT
|
||||
deriving Show
|
||||
|
||||
data ExprF b a = InfixEF b a a
|
||||
|
||||
-- type Expr b = Cofree (ExprF b)
|
||||
|
||||
type RlpExprF b = Sum (Core.ExprF b) (ExprF b)
|
||||
|
||||
type RlpExpr b = Fix (RlpExprF b)
|
||||
|
||||
data Pat b = VarP b
|
||||
deriving Show
|
||||
|
||||
deriveShow1 ''ExprF
|
||||
deriving instance (Show b, Show a) => Show (ExprF b a)
|
||||
|
||||
pattern Finl :: f (Fix (Sum f g)) -> Fix (Sum f g)
|
||||
pattern Finl fa = Fix (InL fa)
|
||||
|
||||
pattern Finr :: g (Fix (Sum f g)) -> Fix (Sum f g)
|
||||
pattern Finr ga = Fix (InR ga)
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
instance (Pretty b, Pretty a) => Pretty (ExprF b a) where
|
||||
prettyPrec = prettyPrec1
|
||||
|
||||
instance Pretty b => Pretty1 (ExprF b) where
|
||||
liftPrettyPrec pr p (InfixEF o a b) = maybeParens (p>0) $
|
||||
pr 1 a <+> pretty o <+> pr 1 b
|
||||
|
||||
instance (Pretty b, Pretty a) => Pretty (Decl b a) where
|
||||
prettyPrec = prettyPrec1
|
||||
|
||||
instance (Pretty b) => Pretty1 (Decl b) where
|
||||
liftPrettyPrec pr p (FunD f as e) = maybeParens (p>0) $
|
||||
hsep [ ttext f, hsep (prettyPrec appPrec1 <$> as)
|
||||
, "=", pr 0 e ]
|
||||
|
||||
instance (Pretty b) => Pretty (Pat b) where
|
||||
prettyPrec p (VarP b) = prettyPrec p b
|
||||
|
||||
instance (Pretty a, Pretty b) => Pretty (Program b a) where
|
||||
prettyPrec = prettyPrec1
|
||||
|
||||
instance (Pretty b) => Pretty1 (Program b) where
|
||||
liftPrettyPrec pr p (Program ds) = vsep $ liftPrettyPrec pr p <$> ds
|
||||
|
||||
@@ -18,7 +18,7 @@ module Rlp.Parse.Types
|
||||
, RlpToken(..), AlexInput(..), Position(..), spanFromPos, LexerAction
|
||||
, Located(..), PsName
|
||||
-- ** Lenses
|
||||
, _TokenLitInt, _TokenVarName, _TokenConName, _TokenVarSym
|
||||
, _TokenLitInt, _TokenVarName, _TokenConName, _TokenVarSym, _TokenConSym
|
||||
, aiPrevChar, aiSource, aiBytes, aiPos, posLine, posColumn
|
||||
|
||||
-- * Error handling
|
||||
@@ -277,7 +277,7 @@ initAlexInput s = AlexInput
|
||||
{ _aiPrevChar = '\0'
|
||||
, _aiSource = s
|
||||
, _aiBytes = []
|
||||
, _aiPos = (1,1,0)
|
||||
, _aiPos = (1,0,0)
|
||||
}
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user