/*
  This file is part of Ingen.
  Copyright 2007-2016 David Robillard <http://drobilla.net/>

  Ingen is free software: you can redistribute it and/or modify it under the
  terms of the GNU Affero General Public License as published by the Free
  Software Foundation, either version 3 of the License, or any later version.

  Ingen is distributed in the hope that it will be useful, but WITHOUT ANY
  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  A PARTICULAR PURPOSE.  See the GNU Affero General Public License for details.

  You should have received a copy of the GNU Affero General Public License
  along with Ingen.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "ingen/Log.hpp"

#include "ingen/ColorContext.hpp"
#include "ingen/Node.hpp"
#include "ingen/URIs.hpp"
#include "ingen/World.hpp"
#include "lv2/core/lv2.h"
#include "lv2/log/log.h"
#include "lv2/urid/urid.h"
#include "raul/Path.hpp"

#include <cstdio>
#include <cstdlib>

namespace ingen {

Log::Log(LV2_Log_Log* log, URIs& uris)
	: _log(log)
	, _uris(uris)
	, _flush(false)
	, _trace(false)
{}

void
Log::rt_error(const char* msg)
{
#ifndef NDEBUG
	va_list args;
	vtprintf(_uris.log_Error, msg, args);
#endif
}

void
Log::error(const std::string& msg)
{
	va_list args;
	vtprintf(_uris.log_Error, msg.c_str(), args);
}

void
Log::warn(const std::string& msg)
{
	va_list args;
	vtprintf(_uris.log_Warning, msg.c_str(), args);
}

void
Log::info(const std::string& msg)
{
	va_list args;
	vtprintf(_uris.log_Note, msg.c_str(), args);
}

void
Log::trace(const std::string& msg)
{
	va_list args;
	vtprintf(_uris.log_Trace, msg.c_str(), args);
}

void
Log::print(FILE* stream, const std::string& msg)
{
	fprintf(stream, "%s", msg.c_str());
	if (_flush) {
		fflush(stdout);
	}
}

int
Log::vtprintf(LV2_URID type, const char* fmt, va_list args)
{
	int ret = 0;
	if (type == _uris.log_Trace && !_trace) {
		return 0;
	} else if (_sink) {
		_sink(type, fmt, args);
	}

	if (_log) {
		ret = _log->vprintf(_log->handle, type, fmt, args);
	} else if (type == _uris.log_Error) {
		ColorContext ctx(stderr, ColorContext::Color::RED);
		ret = vfprintf(stderr, fmt, args);
	} else if (type == _uris.log_Warning) {
		ColorContext ctx(stderr, ColorContext::Color::YELLOW);
		ret = vfprintf(stderr, fmt, args);
	} else if (type == _uris.log_Note) {
		ColorContext ctx(stderr, ColorContext::Color::GREEN);
		ret = vfprintf(stdout, fmt, args);
	} else if (_trace && type == _uris.log_Trace) {
		ColorContext ctx(stderr, ColorContext::Color::GREEN);
		ret = vfprintf(stderr, fmt, args);
	} else {
		fprintf(stderr, "Unknown log type %d\n", type);
		return 0;
	}
	if (_flush) {
		fflush(stdout);
	}
	return ret;
}

static int
log_vprintf(LV2_Log_Handle handle, LV2_URID type, const char* fmt, va_list args)
{
	auto*   f      = static_cast<Log::Feature::Handle*>(handle);
	va_list noargs = {};

	int ret  = f->log->vtprintf(type, f->node->path().c_str(), noargs);
	ret     += f->log->vtprintf(type, ": ", noargs);
	ret     += f->log->vtprintf(type, fmt, args);

	return ret;
}

static int
log_printf(LV2_Log_Handle handle, LV2_URID type, const char* fmt, ...)
{
	va_list args;
	va_start(args, fmt);
	const int ret = log_vprintf(handle, type, fmt, args);
	va_end(args);

	return ret;
}

static void
free_log_feature(LV2_Feature* feature) {
	auto* lv2_log = static_cast<LV2_Log_Log*>(feature->data);
	free(lv2_log->handle);
	free(feature);
}

SPtr<LV2_Feature>
Log::Feature::feature(World& world, Node* block)
{
	auto* handle = static_cast<Handle*>(calloc(1, sizeof(Handle)));
	handle->lv2_log.handle  = handle;
	handle->lv2_log.printf  = log_printf;
	handle->lv2_log.vprintf = log_vprintf;
	handle->log             = &world.log();
	handle->node            = block;

	auto* f = static_cast<LV2_Feature*>(malloc(sizeof(LV2_Feature)));
	f->URI  = LV2_LOG__log;
	f->data = &handle->lv2_log;

	return SPtr<LV2_Feature>(f, &free_log_feature);
}

}  // namespace ingen