/*
 * Copyright (c)2013-2021 ZeroTier, Inc.
 *
 * Use of this software is governed by the Business Source License included
 * in the LICENSE.TXT file in the project's root directory.
 *
 * Change Date: 2026-01-01
 *
 * On the date above, in accordance with the Business Source License, use
 * of this software will be governed by version 2.0 of the Apache License.
 */
/****/

#ifndef ZT_STORE_HPP
#define ZT_STORE_HPP

#include "CallContext.hpp"
#include "Constants.hpp"
#include "Containers.hpp"
#include "Context.hpp"

namespace ZeroTier {

/**
 * Wrapper around API callbacks for data store
 */
class Store {
  public:
    ZT_INLINE Store(const Context& ctx) : m_ctx(ctx)
    {
    }

    /**
     * Get a state object
     *
     * @param type Object type
     * @param id Object ID
     * @param idSize Size of object ID in qwords
     * @return Data or empty vector if not found
     */
    ZT_INLINE Vector<uint8_t>
    get(const CallContext& cc, ZT_StateObjectType type, const uint64_t* const id, unsigned int idSize) const
    {
        Vector<uint8_t> dv;
        void* data = nullptr;
        void (*freeFunc)(void*) = nullptr;
        const int r = m_ctx.cb.stateGetFunction(
            reinterpret_cast<ZT_Node*>(m_ctx.node),
            m_ctx.uPtr,
            cc.tPtr,
            type,
            id,
            idSize,
            &data,
            &freeFunc);
        if (r > 0)
            dv.assign(reinterpret_cast<const uint8_t*>(data), reinterpret_cast<const uint8_t*>(data) + r);
        if ((data) && (freeFunc))
            freeFunc(data);
        return dv;
    }

    /**
     * Store a state object
     *
     * @param type Object type
     * @param id Object ID
     * @param idSize Size of object ID in qwords
     * @param data Data to store
     * @param len Length of data
     */
    ZT_INLINE void
    put(const CallContext& cc,
        ZT_StateObjectType type,
        const uint64_t* const id,
        const unsigned int idSize,
        const void* const data,
        const unsigned int len) noexcept
    {
        m_ctx.cb
            .statePutFunction(reinterpret_cast<ZT_Node*>(this), m_ctx.uPtr, cc.tPtr, type, id, idSize, data, (int)len);
    }

    /**
     * Erase a state object from the object store
     *
     * @param type Object type
     * @param id Object ID
     * @param idSize Size of object ID in qwords
     */
    ZT_INLINE void
    erase(const CallContext& cc, ZT_StateObjectType type, const uint64_t* const id, const unsigned int idSize) noexcept
    {
        m_ctx.cb.statePutFunction(reinterpret_cast<ZT_Node*>(this), m_ctx.uPtr, cc.tPtr, type, id, idSize, nullptr, -1);
    }

  private:
    const Context& m_ctx;
};

}   // namespace ZeroTier

#endif