#ifndef COMPILER_SYNTAX_GRAMMAR_H_
#define COMPILER_SYNTAX_GRAMMAR_H_

#include <vector>
#include <string>
#include <set>
#include "tree_sitter/compiler.h"
#include "compiler/rules/symbol.h"
#include "compiler/variable.h"

namespace tree_sitter {

struct ProductionStep {
  ProductionStep(const rules::Symbol &, int, Associativity);
  ProductionStep(const rules::Symbol &, int, Associativity, int);
  bool operator==(const ProductionStep &) const;

  rules::Symbol symbol;
  int precedence;
  Associativity associativity;
  int rule_id;
};

typedef std::vector<ProductionStep> Production;

struct SyntaxVariable {
  SyntaxVariable(const std::string &, VariableType,
                 const std::vector<Production> &);

  std::string name;
  std::vector<Production> productions;
  VariableType type;
};

typedef std::set<rules::Symbol> ConflictSet;

struct SyntaxGrammar {
  const std::vector<Production> &productions(const rules::Symbol &) const;

  std::vector<SyntaxVariable> variables;
  std::set<rules::Symbol> ubiquitous_tokens;
  std::set<ConflictSet> expected_conflicts;
};

}  // namespace tree_sitter

#endif  // COMPILER_SYNTAX_GRAMMAR_H_