#include "code.h"

#include <string.h>
#include <time.h>

#include "mem.h"

void fill_node_int(struct node** node, int i) {
	free_node(*node, 1, 0);
	(*node)->kind = EXPR_INT;
	(*node)->var1 = my_calloc(1, sizeof(int));
	*((int*) (*node)->var1) = i;
}

void fill_node_name(struct node** node, char* s) {
	free_node(*node, 1, 0);
	(*node)->kind = EXPR_NAME;
	(*node)->var1 = my_calloc(1, strlen(s) + 1);
	strcpy((*node)->var1, s);
}

void code_time(struct node** result) {
	fill_node_int(result, (int) time(NULL));
}

void code_mul(struct node** result, struct node* a, struct node* b) {
	if (a->kind != EXPR_INT || b->kind != EXPR_INT)
		fill_node_name(result, "mul on non-ints");
	else
		fill_node_int(result, *((int*) a->var1) * *((int*) b->var1));
}

void code_sub(struct node** result, struct node* a, struct node* b) {
	if (a->kind != EXPR_INT || b->kind != EXPR_INT)
		fill_node_name(result, "sub on non-ints");
	else
		fill_node_int(result, *((int*) b->var1) - *((int*) a->var1));
}

unsigned char code_find(char* name, void** function) {
	if (!strcmp(name, "time")) {
		*function = (void(*)(void)) code_time;
		return 0;
	} else if (!strcmp(name, "mul")) {
		*function = (void(*)(void)) code_mul;
		return 2;
	} else if (!strcmp(name, "sub")) {
		*function = (void(*)(void)) code_sub;
		return 2;
	}

	*function = NULL;
	return 0;
}