From 10cf9341c54498ac1bdaa7b74a2bac19062a31d7 Mon Sep 17 00:00:00 2001
From: sofia <sofia@teascade.net>
Date: Sun, 13 Jul 2025 18:41:14 +0300
Subject: [PATCH] Add return type for index and array expressions

---
 reid/src/mir/types.rs | 22 ++++++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/reid/src/mir/types.rs b/reid/src/mir/types.rs
index 74766c8..35efbb6 100644
--- a/reid/src/mir/types.rs
+++ b/reid/src/mir/types.rs
@@ -7,6 +7,7 @@ pub enum ReturnTypeOther {
     Set(Metadata),
     EmptyBlock(Metadata),
     NoBlockReturn(Metadata),
+    IndexingNonArray(Metadata),
 }
 
 impl TypeKind {
@@ -86,8 +87,25 @@ impl ReturnType for Expression {
             Block(block) => block.return_type(),
             FunctionCall(fcall) => fcall.return_type(),
             If(expr) => expr.return_type(),
-            Index(expression, _) => todo!("return type for index"),
-            Array(expressions) => todo!("return type for array expression"),
+            Index(expression, _) => {
+                let expr_type = expression.return_type()?;
+                if let (_, TypeKind::Array(elem_ty, _)) = expr_type {
+                    Ok((ReturnKind::Soft, *elem_ty))
+                } else {
+                    Err(ReturnTypeOther::IndexingNonArray(expression.1))
+                }
+            }
+            Array(expressions) => {
+                let first = expressions
+                    .iter()
+                    .next()
+                    .map(|e| e.return_type())
+                    .unwrap_or(Ok((ReturnKind::Soft, TypeKind::Void)))?;
+                Ok((
+                    ReturnKind::Soft,
+                    TypeKind::Array(Box::new(first.1), expressions.len() as u64),
+                ))
+            }
         }
     }
 }