VILLASnode/fpga/tests/unit/logging.cpp

/* Logging utilities for unit test.
 *
 * Author: Steffen Vogel <post@steffenvogel.de>
 * SPDX-FileCopyrightText: 2017 Steffen Vogel <post@steffenvogel.de>
 * SPDX-License-Identifier: Apache-2.0
 */

#include <cstdarg>
#include <memory>

#include <criterion/logging.h>
#include <criterion/options.h>

#include <spdlog/spdlog.h>
#include <villas/log.hpp>

extern "C" {
// We override criterions function here
void criterion_log_noformat(enum criterion_severity severity, const char *msg);
void criterion_plog(enum criterion_logging_level level,
                    const struct criterion_prefix_data *prefix, const char *msg,
                    ...);
void criterion_vlog(enum criterion_logging_level level, const char *msg,
                    va_list args);
}

struct criterion_prefix_data {
  const char *prefix;
  const char *color;
};

static int format_msg(char *buf, size_t buflen, const char *msg, va_list args) {
  int len = vsnprintf(buf, buflen, msg, args);

  // Strip new line
  char *nl = strchr(buf, '\n');
  if (nl)
    *nl = 0;

  return len;
}

void criterion_log_noformat(enum criterion_severity severity, const char *msg) {
  auto logger = villas::Log::get("criterion");

  switch (severity) {
  case CR_LOG_INFO:
    logger->info(msg);
    break;

  case CR_LOG_WARNING:
    logger->warn(msg);
    break;

  case CR_LOG_ERROR:
    logger->error(msg);
    break;
  }
}

void criterion_vlog(enum criterion_logging_level level, const char *msg,
                    va_list args) {
  char formatted_msg[1024];

  if (level < criterion_options.logging_threshold)
    return;

  format_msg(formatted_msg, sizeof(formatted_msg), msg, args);

  auto logger = villas::Log::get("criterion");
  logger->info(formatted_msg);
}

void criterion_plog(enum criterion_logging_level level,
                    const struct criterion_prefix_data *prefix, const char *msg,
                    ...) {
  char formatted_msg[1024];

  va_list args;

  if (level < criterion_options.logging_threshold)
    return;

  va_start(args, msg);
  format_msg(formatted_msg, sizeof(formatted_msg), msg, args);
  va_end(args);

  auto logger = villas::Log::get("criterion");

  if (strstr(formatted_msg, "Warning"))
    logger->warn(formatted_msg);
  else if (strstr(formatted_msg, "Failed"))
    logger->error(formatted_msg);
  else if (!strcmp(prefix->prefix, "----") &&
           !strcmp(prefix->color, "\33[0;34m"))
    logger->info(formatted_msg);
  else if (!strcmp(prefix->prefix, "----") &&
           !strcmp(prefix->color, "\33[1;30m"))
    logger->debug(formatted_msg);
  else if (!strcmp(prefix->prefix, "===="))
    logger->info(formatted_msg);
  else if (!strcmp(prefix->prefix, "RUN "))
    logger->info("Run:  {}", formatted_msg);
  else if (!strcmp(prefix->prefix, "SKIP"))
    logger->info("Skip: {}", formatted_msg);
  else if (!strcmp(prefix->prefix, "PASS"))
    logger->info("Pass: {}", formatted_msg);
  else if (!strcmp(prefix->prefix, "FAIL"))
    logger->error("Fail: {}", formatted_msg);
  else if (!strcmp(prefix->prefix, "WARN"))
    logger->warn(formatted_msg);
  else if (!strcmp(prefix->prefix, "ERR "))
    logger->error(formatted_msg);
}