summaryrefslogtreecommitdiff
path: root/src/Language
diff options
context:
space:
mode:
authorJosh Rahm <joshuarahm@gmail.com>2024-09-25 00:17:19 -0600
committerJosh Rahm <joshuarahm@gmail.com>2024-09-25 00:17:19 -0600
commitfffe42ce4861f53dd86113ab8320e4754f2c570c (patch)
treed9fb492c4c821eec091b2012ffe626cda45f1bde /src/Language
parent0c6ada2f5c8a3ac900fabd0384af558fb6bd334a (diff)
downloadfiddle-fffe42ce4861f53dd86113ab8320e4754f2c570c.tar.gz
fiddle-fffe42ce4861f53dd86113ab8320e4754f2c570c.tar.bz2
fiddle-fffe42ce4861f53dd86113ab8320e4754f2c570c.zip
Split the Ast file into multiple sub files.
Make some more changes to the generic implementation of EasySwitchStage.
Diffstat (limited to 'src/Language')
-rw-r--r--src/Language/Fiddle/Ast.hs566
-rw-r--r--src/Language/Fiddle/Ast/FileInterface.hs33
-rw-r--r--src/Language/Fiddle/Ast/Internal/Generic.hs5
-rw-r--r--src/Language/Fiddle/Ast/Internal/Instances.hs228
-rw-r--r--src/Language/Fiddle/Ast/Internal/Kinds.hs25
-rw-r--r--src/Language/Fiddle/Ast/Internal/Stage.hs43
-rw-r--r--src/Language/Fiddle/Ast/Internal/SyntaxTree.hs537
-rw-r--r--src/Language/Fiddle/Ast/Internal/SyntaxTreeKinds.hs1
-rw-r--r--src/Language/Fiddle/Compiler/Stage1.hs16
-rw-r--r--src/Language/Fiddle/Compiler/Stage2.hs220
-rw-r--r--src/Language/Fiddle/GenericTree.hs99
-rw-r--r--src/Language/Fiddle/Parser.hs8
12 files changed, 1031 insertions, 750 deletions
diff --git a/src/Language/Fiddle/Ast.hs b/src/Language/Fiddle/Ast.hs
index 8680790..7ef12da 100644
--- a/src/Language/Fiddle/Ast.hs
+++ b/src/Language/Fiddle/Ast.hs
@@ -1,570 +1,24 @@
{-# LANGUAGE ConstraintKinds #-}
-{-# LANGUAGE DefaultSignatures #-}
-{-# LANGUAGE DeriveAnyClass #-}
-{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE FlexibleContexts #-}
-{-# LANGUAGE IncoherentInstances #-}
-{-# LANGUAGE TypeOperators #-}
{-# LANGUAGE UndecidableInstances #-}
-module Language.Fiddle.Ast where
+module Language.Fiddle.Ast (module X) where
+import Data.Coerce
import Data.Functor.Identity
import Data.Kind (Type)
import Data.List.NonEmpty
import Data.Proxy
import Data.Text (Text)
import Data.Traversable
+import Data.Type.Bool
+import Data.Type.Equality
import Data.Typeable
+import Data.Void (Void, absurd)
import GHC.Generics
+import qualified GHC.TypeError as TypeError
import GHC.TypeLits
-
--- The Kind types should have for SyntaxTrees.
---
--- A synatx tree kind should take as parameters, an applicative functor for the
--- first argument. This applicative functor allows errors and warnings to be
--- contained to their deferred regions during compilation, so more errors may be
--- reported an once.
---
--- The second argument is the annotation type. Every syntax tree element has an
--- annotation. When initially parsing, this annotation is a list of comments
--- that came before the element and the source position of the element.
-type SynTreeKind = (Type -> Type) -> Type -> Type
-
--- The Type of number during each stage of compilation. When in the first stage,
--- numbers are just strings like anything else. In later stages, numbers get
--- parsed into actual integers. This makes it easier to process later.
-type family NumberType (a :: Stage) :: Type where
- NumberType Stage1 = Text
- NumberType Stage2 = Integer
- NumberType Stage3 = Integer
-
--- The type that represents an import statement. In the early stages of
--- compilation, this is just a string representing the import path, but in later
--- stages of compilation, this actually gets replaced by an abstract
--- representation of the imported material.
-type family ImportType (stage :: Stage) :: SynTreeKind where
- ImportType Stage1 = ImportStatement
- ImportType Stage2 = ImportStatement
- ImportType Stage3 = ImportStatement
-
--- Type-level constraint to determine if a stage is less than some natural
--- ordinal. Used to bound parts of the AST in multiple stages.
-type StageLessThan stage (n :: Natural) = (CmpNat (StageNumber stage) n ~ LT)
-
--- A Name is multiple identifiers separated by dots. It's the way of namespaces
--- to different packages.
-data Name f a where
- Name :: NonEmpty (Identifier f a) -> a -> Name f a
- deriving (Generic, Annotated, Alter, Typeable)
-
--- [[packed, rust: name="field_name"]]
-data Directive f a where
- Directive :: f (DirectiveBody f a) -> a -> Directive f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data DirectiveBody f a where
- DirectiveBody :: [DirectiveElement f a] -> a -> DirectiveBody f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data DirectiveElement f a where
- -- <key>
- DirectiveElementKey ::
- Maybe (Identifier f a) -> Identifier f a -> a -> DirectiveElement f a
- -- (<backend>:)? <key>=<value>
- DirectiveElementKeyValue ::
- Maybe (Identifier f a) ->
- Identifier f a ->
- DirectiveExpression f a ->
- a ->
- DirectiveElement f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data DirectiveExpression f a where
- DirectiveString :: Text -> a -> DirectiveExpression f a
- DirectiveNumber :: Text -> a -> DirectiveExpression f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data Directed t f a where
- Directed :: [Directive f a] -> t f a -> a -> Directed t f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-mapDirected :: (t f a -> t' f a) -> Directed t f a -> Directed t' f a
-mapDirected fn (Directed dr tfa a) = Directed dr (fn tfa) a
-
-mapDirectedM ::
- (Monad m) => (t f a -> m (t' f a)) -> Directed t f a -> m (Directed t' f a)
-mapDirectedM fn (Directed dr tfa a) = Directed dr <$> fn tfa <*> pure a
-
-asDirected :: (Annotated t) => t f a -> Directed t f a
-asDirected tfa = Directed [] tfa (annot tfa)
-
-undirected :: Directed t f a -> t f a
-undirected (Directed _ tfa _) = tfa
-
--- Stage of compilation. Parts of the AST maybe un unavailable with other stages
--- as compilation simplifies the AST.
-data Stage = Stage1 | Stage2 | Stage3
- deriving (Typeable)
-
--- Root of the parse tree. Just contains a list of declarations.
-data FiddleUnit (stage :: Stage) (f :: Type -> Type) a where
- FiddleUnit :: [Directed (FiddleDecl stage) f a] -> a -> FiddleUnit stage f a
- deriving (Generic, Annotated, Typeable)
-
-deriving instance (Alter (ImportType stage)) => Alter (FiddleUnit stage)
-
--- Just an identifier.
-data Identifier f a = Identifier !Text a
- deriving (Generic, Annotated, Alter, Typeable)
-
--- Expression.
-data Expression stage f a where
- -- Just a string. Parsing the number comes in stage2.
- LitNum :: NumberType stage -> a -> Expression stage f a
- Var :: Identifier f a -> a -> Expression stage f a
-
--- Wrapper class because using a type family like "ImportType" means I cannot
--- derive classes. I'd rather only derive for this class than have to derive for
--- the whole PackageBody class.
-data ImportTypeWrapper stage f a where
- ImportTypeWrapper :: ImportType stage f a -> ImportTypeWrapper stage f a
-
-instance (Alter (ImportType stage)) => Alter (ImportTypeWrapper stage) where
- alter ffn fn (ImportTypeWrapper st) =
- ImportTypeWrapper <$> alter ffn fn st
-
-data ImportStatement f a where
- ImportStatement :: Text -> Maybe (ImportList f a) -> a -> ImportStatement f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data ImportList f a where
- ImportList :: [Identifier f a] -> a -> ImportList f a
- deriving (Generic, Annotated, Alter, Typeable)
-
--- Top-level declarations.
-data FiddleDecl (stage :: Stage) (f :: Type -> Type) a where
- {-
- - An option is a key/value pair.
- - option <ident> <ident>;
- -}
- OptionDecl ::
- Identifier f a ->
- Identifier f a ->
- a ->
- FiddleDecl stage f a
- ImportDecl ::
- ImportType stage f a ->
- a ->
- FiddleDecl stage f a
- UsingDecl ::
- Name f a -> a -> FiddleDecl stage f a
- {- Package Statement. Package Name, Package body -}
- PackageDecl ::
- Name f a ->
- f (PackageBody stage f a) ->
- a ->
- FiddleDecl stage f a
- {- location <identifier> = <expr>. -}
- LocationDecl ::
- Identifier f a ->
- Expression stage f a ->
- a ->
- FiddleDecl stage f a
- {- bits <identifier> : <type> -}
- BitsDecl ::
- Identifier f a ->
- BitType stage f a ->
- a ->
- FiddleDecl stage f a
- {- objtype <identifier> : <type> -}
- ObjTypeDecl ::
- Identifier f a ->
- f (ObjTypeBody stage f a) ->
- a ->
- FiddleDecl stage f a
- {- object <ident> at <expr> : <type> -}
- ObjectDecl ::
- Identifier f a ->
- Expression stage f a ->
- ObjType stage f a ->
- a ->
- FiddleDecl stage f a
- deriving (Generic, Annotated, Typeable)
-
-deriving instance (Alter (ImportType stage)) => Alter (FiddleDecl stage)
-
-data ObjTypeBody (stage :: Stage) (f :: Type -> Type) a where
- ObjTypeBody ::
- BodyType f a ->
- [Directed (ObjTypeDecl stage) f a] ->
- a ->
- ObjTypeBody stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data ObjType stage f a where
- -- { <body> }
- -- Anonymous types are only allowed in stage1. Stage2 should have them be
- -- de-anonymized.
- AnonymousObjType :: f (ObjTypeBody 'Stage1 f a) -> a -> ObjType 'Stage1 f a
- -- <type>[<expr>]
- ArrayObjType :: ObjType stage f a -> Expression stage f a -> a -> ObjType stage f a
- -- <identifier>
- ReferencedObjType :: Name f a -> a -> ObjType stage f a
- deriving (Typeable)
-
-data ObjTypeDecl stage f a where
- {- assert_pos(<expr>) -}
- AssertPosStatement ::
- (StageLessThan stage 3) =>
- Expression stage f a ->
- a ->
- ObjTypeDecl stage f a
- {- reg <ident>(<expr>) : <regtype> -}
- RegisterDecl ::
- Maybe (Modifier f a) ->
- Maybe (Identifier f a) ->
- Expression stage f a ->
- Maybe (RegisterBody stage f a) ->
- a ->
- ObjTypeDecl stage f a
- {- reserved(n); -}
- ReservedDecl :: Expression stage f a -> a -> ObjTypeDecl stage f a
- {- <struct|union> { subfields } <name>; -}
- TypeSubStructure ::
- f (ObjTypeBody stage f a) ->
- Maybe (Identifier f a) ->
- a ->
- ObjTypeDecl stage f a
- deriving (Typeable)
-
-data Modifier f a where
- ModifierKeyword :: ModifierKeyword -> a -> Modifier f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data ModifierKeyword = Rw | Ro | Wo deriving (Eq, Ord, Show, Read, Typeable)
-
-data DeferredRegisterBody stage f a where
- DeferredRegisterBody ::
- [Directed (RegisterBitsDecl stage) f a] ->
- a ->
- DeferredRegisterBody stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data BodyType (f :: Type -> Type) a where
- Union :: a -> BodyType f a
- Struct :: a -> BodyType f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data RegisterBody stage f a where
- RegisterBody ::
- BodyType f a ->
- f (DeferredRegisterBody stage f a) ->
- a ->
- RegisterBody stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data RegisterBitsDecl stage f a where
- -- reserved(<expr>)
- ReservedBits :: Expression stage f a -> a -> RegisterBitsDecl stage f a
- -- <modifer> <ident> : <type>
- DefinedBits ::
- Maybe (Modifier f a) ->
- Identifier f a ->
- RegisterBitsTypeRef stage f a ->
- a ->
- RegisterBitsDecl stage f a
- BitsSubStructure ::
- RegisterBody stage f a ->
- Maybe (Identifier f a) ->
- a ->
- RegisterBitsDecl stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data RegisterBitsTypeRef stage f a where
- -- <type>[<expr>]
- RegisterBitsArray ::
- RegisterBitsTypeRef stage f a ->
- Expression stage f a ->
- a ->
- RegisterBitsTypeRef stage f a
- {- Reference to a type. -}
- RegisterBitsReference :: Name f a -> a -> RegisterBitsTypeRef stage f a
- {- enum(<expr>) { <body> }
- Anonymous types are only allowed in stage1.
- Stage2 should de-anonymize these type. -}
- RegisterBitsAnonymousType ::
- AnonymousBitsType Stage1 f a ->
- a ->
- RegisterBitsTypeRef 'Stage1 f a
- {- (<expr>)
- -
- - The expression is just bits ... i.e. an integer.
- -}
- RegisterBitsJustBits ::
- Expression stage f a ->
- a ->
- RegisterBitsTypeRef stage f a
-
-instance Alter (Expression stage) where
- alter ffn fn = \case
- LitNum t a -> LitNum t <$> fn a
- Var i a -> Var <$> alter ffn fn i <*> fn a
-
-instance Annotated (Expression stage) where
- annot = \case
- LitNum _ a -> a
- Var _ a -> a
-
-instance Alter (ObjTypeDecl stage) where
- alter ffn fn = \case
- (AssertPosStatement expr a) ->
- AssertPosStatement <$> alter ffn fn expr <*> fn a
- (RegisterDecl mMod mIdent expr mBody a) ->
- RegisterDecl
- <$> mapM (alter ffn fn) mMod
- <*> mapM (alter ffn fn) mIdent
- <*> alter ffn fn expr
- <*> mapM (alter ffn fn) mBody
- <*> fn a
- (TypeSubStructure mBody mIdent a) ->
- TypeSubStructure
- <$> (ffn =<< mapM (alter ffn fn) mBody)
- <*> mapM (alter ffn fn) mIdent
- <*> fn a
- (ReservedDecl expr a) ->
- ReservedDecl
- <$> alter ffn fn expr
- <*> fn a
-
-instance Annotated (ObjTypeDecl stage) where
- annot = \case
- (AssertPosStatement _ a) -> a
- (RegisterDecl _ _ _ _ a) -> a
-
-instance Alter (ObjType stage) where
- alter ffn fn = \case
- (AnonymousObjType b a) ->
- AnonymousObjType <$> (ffn =<< mapM (alter ffn fn) b) <*> fn a
- (ArrayObjType t e a) ->
- ArrayObjType <$> alter ffn fn t <*> alter ffn fn e <*> fn a
- (ReferencedObjType i a) ->
- ReferencedObjType <$> alter ffn fn i <*> fn a
-
-instance Annotated (ObjType stage) where
- annot = \case
- (AnonymousObjType _ a) -> a
- (ArrayObjType _ _ a) -> a
- (ReferencedObjType _ a) -> a
-
-instance Alter (RegisterBitsTypeRef stage) where
- alter ffn fn = \case
- (RegisterBitsArray ref exp a) ->
- RegisterBitsArray <$> alter ffn fn ref <*> alter ffn fn exp <*> fn a
- (RegisterBitsReference i a) ->
- RegisterBitsReference <$> alter ffn fn i <*> fn a
- (RegisterBitsAnonymousType t a) ->
- RegisterBitsAnonymousType <$> alter ffn fn t <*> fn a
- (RegisterBitsJustBits e a) ->
- RegisterBitsJustBits <$> alter ffn fn e <*> fn a
-
-instance Annotated (RegisterBitsTypeRef stage) where
- annot = \case
- (RegisterBitsArray _ _ a) -> a
- (RegisterBitsReference _ a) -> a
- (RegisterBitsAnonymousType _ a) -> a
- (RegisterBitsJustBits _ a) -> a
-
-data AnonymousBitsType stage f a where
- -- enum(<expr>) { <body> }
- AnonymousEnumBody ::
- Expression stage f a ->
- f (EnumBody stage f a) ->
- a ->
- AnonymousBitsType stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data BitType (stage :: Stage) (f :: Type -> Type) a where
- -- enum(<expr>) { <body> }
- EnumBitType ::
- Expression stage f a ->
- f (EnumBody stage f a) ->
- a ->
- BitType stage f a
- -- (<expr>)
- RawBits :: Expression stage f a -> a -> BitType stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data EnumBody (stage :: Stage) (f :: Type -> Type) a where
- -- <decl>,
- EnumBody :: [Directed (EnumConstantDecl stage) f a] -> a -> EnumBody stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data EnumConstantDecl stage f a where
- -- <ident> = <expr>
- EnumConstantDecl ::
- Identifier f a ->
- Expression stage f a ->
- a ->
- EnumConstantDecl stage f a
- -- reserved = <expr>
- EnumConstantReserved ::
- Expression stage f a ->
- a ->
- EnumConstantDecl stage f a
- deriving (Generic, Annotated, Alter, Typeable)
-
-data PackageBody (stage :: Stage) (f :: Type -> Type) a where
- {- The body of a package -}
- PackageBody :: [Directed (FiddleDecl stage) f a] -> a -> PackageBody stage f a
- deriving (Generic, Annotated, Typeable)
-
-deriving instance (Alter (ImportType stage)) => Alter (PackageBody stage)
-
--- instance Alter (Modifier stage) where
--- alter _ fn (ModifierKeyword m a) = ModifierKeyword m (fn a)
---
--- instance Alter (Identifier) where
--- alter _ fn (Identifier i a) = Identifier i $ fn a
---
--- instance Alter (Expression stage) where
--- alter ffn fn = \case
--- (LitNum t a) -> LitNum t $ fn a
--- (RealNum t a) -> RealNum t $ fn a
--- (Var i a) -> Var (alter ffn fn i) $ fn a
-
-proxyOf :: t f a -> Proxy t
-proxyOf _ = Proxy
-
-class Annotated (t :: SynTreeKind) where
- annot :: t f a -> a
- default annot :: (Generic (t f a), GAnnot a (Rep (t f a))) => t f a -> a
- annot t = gannot (from t)
-
-class GAnnot a r where
- gannot :: r x -> a
-
-instance GAnnot a (Rec0 a) where
- gannot = unK1
-
-instance (GAnnot a r) => GAnnot a (l :*: r) where
- gannot (_ :*: r) = gannot r
-
-instance (GAnnot a r, GAnnot a l) => GAnnot a (l :+: r) where
- gannot (R1 r) = gannot r
- gannot (L1 l) = gannot l
-
-instance (GAnnot a r) => GAnnot a (M1 i c r) where
- gannot (M1 a) = gannot a
-
-class Alter (t :: SynTreeKind) where
- alter ::
- forall f1 f2 a1 a2 m.
- (Monad m, Traversable f1) =>
- (forall z. f1 z -> m (f2 z)) ->
- (a1 -> m a2) ->
- t f1 a1 ->
- m (t f2 a2)
- default alter ::
- forall f1 f2 a1 a2 m.
- ( Generic (t f1 a1),
- Generic (t f2 a2),
- Traversable f1,
- GAlter t f1 f2 a1 a2 (Rep (t f1 a1)) (Rep (t f2 a2)),
- Monad m
- ) =>
- ( forall z.
- f1 z ->
- m (f2 z)
- ) ->
- (a1 -> m a2) ->
- t f1 a1 ->
- m (t f2 a2)
- alter ffn fn t = to <$> galter (proxyOf t) ffn fn (from t)
-
-instance (Alter t, Traversable f) => Functor (t f) where
- fmap f t = runIdentity (alter return (return . f) t)
-
-class GAlter t f1 f2 a1 a2 r1 r2 where
- galter ::
- forall proxy x m.
- (Monad m, Traversable f1) =>
- proxy t ->
- (forall z. f1 z -> m (f2 z)) ->
- (a1 -> m a2) ->
- r1 x ->
- m (r2 x)
-
-{- Altering a record with type a1 will apply the mapping function and produce a
- record with type a2 -}
-instance GAlter t f1 f2 a1 a2 (Rec0 a1) (Rec0 a2) where
- galter _ _ fn k1 = K1 <$> fn (unK1 k1)
-
-{- Base-Case. Altering unrelated leaf types will do nothing. -}
-instance GAlter t f1 f2 a1 a2 (Rec0 u1) (Rec0 u1) where
- galter _ _ _ = return
-
-{- Recursive case. Call alter on sub-structures. -}
-instance
- (Alter u) =>
- GAlter t f1 f2 a1 a2 (Rec0 (u f1 a1)) (Rec0 (u f2 a2))
- where
- galter _ ffn fn k1 =
- K1 <$> alter ffn fn (unK1 k1)
-
-{- Recursive case. Called when there are list of substructures that need to be
- recused. -}
-instance
- (Alter u, Traversable l) =>
- GAlter t f1 f2 a1 a2 (Rec0 (l (u f1 a1))) (Rec0 (l (u f2 a2)))
- where
- galter _ ffn fn k1 =
- K1 <$> mapM (alter ffn fn) (unK1 k1)
-
--- instance GAlter t f1 f2 a1 a2 (Rec0 (f1 z)) (Rec0 (f2 z)) where
--- galter _ ffn _ k1 = K1 <$> ffn (unK1 k1)
-
-{- Generic altering. Descends into the function and alters whatever is inside
- the functor and then transforms the functor using the ffn function. -}
-instance
- (Traversable f1, Alter u) =>
- GAlter t f1 f2 a1 a2 (Rec0 (f1 (u f1 a1))) (Rec0 (f2 (u f2 a2)))
- where
- galter proxy ffn fn k1 = do
- newK <- mapM (alter ffn fn) (unK1 k1)
- K1 <$> ffn newK
-
-instance
- ( GAlter t f1 f2 a1 a2 l1 l2,
- GAlter t f1 f2 a1 a2 r1 r2
- ) =>
- GAlter t f1 f2 a1 a2 (l1 :*: r1) (l2 :*: r2)
- where
- galter proxy ffn fn (a :*: b) = do
- a' <- galter proxy ffn fn a
- b' <- galter proxy ffn fn b
- return (a' :*: b')
-
-instance
- ( GAlter t f1 f2 a1 a2 l1 l2,
- GAlter t f1 f2 a1 a2 r1 r2
- ) =>
- GAlter t f1 f2 a1 a2 (l1 :+: r1) (l2 :+: r2)
- where
- galter proxy ffn fn (R1 r) = R1 <$> galter proxy ffn fn r
- galter proxy ffn fn (L1 l) = L1 <$> galter proxy ffn fn l
-
-instance
- (GAlter t f1 f2 a1 a2 r1 r2) =>
- GAlter t f1 f2 a1 a2 (M1 i c r1) (M1 i c r2)
- where
- galter proxy ffn fn (M1 a) = M1 <$> galter proxy ffn fn a
-
-type family StageNumber (s :: Stage) :: Natural where
- StageNumber Stage1 = 1
- StageNumber Stage2 = 2
- StageNumber Stage3 = 3
-
-{--}
-squeeze :: (Alter t, Traversable f, Monad f) => t f a -> f (t Identity a)
-squeeze = alter (fmap Identity) return
+import Language.Fiddle.Ast.Internal.Instances as X
+import Language.Fiddle.Ast.Internal.Kinds as X
+import Language.Fiddle.Ast.Internal.Stage as X
+import Language.Fiddle.Ast.Internal.SyntaxTree as X
diff --git a/src/Language/Fiddle/Ast/FileInterface.hs b/src/Language/Fiddle/Ast/FileInterface.hs
new file mode 100644
index 0000000..d29fc9d
--- /dev/null
+++ b/src/Language/Fiddle/Ast/FileInterface.hs
@@ -0,0 +1,33 @@
+module Language.Fiddle.Ast.FileInterface where
+
+-- Definitions for file interfaces. These interfaces contain an abstract
+-- representation of the symbols and information exported by a fiddle file.
+-- These interfaces are also serializable, and when compiling a fiddle file, all
+-- the import statements should supply an fdi (fiddle interface) file to speed
+-- up subsequent compilations.
+
+-- import Data.Text
+--
+-- data ObjectType = ObjectType
+-- { objectTypeSize :: Word32
+-- }
+--
+-- data Metatype
+-- = Object
+-- { objectLocation :: Word64,
+-- objectType :: Text
+-- }
+-- | Type
+-- { typeSizeBytes :: Word32
+-- }
+--
+-- data Element a = Element
+-- { elementFullyQualifiedSymbol :: Text,
+-- elementDocumentation :: Maybe Text,
+-- elementMetatype :: Metatype,
+-- elementAnnotation :: a
+-- }
+--
+-- data FileInterface a = FiddleInterface
+-- { exportedElements :: [Element a]
+-- }
diff --git a/src/Language/Fiddle/Ast/Internal/Generic.hs b/src/Language/Fiddle/Ast/Internal/Generic.hs
new file mode 100644
index 0000000..cb075cc
--- /dev/null
+++ b/src/Language/Fiddle/Ast/Internal/Generic.hs
@@ -0,0 +1,5 @@
+{-# LANGUAGE TypeOperators #-}
+module Language.Fiddle.Ast.Internal.Generic where
+
+import GHC.Generics
+import GHC.TypeError as TypeError
diff --git a/src/Language/Fiddle/Ast/Internal/Instances.hs b/src/Language/Fiddle/Ast/Internal/Instances.hs
new file mode 100644
index 0000000..c8c606c
--- /dev/null
+++ b/src/Language/Fiddle/Ast/Internal/Instances.hs
@@ -0,0 +1,228 @@
+{-# LANGUAGE ConstraintKinds #-}
+{-# LANGUAGE DefaultSignatures #-}
+{-# LANGUAGE FlexibleContexts #-}
+{-# LANGUAGE IncoherentInstances #-}
+{-# LANGUAGE TypeOperators #-}
+{-# LANGUAGE UndecidableInstances #-}
+
+module Language.Fiddle.Ast.Internal.Instances where
+
+import Data.Functor.Identity
+import Data.Kind (Type)
+import Data.Typeable
+import GHC.Generics
+import GHC.TypeError as TypeError
+import Language.Fiddle.Ast.Internal.Kinds
+import Language.Fiddle.Ast.Internal.Stage
+
+-- Type class to easily switch between stages if there is no difference in the
+-- syntax tree structure between these stages. Can make things much cleaner and
+-- avoids the boilerplate and bugprone-ness of needing to rote copy everything.
+class
+ (Functor f) =>
+ EasySwitchStage
+ (t :: Stage -> SynTree)
+ (f :: Type -> Type)
+ (fromStage :: Stage)
+ (toStage :: Stage)
+ where
+ switchStage :: t fromStage f a -> t toStage f a
+ default switchStage ::
+ ( Generic (t fromStage f a),
+ Generic (t toStage f a),
+ GEasySwitchStage (Rep (t fromStage f a)) (Rep (t toStage f a))
+ ) =>
+ t fromStage f a ->
+ t toStage f a
+ switchStage t = to $ gSwitchStage (from t)
+
+-- Class for walking a syntax tree under the context of a monad and modifying
+-- the different parts of the SynTree type..
+class Alter (t :: SynTree) where
+ alter ::
+ forall f1 f2 a1 a2 m.
+ (Monad m, Traversable f1) =>
+ (forall z. f1 z -> m (f2 z)) ->
+ (a1 -> m a2) ->
+ t f1 a1 ->
+ m (t f2 a2)
+ default alter ::
+ forall f1 f2 a1 a2 m.
+ ( Generic (t f1 a1),
+ Generic (t f2 a2),
+ Traversable f1,
+ GAlter t f1 f2 a1 a2 (Rep (t f1 a1)) (Rep (t f2 a2)),
+ Monad m
+ ) =>
+ ( forall z.
+ f1 z ->
+ m (f2 z)
+ ) ->
+ (a1 -> m a2) ->
+ t f1 a1 ->
+ m (t f2 a2)
+ alter ffn fn t = to <$> galter (proxyOf t) ffn fn (from t)
+
+-- A syntax tree object is annotated if it has an annotation 'a' as the last
+-- element.
+class Annotated (t :: SynTree) where
+ annot :: t f a -> a
+ default annot :: (Generic (t f a), GAnnot a (Rep (t f a))) => t f a -> a
+ annot t = gannot (from t)
+
+-- Generic implementations of common typeclass for SyntaxTrees.
+--
+-- This is where we try to hide the pig behind the curtain.
+--
+--
+-- ---------------------------------------------------------------
+--
+--
+--
+class GAlter t f1 f2 a1 a2 r1 r2 where
+ galter ::
+ forall proxy x m.
+ (Monad m, Traversable f1) =>
+ proxy t ->
+ (forall z. f1 z -> m (f2 z)) ->
+ (a1 -> m a2) ->
+ r1 x ->
+ m (r2 x)
+
+{- Altering a record with type a1 will apply the mapping function and produce a
+ record with type a2 -}
+instance GAlter t f1 f2 a1 a2 (Rec0 a1) (Rec0 a2) where
+ galter _ _ fn k1 = K1 <$> fn (unK1 k1)
+
+{- Base-Case. Altering unrelated leaf types will do nothing. -}
+instance GAlter t f1 f2 a1 a2 (Rec0 u1) (Rec0 u1) where
+ galter _ _ _ = return
+
+{- Recursive case. Call alter on sub-structures. -}
+instance
+ (Alter u) =>
+ GAlter t f1 f2 a1 a2 (Rec0 (u f1 a1)) (Rec0 (u f2 a2))
+ where
+ galter _ ffn fn k1 =
+ K1 <$> alter ffn fn (unK1 k1)
+
+{- Recursive case. Called when there are list of substructures that need to be
+ recused. -}
+instance
+ (Alter u, Traversable l) =>
+ GAlter t f1 f2 a1 a2 (Rec0 (l (u f1 a1))) (Rec0 (l (u f2 a2)))
+ where
+ galter _ ffn fn k1 =
+ K1 <$> mapM (alter ffn fn) (unK1 k1)
+
+-- instance GAlter t f1 f2 a1 a2 (Rec0 (f1 z)) (Rec0 (f2 z)) where
+-- galter _ ffn _ k1 = K1 <$> ffn (unK1 k1)
+
+{- Generic altering. Descends into the function and alters whatever is inside
+ the functor and then transforms the functor using the ffn function. -}
+instance
+ (Traversable f1, Alter u) =>
+ GAlter t f1 f2 a1 a2 (Rec0 (f1 (u f1 a1))) (Rec0 (f2 (u f2 a2)))
+ where
+ galter proxy ffn fn k1 = do
+ newK <- mapM (alter ffn fn) (unK1 k1)
+ K1 <$> ffn newK
+
+instance
+ ( GAlter t f1 f2 a1 a2 l1 l2,
+ GAlter t f1 f2 a1 a2 r1 r2
+ ) =>
+ GAlter t f1 f2 a1 a2 (l1 :*: r1) (l2 :*: r2)
+ where
+ galter proxy ffn fn (a :*: b) = do
+ a' <- galter proxy ffn fn a
+ b' <- galter proxy ffn fn b
+ return (a' :*: b')
+
+instance
+ ( GAlter t f1 f2 a1 a2 l1 l2,
+ GAlter t f1 f2 a1 a2 r1 r2
+ ) =>
+ GAlter t f1 f2 a1 a2 (l1 :+: r1) (l2 :+: r2)
+ where
+ galter proxy ffn fn (R1 r) = R1 <$> galter proxy ffn fn r
+ galter proxy ffn fn (L1 l) = L1 <$> galter proxy ffn fn l
+
+instance
+ (GAlter t f1 f2 a1 a2 r1 r2) =>
+ GAlter t f1 f2 a1 a2 (M1 i c r1) (M1 i c r2)
+ where
+ galter proxy ffn fn (M1 a) = M1 <$> galter proxy ffn fn a
+
+class GAnnot a r where
+ gannot :: r x -> a
+
+instance GAnnot a (Rec0 a) where
+ gannot = unK1
+
+instance (GAnnot a r) => GAnnot a (l :*: r) where
+ gannot (_ :*: r) = gannot r
+
+instance (GAnnot a r, GAnnot a l) => GAnnot a (l :+: r) where
+ gannot (R1 r) = gannot r
+ gannot (L1 l) = gannot l
+
+instance (GAnnot a r) => GAnnot a (M1 i c r) where
+ gannot (M1 a) = gannot a
+
+instance
+ (GEasySwitchStage s1 s2) =>
+ GEasySwitchStage (M1 i c s1) (M1 i c s2)
+ where
+ gSwitchStage (M1 a) = M1 (gSwitchStage a)
+
+instance
+ (GEasySwitchStage l1 l2, GEasySwitchStage r1 r2) =>
+ GEasySwitchStage (l1 :+: r1) (l2 :+: r2)
+ where
+ gSwitchStage (R1 r) = R1 $ gSwitchStage r
+ gSwitchStage (L1 l) = L1 $ gSwitchStage l
+
+instance
+ (GEasySwitchStage l1 l2, GEasySwitchStage r1 r2) =>
+ (GEasySwitchStage (l1 :*: r1) (l2 :*: r2))
+ where
+ gSwitchStage (l :*: r) = gSwitchStage l :*: gSwitchStage r
+
+instance
+ (EasySwitchStage t f fs ts) =>
+ (GEasySwitchStage (Rec0 (t fs f a)) (Rec0 (t ts f a)))
+ where
+ gSwitchStage (K1 val) = K1 (switchStage val)
+
+instance
+ ( EasySwitchStage t f fs ts,
+ Functor func
+ ) =>
+ (GEasySwitchStage (Rec0 (func (t fs f a))) (Rec0 (func (t ts f a))))
+ where
+ gSwitchStage (K1 val) = K1 (switchStage <$> val)
+
+instance (GEasySwitchStage (Rec0 a) (Rec0 a)) where
+ gSwitchStage = id
+
+instance
+ ( TypeError
+ ( TypeError.Text "Unable to match type "
+ :<>: TypeError.ShowType a
+ :<>: TypeError.Text " with "
+ :<>: TypeError.ShowType b
+ )
+ ) =>
+ (GEasySwitchStage (Rec0 a) (Rec0 b))
+ where
+ gSwitchStage = error "Cannot be called"
+
+class GEasySwitchStage r1 r2 where
+ gSwitchStage :: r1 x -> r2 x
+
+proxyOf :: t f a -> Proxy t
+proxyOf _ = Proxy
+
+instance (Alter t, Traversable f) => Functor (t f) where
+ fmap f t = runIdentity (alter return (return . f) t)
diff --git a/src/Language/Fiddle/Ast/Internal/Kinds.hs b/src/Language/Fiddle/Ast/Internal/Kinds.hs
new file mode 100644
index 0000000..1368c84
--- /dev/null
+++ b/src/Language/Fiddle/Ast/Internal/Kinds.hs
@@ -0,0 +1,25 @@
+
+module Language.Fiddle.Ast.Internal.Kinds where
+
+import Data.Kind (Type)
+import Language.Fiddle.Ast.Internal.Stage
+
+-- The Kind types should have for SyntaxTrees.
+--
+-- A synatx tree kind should take as parameters, an applicative functor for the
+-- first argument. This applicative functor allows errors and warnings to be
+-- contained to their deferred regions during compilation, so more errors may be
+-- reported an once.
+--
+-- The second argument is the annotation type. Every syntax tree element has an
+-- annotation. When initially parsing, this annotation is a list of comments
+-- that came before the element and the source position of the element.
+--
+-- SynTree types are not dependent on the compilation stage. They're the same
+-- regardless of the compilation stage.
+type SynTree = (Type -> Type) -> Type -> Type
+
+-- A syntax tree which is dependent on the stage in compilation. As compilations
+-- progress through stages, types in the syntax tree may change as things become
+-- more and more machine readable.
+type StagedSynTree = Stage -> SynTree
diff --git a/src/Language/Fiddle/Ast/Internal/Stage.hs b/src/Language/Fiddle/Ast/Internal/Stage.hs
new file mode 100644
index 0000000..cfcd0e6
--- /dev/null
+++ b/src/Language/Fiddle/Ast/Internal/Stage.hs
@@ -0,0 +1,43 @@
+{-# LANGUAGE ConstraintKinds #-}
+{-# LANGUAGE FlexibleContexts #-}
+{-# LANGUAGE IncoherentInstances #-}
+{-# LANGUAGE TypeOperators #-}
+{-# LANGUAGE UndecidableInstances #-}
+module Language.Fiddle.Ast.Internal.Stage where
+
+import Data.Type.Bool
+import Data.Type.Equality
+import Data.Typeable
+import qualified GHC.TypeError as TypeError
+import GHC.TypeLits
+
+-- Stage of compilation. Parts of the AST maybe un unavailable with other stages
+-- as compilation simplifies the AST.
+data Stage = Stage1 | Stage2 | Stage3
+ deriving (Typeable)
+
+-- Returns the stage before the given stage.
+type family PreviousStage (s :: Stage) :: Stage where
+ PreviousStage Stage1 = TypeError (TypeError.Text "No Prior stage to Stage1")
+ PreviousStage s = NumberToStage (StageToNumber s - 1)
+
+-- Returns the stage after the give stage.
+type family NextStage (s :: Stage) :: Stage where
+ NextStage s = NumberToStage (StageToNumber s + 1)
+
+-- Converts a stage to a number.
+type family StageToNumber (s :: Stage) :: Natural where
+ StageToNumber Stage1 = 1
+ StageToNumber Stage2 = 2
+ StageToNumber Stage3 = 3
+
+-- Converts a number to a stage.
+type family NumberToStage (n :: Natural) :: Stage where
+ NumberToStage 1 = Stage1
+ NumberToStage 2 = Stage2
+ NumberToStage 3 = Stage3
+ NumberToStage n = TypeError (TypeError.Text "Number in NumberToStage.")
+
+-- Type-level constraint to determine if a stage is less than some natural
+-- ordinal. Used to bound parts of the AST in multiple stages.
+type StageLessThan stage (n :: Natural) = (CmpNat (StageToNumber stage) n == LT)
diff --git a/src/Language/Fiddle/Ast/Internal/SyntaxTree.hs b/src/Language/Fiddle/Ast/Internal/SyntaxTree.hs
new file mode 100644
index 0000000..48852ee
--- /dev/null
+++ b/src/Language/Fiddle/Ast/Internal/SyntaxTree.hs