#include "netbytes.h"


int nb_u8_create( nb_u8 *s, uint8_t val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U8;
	s->val = val; 

	return 0;
}


int nb_u8arr_create( nb_u8arr *s, __NBT_U8ARR_LEN len, uint8_t *val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U8ARRAY;
	s->len = len;
	s->val = val;

	return 0;
}

int nb_u16_create(    nb_u16 *s,    uint16_t val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U16;
	s->val = val;

	return 0;
}


int nb_u16arr_create( nb_u16arr *s, __NBT_U16ARR_LEN len, uint16_t *val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U16ARRAY;
	s->len = len;
	s->val = val;

	return 0;
}


int nb_u32_create(    nb_u32 *s,    uint32_t val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U32;
	s->val = val;

	return 0;
}


int nb_u32arr_create( nb_u32arr *s, __NBT_U32ARR_LEN len, uint32_t *val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U32ARRAY;
	s->len = len;
	s->val = val;

	return 0;
}


int nb_u64_create(    nb_u64 *s,    uint64_t val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U64;
	s->val = val;

	return 0;
}


int nb_u64arr_create( nb_u64arr *s, __NBT_U64ARR_LEN len, uint64_t *val )
{
	if ( s == NULL )
		return -1;

	s->type = NBT_U64ARRAY;
	s->len = len;
	s->val = val;

	return 0;
}


int nb_init( netbyte_store *store )
{
	if ( store == NULL )
		return -1;

	memset( store, 0, sizeof(netbyte_store) );
	
	/* we know that they allready zero, but philosophy more important */
	store->size = 0;
	store->count = 0;

	return 0;
}


int nb_add_u8( netbyte_store *store, nb_u8 *u8 )
{
	nb_u8 *new_u8=NULL;

	if ( store->count >= __NBT_MAX_TYPES - 1 )
	{
		return -1;
	}

	if ( u8->type != NBT_U8 )
	{
		return -2;
	}

	store->types[store->count].type = NBT_U8;

	new_u8 = malloc(sizeof(nb_u8));
	memcpy(new_u8, u8, sizeof(nb_u8)); 
	store->types[store->count].nb_val = (uint8_t *)new_u8;

	store->count += 1;
	return 0;
}	


int nb_add_u8arr( netbyte_store *store, nb_u8arr *u8arr )
{
	nb_u8arr *new_u8arr = NULL;

	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u8arr->type != NBT_U8ARRAY )
		return -2;

	store->types[store->count].type = NBT_U8ARRAY;

	new_u8arr = malloc(sizeof(nb_u8arr));
	memcpy(new_u8arr, u8arr, sizeof(nb_u8arr));
	new_u8arr->val = malloc(sizeof(nb_u8arr)*u8arr->len);
	memcpy(new_u8arr->val, u8arr->val, u8arr->len);
	store->types[store->count].nb_val = (uint8_t *)new_u8arr;

	store->count += 1;
	return 0;
}


int nb_add_u16( netbyte_store *store, nb_u16 *u16 )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u16->type != NBT_U16 )
		return -2;

	store->types[store->count].type = NBT_U16;
	store->types[store->count].nb_val = (uint8_t *)u16;
	store->count += 1;
	return 0;
}


int nb_add_u16arr( netbyte_store *store, nb_u16arr *u16arr )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u16arr->type != NBT_U16ARRAY )
		return -2;

	store->types[store->count].type = NBT_U16ARRAY;
	store->types[store->count].nb_val = (uint8_t *)u16arr;
	store->count += 1;
	return 0;
}


int nb_add_u32( netbyte_store *store, nb_u32 *u32 )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u32->type != NBT_U32 )
		return -2;

	store->types[store->count].type = NBT_U32;
	store->types[store->count].nb_val = (uint8_t *)u32;
	store->count += 1;
	return 0;
}


int nb_add_u32arr( netbyte_store *store, nb_u32arr *u32arr )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u32arr->type != NBT_U32ARRAY )
		return -2;

	store->types[store->count].type = NBT_U32ARRAY;
	store->types[store->count].nb_val = (uint8_t *)u32arr;
	store->count += 1;
	return 0;
}


int nb_add_u64( netbyte_store *store, nb_u64 *u64 )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u64->type != NBT_U64 )
		return -2;

	store->types[store->count].type = NBT_U64;
	store->types[store->count].nb_val = (uint8_t *)u64;
	store->count += 1;
	return 0;
}


int nb_add_u64arr( netbyte_store *store, nb_u64arr *u64arr )
{
	if ( store->count >= __NBT_MAX_TYPES - 1 )
		return -1;

	if ( u64arr->type != NBT_U64ARRAY )
		return -2;

	store->types[store->count].type = NBT_U64ARRAY;
	store->types[store->count].nb_val = (uint8_t *)u64arr;
	store->count += 1;
	return 0;
}


uint8_t *nb_create( netbyte_store *store )
{
	uint8_t *ret = NULL;
	uint8_t *buf = NULL;
	uint8_t *c   = NULL;

	int i;

	if ( store == NULL )
		return NULL;

	if ( store->count < 1 )
		return NULL;

	store->size == 0;

	/* precalc needed size starting from zero not metter what */
	store->size += sizeof(__NBT_SIZE);
	for (i=0; i < store->count; i++)
	{
		__nb_type iter = store->types[i];
		switch( iter.type )
		{
		case NBT_U8:
			store->size += sizeof(__NBT_TYPED) + sizeof(uint8_t);
			break;
		case NBT_U8ARRAY:
			{
				nb_u8arr *u8arr = (nb_u8arr *)iter.nb_val;
				store->size += sizeof(__NBT_TYPED) + 
					sizeof(__NBT_U8ARR_LEN) +
					u8arr->len;
			}
			break;
		case NBT_U16:
			store->size += sizeof(__NBT_TYPED) + sizeof(uint16_t);
			break;
		case NBT_U16ARRAY:
			{
				nb_u16arr *u16arr = (nb_u16arr *)iter.nb_val;
				store->size += sizeof(__NBT_TYPED) + 
					sizeof(__NBT_U16ARR_LEN) +
					u16arr->len*sizeof(uint16_t);
			}
			break;
		case NBT_U32:
			store->size += sizeof(__NBT_TYPED) + sizeof(uint32_t);
			break;
		case NBT_U32ARRAY:
			{
				nb_u32arr *u32arr = (nb_u32arr *)iter.nb_val;
				store->size += sizeof(__NBT_TYPED) + 
					sizeof(__NBT_U32ARR_LEN) +
					u32arr->len*sizeof(uint32_t);
			}
			break;
		case NBT_U64:
			store->size += sizeof(__NBT_TYPED) + sizeof(uint64_t);
			break;
		case NBT_U64ARRAY:
			{
				nb_u64arr *u64arr = (nb_u64arr *)iter.nb_val;
				store->size += sizeof(__NBT_TYPED) + 
					sizeof(__NBT_U64ARR_LEN) +
					u64arr->len*sizeof(uint64_t);
			}
			break;
		default:
			printf("Unknow type be carefull1\n");
		}
	}

	/* create netbyte memory chunk */ 
	buf = malloc( store->size );
	if ( buf == NULL )
	{
		return NULL;
	}
	c = buf;

	/* write size of netbytes */
	memcpy( c, &store->size, sizeof(__NBT_SIZE) );
	c += sizeof(__NBT_SIZE);

	/* per each type copy stuff */
	for ( i = 0; i < store->count; i++ )
	{
		__nb_type iter = store->types[i];
		switch ( iter.type )
		{
		case NBT_U8:
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &((nb_u8 *)(iter.nb_val))->val, sizeof(uint8_t) );
			c += sizeof(uint8_t);
			break;
		case NBT_U8ARRAY:
		{
			nb_u8arr *u8arr = (nb_u8arr *)iter.nb_val;
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &u8arr->len, sizeof(__NBT_U8ARR_LEN));
			c += sizeof(__NBT_U8ARR_LEN);
			memcpy( c, u8arr->val, u8arr->len );
			c += u8arr->len;
			break;
		}
		case NBT_U16:
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &((nb_u16 *)(iter.nb_val))->val, sizeof(uint16_t) );
			c += sizeof(uint16_t);
			break;
		case NBT_U16ARRAY:
		{	
			nb_u16arr *u16arr = (nb_u16arr *)iter.nb_val;
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &u16arr->len, sizeof(__NBT_U16ARR_LEN));
			c += sizeof(__NBT_U16ARR_LEN);
			memcpy( c, u16arr->val, u16arr->len );
			c += u16arr->len*sizeof(uint16_t);
			break;
		}
		case NBT_U32:
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &((nb_u32 *)(iter.nb_val))->val, sizeof(uint32_t) );
			c += sizeof(uint32_t);
			break;
		case NBT_U32ARRAY:
		{
			nb_u32arr *u32arr = (nb_u32arr *)iter.nb_val;
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &u32arr->len, sizeof(__NBT_U32ARR_LEN));
			c += sizeof(__NBT_U32ARR_LEN);
			memcpy( c, u32arr->val, u32arr->len );
			c += u32arr->len*sizeof(uint32_t);
			break;
		}
		case NBT_U64:
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &((nb_u64 *)(iter.nb_val))->val, sizeof(uint64_t) );
			c += sizeof(uint64_t);
			break;
		case NBT_U64ARRAY:
		{
			nb_u64arr *u64arr = (nb_u64arr *)iter.nb_val;
			memcpy( c, &iter.type, sizeof(__NBT_TYPED) );
			c += sizeof(__NBT_TYPED);
			memcpy( c, &u64arr->len, sizeof(__NBT_U64ARR_LEN));
			c += sizeof(__NBT_U64ARR_LEN);
			memcpy( c, u64arr->val, u64arr->len );
			c += u64arr->len*sizeof(uint64_t);
			break;
		}
		default:
			printf("Unknow type be carefull2\n");
		}
	}

	if ( c-buf-1 > store->size )
	{
		printf("Something went wrong while created netbytes\n");
		free(buf);
		return NULL;
	}

	ret = buf;

	return ret;
}


int nb_free(netbyte_store *store)
{
	int i;

	if (!store)
	{
		return -1;
	}

	for (i=0;i<store->count;i++)
	{
		__NBT_TYPED type = store->types[i].type;
		uint8_t *val = store->types[i].nb_val;

		switch(type)
		{
		case NBT_U8:
		{
			nb_u8 *u8 = (nb_u8 *)val;
			free(u8);
			u8=NULL;
			store->types[i].nb_val=NULL;
			store->types[i].type=0;
			break;
		}
		case NBT_U8ARRAY:
		{
			nb_u8arr *u8arr = (nb_u8arr *)val;
			free(u8arr->val);
			free(u8arr);
			store->types[i].nb_val=NULL;
			store->types[i].type=0;
			break;
		}
		default:
			printf("Unknown type\n");
		}
	}
	free(store);

	return 0;
}

int nb_load( netbyte_store *store, uint8_t *data )
{
	__NBT_TYPED type;
	int iter=1;
	uint8_t *c = data;
	int count;

	if ( store == NULL )
	{
		return -1;
	}

	if ( data == NULL )
	{
		return -2;
	}

	__NBT_SIZE size;
	memcpy( &size, c, sizeof(__NBT_SIZE) );
	printf("size: %d\n", size);

	size -= sizeof(__NBT_SIZE);
	c += sizeof(__NBT_SIZE);

	if (c == data)
	{
		store->size = size;
		return 0;
	}

	count = 0;
	memset( store, 0, sizeof(netbyte_store) );
	while ( iter )
	{
		type = (__NBT_TYPED)c[0];
		switch( type )
		{
		case NBT_U8:
		{
			nb_u8 *u8 = malloc( sizeof(nb_u8) );
			
			memcpy( u8, c, sizeof(nb_u8) );
			c += sizeof(nb_u8);
			
			nb_add_u8( store, u8 );

			free(u8);
			
			break;
		}
		case NBT_U8ARRAY:
		{
			nb_u8arr *u8arr = malloc( sizeof(nb_u8arr) );
			
			memcpy( u8arr, c, sizeof(__NBT_TYPED));
			c += sizeof(__NBT_TYPED);
			
			memcpy( &u8arr->len, c, sizeof(__NBT_U8ARR_LEN) );
			c += sizeof(__NBT_U8ARR_LEN);
			
			u8arr->val = c;
			//printf("->2 %s\n",c );
			c += u8arr->len;
			
			nb_add_u8arr( store, u8arr );

			free(u8arr);
			
			break;
		}
		case NBT_U16:
		{
			nb_u16 *u16 = malloc( sizeof(nb_u16) );
			
			memcpy( u16, c, sizeof(nb_u16) );
			c += sizeof(nb_u16);
			
			nb_add_u16( store, u16 );

			break;
		}
		case NBT_U16ARRAY:
		{
			nb_u16arr *u16arr = malloc( sizeof(nb_u16arr) );
			
			memcpy( u16arr, c, sizeof(__NBT_TYPED));
			c += sizeof(__NBT_TYPED);
			
			memcpy( &u16arr->len, c, sizeof(__NBT_U16ARR_LEN) );
			c += sizeof(__NBT_U16ARR_LEN);
			
			u16arr->val = (uint16_t *)c;
			//printf("->2 %s\n",c );
			c += u16arr->len*sizeof(uint16_t);
			
			nb_add_u16arr( store, u16arr );
			break;
		}
		case NBT_U32:
		{
			nb_u32 *u32 = malloc( sizeof(nb_u32) );
			
			memcpy( u32, c, sizeof(nb_u32) );
			c += sizeof(nb_u32);
			
			nb_add_u32( store, u32 );
			break;
		}
		case NBT_U32ARRAY:
		{
			nb_u32arr *u32arr = malloc( sizeof(nb_u32arr) );
			
			memcpy( u32arr, c, sizeof(__NBT_TYPED));
			c += sizeof(__NBT_TYPED);
			
			memcpy( &u32arr->len, c, sizeof(__NBT_U32ARR_LEN) );
			c += sizeof(__NBT_U32ARR_LEN);
			
			u32arr->val = (uint32_t *)c;
			//printf("->2 %s\n",c );
			c += u32arr->len*sizeof(uint32_t);
			
			nb_add_u32arr( store, u32arr );
			break;
		}
		case NBT_U64:
		{
			nb_u64 *u64 = malloc( sizeof(nb_u64) );
			
			memcpy( u64, c, sizeof(nb_u64) );
			c += sizeof(nb_u64);
			
			nb_add_u64( store, u64 );
			break;
		}
		case NBT_U64ARRAY:
		{
			nb_u64arr *u64arr = malloc( sizeof(nb_u64arr) );
			
			memcpy( u64arr, c, sizeof(__NBT_TYPED));
			c += sizeof(__NBT_TYPED);
			
			memcpy( &u64arr->len, c, sizeof(__NBT_U64ARR_LEN) );
			c += sizeof(__NBT_U64ARR_LEN);
			
			u64arr->val = (uint64_t *)c;
			//printf("->2 %s\n",c );
			c += u64arr->len;
			
			nb_add_u64arr( store, u64arr );
			break;
		}
		default:
			printf("Unknown type");
		}
		if ( c >= data + size )
		{
			iter = 0;
		}
	}

	return 0;
}

int nb_count( netbyte_store *store )
{
	if ( store == NULL )
	{
		return -1;
	}

	return store->count;
}


int nb_type( netbyte_store *store, int count, __NBT_TYPED **type )
{
	__NBT_TYPED nbt;
	if ( store == NULL )
	{
		return -1;
	}

	if ( store->count < count )
	{
		return -2;
	}

	//nbt = (__NBT_TYPED)store->types[count].type; //warnign? who cares
	//*type = nbt;
	*type = store->types[count].type;
	return 0;
}

int nb_val( netbyte_store *store, int count, uint8_t **val )
{
	if ( store == NULL )
		return -1;

	if ( store->count < count )
		return -2;

	*val = store->types[count].nb_val;

	return 0;
}


int nb_fread( netbyte_store *store, int fd)
{
	__NBT_SIZE size;
	ssize_t ret;
	off_t old_seek;
	uint8_t *nb;	

	if ( store == NULL )
		return -1;

	if ( fd < 1 )
		return -2;


	nb_init( store );

	old_seek = lseek( fd, 0, SEEK_CUR);
	if ( read( fd, &size, sizeof(__NBT_SIZE) ) != sizeof(__NBT_SIZE) )
	{
		printf("Couldn read netbyte size\n");
		return -3;
	}

	if ( lseek( fd, old_seek, SEEK_SET ) == -1 )
	{
		return -4;
	}

	//different endianess can blow this up muahaha
	nb = malloc( size );

	if ( read( fd, nb, size ) != size )
	{
		return -5;
	}

	if ( nb_load( store, nb ) != 0 )
	{
		printf("Cannot read nebyte from file\n");
		return -6;
	}

	return 0;
}

int nb_print(netbyte_store *store)
{
	int i;

	if (!store)
	{
		printf("nb: NULL\n");
		return -1;
	}

	printf("nb: size=0x%x count=0x%x\n", store->size, store->count);

	for (i=0;i<store->count;i++)
	{
		printf("\t[%x] -> ", i);
		switch (store->types[i].type)
		{
		case NBT_U8:
		{
			nb_u8 *u8 = (nb_u8 *)store->types[i].nb_val;
			printf("u8:    t=0x%x v=0x%x",
				u8->type, u8->val);
			break;
		}
		case NBT_U8ARRAY:
		{
			nb_u8arr *u8arr = (nb_u8arr *)store->types[i].nb_val;
			printf("u8arr: t=0x%x l=0x%x v=0x%x",
				u8arr->type, u8arr->len, u8arr->val);
			break;
		}
		default:
			printf("Unknown");
		}
		printf("\n");
	}

	return 0;
}