yarn should be able to handle cmd now

parser/src/yarn/ast_cmd.c

@@ -6,14 +6,208 @@
 /*   By: maiboyer <maiboyer@student.42.fr>          +#+  +:+       +#+        */
 /*                                                +#+#+#+#+#+   +#+           */
 /*   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);
+}