clientutil.hpp

Go to the documentation of this file.

/*
 * This file is part of libkazv.
 * SPDX-FileCopyrightText: 2021-2023 tusooa <tusooa@kazv.moe>
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */
 
 
#pragma once
#include <libkazv-config.hpp>
 
#include <string>
#include <tuple>
#include <immer/map.hpp>
#include <zug/transducer/filter.hpp>
#include <zug/transducer/eager.hpp>
#include <lager/deps.hpp>
#include <boost/container_hash/hash.hpp>
#include <boost/serialization/string.hpp>
 
#include <cursorutil.hpp>
#include <jobinterface.hpp>
#include <eventinterface.hpp>
#include "thread-safety-helper.hpp"
 
namespace Kazv
{
    struct ClientModel;
 
    template<class K, class V, class List, class Func>
    immer::map<K, V> merge(immer::map<K, V> map, List list, Func keyOf)
    {
        for (auto v : list) {
            auto key = keyOf(v);
            map = std::move(map).set(key, v);
        }
        return map;
    }
 
    inline std::string keyOfPresence(Event e) {
        return e.sender();
    }
 
    inline std::string keyOfAccountData(Event e) {
        return e.type();
    }
 
    inline std::string keyOfTimeline(Event e) {
        return e.id();
    }
 
    inline std::string keyOfEphemeral(Event e) {
        return e.type();
    }
 
 
    struct KeyOfState {
        std::string type;
        std::string stateKey;
        friend bool operator==(const KeyOfState &a, const KeyOfState &b) = default;
    };
    template<class Archive>
    void serialize(Archive &ar, KeyOfState &m, std::uint32_t const /* version */)
    {
        ar & m.type & m.stateKey;
    }
 
    inline KeyOfState keyOfState(Event e) {
        return {e.type(), e.stateKey()};
    }
 
    template<class Context>
    JobInterface &getJobHandler(Context &&ctx)
    {
        return lager::get<JobInterface &>(std::forward<Context>(ctx));
    }
 
    template<class Context>
    EventInterface &getEventEmitter(Context &&ctx)
    {
        return lager::get<EventInterface &>(std::forward<Context>(ctx));
    }
 
    namespace
    {
        template<class ImmerT>
        struct ImmerIterator
        {
            using value_type = typename ImmerT::value_type;
            using reference = typename ImmerT::reference;
            using pointer = const value_type *;
            using difference_type = long int;
            using iterator_category = std::random_access_iterator_tag;
 
            ImmerIterator(const ImmerT &container, std::size_t index)
                : m_container(std::ref(container))
                , m_index(index)
                {}
 
            ImmerIterator &operator+=(difference_type d) {
                m_index += d;
                return *this;
            }
 
            ImmerIterator &operator-=(difference_type d) {
                m_index -= d;
                return *this;
            }
 
            difference_type operator-(ImmerIterator b) const {
                return index() - b.index();
            }
 
            ImmerIterator &operator++() {
                return *this += 1;
            }
 
            ImmerIterator operator++(int) {
                auto tmp = *this;
                *this += 1;
                return tmp;
            }
 
            ImmerIterator &operator--() {
                return *this -= 1;
            }
 
            ImmerIterator operator--(int) {
                auto tmp = *this;
                *this -= 1;
                return tmp;
            }
 
 
            reference &operator*() const {
                return m_container.get().at(m_index);
            }
 
            reference operator[](difference_type d) const;
 
            std::size_t index() const { return m_index; }
 
        private:
 
            std::reference_wrapper<const ImmerT> m_container;
            std::size_t m_index;
        };
 
        template<class ImmerT>
        auto ImmerIterator<ImmerT>::operator[](difference_type d) const -> reference
        {
            return *(*this + d);
        }
 
        template<class ImmerT>
        auto operator+(ImmerIterator<ImmerT> a, long int d)
        {
            return a += d;
        };
 
        template<class ImmerT>
        auto operator+(long int d, ImmerIterator<ImmerT> a)
        {
            return a += d;
        };
 
        template<class ImmerT>
        auto operator-(ImmerIterator<ImmerT> a, long int d)
        {
            return a -= d;
        };
 
        template<class ImmerT>
        auto immerBegin(const ImmerT &c)
        {
            return ImmerIterator<ImmerT>(c, 0);
        }
 
        template<class ImmerT>
        auto immerEnd(const ImmerT &c)
        {
            return ImmerIterator<ImmerT>(c, c.size());
        }
    }
 
    template<class ImmerT1, class RangeT2, class Pred, class Func>
    ImmerT1 sortedUniqueMerge(ImmerT1 base, RangeT2 addon, Pred exists, Func keyOf)
    {
        auto needToAdd = intoImmer(ImmerT1{},
                                   zug::filter([=](auto a) {
                                                   return !exists(a);
                                               }),
                                   addon);
 
        auto cmp = [=](auto a, auto b) {
                       return keyOf(a) < keyOf(b);
                   };
 
        for (auto item : needToAdd) {
            auto it = std::upper_bound(immerBegin(base), immerEnd(base), item, cmp);
            auto index = it.index();
            base = std::move(base).insert(index, item);
        }
 
        return base;
    }
 
    std::string increaseTxnId(std::string cur);
 
    std::string getTxnId(Event event, ClientModel &m);
}
 
namespace std
{
    template<> struct hash<Kazv::KeyOfState>
    {
        std::size_t operator()(const Kazv::KeyOfState & k) const noexcept {
            std::size_t seed = 0;
            boost::hash_combine(seed, k.type);
            boost::hash_combine(seed, k.stateKey);
            return seed;
        }
    };
}
 
#define KAZV_WRAP_ATTR(_type, _d, _attr)          \
    inline auto _attr() const {                   \
        KAZV_VERIFY_THREAD_ID();                  \
        return (_d)[&_type::_attr];               \
    }
 
BOOST_CLASS_VERSION(Kazv::KeyOfState, 0)