#include <libmp/all.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <strings.h>

#include "args.h"
#include "util.h"

void args_add(args_t *args, char *name) {
  if (NULL == args->list)
    args->list = malloc(sizeof(arg_t) * (args->count + 1));
  else
    args->list = realloc(args->list, sizeof(arg_t) * (args->count + 1));

  args->list[args->count].name    = name;
  args->list[args->count++].value = NULL;
}

void args_add_value(args_t *args, char *value) {
  if (NULL != args->list && args->list[args->count - 1].value == NULL) {
    args->list[args->count - 1].value = value;
    return;
  }

  if (NULL == args->list)
    args->list = malloc(sizeof(arg_t) * (args->count + 1));
  else
    args->list = realloc(args->list, sizeof(arg_t) * (args->count + 1));

  args->list[args->count].name    = NULL;
  args->list[args->count++].value = value;
}

args_t *args_parse(int argc, char *argv[]) {
  args_t *args = malloc(sizeof(args_t));
  bzero(args, sizeof(args_t));

  if (argc <= 1)
    return args;

  for (int i = 1; i < argc; i++) {
    if (startswith(argv[i], "--"))
      args_add(args, argv[i] + 2);
    else
      args_add_value(args, argv[i]);
  }

  return args;
}

void args_free(args_t *args) {
  free(args->list);
  free(args);
}

char *args_get_string(args_t *args, char *name) {
  for (int i = 0; i < args->count; i++) {
    if (NULL == args->list[i].name)
      continue;

    if (eq(args->list[i].name, name))
      return args->list[i].value;
  }

  return NULL;
}

bool args_get_bool(args_t *args, char *name) {
  for (int i = 0; i < args->count; i++) {
    if (NULL == args->list[i].name)
      continue;

    if (eq(args->list[i].name, name))
      return true;
  }

  return false;
}

int args_get_int(args_t *args, char *name) {
  for (int i = 0; i < args->count; i++) {
    if (NULL == args->list[i].name)
      continue;

    if (eq(args->list[i].name, name))
      return atoi(args->list[i].value);
  }

  return -1;
}