#ifndef TREE_SITTER_COMPILER_H_
#define TREE_SITTER_COMPILER_H_

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

namespace tree_sitter {

namespace rules {

class Rule;
typedef std::shared_ptr<Rule> rule_ptr;

rule_ptr blank();
rule_ptr choice(const std::vector<rule_ptr> &rules);
rule_ptr repeat(const rule_ptr &content);
rule_ptr seq(const std::vector<rule_ptr> &rules);
rule_ptr sym(const std::string &name);
rule_ptr pattern(const std::string &value);
rule_ptr str(const std::string &value);
rule_ptr keyword(const std::string &value);
rule_ptr keypattern(const std::string &value);
rule_ptr err(const rule_ptr &rule);
rule_ptr prec(int precedence, rule_ptr rule);
rule_ptr token(rule_ptr rule);

std::ostream &operator<<(std::ostream &stream, const rules::rule_ptr &rule);

}  // namespace rules

class Grammar {
  const std::vector<std::pair<std::string, rules::rule_ptr>> rules_;
  std::set<rules::rule_ptr> ubiquitous_tokens_;

 public:
  explicit Grammar(const std::vector<std::pair<std::string, rules::rule_ptr>> &);
  bool operator==(const Grammar &) const;
  std::string start_rule_name() const;
  const rules::rule_ptr rule(const std::string &) const;
  const std::vector<std::pair<std::string, rules::rule_ptr>> &rules() const;
  const std::set<rules::rule_ptr> &ubiquitous_tokens() const;
  Grammar &ubiquitous_tokens(const std::set<rules::rule_ptr> &);
};

struct Conflict {
  explicit Conflict(std::string description);
  std::string description;
  bool operator==(const Conflict &other) const;
  bool operator<(const Conflict &other) const;
};

enum GrammarErrorType {
  GrammarErrorTypeRegex,
  GrammarErrorTypeUndefinedSymbol,
  GrammarErrorTypeInvalidUbiquitousToken
};

class GrammarError {
 public:
  GrammarError(GrammarErrorType type, std::string message);
  bool operator==(const GrammarError &other) const;
  GrammarErrorType type;
  std::string message;
};

std::tuple<std::string, std::vector<Conflict>, const GrammarError *> compile(
    const Grammar &grammar, std::string name);

std::ostream &operator<<(std::ostream &stream, const Grammar &grammar);
std::ostream &operator<<(std::ostream &stream, const Conflict &conflict);
std::ostream &operator<<(std::ostream &stream, const GrammarError *error);

}  // namespace tree_sitter

#endif  // TREE_SITTER_COMPILER_H_