/*
 *
 *    Copyright (c) 2021 Project CHIP Authors
 *
 *    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 <lib/support/logging/CHIPLogging.h>
#include <stdio.h>

namespace {

constexpr size_t kMaxLogMessageLength = 256;

using PythonLogCallback = void (*)(uint8_t category, const char * module, const char * message);

PythonLogCallback sPythonLogCallback;

void ENFORCE_FORMAT(3, 0) NativeLoggingCallback(const char * module, uint8_t category, const char * msg, va_list args)
{
    if (sPythonLogCallback == nullptr)
    {
        return;
    }
    char buffer[kMaxLogMessageLength];
    vsnprintf(buffer, sizeof(buffer), msg, args);
    buffer[sizeof(buffer) - 1] = 0;

    sPythonLogCallback(category, module, buffer);
}

} // namespace

extern "C" void pychip_logging_set_callback(PythonLogCallback callback)
{
    if (callback == nullptr)
    {
        chip::Logging::SetLogRedirectCallback(nullptr);
        sPythonLogCallback = callback;
    }
    else
    {
        sPythonLogCallback = callback;
        chip::Logging::SetLogRedirectCallback(NativeLoggingCallback);
    }
}