#include "fud_assert.hpp"

#include "fud_format.hpp"
#include "fud_string.hpp"

#include <climits>
#include <cstdio>
#include <exception>

namespace fud {

struct BufferSink {
    DrainResult drain(StringView source);
};

DrainResult BufferSink::drain(StringView source)
{
    DrainResult result{0, FudStatus::Success};
    if (source.m_data == nullptr) {
        result.status = FudStatus::NullPointer;
        return result;
    }
    if (source.m_length == 0) {
        result.status = FudStatus::Success;
        return result;
    }
    result.bytesWritten = fwrite(reinterpret_cast<const char*>(source.m_data), 1, source.m_length, stderr);
    if (result.bytesWritten != source.m_length) {
        result.status = FudStatus::Full;
    }
    return result;
}

namespace impl {
void assertFailMessage(const char* assertion, const std::source_location sourceLocation)
{
    BufferSink sink;
    const char* fileName = sourceLocation.file_name();
    if (fileName == nullptr) {
        fileName = "Unknown file";
    }
    const char* functionName = sourceLocation.function_name();
    if (functionName == nullptr) {
        functionName = "Unknown Function";
    }
    format(sink, FormatCharMode::Unchecked, "{}:{}:{}: {}\n", fileName, functionName, sourceLocation.line(), assertion);
}

} // namespace impl

[[noreturn]] void assertFail(const char* assertion, const std::source_location sourceLocation)
{
    impl::assertFailMessage(assertion, sourceLocation);

    std::terminate();
}

} // namespace fud