From 7f23772ec9f61967f9e7e547b93bfa2817a67e10 Mon Sep 17 00:00:00 2001
From: Leo Dev <klesti.selimay@icloud.com>
Date: Sat, 27 Jul 2024 04:40:48 +0200
Subject: [PATCH] pointers

---
 src/transpiler.rs | 58 +++++++++++++++++++++++++++++++++++++++++++----
 src/wyst.pest     | 19 +++++++++++++---
 2 files changed, 70 insertions(+), 7 deletions(-)

diff --git a/src/transpiler.rs b/src/transpiler.rs
index 9d6fe9a..57c8f16 100644
--- a/src/transpiler.rs
+++ b/src/transpiler.rs
@@ -152,23 +152,48 @@ impl Transpiler {
         match pair.as_rule() {
             Rule::func_def => {
                 res += format!(
-                    "pub fn {}({}) -> {} {}\n",
+                    "pub fn {}({}) -> {} \n{}\n",
                     self.transpile(tokens[1].clone()),
                     self.transpile_pairs(tokens[2].clone().into_inner())
-                        .as_str(),
+                        .as_str()
+                        .replace("let mut", ""),
                     self.transpile(tokens[0].clone()),
                     self.transpile(tokens[3].clone())
                 )
                 .as_str();
             }
-            Rule::var_def => {
+            Rule::ptr_fn_def => {
                 res += format!(
-                    "let mut {}: {}",
+                    "pub fn {}({}) -> &mut {} \n{}\n",
+                    self.transpile(tokens[1].clone()),
+                    self.transpile_pairs(tokens[2].clone().into_inner())
+                        .as_str()
+                        .replace("let mut", ""),
+                    self.transpile(tokens[0].clone()),
+                    self.transpile(tokens[3].clone())
+                )
+                .as_str();
+            }
+            Rule::var_def_ptr => {
+                res += format!(
+                    "let mut {}: &mut {}",
                     self.transpile(tokens[1].clone()),
                     self.transpile(tokens[0].clone())
                 )
                 .as_str();
             }
+            Rule::var_def => {
+                if tokens.len() == 1 {
+                    res += self.transpile(tokens[0].clone()).as_str();
+                } else {
+                    res += format!(
+                        "let mut {}: {}",
+                        self.transpile(tokens[1].clone()),
+                        self.transpile(tokens[0].clone())
+                    )
+                    .as_str();
+                }
+            }
             Rule::var_def_set => {
                 res += format!(
                     "{} = {}",
@@ -276,6 +301,31 @@ impl Transpiler {
                 res += ",";
             }
 
+            Rule::ptr_set => {
+                res += format!(
+                    "*{} = {}",
+                    self.transpile(tokens[0].clone()),
+                    self.transpile(tokens[1].clone()),
+                )
+                .as_str();
+            }
+
+            Rule::int => {
+                res += " ";
+                res += pair.as_str();
+                res += " ";
+            }
+
+            Rule::refrence => {
+                res += "&mut ";
+                res += self.transpile(tokens[0].clone()).as_str();
+            }
+
+            Rule::return_stm => {
+                res += "return ";
+                res += self.transpile(tokens[0].clone()).as_str();
+            }
+
             _ => {
                 println!("\x1b[33mUnh: {}\x1b[0m", pair);
                 res += " ";
diff --git a/src/wyst.pest b/src/wyst.pest
index 2b9bf25..2136e16 100644
--- a/src/wyst.pest
+++ b/src/wyst.pest
@@ -17,7 +17,7 @@ raw_string = { (!("\\" | "\"") ~ ANY)+ }
 string = { "\"" ~ (raw_string | escape )* ~ "\"" }
 char = { "'" ~ (escape | ANY) ~ "'" }
 bool = @{"true" | "false"}
-keyword = @{"struct"}
+keyword = @{"struct"|"return"}
 map = { "#" ~ "map" ~ identifier ~ comma ~ identifier }
 
 // [numbers]
@@ -32,7 +32,8 @@ def_identifier = {identifier ~ ("<" ~ ((def_identifier ~ comma)* ~ def_identifie
 // [definitions]
 round_def = { "(" ~ ((var_def ~ comma)* ~ var_def?) ~ ")" }
 curly_def = { "{" ~ (var_def ~ semicolon)* ~ "}" }
-var_def = { def_identifier ~ identifier }
+var_def = { (def_identifier ~ identifier) | var_def_ptr }
+var_def_ptr = { def_identifier ~ "*" ~ identifier }
 var_def_set = { var_def ~ "=" ~ expr }
 var_set = { def_identifier ~ "=" ~ expr }
 func_def = { def_identifier ~ identifier ~ round_def ~ curly }
@@ -41,6 +42,14 @@ ecall = { identifier ~ ( "(" ~ ((def_identifier ~ comma)* ~ def_identifier?) ~ "
 ecurly = { "{" ~ ((ecall ~ comma)* ~ ecall?) ~ "}" }
 enum_def = { "enum" ~ identifier ~ ecurly }
 
+// [Ptr]
+ptr_fn_def = { def_identifier ~ "*" ~ identifier ~ round_def ~ curly }
+ptr_set = { "*" ~ identifier ~ "=" ~ expr }
+refrence = { "&" ~ identifier }
+
+// [Statement]
+return_stm = { "return" ~ expr }
+
 rust_macro_lint = { "macro_rules" ~ "!" ~ identifier }
 rust_code = { (rust_macro_lint | ANY)* }
 
@@ -58,7 +67,8 @@ expr_ = {
     int |
     bool |
     identifier |
-    string
+    string |
+    refrence
 }
 
 expr = {
@@ -69,9 +79,11 @@ expr = {
 
 code_expr = {
     (
+        return_stm |
         var_def_set|
         var_def |
         var_set |
+        ptr_set |
         expr
     ) ~ semicolon
 }
@@ -81,6 +93,7 @@ top_expr = {
         include |
         include_global |
         func_def |
+        ptr_fn_def |
         struct_def |
         enum_def |
         map