ref.hpp

Go to the documentation of this file.

// Copyright 2020 Arthur Sonzogni. All rights reserved.
// Use of this source code is governed by the MIT license that can be found in
// the LICENSE file.
#ifndef FTXUI_UTIL_REF_HPP
#define FTXUI_UTIL_REF_HPP
 
#include <ftxui/screen/string.hpp>
#include <memory>
#include <string>
#include <string_view>
#include <variant>
#include <vector>
 
namespace ftxui {
 
/// @brief An adapter. Own or reference an immutable object.
template <typename T>
class ConstRef {
 public:
  ConstRef() = default;
  ConstRef(T t) : variant_(std::move(t)) {}  // NOLINT
  ConstRef(const T* t) : variant_(t) {}      // NOLINT
  ConstRef& operator=(ConstRef&&) noexcept = default;
  ConstRef(const ConstRef<T>&) = default;
  ConstRef(ConstRef<T>&&) noexcept = default;
  ~ConstRef() = default;
 
  // Make a "reseatable" reference
  ConstRef<T>& operator=(const ConstRef<T>&) = default;
 
  // Accessors:
  const T& operator()() const { return *Address(); }
  const T& operator*() const { return *Address(); }
  const T* operator->() const { return Address(); }
 
 private:
  std::variant<T, const T*> variant_ = T{};
 
  const T* Address() const {
    return std::holds_alternative<T>(variant_) ? &std::get<T>(variant_)
                                               : std::get<const T*>(variant_);
  }
};
 
/// @brief An adapter. Own or reference an mutable object.
template <typename T>
class Ref {
 public:
  Ref() = default;
  Ref(T t) : variant_(std::move(t)) {}  // NOLINT
  Ref(T* t) : variant_(t) {}            // NOLINT
  ~Ref() = default;
  Ref& operator=(Ref&&) noexcept = default;
  Ref(const Ref<T>&) = default;
  Ref(Ref<T>&&) noexcept = default;
 
  // Make a "reseatable" reference.
  Ref<T>& operator=(const Ref<T>&) = default;
 
  // Accessors:
  T& operator()() { return *Address(); }
  T& operator*() { return *Address(); }
  T* operator->() { return Address(); }
  const T& operator()() const { return *Address(); }
  const T& operator*() const { return *Address(); }
  const T* operator->() const { return Address(); }
 
 private:
  std::variant<T, T*> variant_ = T{};
 
  const T* Address() const {
    return std::holds_alternative<T>(variant_) ? &std::get<T>(variant_)
                                               : std::get<T*>(variant_);
  }
  T* Address() {
    return std::holds_alternative<T>(variant_) ? &std::get<T>(variant_)
                                               : std::get<T*>(variant_);
  }
};
 
/// @brief An adapter. Own or reference a constant string. For convenience, this
/// class convert multiple mutable string toward a shared representation.
class StringRef : public Ref<std::string> {
 public:
  using Ref<std::string>::Ref;
 
  StringRef(const wchar_t* ref)  // NOLINT
      : StringRef(to_string(std::wstring(ref))) {}
  StringRef(const char* ref)  // NOLINT
      : StringRef(std::string(ref)) {}
};
 
/// @brief An adapter. Own or reference a constant string. For convenience, this
/// class convert multiple immutable string toward a shared representation.
class ConstStringRef : public ConstRef<std::string> {
 public:
  using ConstRef<std::string>::ConstRef;
 
  ConstStringRef(const std::wstring* ref)  // NOLINT
      : ConstStringRef(to_string(*ref)) {}
  ConstStringRef(const std::wstring ref)  // NOLINT
      : ConstStringRef(to_string(ref)) {}
  ConstStringRef(const wchar_t* ref)  // NOLINT
      : ConstStringRef(to_string(std::wstring(ref))) {}
  ConstStringRef(const char* ref)  // NOLINT
      : ConstStringRef(std::string(ref)) {}
};
 
/// @brief An adapter. Reference a list of strings.
///
/// Supported input:
/// - `std::vector<std::string>`
/// - `std::vector<std::string>*`
/// - `std::vector<std::wstring>*`
/// - `Adapter*`
/// - `std::unique_ptr<Adapter>`
class ConstStringListRef {
 public:
  // Bring your own adapter:
  class Adapter {
   public:
    Adapter() = default;
    Adapter(const Adapter&) = default;
    Adapter& operator=(const Adapter&) = default;
    Adapter(Adapter&&) = default;
    Adapter& operator=(Adapter&&) = default;
    virtual ~Adapter() = default;
    virtual size_t size() const = 0;
    virtual std::string_view operator[](size_t i) const = 0;
  };
  using Variant = std::variant<const std::vector<std::string>,        //
                               const std::vector<std::string>*,       //
                               const std::vector<std::string_view>,   //
                               const std::vector<std::string_view>*,  //
                               const std::vector<std::wstring>*,      //
                               Adapter*,                              //
                               std::unique_ptr<Adapter>               //
                               >;
 
  ConstStringListRef() = default;
  ~ConstStringListRef() = default;
  ConstStringListRef& operator=(const ConstStringListRef&) = default;
  ConstStringListRef& operator=(ConstStringListRef&&) = default;
  ConstStringListRef(ConstStringListRef&&) = default;
  ConstStringListRef(const ConstStringListRef&) = default;
 
  ConstStringListRef(std::vector<std::string> value)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(value);
  }
  ConstStringListRef(const std::vector<std::string>* value)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(value);
  }
  ConstStringListRef(std::vector<std::string_view> value)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(value);
  }
  ConstStringListRef(const std::vector<std::string_view>* value)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(value);
  }
  ConstStringListRef(const std::vector<std::wstring>* value)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(value);
  }
  ConstStringListRef(Adapter* adapter)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(adapter);
  }
  template <typename AdapterType>
  ConstStringListRef(std::unique_ptr<AdapterType> adapter)  // NOLINT
  {
    variant_ = std::make_shared<Variant>(
        static_cast<std::unique_ptr<Adapter>>(std::move(adapter)));
  }
 
  size_t size() const {
    return variant_ ? std::visit(SizeVisitor(), *variant_) : 0;
  }
 
  std::string operator[](size_t i) const {
    if (!variant_) {
      return "";
    }
    auto& v = *variant_;
    if (std::holds_alternative<const std::vector<std::string>>(v)) {
      return std::get<const std::vector<std::string>>(v)[i];
    }
    if (std::holds_alternative<const std::vector<std::string>*>(v)) {
      return (*std::get<const std::vector<std::string>*>(v))[i];
    }
    if (std::holds_alternative<const std::vector<std::string_view>>(v)) {
      return std::string(std::get<const std::vector<std::string_view>>(v)[i]);
    }
    if (std::holds_alternative<const std::vector<std::string_view>*>(v)) {
      return std::string(
          (*std::get<const std::vector<std::string_view>*>(v))[i]);
    }
    if (std::holds_alternative<const std::vector<std::wstring>*>(v)) {
      return to_string((*std::get<const std::vector<std::wstring>*>(v))[i]);
    }
    if (std::holds_alternative<Adapter*>(v)) {
      return std::string((*std::get<Adapter*>(v))[i]);
    }
    if (std::holds_alternative<std::unique_ptr<Adapter>>(v)) {
      return std::string((*std::get<std::unique_ptr<Adapter>>(v))[i]);
    }
    return "";
  }
 
  std::string_view at(size_t i) const {
    if (!variant_) {
      return "";
    }
    auto& v = *variant_;
    if (std::holds_alternative<const std::vector<std::string>>(v)) {
      return std::get<const std::vector<std::string>>(v)[i];
    }
    if (std::holds_alternative<const std::vector<std::string>*>(v)) {
      return (*std::get<const std::vector<std::string>*>(v))[i];
    }
    if (std::holds_alternative<const std::vector<std::string_view>>(v)) {
      return std::get<const std::vector<std::string_view>>(v)[i];
    }
    if (std::holds_alternative<const std::vector<std::string_view>*>(v)) {
      return (*std::get<const std::vector<std::string_view>*>(v))[i];
    }
    if (std::holds_alternative<const std::vector<std::wstring>*>(v)) {
      return {};
    }
    if (std::holds_alternative<Adapter*>(v)) {
      return (*std::get<Adapter*>(v))[i];
    }
    if (std::holds_alternative<std::unique_ptr<Adapter>>(v)) {
      return (*std::get<std::unique_ptr<Adapter>>(v))[i];
    }
    return {};
  }
 
 private:
  struct SizeVisitor {
    size_t operator()(const std::vector<std::string>& v) const {
      return v.size();
    }
    size_t operator()(const std::vector<std::string>* v) const {
      return v->size();
    }
    size_t operator()(const std::vector<std::string_view>& v) const {
      return v.size();
    }
    size_t operator()(const std::vector<std::string_view>* v) const {
      return v->size();
    }
    size_t operator()(const std::vector<std::wstring>* v) const {
      return v->size();
    }
    size_t operator()(const Adapter* v) const { return v->size(); }
    size_t operator()(const std::unique_ptr<Adapter>& v) const {
      return v->size();
    }
  };
 
  std::shared_ptr<Variant> variant_;
};
 
}  // namespace ftxui
 
#endif /* end of include guard: FTXUI_UTIL_REF_HPP */