nonstd.h

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the Apache 2.0 License.
#pragma once
 
#include <array>
#include <cctype>
#include <regex>
#include <string>
#include <string_view>
#include <type_traits>
#include <unistd.h>
#include <vector>
 
#define FMT_HEADER_ONLY
#include <fmt/format.h>
 
namespace ccf::nonstd
{
  template <typename T, template <typename...> class U>
  struct is_specialization : std::false_type
  {};
 
  template <template <typename...> class T, typename... Args>
  struct is_specialization<T<Args...>, T> : std::true_type
  {};
 
  template <typename T>
  struct is_std_array : std::false_type
  {};
 
  template <typename T, size_t N>
  struct is_std_array<std::array<T, N>> : public std::true_type
  {};
 
  template <typename T>
  struct is_std_vector : std::false_type
  {};
 
  template <typename T>
  struct is_std_vector<std::vector<T>> : public std::true_type
  {};
 
  template <typename T>
  struct dependent_false : public std::false_type
  {};
 
  template <typename T>
  static constexpr bool dependent_false_v = dependent_false<T>::value;
 
  template <typename T, T>
  struct value_dependent_false : public std::false_type
  {};
 
  template <typename T, T t>
  static constexpr bool value_dependent_false_v = dependent_false<T>::value;
 
  static inline std::vector<std::string_view> split(
    const std::string_view& s,
    const std::string_view& separator = " ",
    size_t max_split = SIZE_MAX)
  {
    std::vector<std::string_view> result;
 
    auto separator_end = 0;
    auto next_separator_start = s.find(separator);
    while (next_separator_start != std::string_view::npos &&
           result.size() < max_split)
    {
      result.push_back(
        s.substr(separator_end, next_separator_start - separator_end));
 
      separator_end = next_separator_start + separator.size();
      next_separator_start = s.find(separator, separator_end);
    }
 
    result.push_back(s.substr(separator_end));
 
    return result;
  }
 
  /* split_1 wraps split and allows writing things like:
   * auto [host, port] = ccf::nonstd::split_1("1.2.3.4:8000", ":")
   */
  static inline std::tuple<std::string_view, std::string_view> split_1(
    const std::string_view& s, const std::string_view& separator)
  {
    const auto v = split(s, separator, 1);
    if (v.size() == 1)
    {
      // If separator is not present, return {s, ""};
      return std::make_tuple(v[0], "");
    }
 
    return std::make_tuple(v[0], v[1]);
  }
 
  static inline std::vector<std::string_view> rsplit(
    const std::string_view& s,
    const std::string_view& separator = " ",
    size_t max_split = SIZE_MAX)
  {
    std::vector<std::string_view> result;
 
    auto prev_separator_start = s.size();
    auto next_separator_start = s.rfind(separator);
    while (next_separator_start != std::string_view::npos &&
           result.size() < max_split)
    {
      auto separator_end = next_separator_start + separator.size();
 
      result.push_back(
        s.substr(separator_end, prev_separator_start - separator_end));
 
      prev_separator_start = next_separator_start;
 
      if (next_separator_start == 0)
      {
        break;
      }
      else
      {
        next_separator_start = s.rfind(separator, prev_separator_start - 1);
      }
    }
 
    result.push_back(s.substr(0, prev_separator_start));
 
    return result;
  }
 
  /* rsplit_1 wraps rsplit _and reverses the result order_ and allows writing
   * things like:
   * auto [host, port] = ccf::nonstd::rsplit_1("[1:2:3:4]:8000", ":")
   */
  static inline std::tuple<std::string_view, std::string_view> rsplit_1(
    const std::string_view& s, const std::string_view& separator)
  {
    const auto v = rsplit(s, separator, 1);
    if (v.size() == 1)
    {
      // If separator is not present, return {"", s};
      return std::make_tuple("", v[0]);
    }
 
    return std::make_tuple(v[1], v[0]);
  }
 
  static inline void to_upper(std::string& s)
  {
    std::transform(s.begin(), s.end(), s.begin(), [](unsigned char c) {
      return std::toupper(c);
    });
  }
  static inline void to_lower(std::string& s)
  {
    std::transform(s.begin(), s.end(), s.begin(), [](unsigned char c) {
      return std::tolower(c);
    });
  }
 
  static inline std::string_view trim(
    std::string_view s, std::string_view trim_chars = " \t\r\n")
  {
    const auto start = std::min(s.find_first_not_of(trim_chars), s.size());
    const auto end = std::min(s.find_last_not_of(trim_chars) + 1, s.size());
    return s.substr(start, end - start);
  }
 
  template <size_t I = 0, typename F, typename... Ts>
  static void tuple_for_each(const std::tuple<Ts...>& t, const F& f)
  {
    if constexpr (I < sizeof...(Ts))
    {
      f(std::get<I>(t));
      tuple_for_each<I + 1>(t, f);
    }
  }
 
  static void close_fd(int* fd)
  {
    if (fd != nullptr && *fd >= 0)
    {
      close(*fd);
      *fd = -1;
    }
  }
  using CloseFdGuard = std::unique_ptr<int, decltype(&close_fd)>;
  static inline CloseFdGuard make_close_fd_guard(int* fd)
  {
    return CloseFdGuard(fd, close_fd);
  }
}