From 8e9199f453114e7a1ffcdcf0a7124468ff120f5a Mon Sep 17 00:00:00 2001
From: Christoffer Lerno <christoffer@aegik.com>
Date: Sat, 7 Sep 2024 14:26:42 +0200
Subject: [PATCH] Untyped splat.

---
 src/compiler/sema_expr.c                    | 12 +++++++
 test/test_suite/functions/splat_untyped.c3t | 39 +++++++++++++++++++++
 2 files changed, 51 insertions(+)
 create mode 100644 test/test_suite/functions/splat_untyped.c3t

diff --git a/src/compiler/sema_expr.c b/src/compiler/sema_expr.c
index 31a245b18..920c3b9ae 100644
--- a/src/compiler/sema_expr.c
+++ b/src/compiler/sema_expr.c
@@ -1460,6 +1460,18 @@ INLINE bool sema_call_splat_vasplat(SemaContext *context, Expr *arg, Expr ***arg
 
 INLINE Expr **sema_splat_arraylike_append(SemaContext *context, Expr **args, Expr *arg, ArrayIndex len)
 {
+	if (expr_is_const(arg))
+	{
+		for (ArrayIndex i = 0; i < len; i++)
+		{
+			Expr *expr = expr_copy(arg);
+			Expr *subscript = expr_new_expr(EXPR_SUBSCRIPT, expr);
+			subscript->subscript_expr.range.start = exprid(expr_new_const_int(arg->span, type_usz, i));
+			subscript->subscript_expr.expr = exprid(expr);
+			vec_add(args, subscript);
+		}
+		return args;
+	}
 	Decl *temp = decl_new_generated_var(arg->type, VARDECL_LOCAL, arg->span);
 	Expr *decl_expr = expr_generate_decl(temp, arg);
 	Expr *list = expr_new_expr(EXPR_EXPRESSION_LIST, arg);
diff --git a/test/test_suite/functions/splat_untyped.c3t b/test/test_suite/functions/splat_untyped.c3t
new file mode 100644
index 000000000..ab196aca6
--- /dev/null
+++ b/test/test_suite/functions/splat_untyped.c3t
@@ -0,0 +1,39 @@
+// #target: macos-x64
+module test;
+import std;
+fn void main()
+{
+	var $a = { 1, 2, "abc" };
+	io::printfn("Hello %s %s %s %s", 1, ...$a);
+}
+
+/* #expect: test.ll
+
+  %varargslots = alloca [4 x %any], align 16
+  %taddr = alloca i32, align 4
+  %taddr1 = alloca i32, align 4
+  %taddr2 = alloca i32, align 4
+  %taddr4 = alloca %"char[]", align 8
+  %retparam = alloca i64, align 8
+  store i32 1, ptr %taddr, align 4
+  %0 = insertvalue %any undef, ptr %taddr, 0
+  %1 = insertvalue %any %0, i64 ptrtoint (ptr @"$ct.int" to i64), 1
+  store %any %1, ptr %varargslots, align 16
+  store i32 1, ptr %taddr1, align 4
+  %2 = insertvalue %any undef, ptr %taddr1, 0
+  %3 = insertvalue %any %2, i64 ptrtoint (ptr @"$ct.int" to i64), 1
+  %ptradd = getelementptr inbounds i8, ptr %varargslots, i64 16
+  store %any %3, ptr %ptradd, align 16
+  store i32 2, ptr %taddr2, align 4
+  %4 = insertvalue %any undef, ptr %taddr2, 0
+  %5 = insertvalue %any %4, i64 ptrtoint (ptr @"$ct.int" to i64), 1
+  %ptradd3 = getelementptr inbounds i8, ptr %varargslots, i64 32
+  store %any %5, ptr %ptradd3, align 16
+  store %"char[]" { ptr @.str.1, i64 3 }, ptr %taddr4, align 8
+  %6 = insertvalue %any undef, ptr %taddr4, 0
+  %7 = insertvalue %any %6, i64 ptrtoint (ptr @"$ct.String" to i64), 1
+  %ptradd5 = getelementptr inbounds i8, ptr %varargslots, i64 48
+  store %any %7, ptr %ptradd5, align 16
+  %8 = call i64 @std.io.printfn(ptr %retparam, ptr @.str, i64 17, ptr %varargslots, i64 4)
+  ret void
+}