// SPDX-License-Identifier: MIT
#pragma once

#include <FEXCore/fextl/map.h>
#include <FEXCore/fextl/unordered_map.h>
#include <FEXCore/fextl/string.h>
#include <FEXCore/fextl/vector.h>

#include "Linux/Utils/ELFContainer.h"

#include <cstdint>
#include <functional>
#include <utility>

namespace ELFLoader {
class ELFSymbolDatabase final {
public:
  ELFSymbolDatabase(::ELFLoader::ELFContainer* file);
  ~ELFSymbolDatabase();

  uint64_t GetElfBase() const;

  void MapMemoryRegions(std::function<void*(uint64_t, uint64_t, bool)> Mapper);
  void WriteLoadableSections(::ELFLoader::ELFContainer::MemoryWriter Writer);

  uint64_t DefaultRIP() const;

  using RangeType = std::pair<uint64_t, uint64_t>;
  const ::ELFLoader::ELFSymbol* GetSymbolInRange(RangeType Address);
  const ::ELFLoader::ELFSymbol* GetGlobalSymbolInRange(RangeType Address);
  const ::ELFLoader::ELFSymbol* GetNoWeakSymbolInRange(RangeType Address);

  void GetInitLocations(fextl::vector<uint64_t>* Locations);

private:
  ::ELFLoader::ELFContainer* File;

  struct ELFInfo {
    fextl::string Name;
    ::ELFLoader::ELFContainer* Container;
    ::ELFLoader::ELFContainer::MemoryLayout CustomLayout;
    void* ELFBase;
    uint64_t GuestBase;
  };

  ELFInfo LocalInfo;
  fextl::vector<ELFInfo*> DynamicELFInfo;
  fextl::vector<ELFInfo*> InitializationOrder;
  uint64_t ELFMemorySize {};
  bool FixedNoReplace {true};
  void* ELFBase {};

  fextl::unordered_map<fextl::string, ELFInfo*> NameToELF;
  fextl::vector<fextl::string> LibrarySearchPaths;

  // Symbols
  fextl::vector<ELFLoader::ELFSymbol*> Symbols;
  using SymbolTableType = fextl::unordered_map<fextl::string, ELFLoader::ELFSymbol*>;
  SymbolTableType SymbolMap;
  SymbolTableType SymbolMapGlobalOnly;
  SymbolTableType SymbolMapNoWeak;
  SymbolTableType SymbolMapNoMain;
  SymbolTableType SymbolMapNoMainNoWeak;
  fextl::map<uint64_t, ELFLoader::ELFSymbol*> SymbolMapByAddress;

  bool FindLibraryFile(fextl::string* Result, const char* Library);
  void FillLibrarySearchPaths();
  void FillMemoryLayouts(uint64_t DefinedBase);
  void FillInitializationOrder();
  void FillSymbols();

  void HandleRelocations();

  const ::ELFLoader::ELFSymbol* GetSymbolFromTable(RangeType Address, fextl::unordered_map<fextl::string, ELFLoader::ELFSymbol*>& Table);
};
} // namespace ELFLoader