grammar_registry.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/types/config.h>
 
#include <filesystem>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace entropic {
 
class GrammarRegistry {
public:
    size_t load_bundled(const std::filesystem::path& grammar_dir);
 
    bool register_grammar(const std::string& key,
                          const std::string& gbnf_content,
                          const std::string& source = "runtime");
 
    bool register_from_file(const std::string& key,
                            const std::filesystem::path& path);
 
    bool deregister(const std::string& key);
 
    std::string get(const std::string& key) const;
 
    bool has(const std::string& key) const;
 
    GrammarEntry entry(const std::string& key) const;
 
    std::vector<GrammarEntry> list() const;
 
    static std::string validate(const std::string& gbnf_content);
 
    size_t size() const;
 
    void clear();
 
private:
    std::unordered_map<std::string, GrammarEntry> grammars_;
 
    mutable std::mutex registry_mutex_;
};
 
} // namespace entropic