From 284bc61f24db0cdf49cf1c5fe2cf76e81f6d7293 Mon Sep 17 00:00:00 2001
From: Toni <antpiasecki@gmail.com>
Date: Wed, 13 May 2026 14:11:05 +0200
Subject: [PATCH] typecheck chained pointer operations

---
 src/codegen_x86_64.rs | 29 ++++++-----------------------
 src/std/std.zr        | 18 ++++++------------
 2 files changed, 12 insertions(+), 35 deletions(-)

diff --git a/src/codegen_x86_64.rs b/src/codegen_x86_64.rs
index 5b7f759..c2d5fc9 100644
--- a/src/codegen_x86_64.rs
+++ b/src/codegen_x86_64.rs
@@ -8,6 +8,7 @@ use crate::{
 
 struct Var {
     pub stack_offset: usize,
+    #[allow(unused)]
     pub var_type: String,
 }
 
@@ -599,7 +600,7 @@ _builtin_environ:
                     ExprKind::MemberAccess { left, field } => {
                         emit!(&mut self.output, "    push rax");
 
-                        let offset = self.get_field_offset(env, left, field)?;
+                        let offset = self.get_field_offset(left, field)?;
 
                         self.compile_expr(env, left)?;
                         emit!(&mut self.output, "    pop rbx");
@@ -742,7 +743,7 @@ _builtin_environ:
                 emit!(&mut self.output, "    pop rax");
             }
             ExprKind::MemberAccess { left, field } => {
-                let offset = self.get_field_offset(env, left, field)?;
+                let offset = self.get_field_offset(left, field)?;
                 self.compile_expr(env, left)?;
                 emit!(&mut self.output, "    mov rax, QWORD [rax+{}]", offset);
             }
@@ -753,28 +754,10 @@ _builtin_environ:
         Ok(())
     }
 
-    fn get_field_offset(
-        &self,
-        env: &mut Env,
-        left: &Expr,
-        field: &Token,
-    ) -> Result<usize, ZernError> {
-        let struct_name = match &left.kind {
-            ExprKind::Variable(name) => match env.get_var(&name.lexeme) {
-                Some(v) => v.var_type.clone(),
-                None => {
-                    return error!(name.loc, format!("undefined variable: {}", &name.lexeme));
-                }
-            },
-            _ => {
-                return error!(
-                    &field.loc,
-                    "cannot determine struct type for member assignment"
-                );
-            }
-        };
+    fn get_field_offset(&self, left: &Expr, field: &Token) -> Result<usize, ZernError> {
+        let struct_name = &self.expr_types[&left.id];
 
-        let fields = match self.symbol_table.structs.get(&struct_name) {
+        let fields = match self.symbol_table.structs.get(struct_name) {
             Some(f) => f,
             None => {
                 return error!(&field.loc, format!("unknown struct type: {}", struct_name));
diff --git a/src/std/std.zr b/src/std/std.zr
index b256ccd..e9b8349 100644
--- a/src/std/std.zr
+++ b/src/std/std.zr
@@ -55,8 +55,7 @@ func mem._split_block[blk: mem.Block, needed: i64] : void
         new_blk->prev = blk
 
         if blk->next as ptr
-            let next = blk->next
-            next->prev = new_blk
+            blk->next->prev = new_blk
         else
             mem.write64(_builtin_heap_tail(), new_blk)
 
@@ -133,8 +132,7 @@ func mem.free[x: any] : void
             blk->size = blk->size + MEM_BLOCK_SIZE + next->size
             blk->next = next->next
             if next->next as ptr
-                let b = next->next
-                b->prev = blk
+                next->next->prev = blk
             if mem.read64(_builtin_heap_tail()) == next as ptr
                 mem.write64(_builtin_heap_tail(), blk)
 
@@ -145,8 +143,7 @@ func mem.free[x: any] : void
             prev->size = prev->size + MEM_BLOCK_SIZE + blk->size
             prev->next = blk->next
             if blk->next as ptr
-                let b = blk->next
-                b->prev = prev
+                blk->next->prev = prev
             if mem.read64(_builtin_heap_tail()) == blk as ptr
                 mem.write64(_builtin_heap_tail(), prev)
             blk = prev
@@ -154,13 +151,11 @@ func mem.free[x: any] : void
     let block_total = blk->size + MEM_BLOCK_SIZE
     if (blk as i64 & 4095) == 0 && (block_total & 4095) == 0
         if blk->prev as ptr
-            let b = blk->prev
-            b->next = blk->next
+            blk->prev->next = blk->next
         else
             mem.write64(_builtin_heap_head(), blk->next)
         if blk->next as ptr
-            let b = blk->next
-            b->prev = blk->prev
+            blk->next->prev = blk->prev
         else
             mem.write64(_builtin_heap_tail(), blk->prev)
         _builtin_syscall(SYS_munmap, blk, block_total)
@@ -194,8 +189,7 @@ func mem.realloc?[x: ptr, new_size: i64] : ptr
             blk->size = combined
             blk->next = next->next
             if next->next as ptr
-                let b = next->next
-                b->prev = blk
+                next->next->prev = blk
             if mem.read64(_builtin_heap_tail()) == next as ptr
                 mem.write64(_builtin_heap_tail(), blk)
             mem._split_block(blk, new_size)