/*
 *	UCW Library -- Test of Extended Types
 *
 *	(c) 2014 Robert Kessl <robert.kessl@economia.cz>
 *
 *	This software may be freely distributed and used according to the terms
 *	of the GNU Lesser General Public License.
 */

#include <ucw/lib.h>
#include <ucw/mempool.h>
#include <ucw/xtypes.h>
#include <ucw/xtypes-extra.h>

#include <errno.h>
#include <inttypes.h>
#include <stdlib.h>
#include <time.h>

static void test_size_parse_correct(struct fastbuf *out)
{
  static const char *size_strs[] = {
    "4",
    "4KB",
    "4MB",
    "4GB",
    "4TB",
    NULL
  };

  static u64 size_parsed[] = {
    4LLU,
    4 * 1024LLU,
    4 * 1024LLU * 1024LLU,
    4 * 1024LLU * 1024LLU * 1024LLU,
    4 * 1024LLU * 1024LLU * 1024LLU * 1024LLU
  };

  struct mempool *pool = mp_new(4096);

  uint i = 0;
  while(size_strs[i] != NULL) {
    u64 result;
    const char *parse_err = xt_size.parse(size_strs[i], &result, pool);

    if(parse_err != NULL) {
      die("Unexpected error in xt_size.parse");
    }
    if(size_parsed[i] != result) {
      die("xt_size.parse parsed an incorrect value.");
    }

    const char *result_str = xt_size.format(&result, XT_SIZE_FMT_UNIT(i), pool);
    bprintf(out, "%s %s\n", size_strs[i], result_str);

    i++;
  }

  mp_delete(pool);
}

static void test_size_parse_errors(struct fastbuf *out)
{
  static const char *size_strs[] = {
    "1X",
    "KB",
    "X1KB",
    "1XKB",
    "1KBX",
    "\0",
    NULL
  };

  uint i = 0;
  struct mempool *pool = mp_new(4096);

  while(size_strs[i] != NULL) {
    u64 result;
    const char *parse_err = xt_size.parse(size_strs[i], &result, pool);
    if(parse_err == NULL) {
      bprintf(out, "xt_size.parse incorrectly did not result in error while parsing: '%s'.\n", size_strs[i]);
    } else {
      bprintf(out, "xt_size.parse error: '%s'.\n", parse_err);
    }

    i++;
  }

  mp_delete(pool);
}

static void test_bool_parse_correct(struct fastbuf *out)
{
  static const char *bool_strs[] = {
    "0",
    "1",
    "false",
    "true",
    NULL
  };

  static bool bool_parsed[] = {
    false,
    true,
    false,
    true
  };

  struct mempool *pool = mp_new(4096);
  uint i = 0;

  while(bool_strs[i] != NULL) {
    bool result;
    const char *err_str = xt_bool.parse(bool_strs[i], &result, pool);
    if(err_str != NULL) {
      die("Unexpected error in xt_bool.parse %s", err_str);
    }
    if(bool_parsed[i] != result) {
      die("xt_bool.parse parsed an incorrect value.");
    }
    bprintf(out, "%s %s\n", bool_strs[i], result ? "true" : "false");
    i++;
  }

  mp_delete(pool);
}

static void test_timestamp_parse_correct(struct fastbuf *out)
{
  static const char *timestamp_strs[] = {
    "1403685533",
    "2014-06-25 08:38:53",
    NULL
  };

  static u64 timestamp_parsed[] = {
    1403685533,
    0,
  };

  // fill the value of timestamp_parsed as the timestamp in the
  // current time zone.
  {
    struct tm parsed_time;
    strptime(timestamp_strs[1], "%F %T", &parsed_time);
    timestamp_parsed[1] = mktime(&parsed_time);
  }

  struct mempool *pool = mp_new(4096);
  uint i = 0;

  while(timestamp_strs[i]) {
    u64 result;
    const char *err_str = xt_timestamp.parse(timestamp_strs[i], &result, pool);
    if(err_str != NULL) {
      die("Unexpected error in xt_timestamp.parse: %s", err_str);
    }
    if(timestamp_parsed[i] != result) {
      die("Expected: %" PRIu64 " but got %" PRIu64, timestamp_parsed[i], result);
    }

    bprintf(out, "%" PRIu64 " %" PRIu64 "\n", timestamp_parsed[i], result);

    i++;
  }

  mp_delete(pool);
}

static void test_timestamp_parse_errors(struct fastbuf *out)
{
  static const char *timestamp_strs[] = {
    "1403685533X",
    "2014X-06-25 08:38:53",
    "2X014-06-25 08:38:53",
    "2014-06-25 08:38:53X",
    "X2014-06-25 08:38:53",
    "X1403685533",
    "14X03685533",
    "1403685533X",
    NULL
  };

  struct mempool *pool = mp_new(4096);
  uint i = 0;

  while(timestamp_strs[i]) {
    u64 result;
    const char *err_str = xt_timestamp.parse(timestamp_strs[i], &result, pool);

    if(err_str == NULL) {
      bprintf(out, "xt_timestamp.parse incorrectly did not result in error while parsing: '%s'.\n", timestamp_strs[i]);
    } else {
      bprintf(out, "xt_timestamp.parse error: '%s'.\n", err_str);
    }

    i++;
  }

  mp_delete(pool);
}

int main(void)
{
  setenv("TZ", "Europe/Prague", 1);
  struct fastbuf *out;
  out = bfdopen_shared(1, 4096);

  test_size_parse_correct(out);
  test_size_parse_errors(out);
  test_bool_parse_correct(out);
  test_timestamp_parse_correct(out);
  test_timestamp_parse_errors(out);
  bclose(out);

  return 0;
}