wbArg.cpp

#include "wb.h"

#ifdef WB_USE_SESSION_ID
static char * _sessionId = nullptr;
char * _envSessionId() {
#ifdef WB_USE_UNIX
  if (_sessionId != nullptr) {
    char *env = std::getenv("SESSION_ID");
    if (env) {
      _sessionId = wbString_duplicate(env);
    }
  }
#endif /* WB_USE_UNIX */
  return wbString_duplicate(_sessionId);
}
char * sessionId() {
  if (_sessionId != "") {
    return wbString_duplicate(_sessionId);
  }
  return wbString_duplicate(_envSessionId());
}
#else /* WB_USE_SESSION_ID */
char * _envSessionId() {
  return wbString_duplicate("session_id_disabled");
}
char * sessionId() {
  return wbString_duplicate("session_id_disabled");
}
#endif /* WB_USE_SESSION_ID */

wbArg_t wbArg_new(int *argc, char ***argv) {
  wbArg_t arg;

  wb_init(argc, argv);

  wbArg_setSessionId(arg, _envSessionId());
  wbArg_setInputCount(arg, 0);
  wbArg_setInputFiles(arg, NULL);
  wbArg_setOutputFile(arg, NULL);
  wbArg_setType(arg, NULL);
  wbArg_setExpectedOutputFile(arg, NULL);
  return arg;
}

void wbArg_delete(wbArg_t arg) {
  if (wbArg_getInputCount(arg) > 0 && wbArg_getInputFiles(arg) != nullptr) {
    int ii;
    for (ii = 0; ii < wbArg_getInputCount(arg); ii++) {
      wbDelete(wbArg_getInputFile(arg, ii));
    }
    wbDelete(wbArg_getInputFiles(arg));
    wbArg_setInputCount(arg, 0);
    wbArg_setInputFiles(arg, NULL);
  }
  if (wbArg_getOutputFile(arg)) {
    wbDelete(wbArg_getOutputFile(arg));
    wbArg_setOutputFile(arg, NULL);
  }
  if (wbArg_getExpectedOutputFile(arg)) {
    wbDelete(wbArg_getExpectedOutputFile(arg));
    wbArg_setExpectedOutputFile(arg, NULL);
  }
  if (wbArg_getType(arg)) {
    wbDelete(wbArg_getType(arg));
    wbArg_setType(arg, NULL);
  }
  return;
}

static int getInputFileCount(char *arg) {
  int count = 1;
  while (*arg != '\0' && *arg != '-') {
    if (*arg == ',') {
      count++;
    }
    arg++;
  }
  return count;
}

static char **parseInputFiles(char *arg, int *resCount) {
  int count;
  int ii = 0;
  char **files;
  char *token;

  count = getInputFileCount(arg);

  files = wbNewArray(char *, count);

  token = strtok(arg, ",");
  while (token != nullptr) {
    files[ii++] = wbString_duplicate(token);
    token       = strtok(NULL, ",");
  }
  *resCount = ii;
  return files;
}

static char *parseString(char *arg) {
  return wbString_duplicate(arg);
}

static void parseSessionId(char *arg) {
#ifdef WB_USE_SESSION_ID
  _sessionId = std::string(arg);
#endif /* WB_USE_SESSION_ID */
}

wbArg_t wbArg_read(int argc, char **argv) {
  int ii;
  wbArg_t arg;

  arg = wbArg_new(&argc, &argv);
  for (ii = 0; ii < argc; ii++) {
    if (wbString_startsWith(argv[ii], "-s")) {
      parseSessionId(argv[++ii]);
      wbArg_setSessionId(arg, sessionId());
    } else if (wbString_startsWith(argv[ii], "-i")) {
      int fileCount;
      char **files;

      files = parseInputFiles(argv[++ii], &fileCount);

      wbArg_setInputCount(arg, fileCount);
      wbArg_setInputFiles(arg, files);
    } else if (wbString_startsWith(argv[ii], "-o")) {
      char *file = parseString(argv[++ii]);
      wbArg_setOutputFile(arg, file);
    } else if (wbString_startsWith(argv[ii], "-e")) {
      char *file = parseString(argv[++ii]);
      wbArg_setExpectedOutputFile(arg, file);
    } else if (wbString_startsWith(argv[ii], "-t")) {
      char *type = parseString(argv[++ii]);
      wbArg_setType(arg, type);
    } else if (argv[ii][0] == '-') {
      wbLog(ERROR, "Unexpected program option ", argv[ii]);
    }
  }

  return arg;
}