/*
 * libfud
 * Copyright 2024 Dominick Allen
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "fud_assert.hpp"

#include "fud_drain.hpp"
#include "fud_format.hpp"
#include "fud_string_view.hpp"

#include <cstdio>
#include <exception>

namespace fud {

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

DrainResult StdErrSink::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;
    }
    /* TODO: give users control over this functionality */
    result.bytesDrained = fwrite(reinterpret_cast<const char*>(source.m_data), 1, source.m_length, stderr);
    if (result.bytesDrained != source.m_length) {
        result.status = FudStatus::Full;
    }
    return result;
}

namespace impl {
void assertFailMessage(const char* assertion, const std::source_location sourceLocation)
{
    StdErrSink 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";
    }
    static_cast<void>(format(
        sink,
        FormatCharMode::Unchecked,
        u8"{}:{}:{}: {}\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