very small change

This commit is contained in:
2024-04-13 20:40:33 +02:00
parent 3e545e462d
commit e0c88f1d60
2 changed files with 23 additions and 27 deletions
+23 -20
View File
@@ -1,32 +1,13 @@
use crate::lexer::{lex, TokenType};
use crate::parser::{Parser, AstType};
pub fn auto_strip(input: String) -> String {
match input.chars().next().unwrap() {
'(' => {
return input.strip_prefix("(").unwrap().strip_suffix(")").unwrap().to_string()
}
'{' => {
return input.strip_prefix("{").unwrap().strip_suffix("}").unwrap().to_string()
}
_ => {
return input
}
}
}
pub fn transpile(input: String, indent: u32) -> String {
let mut input = input;
if indent > 0 {
input = auto_strip(input);
}
let mut result = String::new();
let tokens = lex(input.as_str(), false);
println!("\n\n\n\n");
let mut full_ast = Parser::new(tokens.clone());
while full_ast.tokens.len() > full_ast.index as usize {
let ast = full_ast.next();
println!("{}", ast);
if ast.ast_type == AstType::FunctionDeceleration {
result += format!("fn {}{} -> {} {}", ast.tokens[1].value, ast.tokens[2].value, ast.tokens[0].value, transpile(ast.tokens[3].value.clone(), indent+1)).as_str();
} else if ast.ast_type == AstType::VoidFunctionDeceleration {
@@ -50,7 +31,28 @@ pub fn transpile(input: String, indent: u32) -> String {
}
result += ")";
} else {
result += format!("{}: {}", ast.tokens[1].value, ast.tokens[0].value).as_str();
result += format!("let {}: {}", ast.tokens[1].value, ast.tokens[0].value).as_str();
}
} else if ast.ast_type == AstType::MutVariableDeceleration {
let _o = String::new();
let mut oy = ast.tokens.clone();
oy.remove(0);
if oy.len() > 1 {
result += "(";
for t in &oy {
result+=t.value.as_str()
}
result += "): (";
for t in oy {
if t.value == "," {
result+=", "
} else {
result+=ast.tokens[0].value.as_str()
}
}
result += ")";
} else {
result += format!("let mut {}: {}", ast.tokens[1].value, ast.tokens[0].value).as_str();
}
} else {
if ast.tokens[0].token_type==TokenType::Newline {
@@ -60,6 +62,7 @@ pub fn transpile(input: String, indent: u32) -> String {
result += ast.tokens[0].value.as_str()
}
}
}
result = result.trim_end().to_string();
if indent > 0 {