#pragma once

#include "ObjContainer/ObjContainerReferenceable.h"
#include "ObjContainer/ObjContainerRepository.h"
#include "ObjContainer/SoundBank/SoundBankTypes.h"
#include "SearchPath/ISearchPath.h"
#include "Utils/ClassUtils.h"
#include "Utils/FileUtils.h"
#include "Utils/ObjStream.h"
#include "Zone/Zone.h"

#include <istream>

class SoundBankEntryInputStream
{
public:
    std::unique_ptr<std::istream> m_stream;
    SoundAssetBankEntry m_entry;

    SoundBankEntryInputStream();
    SoundBankEntryInputStream(std::unique_ptr<std::istream> stream, SoundAssetBankEntry entry);

    _NODISCARD bool IsOpen() const;
};

class SoundBank final : public ObjContainerReferenceable
{
    static constexpr uint32_t MAGIC = FileUtils::MakeMagic32('2', 'U', 'X', '#');
    static constexpr uint32_t VERSION = 14u;

    std::string m_file_name;
    std::unique_ptr<std::istream> m_stream;
    int64_t m_file_size;

    bool m_initialized;
    SoundAssetBankHeader m_header;
    std::vector<std::string> m_dependencies;
    std::vector<SoundAssetBankEntry> m_entries;
    std::vector<SoundAssetBankChecksum> m_checksums;
    std::unordered_map<unsigned int, size_t> m_entries_by_id;

    bool ReadHeader();
    bool ReadEntries();
    bool ReadChecksums();

public:
    static ObjContainerRepository<SoundBank, Zone> Repository;

    static std::string GetFileNameForDefinition(bool streamed, const char* zone, const char* language);

    SoundBank(std::string fileName, std::unique_ptr<std::istream> stream, int64_t fileSize);
    ~SoundBank() override = default;
    SoundBank(const SoundBank& other) = delete;
    SoundBank(SoundBank&& other) noexcept = default;
    SoundBank& operator=(const SoundBank& other) = delete;
    SoundBank& operator=(SoundBank&& other) noexcept = default;

    std::string GetName() override;

    bool Initialize();
    _NODISCARD const std::vector<std::string>& GetDependencies() const;

    _NODISCARD bool VerifyChecksum(const SoundAssetBankChecksum& checksum) const;
    _NODISCARD SoundBankEntryInputStream GetEntryStream(unsigned int id) const;
};