#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "lily-test.h"

struct lily_globals_t _lily_globals;


void lily_init()
{
   _lily_globals.error_msg_len = 0;
   _lily_globals.error_msg = NULL;
   _lily_globals.error_location = "";
}


static void _lily_assert_msg(bool statement, const char *location,
			     const char *format_string, va_list args)
{
   if (statement) {
      va_end(args);
      return; // no error, return
   }

   _lily_globals.error_location = location;

   va_list args_len;
   va_copy(args_len, args);
   size_t length = vsnprintf(NULL, 0, format_string, args_len);
   va_end(args_len);

   if (_lily_globals.error_msg_len < length+1 ||
       _lily_globals.error_msg == NULL) {
      if (_lily_globals.error_msg != NULL)
	 free(_lily_globals.error_msg);

      char *msg = malloc((length+2) * sizeof(char));

      if (msg == NULL) {
	 fprintf(stderr, "WARNING: failed to allocate memory for failed test message!\n");
	 _lily_globals.error_msg = NULL;
	 va_end(args);
	 longjmp(_lily_globals.env, 1);
	 return;
      }
      else {
	 _lily_globals.error_msg = msg;
	 _lily_globals.error_msg_len = length+1;
      }
   }

   vsnprintf(_lily_globals.error_msg, length+1, format_string, args);

   va_end(args);
   longjmp(_lily_globals.env, 1);
}


void lily_assert_msg(bool statement, const char *location,
		     const char *format_string, ...)
{
   va_list args;
   va_start(args, format_string);
   // _lily_assert_msg may long jump, so it takes calls va_end(args) for you
   _lily_assert_msg(statement, location, format_string, args);
}


void _lily_assert_true(const char *statement, bool value, const char *location)
{
   lily_assert_msg(value, location,
		   "%s is not true",
		   statement);
}


void _lily_assert_false(const char *statement, bool value, const char *location)
{
   lily_assert_msg(!value, location,
		   "%s is not false",
		   statement);
}


void _lily_assert_not_null(const char *name, void *ptr, const char *location)
{
   lily_assert_msg(ptr != NULL, location,
		   "%s is NULL",
		   name);
}


void _lily_assert_null(const char *name, void *ptr, const char *location)
{
   lily_assert_msg(ptr == NULL, location,
		   "%s (%p) is not NULL",
		   name, ptr);
}


void _lily_assert_ptr_equal(const char *name_a, const char *name_b,
			    void *a, void *b, const char *location)
{
   lily_assert_msg(a == b, location,
		   "%s (%p) is not equal to %s (%p)",
		   name_a, a, name_b, b);
}


void _lily_assert_ptr_not_equal(const char *name_a, const char *name_b,
			    void *a, void *b, const char *location)
{
   lily_assert_msg(a != b, location,
		   "%s (%p) is equal to %s",
		   name_a, a, name_b);
}


void _lily_assert_int_equal(const char *name_a, const char *name_b,
			    intmax_t a, intmax_t b, const char *location)
{
   lily_assert_msg(a == b, location,
		   "%s (%d) is not equal to %s (%d)",
		   name_a, a, name_b, b);
}


void _lily_assert_int_not_equal(const char *name_a, const char *name_b,
			    intmax_t a, intmax_t b, const char *location)
{
   lily_assert_msg(a != b, location,
		   "%s (%d) is equal to %s",
		   name_a, a, name_b);
}


void _lily_assert_float_equal(const char *name_a, const char *name_b,
			      double a, double b, double epsilon, const char *location)
{
   lily_assert_msg(abs(a - b) <= epsilon,
		   "%s (%f) is not equal to %s (%f) (epsilon: %f)",
		   name_a, a, name_b, b, epsilon);
}


void _lily_assert_float_not_equal(const char *name_a, const char *name_b,
				  double a, double b, double epsilon, const char *location)
{
   lily_assert_msg(abs(a - b) > epsilon,
		   "%s (%f) is equal to %s (%f) (epsilon: %f)",
		   name_a, a, name_b, b, epsilon);
}


void _lily_assert_string_equal(const char *name_a, const char *name_b,
			       char *a, char *b, const char *location)
{
   lily_assert_msg(strcmp(a, b) == 0,
		   "%s ('%s') is not equal to %s ('%s')",
		   name_a, a, name_b, b);
}


void _lily_assert_string_not_equal(const char *name_a, const char *name_b,
				   char *a, char *b, const char *location)
{
   lily_assert_msg(strcmp(a, b) != 0,
		   "%s ('%s') is equal to %s",
		   name_a, a, name_b);
}


void _lily_assert_memory_equal(const char *name_a, const char *name_b,
			       void *a, void *b, size_t size, const char *location)
{
   lily_assert_msg(memcmp(a, b, size) == 0,
		   "%s and %s contain different data", name_a, name_b);
}


void _lily_assert_memory_not_equal(const char *name_a, const char *name_b,
				   void *a, void *b, size_t size, const char *location)
{
   lily_assert_msg(memcmp(a, b, size) == 0,
		   "%s contains the same data s %s", name_a, name_b);
}


lily_queue_t* lily_queue_create()
{
   const size_t size = 256 * sizeof(uint8_t);

   lily_queue_t *q = malloc(sizeof(lily_queue_t));
   q->buf_size = size;
   q->buf = malloc(size);
   q->front = q->buf;
   q->back = q->front;

   return q;
}


void lily_queue_destroy(lily_queue_t *q)
{
   free(q->buf);
   free(q);
}


void _lily_enqueue(lily_queue_t *q, size_t size, uint8_t *data)
{
   size_t used = q->back - q->buf;
   size_t remaining = q->buf_size - used;

   if (remaining < size) {
      /* re-allocate bigger buffer */
      size_t size_new = 2 * q->buf_size;
      uint8_t *buf_new = realloc(q->buf, size_new);
      if (buf_new == NULL) {
	 fprintf(stderr, "failed to allocated %lu bytes for queue %p!\n",
		 size_new, q);
	 return;
      }
      q->buf = buf_new;
   }

   memcpy(q->back, data, size);
   q->back += size;
}


void _lily_dequeue(lily_queue_t *q, size_t size, uint8_t *ptr)
{
   size_t dist = q->back - q->front;
   if (dist < size) {
      fprintf(stderr, "attempted to read %lu bytes out of queue %p, "
	      "which has %lu bytes presently queued\n", size, q, dist);
      return;
   }

   memcpy(ptr, q->front, size);
   q->front += size;
}