diff --git a/src/transpiler.rs b/src/transpiler.rs new file mode 100644 index 0000000..b97b4db --- /dev/null +++ b/src/transpiler.rs @@ -0,0 +1,72 @@ +use crate::lexer::{lex, Token, 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 { + result += format!("fn {}{} {}", ast.tokens[1].value, ast.tokens[2].value, transpile(ast.tokens[3].value.clone(), indent+1)).as_str(); + } else if ast.ast_type == AstType::VariableDeceleration { + let mut 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!("{}: {}", ast.tokens[1].value, ast.tokens[0].value).as_str(); + } + } else { + if ast.tokens[0].token_type==TokenType::Newline { + result += (ast.tokens[0].value.as_str().to_owned() + (" ".repeat((indent as usize)*2).as_str())).as_str() + } + else { + result += ast.tokens[0].value.as_str() + } + } + } + result = result.trim_end().to_string(); + if indent > 0 { + result += "\n"; + result += " ".repeat((indent as usize-1)*2).as_str(); + return "{".to_owned()+result.as_str()+"}" + } else { + return result + } +}