refactor: Cleanup

Socordia-Org · Dec 23, 2024 · 0f4982c · 0f4982c
1 parent 25d3cff
commit 0f4982c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/NewSource/SocordiaC/Compilation/CollectFunctionsListener.cs b/NewSource/SocordiaC/Compilation/CollectFunctionsListener.cs
@@ -1,8 +1,12 @@
 using System.Reflection;
 using DistIL.AsmIO;
+using DistIL.CodeGen.Cil;
+using DistIL.IR;
+using DistIL.IR.Utils;
 using MrKWatkins.Ast.Listening;
 using Socordia.CodeAnalysis.AST;
 using Socordia.CodeAnalysis.AST.Declarations;
+using MethodBody = DistIL.IR.MethodBody;
 
 namespace SocordiaC.Compilation;
 
@@ -15,6 +19,11 @@ protected override void ListenToNode(Driver context, FunctionDefinition node)
         var method = type.CreateMethod(node.Signature.Name.Name,
             Utils.GetTypeFromNode(node.Signature.ReturnType, type), [], attrs);
 
+        if (!node.Modifiers.Contains(Modifier.Extern))
+        {
+            method.Body = new MethodBody(method);
+        }
+
         if (type == context.FunctionsType && method.IsStatic && method.Name == "main")
         {
             context.Compilation.Module.EntryPoint = method;