erased_string.h

Go to the documentation of this file.

#pragma once
 
#include "di/container/string/constant_string_interface.h"
#include "di/container/string/utf8_encoding.h"
#include "di/meta/util.h"
#include "di/vocab/span/prelude.h"
 
namespace di::container {
namespace detail {
    struct IntoErasedStringFunction {
        template<typename T>
        requires(concepts::TagInvocable<IntoErasedStringFunction, T>)
        constexpr auto operator()(T&& value) const {
            return function::tag_invoke(*this, util::forward<T>(value));
        }
    };
}
 
constexpr inline auto into_erased_string = detail::IntoErasedStringFunction {};
 
class ErasedString : public string::ConstantStringInterface<ErasedString, string::Utf8Encoding> {
public:
    using Encoding = string::Utf8Encoding;
 
    constexpr auto encoding() const { return Encoding {}; }
    constexpr auto span() const { return m_data; }
 
    enum class ThunkOp { Move, Destroy };
 
    using ThunkFunction = void (*)(ErasedString* dest, ErasedString* src, ThunkOp op);
 
    Span<c8 const> m_data;
    void* m_state[3] {};
    ThunkFunction const m_thunk { nullptr };
 
    constexpr explicit ErasedString(Span<c8 const> data = {}, void* state0 = nullptr, void* state1 = nullptr,
                                    void* state2 = nullptr, ThunkFunction thunk = nullptr)
        : m_data({ data.data(), data.size() - 1 }), m_state(state0, state1, state2), m_thunk(thunk) {}
 
    constexpr ErasedString(ErasedString&& other)
        : m_data(other.m_data), m_state(other.m_state[0], other.m_state[1], other.m_state[2]), m_thunk(other.m_thunk) {
        if (this->m_thunk) {
            m_thunk(this, &other, ThunkOp::Move);
        }
    }
 
    template<typename T>
    requires(!concepts::RemoveCVRefSameAs<T, ErasedString> && requires { into_erased_string(util::declval<T>()); })
    constexpr ErasedString(T&& value) : ErasedString(into_erased_string(util::forward<T>(value))) {}
 
    constexpr auto operator=(ErasedString&& other) -> ErasedString& {
        if (this != &other) {
            util::destroy_at(this);
            util::construct_at(this, util::move(other));
        }
        return *this;
    }
 
    constexpr ~ErasedString() {
        if (m_thunk) {
            m_thunk(this, this, ThunkOp::Destroy);
        }
        m_data = {};
    }
};
}
 
namespace di {
using container::ErasedString;
}