Abbie's Haskell compiler
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

143 lines
3.5 KiB

module Frontend.Syntax where
import Frontend.Parser.Posn
import qualified Data.Text as T
import Data.Text (Text)
data FeExpr var
= Ref var
| Con var
| App (FeExpr var) (FeExpr var)
| Lam (FePat var) (FeExpr var)
| Let [FeDecl var] (FeExpr var)
| Tuple [FeExpr var]
| Annot (FeExpr var) (FeType var)
| Literal Literal
| ParenExp (FeExpr var)
| SPExpr (FeExpr var) Posn Posn
deriving (Eq, Show, Ord)
instance HasPosn (FeExpr var) where
startPosn (SPExpr _ s _) = s
startPosn _ = error "no start posn in parsed expression?"
endPosn (SPExpr _ _ e) = e
endPosn _ = error "no end posn in parsed expression?"
span sp ep (SPExpr x _ _) = SPExpr x (startPosn sp) (endPosn ep)
span sp ep x = SPExpr x (startPosn sp) (endPosn ep)
data FePat var
= Var var
| Wildcard
| TupPat [FePat var]
| LitPat Literal
| ParenPat (FePat var) -- parsed parentheses
| SPPat (FePat var) Posn Posn
deriving (Eq, Show, Ord)
instance HasPosn (FePat var) where
startPosn (SPPat _ s _) = s
startPosn _ = error "no start posn in parsed expression?"
endPosn (SPPat _ _ e) = e
endPosn _ = error "no end posn in parsed pattern?"
span sp ep (SPPat x _ _) = SPPat x (startPosn sp) (endPosn ep)
span sp ep x = SPPat x (startPosn sp) (endPosn ep)
data FeType var
= Tyvar var
| Tycon var
| Tyapp (FeType var) (FeType var)
| Tyarr (FeType var) (FeType var)
| Tytup [FeType var]
| ParenType (FeType var) -- parsed parentheses
| SPType (FeType var) Posn Posn
deriving (Eq, Show, Ord)
instance HasPosn (FeType var) where
startPosn (SPType _ s _) = s
startPosn _ = error "no start posn in parsed type?"
endPosn (SPType _ _ e) = e
endPosn _ = error "no end posn in parsed type?"
span sp ep (SPType x _ _) = SPType x (startPosn sp) (endPosn ep)
span sp ep x = SPType x (startPosn sp) (endPosn ep)
data FeDecl var
= PatDecl (FePat var) (FeRhs var)
| FunDecl var [FePat var] (FeRhs var)
| TySig [var] (FeType var)
deriving (Eq, Show, Ord)
data FeRhs var
= BareRhs (FeExpr var) [FeDecl var]
deriving (Eq, Show, Ord)
data Literal
= LitString T.Text
| LitNumber Integer
deriving (Eq, Show, Ord)
data FeModule var
= Module { moduleName :: var
, moduleExports :: Maybe [NamespacedItem var]
, moduleItems :: [ModuleItem var]
}
deriving (Eq, Show, Ord)
data ModuleImport var
= Import { importMod :: var
, importList :: Maybe [NamespacedItem var]
, importQualified :: Bool
, importAlias :: Maybe var
}
deriving (Eq, Show, Ord)
data NamespacedItem var
= IEVar var
| IECon var
| IEModule var
deriving (Eq, Show, Ord)
data ModuleItem var
= ModDecl (FeDecl var)
| ModImport (ModuleImport var)
deriving (Eq, Show, Ord)
data ParsedVar
= UnqualVar
{ varId :: Text
, varBegin :: Posn
, varEnd :: Posn
}
| QualVar
{ varId :: Text
, varPrefix :: Text
, varBegin :: Posn
, varEnd :: Posn
}
| ModId
{ varId :: Text
, varBegin :: Posn
, varEnd :: Posn
}
deriving (Eq, Show, Ord)
toModId :: ParsedVar -> ParsedVar
toModId x@ModId{} = x
toModId (UnqualVar x y z) = ModId x y z
toModId (QualVar id pref b e) = ModId (pref <> T.singleton '.' <> id) b e
instance HasPosn ParsedVar where
startPosn = varBegin
endPosn = varEnd
span sp ep s = s { varBegin = startPosn sp, varEnd = endPosn ep }