CommandParseHelpers.h

#pragma once
 
#include "Utilities/Parsing.h"
 
#include <ctime>
#include <iomanip>
#include <sstream>
#include <string>
#include <vector>
 
namespace Cogs::ComponentModel
{
  class Entity;
}
 
namespace Cogs::Core
{
  inline StringView getOption(const std::vector<ParsedValue> & options, const StringView & key, const StringView & defaultValue = "")
  {
    for (auto & o : options) {
      if (key == o.key) {
        return o.value;
      }
    }
 
    return defaultValue;
  }
 
  inline size_t getOption(const std::vector<ParsedValue> & options, const StringView & key, const size_t defaultValue)
  {
    for (auto & o : options) {
      if (key == o.key) {
        int value;
        if (!o.asInt(value)) return defaultValue;
        return size_t(value);
      }
    }
 
    return defaultValue;
  }
 
  inline float getOption(const std::vector<ParsedValue> & options, const StringView & key, const float defaultValue)
  {
    for (auto & o : options) {
      if (key == o.key) {
        float value;
        if (!o.asFloat(value)) value = defaultValue;
        return value;
      }
    }
 
    return defaultValue;
  }
 
  inline bool replaceToken(std::string & str, const std::vector<ParsedValue> & options, const std::string token, const StringView & option, const StringView & direct)
  {
    auto fPos = str.find(token);
 
    if (fPos != std::string::npos) {
      auto replacement = option.empty() ? direct : getOption(options, option);
 
      if (replacement.empty()) {
        return false;
      }
 
      str.replace(fPos, token.length(), replacement.to_string());
    }
 
    return true;
  }
 
  inline std::string makeDateString(std::time_t & t, const StringView & fmt)
  {
#ifdef WIN32
    struct tm tm;
    localtime_s(&tm, &t);
#else
    auto & tm = *localtime(&t);
#endif
 
    std::ostringstream oss;
 
    oss << std::put_time(&tm, fmt.data());
    auto year = oss.str();
 
    return oss.str();
  }
 
  inline std::string replaceFileMacros(std::string outputName, const std::vector<ParsedValue> & options)
  {
    auto t = std::time(nullptr);
 
    auto year = makeDateString(t, "%Y");
    auto month = makeDateString(t, "%m");
    auto day = makeDateString(t, "%d");
    auto date = makeDateString(t, "%Y-%m-%d");
 
    struct TokenReplacementPair
    {
      const char * token;
      const char * optionName;
      const char * direct;
    } tokens[] = {
      { "$FileName", "stem", nullptr},        // Value of the "stem" option.
      { "$Extension", "extension", nullptr },  // Value of the "extension" option.
      { "$Year", "", year.c_str() },  // Value of Current date year.
      { "$Month", "", month.c_str() },
      { "$Day", "", day.c_str() },
      { "$Date", "", date.c_str() },
    };
 
    for (auto & p : tokens) {
      replaceToken(outputName, options, p.token, p.optionName, p.direct);
    }
 
    return outputName;
  }
 
  inline std::string fixInvalidPathChars(const std::string & filename)
  {
    std::string output;
    for (auto c : filename) {
      switch (c) {
        case '<':
        case '>':
        case ':':
        case '"':
        case '/':
        case '\\':
        case '|':
        case '?':
        case '*':
          break;
        default:
          output.push_back(c);
      }
    }
 
    return output;
  }
}