#pragma once
#include <functional>
#include <map>
#include <vector>

#include <decomp/symbol.hpp>

namespace theo::comp {
class symbol_table_t {
 public:
  symbol_table_t();
  symbol_table_t(const std::vector<decomp::symbol_t>&& syms);

  void add_symbol(decomp::symbol_t& sym);
  void add_symbols(std::vector<decomp::symbol_t>& syms);

  void update(std::string& name, decomp::symbol_t& sym);
  void update(std::string& name, std::uintptr_t location);

  void for_each(std::function<bool(decomp::symbol_t& sym)> fn);
 private:
  std::map<std::string, decomp::symbol_t> m_table;
};
}  // namespace theo::comp