Kelsidavis-WoWee/include/pipeline/mpq_manager.hpp

#pragma once

#include <string>
#include <vector>
#include <cstdint>
#include <memory>
#include <map>

// Forward declare StormLib handle
typedef void* HANDLE;

namespace wowee {
namespace pipeline {

/**
 * MPQManager - Manages MPQ archive loading and file reading
 *
 * WoW 3.3.5a stores all game assets in MPQ archives.
 * This manager loads multiple archives and provides unified file access.
 */
class MPQManager {
public:
    MPQManager();
    ~MPQManager();

    /**
     * Initialize the MPQ system
     * @param dataPath Path to WoW Data directory
     * @return true if initialization succeeded
     */
    bool initialize(const std::string& dataPath);

    /**
     * Shutdown and close all archives
     */
    void shutdown();

    /**
     * Load a single MPQ archive
     * @param path Full path to MPQ file
     * @param priority Priority for file resolution (higher = checked first)
     * @return true if archive loaded successfully
     */
    bool loadArchive(const std::string& path, int priority = 0);

    /**
     * Check if a file exists in any loaded archive
     * @param filename Virtual file path (e.g., "World\\Maps\\Azeroth\\Azeroth.wdt")
     * @return true if file exists
     */
    bool fileExists(const std::string& filename) const;

    /**
     * Read a file from MPQ archives
     * @param filename Virtual file path
     * @return File contents as byte vector (empty if not found)
     */
    std::vector<uint8_t> readFile(const std::string& filename) const;

    /**
     * Get file size without reading it
     * @param filename Virtual file path
     * @return File size in bytes (0 if not found)
     */
    uint32_t getFileSize(const std::string& filename) const;

    /**
     * Check if MPQ system is initialized
     */
    bool isInitialized() const { return initialized; }

    /**
     * Get list of loaded archives
     */
    const std::vector<std::string>& getLoadedArchives() const { return archiveNames; }

private:
    struct ArchiveEntry {
        HANDLE handle;
        std::string path;
        int priority;
    };

    bool initialized = false;
    std::string dataPath;
    std::vector<ArchiveEntry> archives;
    std::vector<std::string> archiveNames;

    /**
     * Find archive containing a file
     * @param filename File to search for
     * @return Archive handle or nullptr if not found
     */
    HANDLE findFileArchive(const std::string& filename) const;

    /**
     * Load patch archives (e.g., patch.MPQ, patch-2.MPQ, etc.)
     */
    bool loadPatchArchives();

    /**
     * Load locale-specific archives
     * @param locale Locale string (e.g., "enUS")
     */
    bool loadLocaleArchives(const std::string& locale);
};

} // namespace pipeline
} // namespace wowee