#pragma once
#include <comp/symbol_table.hpp>
#include <decomp/decomp.hpp>

namespace theo::comp {
using resolver_t = std::function<std::uintptr_t(std::string)>;
using copier_t = std::function<void(std::uintptr_t, void*, std::uint32_t)>;
using allocator_t =
    std::function<std::uintptr_t(std::uint32_t,
                                 coff::section_characteristics_t)>;

class comp_t {
 public:
  explicit comp_t(decomp::decomp_t* dcmp);
  explicit comp_t(decomp::decomp_t* dcmp,
                  allocator_t alloc,
                  copier_t copy,
                  resolver_t resolve);

  void compose();
  void allocator(allocator_t alloc);
  void copier(copier_t copy);
  void resolver(resolver_t resolve);
  std::uintptr_t resolve(const std::string&& sym);

 private:
  void gen_reloc_trans(decomp::symbol_t* sym);
  decomp::decomp_t* m_dcmp;
  resolver_t m_resolver;
  copier_t m_copier;
  allocator_t m_allocator;
};
}  // namespace theo::comp