minishell/exec/src/builtins/export.c

106 lines
3.1 KiB
C

/* ************************************************************************** */
/* */
/* ::: :::::::: */
/* export.c :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: rparodi <rparodi@student.42.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2024/08/07 14:13:41 by rparodi #+# #+# */
/* Updated: 2024/08/30 19:25:49 by maiboyer ### ########.fr */
/* */
/* ************************************************************************** */
#include "exec/builtins.h"
#include "me/hashmap/hashmap_env.h"
#include "me/printf/printf.h"
#include "me/str/str.h"
#include "me/string/string.h"
#include "me/types.h"
#include "me/vec/vec_str.h"
t_error _append_key_to_vec(t_usize _, const t_str *key, t_str *v, void *vec);
bool _sort_str(t_str *_lhs, t_str *_rhs);
t_error get_uniq_keys(t_state *state, t_vec_str *out);
static void _assign_export(t_usize idx, t_str *arg, void *vctx)
{
struct s_assign_export_state *ctx;
const char *first_eq;
t_str key;
t_str value;
if (arg == NULL || *arg == NULL || idx == 0)
return ;
ctx = vctx;
first_eq = str_find_chr(*arg, '=');
if (first_eq == NULL || first_eq == *arg)
return ;
key = str_substring(*arg, 0, first_eq - *arg);
value = str_substring(first_eq, 1, ~0llu);
if (hmap_env_insert(ctx->state->env, key, value))
ctx->err = ERROR;
}
static t_error handle_quotes(t_str raw, t_string *out)
{
t_usize i;
t_string ret;
i = 0;
if (!raw)
return (ERROR);
ret = string_new(16);
while (raw[i] != '\0')
{
if (raw[i] == '\'')
string_push(&ret, "'\"'\"'");
else
string_push_char(&ret, raw[i]);
i++;
}
return (*out = ret, NO_ERROR);
}
t_error _print_export_env(\
t_state *state, t_builtin_spawn_info info, t_i32 *exit_code)
{
t_vec_str uniq;
t_usize i;
t_str *value;
t_string buf;
if (get_uniq_keys(state, &uniq))
return (ERROR);
i = 0;
while (i < uniq.len)
{
value = hmap_env_get(state->tmp_var, &uniq.buffer[i]);
if (value == NULL)
value = hmap_env_get(state->env, &uniq.buffer[i]);
if (value == NULL || *value == NULL)
me_printf_fd(info.stdout, "export %s\n", uniq.buffer[i]);
else
{
if (!handle_quotes(*value, &buf))
(me_printf_fd(\
info.stdout, "export %s='%s'\n", uniq.buffer[i], buf.buf), string_free(buf));
}
i++;
}
return (vec_str_free(uniq), *exit_code = 0, NO_ERROR);
}
t_error builtin_export(\
t_state *state, t_builtin_spawn_info info, t_i32 *exit_code)
{
struct s_assign_export_state assign_ctx;
(void)(state);
assign_ctx.info = &info;
assign_ctx.state = state;
assign_ctx.err = NO_ERROR;
if (info.args.len == 1)
return (_print_export_env(state, info, exit_code));
return (*exit_code = 0, vec_str_iter(\
&info.args, _assign_export, &assign_ctx), assign_ctx.err);
}