#ifndef __TreeSitter__grammar__
#define __TreeSitter__grammar__

#include <map>
#include <vector>
#include "rules.h"

namespace tree_sitter {
    class Grammar {
        typedef std::initializer_list<std::pair<const std::string, const rules::rule_ptr>> rule_map_init_list;
        typedef std::map<const std::string, const rules::rule_ptr> rule_map;

    public:
        Grammar(const rule_map_init_list &rules);
        Grammar(std::string start_rule_name, const rule_map &rules);
        Grammar(std::string start_rule_name, const rule_map &rules, const rule_map &aux_rules);

        const std::string start_rule_name;
        std::vector<std::string> rule_names() const;
        bool operator==(const Grammar &other) const;
        bool has_definition(const rules::Symbol &symbol) const;
        const rules::rule_ptr rule(const rules::Symbol &symbol) const;

        rule_map rules;
        rule_map aux_rules;
    };
    
    std::ostream& operator<<(std::ostream &stream, const Grammar &grammar);
}

#endif