/* ************************************************************************** */
/*                                                                            */
/*                                                        :::      ::::::::   */
/*   hashmap_env.c                              :+:      :+:    :+:   */
/*                                                    +:+ +:+         +:+     */
/*   By: maiboyer <maiboyer@student.42.fr>          +#+  +:+       +#+        */
/*                                                +#+#+#+#+#+   +#+           */
/*   Created: 2023/12/06 10:58:20 by maiboyer          #+#    #+#             */
/*   Updated: 2023/12/11 15:32:51 by maiboyer         ###   ########.fr       */
/*                                                                            */
/* ************************************************************************** */

#include "me/hash/hasher.h"
#include "me/hash/sip.h"
#include "me/hashmap/hashmap_env.h"
#include "me/mem/mem_alloc.h"
#include "me/mem/mem_alloc_array.h"
#include "me/mem/mem_copy.h"
#include "me/types.h"
#include <stdlib.h>

t_hashmap_env *new_hashmap_env(t_hash_env_fn hfunc,
											   t_eq_env_fn	 cfunc,
											   t_drop_env_fn drop)
{
	return (new_hashmap_with_buckets_env(hfunc, cfunc, drop,
												 DEFAULT_BUCKETS));
}

t_hashmap_env *new_hashmap_with_buckets_env(
	t_hash_env_fn hfunc, t_eq_env_fn cfunc,
	t_drop_env_fn drop, t_usize buckets)
{
	t_hashmap_env *hmap;

	hmap = mem_alloc(sizeof(*hmap));
	if (hmap == NULL)
		return (NULL);
	hmap->buckets = mem_alloc_array(buckets, sizeof(t_entry_env *));
	hmap->num_buckets = buckets;
	hmap->hasher = hasher_sip13_new();
	hmap->hfunc = hfunc;
	hmap->cfunc = cfunc;
	hmap->drop = drop;
	if (hmap->buckets == NULL)
		return ((void)me_free(hmap), NULL);
	return (hmap);
}

void drop_hashmap_env(t_hashmap_env *hmap)
{
	t_usize index;

	index = 0;
	while (index < hmap->num_buckets)
	{
		if (hmap->buckets[index])
		{
			hmap->drop(hmap->buckets[index]->kv);
			me_free(hmap->buckets[index]);
		}
		index++;
	}
	hasher_finish(&hmap->hasher);
	me_free(hmap->buckets);
	me_free(hmap);
}

t_entry_env *hashmap_get_entry_env(t_hashmap_env *hmap,
												   t_usize		 hashed_key,
												   t_str *key,
												   t_entry_env **prev)
{
	t_entry_env *entry;

	entry = hmap->buckets[hashed_key % hmap->num_buckets];
	while (entry != NULL)
	{
		if (!hmap->cfunc(&entry->kv.key, key))
		{
			*prev = entry;
			entry = entry->next;
		}
		else
		{
			return (entry);
		}
	}
	return (NULL);
}

void insert_hashmap_env(t_hashmap_env *hmap, t_str key,
								t_str value)
{
	t_usize				 hashed_key;
	t_entry_env *prev;
	t_entry_env *entry;

	hmap->hfunc(&hmap->hasher, &key);
	hashed_key = hasher_reset_and_finish(&hmap->hasher);
	prev = NULL;
	entry = hashmap_get_entry_env(hmap, hashed_key, &key, &prev);
	if (entry == NULL)
	{
		entry = mem_alloc(sizeof(t_entry_env));
		entry->hash_id = hashed_key;
		entry->kv = (t_kv_env){.key = key, .val = value};
		entry->next = NULL;
		if (prev == NULL)
			hmap->buckets[hashed_key % hmap->num_buckets] = entry;
		else
			prev->next = entry;
	}
	else
		entry->kv.val = value;
}