#include <stdio.h>

#include "bootlib.h"
#include "stack.h"

void stack_push(my_stack_t *stack, void *obj) {
	if (stack->count == stack->capacity) {
		// Double stack capacity to avoid reallocing often
		stack->capacity *= 2;
		stack->data = boot_realloc(stack->data, stack->capacity * sizeof(void *));
		if (stack->data == NULL) {
			// Unable to realloc, just exit :) get gud
			exit(1);
		}
	}

	stack->data[stack->count] = obj;
	stack->count++;

	return;
}

void *stack_pop(my_stack_t *stack) {
	if (stack->count == 0) {
		return NULL;
	}

	stack->count--;
	return stack->data[stack->count];
}

void stack_free(my_stack_t *stack) {
	if (stack == NULL) {
		return;
	}

	if (stack->data != NULL) {
		boot_free(stack->data);
	}

	boot_free(stack);
}

void stack_remove_nulls(my_stack_t *stack) {
	size_t new_count = 0;

	// Iterate through the stack and compact non-NULL pointers.
	for (size_t i = 0; i < stack->count; ++i) {
		if (stack->data[i] != NULL) {
			stack->data[new_count++] = stack->data[i];
		}
	}

	// Update the count to reflect the new number of elements.
	stack->count = new_count;

	// Optionally, you might want to zero out the remaining slots.
	for (size_t i = new_count; i < stack->capacity; ++i) {
		stack->data[i] = NULL;
	}
}

my_stack_t *stack_new(size_t capacity) {
	my_stack_t *stack = boot_malloc(sizeof(my_stack_t));
	if (stack == NULL) {
		return NULL;
	}

	stack->count = 0;
	stack->capacity = capacity;
	stack->data = boot_malloc(stack->capacity * sizeof(void *));
	if (stack->data == NULL) {
		boot_free(stack);
		return NULL;
	}

	return stack;
}