#ifndef tree_sitter_compiler_h
#define tree_sitter_compiler_h

#include <vector>
#include <map>
#include <set>
#include <string>
#include <memory>

namespace tree_sitter {
    namespace rules {
        class Rule;
        class Symbol;
        
        struct CharacterRange {
            char min;
            char max;
            CharacterRange(char);
            CharacterRange(char, char);
            bool operator==(const CharacterRange &) const;
            bool operator<(const CharacterRange &) const;
            std::string to_string() const;
        };
    }
}

namespace std {
    template<>
    struct hash<tree_sitter::rules::CharacterRange> {
        size_t operator()(const tree_sitter::rules::CharacterRange &range) const {
            return (hash<char>()(range.min) ^ hash<char>()(range.max));
        }
    };
}

namespace tree_sitter {
    namespace rules {
        typedef std::shared_ptr<Rule> rule_ptr;
        std::ostream& operator<<(std::ostream& stream, const rule_ptr &rule);
        
        rule_ptr blank();
        rule_ptr character(const std::set<CharacterRange> &matches);
        rule_ptr character(const std::set<CharacterRange> &matches, bool);
        rule_ptr choice(const std::vector<rule_ptr> &rules);
        rule_ptr pattern(const std::string &value);
        rule_ptr repeat(const rule_ptr content);
        rule_ptr seq(const std::vector<rule_ptr> &rules);
        rule_ptr str(const std::string &value);
        rule_ptr sym(const std::string &name);
        rule_ptr _sym(const std::string &name);
    }
}

namespace tree_sitter {
    class Grammar {
    public:
        Grammar(std::string start_rule_name, const std::map<const std::string, const rules::rule_ptr> &rules);
        bool operator==(const Grammar &other) const;
        const std::string start_rule_name;
        const std::map<const std::string, const rules::rule_ptr> rules;
    };
    
    std::ostream& operator<<(std::ostream &stream, const Grammar &grammar);
    
    std::string compile(const Grammar &grammar, std::string name);
}

#endif