#ifndef COMPILER_PARSE_TABLE_H_
#define COMPILER_PARSE_TABLE_H_

#include <map>
#include <set>
#include <utility>
#include <vector>
#include "compiler/lex_table.h"
#include "compiler/rules/symbol.h"
#include "compiler/rules/metadata.h"

namespace tree_sitter {

typedef uint64_t ParseStateId;

typedef enum {
  ParseActionTypeError,
  ParseActionTypeReduceExtra,
  ParseActionTypeShiftExtra,

  ParseActionTypeShift,
  ParseActionTypeReduce,
  ParseActionTypeAccept,
} ParseActionType;

class ParseAction {
  ParseAction(ParseActionType type, ParseStateId state_index,
              rules::Symbol symbol, size_t consumed_symbol_count,
              std::set<int> precedence_values, Associativity, int production_id);

 public:
  ParseAction();
  static ParseAction Accept();
  static ParseAction Error();
  static ParseAction Shift(ParseStateId state_index,
                           std::set<int> precedence_values);
  static ParseAction Reduce(rules::Symbol symbol, size_t consumed_symbol_count,
                            int precedence, Associativity, int production_id);
  static ParseAction ShiftExtra();
  static ParseAction ReduceExtra(rules::Symbol symbol);
  bool operator==(const ParseAction &) const;
  bool operator<(const ParseAction &) const;

  ParseActionType type;
  rules::Symbol symbol;
  ParseStateId state_index;
  size_t consumed_symbol_count;
  std::set<int> precedence_values;
  Associativity associativity;
  int production_id;
};

}  // namespace tree_sitter

namespace std {

template <>
struct hash<tree_sitter::ParseAction> {
  size_t operator()(const tree_sitter::ParseAction &action) const {
    return (hash<int>()(action.type) ^
            hash<tree_sitter::rules::Symbol>()(action.symbol) ^
            hash<size_t>()(action.state_index) ^
            hash<size_t>()(action.consumed_symbol_count) ^
            hash<size_t>()(action.production_id));
  }
};

}  // namespace std

namespace tree_sitter {

class ParseState {
 public:
  ParseState();
  std::map<rules::Symbol, std::vector<ParseAction>> actions;
  std::set<rules::Symbol> expected_inputs() const;
  LexStateId lex_state_id;
};

class ParseTable {
 public:
  ParseStateId add_state();
  ParseAction &set_action(ParseStateId state_id, rules::Symbol symbol,
                          ParseAction action);
  ParseAction &add_action(ParseStateId state_id, rules::Symbol symbol,
                          ParseAction action);

  std::vector<ParseState> states;
  std::set<rules::Symbol> symbols;
  std::set<int> fragile_production_ids;
};

}  // namespace tree_sitter

#endif  // COMPILER_PARSE_TABLE_H_