Added ability to parse polymorphic types

IsaacShelton · Oct 24, 2024 · dcb5b7f · dcb5b7f
1 parent e56432a
commit dcb5b7f
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 8 deletions.
diff --git a/src/ast/datatype/kind/display.rs b/src/ast/datatype/kind/display.rs
@@ -48,6 +48,9 @@ impl Display for &TypeKind {
             TypeKind::FunctionPointer(_function) => {
                 write!(f, "(function pointer type)")?;
             }
+            TypeKind::Polymorph(polymorph) => {
+                write!(f, "${}", polymorph)?;
+            }
         }
 
         Ok(())

diff --git a/src/ast/datatype/kind/mod.rs b/src/ast/datatype/kind/mod.rs
@@ -21,6 +21,7 @@ pub enum TypeKind {
     AnonymousUnion(AnoymousUnion),
     AnonymousEnum(AnonymousEnum),
     FunctionPointer(FunctionPointer),
+    Polymorph(String),
 }
 
 impl TypeKind {

diff --git a/src/parser/parse_type.rs b/src/parser/parse_type.rs
@@ -20,14 +20,18 @@ impl<'a, I: Inflow<Token>> Parser<'a, I> {
         let token = self.input.peek().clone();
 
         let Ok(name) = self.parse_name(None::<&str>) else {
-            return Err(ParseError {
-                kind: ParseErrorKind::ExpectedType {
-                    prefix: prefix.map(|prefix| prefix.to_string()),
-                    for_reason: for_reason.map(|for_reason| for_reason.to_string()),
-                    got: token.to_string(),
-                },
-                source,
-            });
+            if !token.kind.is_polymorph() {
+                return Err(ParseError {
+                    kind: ParseErrorKind::ExpectedType {
+                        prefix: prefix.map(|prefix| prefix.to_string()),
+                        for_reason: for_reason.map(|for_reason| for_reason.to_string()),
+                        got: token.to_string(),
+                    },
+                    source,
+                });
+            }
+
+            return Ok(TypeKind::Polymorph(token.kind.unwrap_polymorph()).at(source));
         };
 
         let generics = self.parse_generics()?;

diff --git a/src/resolve/mod.rs b/src/resolve/mod.rs
@@ -647,6 +647,10 @@ impl<'a> ResolveTypeCtx<'a> {
                     },
                 ))
             }
+            ast::TypeKind::Polymorph(polymorph) => Err(ResolveErrorKind::Other {
+                message: format!("Unresolved polymorphic type '${}'", polymorph),
+            }
+            .at(ast_type.source)),
         }
         .map(|kind| kind.at(ast_type.source))
     }