#include "ast.h"

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

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

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

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

struct ast_cmd *ast_get_cmd(struct ast *node)
{
    assert(node != NULL);
    assert(node->type == AST_CMD);
    return (struct ast_cmd *)node->data;
}

static struct ast *ast_create(enum ast_type type, void *data)
{
    struct ast *node = malloc(sizeof(struct ast));
    if (!node)
        return NULL;

    node->type = type;
    node->data = data;

    return node;
}

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 *ast_create_cmd(struct list *cmd)
{
    struct ast_cmd *cmd_data = malloc(sizeof(struct ast_cmd));
    if (!cmd_data)
        return NULL;

    cmd_data->cmd = cmd;

    return ast_create(AST_CMD, cmd_data);
}