#ifndef PS2RECOMP_ELF_PARSER_H
#define PS2RECOMP_ELF_PARSER_H

#include "ps2recomp/types.h"
#include <elfio/elfio.hpp>
#include <string>
#include <vector>
#include <memory>

namespace ps2recomp
{

    class ElfParser
    {
    public:
        ElfParser(const std::string &filePath);
        ~ElfParser();

        bool parse();

        std::vector<Function> extractFunctions();
        std::vector<Symbol> extractSymbols();
        std::vector<Section> getSections();
        std::vector<Relocation> getRelocations();

        // Helper methods
        bool isValidAddress(uint32_t address) const;
        uint32_t readWord(uint32_t address) const;
        uint8_t *getSectionData(const std::string &sectionName);
        uint32_t getSectionAddress(const std::string &sectionName);
        uint32_t getSectionSize(const std::string &sectionName);

    private:
        std::string m_filePath;
        std::unique_ptr<ELFIO::elfio> m_elf;

        std::vector<Section> m_sections;
        std::vector<Symbol> m_symbols;
        std::vector<Relocation> m_relocations;

        void loadSections();
        void loadSymbols();
        void loadRelocations();
        bool isExecutableSection(const ELFIO::section *section) const;
        bool isDataSection(const ELFIO::section *section) const;
    };

} // namespace ps2recomp

#endif // PS2RECOMP_ELF_PARSER_H