VILLASnode/include/villas/nodes/redis_helpers.hpp

/* Redis node-type helpers.
 *
 * Author: Steffen Vogel <post@steffenvogel.de>
 * SPDX-FileCopyrightText: 2014-2023 Institute for Automation of Complex Power Systems, RWTH Aachen University
 * SPDX-License-Identifier: Apache-2.0
 */

#pragma once

#include <chrono>
#include <fmt/ostream.h>
#include <functional>
#include <sw/redis++/connection.h>
#include <sw/redis++/redis++.h>
#include <villas/node/config.hpp>

namespace std {

template <typename _rep, typename ratio>
struct hash<std::chrono::duration<_rep, ratio>> {
  std::size_t operator()(std::chrono::duration<_rep, ratio> const &s) const {
    return std::hash<_rep>{}(s.count());
  }
};

template <> struct hash<sw::redis::tls::TlsOptions> {
  std::size_t operator()(const sw::redis::tls::TlsOptions &t) const {
#ifdef REDISPP_WITH_TLS
    return hash<bool>()(t.enabled) ^ hash<std::string>()(t.cacert) ^
           hash<std::string>()(t.cacertdir) ^ hash<std::string>()(t.cert) ^
           hash<std::string>()(t.key) ^ hash<std::string>()(t.sni);
#else
    return 0;
#endif // REDISPP_WITH_TLS
  }
};

template <> struct hash<sw::redis::ConnectionOptions> {
  std::size_t operator()(const sw::redis::ConnectionOptions &o) const {
    return hash<int>()(static_cast<int>(o.type)) ^ hash<std::string>()(o.host) ^
           hash<int>()(o.port) ^ hash<std::string>()(o.path) ^
           hash<std::string>()(o.user) ^ hash<std::string>()(o.password) ^
           hash<int>()(o.db) ^ hash<bool>()(o.keep_alive) ^
           hash<std::chrono::milliseconds>()(o.connect_timeout) ^
           hash<std::chrono::milliseconds>()(o.socket_timeout) ^
           hash<sw::redis::tls::TlsOptions>()(o.tls);
  }
};

} // namespace std

namespace sw {
namespace redis {

inline bool operator==(const tls::TlsOptions &o1, const tls::TlsOptions &o2) {
#ifdef REDISPP_WITH_TLS
  return o1.enabled == o2.enabled && o1.cacert == o2.cacert &&
         o1.cacertdir == o2.cacertdir && o1.cert == o2.cert &&
         o1.key == o2.key && o1.sni == o2.sni;
#else
  return true;
#endif // REDISPP_WITH_TLS
}

inline bool operator==(const ConnectionOptions &o1,
                       const ConnectionOptions &o2) {
  return o1.type == o2.type && o1.host == o2.host && o1.port == o2.port &&
         o1.path == o2.path && o1.user == o2.user &&
         o1.password == o2.password && o1.db == o2.db &&
         o1.keep_alive == o2.keep_alive &&
         o1.connect_timeout == o2.connect_timeout &&
         o1.socket_timeout == o2.socket_timeout && o1.tls == o2.tls;
}

#ifdef REDISPP_WITH_TLS
namespace tls {
std::ostream &operator<<(std::ostream &os, const TlsOptions &t) {
  os << "tls.enabled=" << (t.enabled ? "yes" : "no");

  if (t.enabled) {
    if (!t.cacert.empty())
      os << ", tls.cacert=" << t.cacert;

    if (!t.cacertdir.empty())
      os << ", tls.cacertdir=" << t.cacertdir;

    if (!t.cert.empty())
      os << ", tls.cert=" << t.cert;

    if (!t.key.empty())
      os << ", tls.key=" << t.key;

    if (!t.sni.empty())
      os << ", tls.sni=" << t.sni;
  }

  return os;
}
} // namespace tls
#endif // REDISPP_WITH_TLS

inline std::ostream &operator<<(std::ostream &os, const ConnectionType &t) {
  switch (t) {
  case ConnectionType::TCP:
    os << "tcp";
    break;

  case ConnectionType::UNIX:
    os << "unix";
    break;
  }

  return os;
}

inline std::ostream &operator<<(std::ostream &os, const ConnectionOptions &o) {
  os << "type=" << o.type;

  switch (o.type) {
  case ConnectionType::TCP:
    os << ", host=" << o.host << ", port=" << o.port;
    break;

  case ConnectionType::UNIX:
    os << ", path=" << o.path;
    break;
  }

  os << ", db=" << o.db << ", user=" << o.user
     << ", keepalive=" << (o.keep_alive ? "yes" : "no");

#ifdef REDISPP_WITH_TLS
  os << ", " << o.tls;
#endif

  return os;
}

} // namespace redis
} // namespace sw

namespace villas {
namespace node {
#ifdef REDISPP_WITH_URI
inline sw::redis::ConnectionOptions
make_redis_connection_options(char const *uri) {
  auto u = sw::redis::Uri{uri};
  return u.connection_options();
}
#else
inline sw::redis::ConnectionOptions
make_redis_connection_options(char const *uri) {
  return sw::redis::ConnectionOptions{uri};
}
#endif
} // namespace node
} // namespace villas

#ifndef FMT_LEGACY_OSTREAM_FORMATTER
template <>
class fmt::formatter<sw::redis::ConnectionType>
    : public fmt::ostream_formatter {};
template <>
class fmt::formatter<sw::redis::ConnectionOptions>
    : public fmt::ostream_formatter {};
#ifdef REDISPP_WITH_TLS
template <>
class fmt::formatter<sw::redis::tls::TlsOptions>
    : public fmt::ostream_formatter {};
#endif
#endif