From 5c7c999da00b394b1994e79f342ab7d7a09602b0 Mon Sep 17 00:00:00 2001 From: maix0 Date: Wed, 9 Oct 2024 21:55:55 +0200 Subject: [PATCH] yarn should be able to handle cmd now --- parser/src/yarn/ast_cmd.c | 225 +++++++++++++++++++++++++++++++++++++- 1 file changed, 222 insertions(+), 3 deletions(-) diff --git a/parser/src/yarn/ast_cmd.c b/parser/src/yarn/ast_cmd.c index dc69f5a1..70160538 100644 --- a/parser/src/yarn/ast_cmd.c +++ b/parser/src/yarn/ast_cmd.c @@ -6,14 +6,208 @@ /* By: maiboyer +#+ +:+ +#+ */ /* +#+#+#+#+#+ +#+ */ /* Created: 2024/10/09 12:44:53 by maiboyer #+# #+# */ -/* Updated: 2024/10/09 12:45:32 by maiboyer ### ########.fr */ +/* Updated: 2024/10/09 21:55:35 by maiboyer ### ########.fr */ /* */ /* ************************************************************************** */ #include "ast/ast.h" -#include "parser/token.h" +#include "me/str/str.h" +#include "me/string/string.h" +#include "me/types.h" #include "me/vec/vec_ast.h" #include "me/vec/vec_token.h" +#include "parser/token.h" + +t_error handle_tok_word_inner(t_token *tok, t_ast_node *out); + +static bool _is_carret(enum e_token type) +{ + return (type == TOK_DLCARRET || type == TOK_DRCARRET || type == TOK_LCARRET || type == TOK_RCARRET); +} + +static t_error _create_ast_redir(enum e_token ty, t_ast_node *out) +{ + t_ast_node ret; + ret = NULL; + if (ty == TOK_DLCARRET) + { + ret = ast_alloc(AST_HEREDOC_REDIRECTION); + ret->data.heredoc_redirection.op = AST_REDIR_HEREDOC; + } + else if (ty == TOK_DRCARRET || ty == TOK_LCARRET || ty == TOK_DRCARRET) + ret = ast_alloc(AST_FILE_REDIRECTION); + else + return (ERROR); + if (ty == TOK_DRCARRET) + ret->data.file_redirection.op = AST_REDIR_APPEND; + else if (ty == TOK_LCARRET) + ret->data.file_redirection.op = AST_REDIR_INPUT; + else if (ty == TOK_DRCARRET) + ret->data.file_redirection.op = AST_REDIR_OUTPUT; + return (*out = ret, NO_ERROR); +} + +t_const_str _token_to_string(t_token *arg) +{ + t_usize i; + t_string s; + t_const_str tmp; + + if (arg == NULL || arg->type == TOK_NONE) + return (NULL); + if (arg->string.buf != NULL) + return (str_clone(arg->string.buf)); + else if (arg->subtokens.buffer != NULL) + { + i = 0; + s = string_new(16); + while (i < arg->subtokens.len) + { + tmp = _token_to_string(&arg->subtokens.buffer[i++]); + string_push(&s, tmp); + str_free((t_str)tmp); + } + return (s.buf); + } + return (NULL); +} + +static t_error _ast_set_redir_arg(t_ast_node node, t_token *arg) +{ + if (node == NULL || arg == NULL || (node->kind != AST_HEREDOC_REDIRECTION && node->kind != AST_FILE_REDIRECTION)) + return (ERROR); + if (node->kind == AST_HEREDOC_REDIRECTION) + node->data.heredoc_redirection.delimiter = (t_str)_token_to_string(arg); + else if (handle_tok_word_inner(arg, &node->data.file_redirection.output)) + return (ERROR); + return (NO_ERROR); +} + +t_error handle_tok_word(t_ast_node cmd, t_token *tok) +{ + t_ast_node word; + + if (cmd == NULL || cmd->kind != AST_COMMAND || tok == NULL || tok->type != TOK_WORD) + return (ERROR); + if (handle_tok_word_inner(tok, &word)) + return (ERROR); + vec_ast_push(&cmd->data.command.cmd_word, word); + return (NO_ERROR); +} + +t_error _tok_word(t_token *tok, t_ast_node *out); +t_error _tok_word_expansion(t_token *tok, t_ast_node *out) +{ + t_ast_node ret; + + ret = ast_alloc(AST_EXPANSION); + ret->data.expansion.var_name = (t_str)_token_to_string(tok); + return (*out = ret, NO_ERROR); +} +t_error _tok_word_nquote(t_token *tok, t_ast_node *out) +{ + t_ast_node ret; + t_ast_node tmp; + + ret = ast_alloc(AST_WORD); + ret->data.word.kind = AST_WORD_NO_QUOTE; + tmp = ast_alloc(AST_RAW_STRING); + tmp->data.raw_string.str = (t_str)_token_to_string(tok); + vec_ast_push(&ret->data.word.inner, tmp); + return (*out = ret, NO_ERROR); +} +t_error _tok_word_squote(t_token *tok, t_ast_node *out) +{ + t_ast_node ret; + t_ast_node tmp; + + ret = ast_alloc(AST_WORD); + ret->data.word.kind = AST_WORD_SINGLE_QUOTE; + tmp = ast_alloc(AST_RAW_STRING); + tmp->data.raw_string.str = (t_str)_token_to_string(tok); + vec_ast_push(&ret->data.word.inner, tmp); + return (*out = ret, NO_ERROR); +} +t_error _tok_word_dquote(t_token *tok, t_ast_node *out) +{ + t_ast_node ret; + t_ast_node tmp; + t_usize i; + + ret = ast_alloc(AST_WORD); + ret->data.word.kind = AST_WORD_DOUBLE_QUOTE; + i = 0; + while (i < tok->subtokens.len) + { + if (_tok_word(&tok->subtokens.buffer[i], &tmp)) + return (ast_free(ret), ERROR); + vec_ast_push(&ret->data.word.inner, tmp); + } + return (*out = ret, NO_ERROR); +} + +t_error _tok_word(t_token *tok, t_ast_node *out) +{ + if (tok->type == TOK_EXPENSION) + { + if (_tok_word_expansion(tok, out)) + return (ERROR); + } + else if (tok->type == TOK_NQUOTE) + { + if (_tok_word_nquote(tok, out)) + return (ERROR); + } + else if (tok->type == TOK_SQUOTE) + { + if (_tok_word_squote(tok, out)) + return (ERROR); + } + else if (tok->type == TOK_DQUOTE) + { + if (_tok_word_dquote(tok, out)) + return (ERROR); + } + else + return (ERROR); + return (NO_ERROR); +} + +t_error handle_tok_word_inner(t_token *tok, t_ast_node *out) +{ + t_usize i; + t_ast_node ret; + t_ast_node tmp; + + if (tok == NULL || out == NULL || tok->type != TOK_WORD) + return (ERROR); + i = 0; + ret = ast_alloc(AST_WORD); + ret->data.word.kind = AST_WORD_NO_QUOTE; + while (i < tok->subtokens.len) + { + if (_tok_word(&tok->subtokens.buffer[i++], &tmp)) + return (ast_free(ret), ERROR); + vec_ast_push(&ret->data.word.inner, tmp); + } + return (*out = ret, NO_ERROR); +} + +t_error handle_tok_redir(t_ast_node cmd, t_token *tok) +{ + t_ast_node redir; + + if (cmd == NULL || tok == NULL || tok->type != TOK_REDIR || cmd->kind != AST_COMMAND) + return (ERROR); + if (!(tok->subtokens.len == 2 && _is_carret(tok->subtokens.buffer[0].type) && tok->subtokens.buffer[1].type == TOK_WORD)) + return (ERROR); + if (_create_ast_redir(tok->subtokens.buffer[0].type, &redir)) + return (ERROR); + if (_ast_set_redir_arg(redir, &tok->subtokens.buffer[1])) + return (ast_free(redir), ERROR); + vec_ast_push(&cmd->data.command.prefixes, redir); + return (NO_ERROR); +} /// la fonction doit prendre une t_token de type TOK_CMD qui contient que deux type de subtokens TOK_WORD ou TOK_REDIR /// un TOK_WORD == un arguemnt @@ -24,4 +218,29 @@ /// les noms peuvent etre different idk /// a terme la fonction utilisera t_error et tt; /// struct s_ast_command `ast/include/ast/_raw_structs.h` -t_ast_node ast_from_cmd(t_token tok); +t_ast_node ast_from_cmd(t_token tok) +{ + t_ast_node ret; + t_usize i; + + if (tok.type != TOK_CMD) + me_abort("tok.type != TOK_CMD"); + ret = ast_alloc(AST_COMMAND); + i = 0; + while (i < tok.subtokens.len) + { + if (tok.subtokens.buffer[i].type == TOK_REDIR) + { + if (handle_tok_redir(ret, &tok.subtokens.buffer[i])) + me_abort("handle_tok_redir error"); + } + else if (tok.subtokens.buffer[i].type == TOK_WORD) + { + if (handle_tok_word(ret, &tok.subtokens.buffer[i])) + me_abort("handle_tok_word error"); + } + else + me_abort("handle_tok_cmd not word|redir"); + } + return (ret); +}