fix: methods are now callable locally

PLC-lang · Nov 18, 2024 · c376416 · c376416
1 parent 712677c
commit c376416
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/codegen/generators/expression_generator.rs b/src/codegen/generators/expression_generator.rs
@@ -862,7 +862,14 @@ impl<'ink, 'b> ExpressionCodeGenerator<'ink, 'b> {
             // no function
             let (class_ptr, call_ptr) = match pou {
                 PouIndexEntry::Method { .. } => {
-                    let class_ptr = self.generate_lvalue(operator)?;
+                    let class_ptr = self.generate_lvalue(operator).or_else(|_| {
+                        // this might be a local method
+                        function_context
+                            .function
+                            .get_first_param()
+                            .map(|class_ptr| class_ptr.into_pointer_value())
+                            .ok_or_else(|| Diagnostic::cannot_generate_call_statement(operator))
+                    })?;
                     let call_ptr =
                         self.allocate_function_struct_instance(implementation.get_call_name(), operator)?;
                     (Some(class_ptr), call_ptr)