Allow for cloning inner values

reid/src/codegen.rs

@@ -265,7 +265,7 @@ impl mir::Statement {
     ) -> Option<InstructionValue> {
         match &self.0 {
             mir::StmtKind::Let(NamedVariableRef(ty, name, _), mutable, expression) => {
-                let value = expression.codegen(scope, state).unwrap();
+                let value = expression.codegen(scope, &state.load(true)).unwrap();
                 scope.stack_values.insert(
                     name.clone(),
                     StackValue(
@@ -415,8 +415,14 @@ impl mir::Expression {
                 }
             }
             mir::ExprKind::Indexed(expression, val_t, idx_expr) => {
-                let array = expression.codegen(scope, state)?;
-                let idx = idx_expr.codegen(scope, state)?;
+                dbg!(&expression, &idx_expr);
+                let array = expression
+                    .codegen(scope, state)
+                    .expect("array returned none!");
+                let idx = idx_expr
+                    .codegen(scope, state)
+                    .expect("index returned none!");
+
                 let mut ptr = scope
                     .block
                     .build(Instr::GetElemPtr(array, vec![idx]))
@@ -468,14 +474,19 @@ impl mir::Expression {
                 Some(array)
             }
             mir::ExprKind::Accessed(expression, type_kind, field) => {
-                let struct_val = expression.codegen(scope, &mut state.load(true))?;
+                let struct_val = expression.codegen(scope, &mut state.load(true)).unwrap();
 
-                let struct_ty = expression.return_type().ok()?.1.known().ok()?;
-                let TypeKind::CustomType(name) = struct_ty else {
-                    return None;
+                let struct_ty = expression
+                    .return_type()
+                    .map(|r| r.1.known())
+                    .unwrap()
+                    .unwrap();
+
+                let TypeKind::CustomType(name) = struct_ty.deref_borrow() else {
+                    panic!("tried accessing non-custom-type");
                 };
-                let TypeDefinitionKind::Struct(struct_ty) = scope.get_typedef(&name)?;
-                let idx = struct_ty.find_index(field)?;
+                let TypeDefinitionKind::Struct(struct_ty) = scope.get_typedef(&name).unwrap();
+                let idx = struct_ty.find_index(field).unwrap();
 
                 let mut value = scope
                     .block
@@ -644,6 +655,9 @@ impl TypeKind {
                     TypeDefinitionKind::Struct(_) => Type::Ptr(Box::new(custom_t)),
                 }
             }
+            TypeKind::Borrow(type_kind) => {
+                Type::Ptr(Box::new(type_kind.get_type(type_vals, typedefs)))
+            }
         }
     }
 }

reid/src/mir/impl.rs

@@ -149,7 +149,7 @@ impl Expression {
             Indexed(expression, _, _) => {
                 let expr_type = expression.return_type()?;
                 if let (_, TypeKind::Array(elem_ty, _)) = expr_type {
-                    Ok((ReturnKind::Soft, *elem_ty))
+                    Ok((ReturnKind::Soft, TypeKind::Borrow(Box::new(*elem_ty))))
                 } else {
                     Err(ReturnTypeOther::IndexingNonArray(expression.1))
                 }
@@ -165,7 +165,10 @@ impl Expression {
                     TypeKind::Array(Box::new(first.1), expressions.len() as u64),
                 ))
             }
-            Accessed(_, type_kind, _) => Ok((ReturnKind::Soft, type_kind.clone())),
+            Accessed(_, type_kind, _) => Ok((
+                ReturnKind::Soft,
+                TypeKind::Borrow(Box::new(type_kind.clone())),
+            )),
             Struct(name, _) => Ok((ReturnKind::Soft, TypeKind::CustomType(name.clone()))),
         }
     }
@@ -267,6 +270,13 @@ impl TypeKind {
             _ => resolved,
         }
     }
+
+    pub fn deref_borrow(&self) -> TypeKind {
+        match self {
+            TypeKind::Borrow(type_kind) => *type_kind.clone(),
+            _ => self.clone(),
+        }
+    }
 }
 
 #[derive(Debug, Clone, thiserror::Error)]

reid/src/mir/mod.rs

@@ -65,6 +65,8 @@ pub enum TypeKind {
     StringPtr,
     #[error("[{0}; {1}]")]
     Array(Box<TypeKind>, u64),
+    #[error("Borrow({0})")]
+    Borrow(Box<TypeKind>),
     #[error("{0}")]
     CustomType(String),
     #[error(transparent)]

reid/src/mir/typecheck.rs

@@ -503,13 +503,17 @@ impl Expression {
                 Ok((_, ty)) => Ok(ty),
                 Err(e) => Err(e),
             },
-            ExprKind::Indexed(expression, elem_ty, _) => {
+            ExprKind::Indexed(expression, elem_ty, idx_expr) => {
                 // Try to unwrap hint type from array if possible
                 let hint_t = hint_t.map(|t| match t {
                     TypeKind::Array(type_kind, _) => &type_kind,
                     _ => t,
                 });
 
+                // Typecheck and narrow index-expression
+                let idx_expr_res = idx_expr.typecheck(state, typerefs, Some(&TypeKind::U32));
+                state.ok(idx_expr_res, idx_expr.1);
+
                 // TODO it could be possible to check length against constants..
 
                 let expr_t = expression.typecheck(state, typerefs, hint_t)?;

reid/src/mir/typeinference.rs

@@ -105,6 +105,7 @@ impl Block {
                         (var_ref.as_mut(), expr_ty_ref.as_mut())
                     {
                         var_ref.narrow(&expr_ty_ref);
+                        dbg!(var_ref);
                     }
                 }
                 StmtKind::Set(lhs, rhs) => {
@@ -252,9 +253,13 @@ impl Expression {
                     ReturnKind::Soft => Ok(block_ref.1),
                 }
             }
-            ExprKind::Indexed(expression, index_ty, _) => {
+            ExprKind::Indexed(expression, index_ty, idx_expr) => {
                 let expr_ty = expression.infer_types(state, type_refs)?;
 
+                // Infer and narrow types used for indexing
+                let mut idx_ty = idx_expr.infer_types(state, type_refs)?;
+                idx_ty.narrow(&type_refs.from_type(&U32).unwrap());
+
                 // Check that the resolved type is at least an array, no
                 // need for further resolution.
                 let kind = expr_ty.resolve_weak().unwrap();

reid_src/array.reid

@@ -1,13 +1,13 @@
 // Arithmetic, function calls and imports!
 
-fn array() -> [[[u16; 4]; 1]; 1] {
-    return [[[10, 15, 7, 9]]];
+fn array() -> [u16; 4] {
+    return [10, 15, 7, 9];
 }
 
 fn main() -> u16 {
     let mut list = array();
 
-    list[0][0][3] = 5;
+    let a = 1;
 
-    return list[0][0][3];
+    return list[a + 1];
 }

reid_src/array_structs.reid

@@ -11,7 +11,13 @@ fn main() -> u32 {
         second: [6, 3, 17, 8],
     }];
 
-    value[0].second[2] = 99;
+    let val1 = 0;
+    let val2 = 1;
+
+    // value[val1].second[val2 + 1] = 99;
+
+    let mut b = value[val1];
+    b.second[2] = 99;
 
     return value[0].second[2];
 }