File registry.h¶

File List > endstone > registry.h
// Copyright (c) 2024, The Endstone Project. (https://endstone.dev) All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once

#include <functional>
#include <stdexcept>

#include <fmt/format.h>

#include "identifier.h"

namespace endstone {
class IRegistry {
public:
    virtual ~IRegistry() = default;
};

template <typename T>
class Registry : public IRegistry {
public:
    virtual T *get(Identifier<T> id) noexcept = 0;

    virtual const T *get(Identifier<T> id) const noexcept = 0;

    virtual T &getOrThrow(Identifier<T> id)
    {
        if (auto *p = get(id)) {
            return *p;
        }
        throw std::invalid_argument(fmt::format("No registry entry found for identifier: {}", id));
    }

    virtual const T &getOrThrow(Identifier<T> id) const
    {
        if (auto *p = get(id)) {
            return *p;
        }
        throw std::invalid_argument(fmt::format("No registry entry found for identifier: {}", id));
    }

    virtual void forEach(std::function<bool(const T &)> func) const = 0;
};
}  // namespace endstone

#define ENDSTONE_REGISTRY_TYPE(type)                            \
    static constexpr auto RegistryType = #type;                 \
                                                                \
    static const type *get(Identifier<type> id)                 \
    {                                                           \
        return detail::getServer().getRegistry<type>().get(id); \
    }