#include <gargoyle.h>
#include <gargoyle/scribe.h>
#include <gargoyle/sleuth.h>
#include <gargoyle/twine.h>
#include <stddef.h>
#include <stdio.h>
#include <string.h>

uint8_t gargoyle_digest_argv(uint16_t optc, struct gargoyle_opt *optv, int *argc, char ***argv, struct gargoyle_err *err, gargoyle_flag_type flags) {
  if(!(flags & GARGOYLE_FLG_GREED)) {
    *argc -= 1;
    *argv += 1;
  }

  if(flags & GARGOYLE_FLG_BCASE) {
    flags |= GARGOYLE_FLG_ICASE;
  }

  GARGOYLE_NEW_ERR(err);

  while(1) {
    const char *arg = **argv;

    if(!arg || *arg != '-' || !*(arg + 1)) {
      return GARGOYLE_ERR_SUCCESS;
    } else if(*arg == '-' && *(arg + 1) == '-' && !*(arg + 2)) {
      *argc -= 1;
      *argv += 1;
      return GARGOYLE_ERR_SUCCESS;
    }

    struct gargoyle_opt *opt = NULL;

    if(*(arg + 1) == '-') {
      const char *brand = arg + 2;
      opt = gargoyle_find_brand(optc, optv, brand, NULL, 0, flags);

      if(!opt) {
        if(flags & GARGOYLE_FLG_STRCT) {
          char *eql = index(brand, '=');
          GARGOYLE_MK_ERR(err, GARGOYLE_ERR_UNKNOWN_OPT, "unknown option '--%.*s'", (int) (eql - brand), brand);
          return err->code;
        } else {
          *argc -= 1;
          *argv += 1;
          continue;
        }
      }

      uint8_t res = 0;
      const char *val = NULL;

      if(opt->type & GARGOYLE_TYPE_BOOL) {
        res = gargoyle_from_bool(opt, brand, "no-", 3, flags);
      } else if(*(brand + opt->brand_sz) == '=') {
        val = brand + opt->brand_sz + 1;
        res = gargoyle_from_rope(opt, val, flags);
      } else {
        if(!*(*argv + 1)) {
          GARGOYLE_MK_ERR(err, GARGOYLE_ERR_VALUE_REQUIRED, "option '--%s' requires a value", opt->brand);
          return err->code;
        }

        *argc -= 1;
        *argv += 1;
        val = **argv;
        res = gargoyle_from_rope(opt, val, flags);
      }

      if(res) {
        if(val) {
          GARGOYLE_MK_ERR(err, res, "invalid value '%s' for option '--%s'", val, opt->brand);
        } else {
          GARGOYLE_MK_ERR(err, res, "invalid value for option '--%s'", opt->brand);
        }

        return err->code;
      }
    } else {
      const char *idx = arg + 1;

      for(; *idx; idx += 1) {
        opt = gargoyle_find_emblem(optc, optv, *idx, flags);

        if(!opt) {
          if(flags & GARGOYLE_FLG_STRCT) {
            GARGOYLE_MK_ERR(err, GARGOYLE_ERR_UNKNOWN_OPT, "unknown option '-%c'", *idx);
            return err->code;
          } else {
            idx += 1;
            continue;
          }
        }

        uint8_t res = 0;
        const char *val = NULL;

        if(opt->type & GARGOYLE_TYPE_BOOL) {
          res = gargoyle_from_bool(opt, "", "*", 1, flags);
        } else if(*(idx + 1)) {
          val = idx + 1;
          res = gargoyle_from_rope(opt, val, flags);
          break;
        } else {
          if(!*(*argv + 1)) {
            GARGOYLE_MK_ERR(err, GARGOYLE_ERR_VALUE_REQUIRED, "option '-%c' requires a value", opt->emblem);
            return err->code;
          }

          *argc -= 1;
          *argv += 1;
          val = **argv;
          res = gargoyle_from_rope(opt, val, flags);
        }

        if(res) {
          if(val) {
            GARGOYLE_MK_ERR(err, res, "invalid value '%s' for option '-%c'", val, opt->emblem);
          } else {
            GARGOYLE_MK_ERR(err, res, "invalid value for option '-%c'", opt->emblem);
          }

          return err->code;
        }
      }
    }

    *argc -= 1;
    *argv += 1;
  }

  return GARGOYLE_ERR_SUCCESS;
}

uint8_t gargoyle_digest_envh(uint16_t optc, struct gargoyle_opt *optv, const char *prefix, uint16_t prefix_sz, char ***envh, struct gargoyle_err *err, gargoyle_flag_type flags) {
  if(flags & GARGOYLE_FLG_BCASE) {
    flags |= GARGOYLE_FLG_ICASE;
  }

  for(; **envh; *envh += 1) {
    const char *var = **envh;

    if(!gargoyle_cmp(prefix, var, prefix_sz, flags) || *(var + prefix_sz) != '_') {
      continue;
    }

    const char *brand = var + prefix_sz + 1;
    struct gargoyle_opt *opt = gargoyle_find_brand(optc, optv, brand, "NO_", 3, flags);

    if(!opt) {
      if(flags & GARGOYLE_FLG_STRCT) {
        err->code = GARGOYLE_ERR_UNKNOWN_OPT;
        char *eql = index(brand, '=');
        snprintf(err->msg, GARGOYLE_MAX_ERR_SZ, "unknown option %.*s", (int) (eql - brand), brand);
        return GARGOYLE_ERR_UNKNOWN_OPT;
      } else {
        continue;
      }
    }

    uint8_t res = 0;

    if(opt->type & GARGOYLE_TYPE_BOOL) {
      res = gargoyle_from_bool(opt, brand, "NO_", 3, flags);
    } else {
      res = gargoyle_from_rope(opt, brand + opt->brand_sz + 1, flags);
    }

    if(res) {
      return res;
    }
  }

  return GARGOYLE_ERR_SUCCESS;
}