#pragma once

#define MEMORY_CATCH(expr)                 \
try {                                      \
    (expr);                                \
} catch (MemoryException exc) {            \
    MemoryException::HandleException(exc); \
}                                          \
do {} while (0)

#define MEMORY_THROW(...) throw MemoryException(__func__, __LINE__, ##__VA_ARGS__);

class MemoryException : public std::exception {
public:
    inline MemoryException(const char* func, int32_t line, const char* message) noexcept
        : MemoryException(func, line, message, {}, 0) {}
    inline MemoryException(const char* func, int32_t line, const char* message, const std::vector<int>& offsets) noexcept
        : MemoryException(func, line, message, offsets, 0) {}
    inline MemoryException(const char* func, int32_t line, const char* message, const std::vector<int>& offsets, size_t numItems) noexcept
        : _func(func), _line(line), _message(message), _offsets(offsets), _numItems(numItems) {}
    
    ~MemoryException() = default;
    inline const char* what() const noexcept {
        return _message;
    }
    static void HandleException(const MemoryException& exc) noexcept {
        std::string msg = "MemoryException thrown in function ";
        msg += exc._func;
        msg += " on line " + std::to_string(exc._line) + ":\n" + exc._message + "\nOffsets:";
        for (int offset : exc._offsets) {
            msg += " " + std::to_string(offset);
        }
        msg += "\n";
        if (exc._numItems != 0) {
            msg += "Num Items: " + std::to_string(exc._numItems) + "\n";
        }
        OutputDebugStringA(msg.c_str());
#ifndef NDEBUG
        MessageBoxA(NULL, msg.c_str(), "Memory Exception Thrown", MB_OK);
#endif
    }

private:
    const char* _func;
    int32_t _line;
    const char* _message;
    const std::vector<int> _offsets;
    size_t _numItems = 0;
};