Correct mutability for methods

Author: vladbelousoff

examples/017_struct_methods.tea

@@ -27,7 +27,7 @@ impl Point {
     }
 
     // Method that modifies the point (moves it)
-    fn move_by(dx: f32, dy: f32) {
+    fn mut move_by(dx: f32, dy: f32) {
         self.x = self.x + dx;
         self.y = self.y + dy;
     }
@@ -73,7 +73,7 @@ impl Rectangle {
     }
 
     // Scale the rectangle
-    fn scale(factor: f32) {
+    fn mut scale(factor: f32) {
         self.width = self.width * factor;
         self.height = self.height * factor;
     }
@@ -82,7 +82,7 @@ impl Rectangle {
 // Implement methods for Counter
 impl Counter {
     // Increment counter
-    fn increment() {
+    fn mut increment() {
         if self.value < self.max_value {
             self.value = self.value + 1;
         }
@@ -96,7 +96,7 @@ impl Counter {
     }
 
     // Reset counter
-    fn reset() {
+    fn mut reset() {
         self.value = 0;
     }
 

include/tea_scope.h

@@ -8,7 +8,6 @@ typedef struct {
   tea_list_entry_t fns;
   tea_list_entry_t nfns;
   tea_list_entry_t structs;
-
   tea_list_entry_t vars;
 } tea_ctx_t;
 
@@ -37,4 +36,4 @@ tea_var_t *tea_scope_find_local(const tea_scope_t *scp, const char *name);
 tea_var_t *tea_scope_find(const tea_scope_t *scp, const char *name);
 
 bool tea_decl_var(tea_ctx_t *ctx, tea_scope_t *scp, const char *name,
-                  unsigned int flags, const char *type, const tea_node_t *init);
+                  unsigned int flags, const char *type, const tea_node_t *initial_value);

src/tea_fn.c

@@ -264,12 +264,15 @@ tea_val_t tea_eval_fn_call(tea_ctx_t *ctx, tea_scope_t *scp,
       return tea_val_undef();
     }
 
-    // declare 'self' for the scope
-    tea_decl_var(ctx, &inner_scope, "self", 0, NULL, object_node);
-
     function_name = field_token->buf;
     function = tea_ctx_find_fn(&struct_declaration->fns, function_name);
 
+    if (function) {
+      // declare 'self' for the scope
+      tea_decl_var(ctx, &inner_scope, "self",
+        function->mut ? TEA_VAR_MUT : 0,
+        NULL, object_node);
+    }
   } else {
     const tea_tok_t *token = node->tok;
     if (token) {
@@ -377,6 +380,10 @@ tea_val_t tea_eval_fn_call(tea_ctx_t *ctx, tea_scope_t *scp,
     return return_context.ret_val;
   }
 
+  if (!result) {
+    exit(-1);
+  }
+
   return tea_val_undef();
 }