#pragma once
#include <cstddef>
#include <cstdint>
#include <obf/transform/transform.hpp>

namespace theo::recomp {

/// <summary>
/// meta data about a relocation for a symbol
/// </summary>
class reloc_t {
 public:
  /// <summary>
  /// explicit constructor for this class.
  /// </summary>
  /// <param name="offset">offset into the symbol data where the relocation is
  /// at. all relocations are assumed to be linear virtual addresses of the
  /// symbol.</param>
  /// <param name="hash">hash of the symbol to which the relocation is
  /// of.</param> <param name="sym_name">the name of the symbol to which the
  /// relocation is of.</param>
  explicit reloc_t(std::uint32_t offset,
                   std::size_t hash,
                   const std::string&& sym_name)
      : m_offset(offset), m_hash(hash), m_sym_name(sym_name) {}

  /// <summary>
  /// returns the hash of the relocation symbol.
  /// </summary>
  /// <returns>hash of the relocation symbol</returns>
  std::size_t hash() { return m_hash; }

  /// <summary>
  /// returns the name of the relocation symbol.
  /// </summary>
  /// <returns>returns the name of the relocation symbol.</returns>
  std::string name() { return m_sym_name; }

  /// <summary>
  /// returns the offset into the symbol to which the relocation will be
  /// applied. the offset is in bytes. zero based.
  /// </summary>
  /// <returns>returns the offset into the symbol to which the relocation will
  /// be applied. the offset is in bytes. zero based.</returns>
  std::uint32_t offset() { return m_offset; }

  /// <summary>
  /// sets the offset to which the relocation gets applied too.
  /// </summary>
  /// <param name="offset">offset to which the relocation gets applied
  /// too.</param>
  void offset(std::uint32_t offset) { m_offset = offset; }

  /// <summary>
  /// adds a transformation to be applied to the relocation prior to writing it
  /// into the symbol.
  /// </summary>
  /// <param name="entry">a pair containing a lambda function that when executed
  /// transforms a relocation. the second value in the pair is a random value
  /// which is passed to the lambda.</param>
  void add_transform(
      std::pair<obf::transform::transform_t*, std::uint32_t> entry) {
    m_transforms.push_back(entry);
  }

  /// <summary>
  /// gets the vector of transformation.
  /// </summary>
  /// <returns>returns the vector of transformations.</returns>
  std::vector<std::pair<obf::transform::transform_t*, std::uint32_t>>&
  get_transforms() {
    return m_transforms;
  }

 private:
  std::vector<std::pair<obf::transform::transform_t*, std::uint32_t>>
      m_transforms;
  std::string m_sym_name;
  std::size_t m_hash;
  std::uint32_t m_offset;
};
}  // namespace theo::recomp