42sh/src/utils/ast/ast_if.c

#include "utils/ast/ast_if.h"

#include <stdlib.h>
#include <assert.h>
#include <stdbool.h>

struct ast *ast_create_if(struct ast *condition, struct ast *then_clause,
                          struct ast *else_clause)
{
    struct ast_if *if_data = malloc(sizeof(struct ast_if));
    if (!if_data)
        return NULL;

    if_data->condition = condition;
    if_data->then_clause = then_clause;
    if_data->else_clause = else_clause;

    return ast_create(AST_IF, if_data);
}

struct ast_if *ast_get_if(struct ast *node)
{
    assert(node != NULL);
    assert(node->type == AST_IF);
    return (struct ast_if *)node->data;
}

bool ast_is_if(struct ast *node)
{
    assert(node != NULL);
    return node->type == AST_IF;
}

void ast_free_if(struct ast_if *if_data)
{
    if (if_data == NULL)
        return;

    ast_free(if_data->condition);
    ast_free(if_data->then_clause);
    ast_free(if_data->else_clause);

    free(if_data);
}