Fix generic functions

ZenCodeLang · Nov 1, 2024 · 3fea2d4 · 3fea2d4
1 parent 9219731
commit 3fea2d4
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/Parser/src/main/java/org/openzen/zenscript/parser/definitions/ParsedFunction.java b/Parser/src/main/java/org/openzen/zenscript/parser/definitions/ParsedFunction.java
@@ -2,11 +2,13 @@
 
 import org.openzen.zencode.shared.CodePosition;
 import org.openzen.zencode.shared.CompileException;
+import org.openzen.zenscript.codemodel.FunctionHeader;
 import org.openzen.zenscript.codemodel.HighLevelDefinition;
 import org.openzen.zenscript.codemodel.Modifiers;
 import org.openzen.zenscript.codemodel.compilation.*;
 import org.openzen.zenscript.codemodel.context.CompilingPackage;
 import org.openzen.zenscript.codemodel.definition.FunctionDefinition;
+import org.openzen.zenscript.codemodel.generic.TypeParameter;
 import org.openzen.zenscript.codemodel.identifiers.TypeSymbol;
 import org.openzen.zenscript.codemodel.type.BasicTypeID;
 import org.openzen.zenscript.lexer.ParseException;
@@ -94,8 +96,17 @@ public boolean isInner() {
 
 		@Override
 		public void linkTypes() {
-			if (compiled.header == null)
-				compiled.setHeader(header.compile(compiler.types()));
+			if (compiled.header == null) {
+				FunctionHeader compiledHeader = header.compile(compiler.types());
+				compiled.setTypeParameters(compiledHeader.typeParameters);
+
+				FunctionHeader withoutTypeParameters = new FunctionHeader(
+						TypeParameter.NONE,
+						compiledHeader.getReturnType(),
+						compiledHeader.thrownType,
+						compiledHeader.parameters);
+				compiled.setHeader(withoutTypeParameters);
+			}
 		}
 
 		@Override