/*
    Ousía
    Copyright (C) 2014, 2015  Benjamin Paaßen, Andreas Stöckel

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program 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 General Public License for more details.

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

#include <core/common/Exceptions.hpp>
#include <core/common/Utils.hpp>
#include <core/common/VariantWriter.hpp>
#include <core/model/Ontology.hpp>
#include <core/model/Typesystem.hpp>
#include <core/model/RootNode.hpp>

#include "ParserScope.hpp"

#define SCOPE_DEBUG_OUTPUT 0
#if SCOPE_DEBUG_OUTPUT
#include <iostream>
#endif

namespace ousia {

/* Class ParserScopeBase */

ParserScopeBase::ParserScopeBase() {}

ParserScopeBase::ParserScopeBase(const ManagedVector<Node> &nodes)
    : nodes(nodes)
{
}

Rooted<Node> ParserScopeBase::resolve(const Rtti *type,
                                      const std::vector<std::string> &path,
                                      Logger &logger)
{
	// Go up the stack and try to resolve the
	for (auto it = nodes.rbegin(); it != nodes.rend(); it++) {
		std::vector<ResolutionResult> res = (*it)->resolve(type, path);

		// Abort if the object could not be resolved
		if (res.empty()) {
			continue;
		}

		// Log an error if the object is not unique
		if (res.size() > 1) {
			logger.error(std::string("The reference \"") +
			             Utils::join(path, ".") + ("\" is ambigous!"));
			logger.note("Referenced objects are:", SourceLocation{},
			            MessageMode::NO_CONTEXT);
			for (const ResolutionResult &r : res) {
				logger.note(Utils::join(r.path(), "."), *(r.node));
			}
		}
		return res[0].node;
	}
	return nullptr;
}

bool ParserScopeBase::isEmpty() const { return nodes.empty(); }

Rooted<Node> ParserScopeBase::getRoot() const { return nodes.front(); }

Rooted<Node> ParserScopeBase::getLeaf() const { return nodes.back(); }

const ManagedVector<Node> &ParserScopeBase::getStack() const { return nodes; }

std::vector<Rtti const *> ParserScopeBase::getStackTypeSignature() const
{
	std::vector<Rtti const *> res;
	res.reserve(nodes.size());
	for (size_t i = 0; i < nodes.size(); i++) {
		res.push_back(nodes[i]->type());
	}
	return res;
}

Rooted<Node> ParserScopeBase::select(RttiSet types, int maxDepth)
{
	ssize_t minDepth = 0;
	if (maxDepth >= 0) {
		minDepth = static_cast<ssize_t>(nodes.size()) - (maxDepth + 1);
	}
	for (ssize_t i = nodes.size() - 1; i >= minDepth; i--) {
		if (nodes[i]->type()->isOneOf(types)) {
			return nodes[i];
		}
	}
	return nullptr;
}

Rooted<Node> ParserScopeBase::selectOrThrow(RttiSet types, int maxDepth)
{
	Rooted<Node> res = select(types, maxDepth);
	if (res == nullptr) {
		std::vector<std::string> typenames;
		for (auto type : types) {
			typenames.push_back(type->name);
		}
		throw LoggableException{std::string(
		                            "Expected to be inside an element of one "
		                            "of the internal types ") +
		                        Utils::join(typenames, "\", \"", "\"", "\"")};
	}
	return res;
}

/* Class DeferredResolution */

DeferredResolution::DeferredResolution(const ManagedVector<Node> &nodes,
                                       const std::vector<std::string> &path,
                                       const Rtti *type,
                                       ResolutionResultCallback resultCallback,
                                       Handle<Node> owner)
    : scope(nodes),
      resultCallback(resultCallback),
      path(path),
      type(type),
      owner(owner)
{
}

bool DeferredResolution::resolve(
    const std::unordered_multiset<const Node *> &ignore, Logger &logger)
{
	// Fork the logger to prevent error messages from being shown if we actively
	// ignore the resolution result
	LoggerFork loggerFork = logger.fork();
	Rooted<Node> res = scope.resolve(type, path, loggerFork);
	if (res != nullptr) {
		if (!ignore.count(res.get())) {
			loggerFork.commit();
			try {
				// Push the location onto the logger default location stack
				GuardedLogger loggerGuard(logger, *owner);
				resultCallback(res, owner, logger);
			}
			catch (LoggableException ex) {
				logger.log(ex);
			}
			return true;
		}
	} else {
		loggerFork.commit();
	}
	return false;
}

void DeferredResolution::fail(Logger &logger)
{
	try {
		resultCallback(nullptr, owner, logger);
	}
	catch (LoggableException ex) {
		logger.log(ex);
	}
}

/* Class ParserScope */

ParserScope::ParserScope(const ManagedVector<Node> &nodes,
                         const std::vector<ParserFlagDescriptor> &flags)
    : ParserScopeBase(nodes), flags(flags), topLevelDepth(nodes.size())
{
}

ParserScope::ParserScope() : topLevelDepth(0) {}

bool ParserScope::checkUnwound(Logger &logger) const
{
	if (nodes.size() > topLevelDepth) {
		logger.error("Not all open elements have been closed!",
		             SourceLocation{}, MessageMode::NO_CONTEXT);
		logger.note("Still open elements are: ", SourceLocation{},
		            MessageMode::NO_CONTEXT);
		for (size_t i = topLevelDepth; i < nodes.size(); i++) {
			logger.note(std::string("Element of interal type ") +
			                nodes[i]->type()->name +
			                std::string(" defined here:"),
			            *nodes[i]);
		}
		return false;
	}
	return true;
}

ParserScope ParserScope::fork() { return ParserScope{nodes, flags}; }

bool ParserScope::join(const ParserScope &fork, Logger &logger)
{
	// Make sure the fork has been unwound
	if (!fork.checkUnwound(logger)) {
		return false;
	}

	// Insert the deferred resolutions of the fork into our own deferred
	// resolution list
	deferred.insert(deferred.end(), fork.deferred.begin(), fork.deferred.end());
	awaitingResolution.insert(fork.awaitingResolution.begin(),
	                          fork.awaitingResolution.end());
	return true;
}

void ParserScope::push(Handle<Node> node)
{
	const size_t currentDepth = nodes.size();
	if (currentDepth == topLevelDepth) {
		topLevelNodes.push_back(node);
	}
	nodes.push_back(node);

#if SCOPE_DEBUG_OUTPUT
	std::cout << "SCOPE: pushed " << node->type()->name << std::endl;
#endif
}

void ParserScope::pop(Logger &logger)
{
	// Make sure pop is not called without an element on the stack
	const size_t currentDepth = nodes.size();
	if (currentDepth == topLevelDepth) {
		throw LoggableException{"No element here to end!"};
	}

	// Remove all flags from the stack that were set for higher stack depths.
	size_t newLen = 0;
	for (ssize_t i = flags.size() - 1; i >= 0; i--) {
		if (flags[i].depth < currentDepth) {
			newLen = i + 1;
			break;
		}
	}
	flags.resize(newLen);
	// Whenever a RootNode is popped from the stack, we have to perform deferred
	// resolution and validate the subtree
	Rooted<Node> node = nodes.back();
	if (node->isa(&RttiTypes::RootNode)) {
		// Perform pending resolutions -- do not issue errors now
		performDeferredResolution(logger, true);

		// Perform validation of the subtree.
		node->validate(logger);
	}

	// Remove the element from the stack
	nodes.pop_back();

#if SCOPE_DEBUG_OUTPUT
	std::cout << "SCOPE: popped " << node->type()->name << std::endl;
#endif
}

ManagedVector<Node> ParserScope::getTopLevelNodes() const
{
	return topLevelNodes;
}

void ParserScope::setFlag(ParserFlag flag, bool value)
{
	// Fetch the current stack depth
	const size_t currentDepth = nodes.size();

	// Try to change the value of the flag if it was already set on the same
	// stack depth
	for (auto it = flags.rbegin(); it != flags.rend(); it++) {
		if (it->depth == currentDepth) {
			if (it->flag == flag) {
				it->value = value;
				return;
			}
		} else {
			break;
		}
	}

	// Insert a new element into the flags list
	flags.emplace_back(currentDepth, flag, value);
}

bool ParserScope::getFlag(ParserFlag flag)
{
	for (auto it = flags.crbegin(); it != flags.crend(); it++) {
		if (it->flag == flag) {
			return it->value;
		}
	}
	return false;
}

bool ParserScope::resolve(const Rtti *type,
                          const std::vector<std::string> &path,
                          Handle<Node> owner, Logger &logger,
                          ResolutionImposterCallback imposterCallback,
                          ResolutionResultCallback resultCallback)
{
	if (!resolve(type, path, owner, logger, resultCallback)) {
		resultCallback(imposterCallback(), owner, logger);
		return false;
	}
	return true;
}

bool ParserScope::resolve(const Rtti *type,
                          const std::vector<std::string> &path,
                          Handle<Node> owner, Logger &logger,
                          ResolutionResultCallback resultCallback)
{
	// Try to directly resolve the node
	Rooted<Node> res = ParserScopeBase::resolve(type, path, logger);
	if (res != nullptr && !awaitingResolution.count(res.get())) {
		try {
			resultCallback(res, owner, logger);
		}
		catch (LoggableException ex) {
			logger.log(ex, *owner);
		}
		return true;
	}

	// Mark the owner as "awaitingResolution", preventing it from being returned
	// as resolution result
	if (owner != nullptr) {
		awaitingResolution.insert(owner.get());
	}
	deferred.emplace_back(nodes, path, type, resultCallback, owner);
	return false;
}

bool ParserScope::resolveType(const std::vector<std::string> &path,
                              Handle<Node> owner, Logger &logger,
                              ResolutionResultCallback resultCallback)
{
	// Check whether the given path denotes an array, if yes recursively resolve
	// the inner type and wrap it in an array type (this allows multi
	// dimensional arrays). Perform a similar procedure for reference types,
	// denoted using '@'.
	if (!path.empty()) {
		const std::string &last = path.back();
		const std::string &first = path.front();
		if (last.size() >= 2 && last.substr(last.size() - 2, 2) == "[]") {
			// Type ends with "[]", remove this from the last element in the
			// list
			std::vector<std::string> p = path;
			p.back() = last.substr(0, last.size() - 2);

			// Resolve the rest of the type
			return resolveType(p, owner, logger,
			                   [resultCallback](Handle<Node> resolved,
			                                    Handle<Node> owner,
			                                    Logger &logger) {
				if (resolved != nullptr) {
					Rooted<ArrayType> arr{new ArrayType(resolved.cast<Type>())};
					resultCallback(arr, owner, logger);
				} else {
					resultCallback(nullptr, owner, logger);
				}
			});
		} else if (first.size() >= 1 && first[0] == '@') {
			// If only the '@' is given, create and return an empty reference
			// type.
			if (first.size() == 1 && path.size() == 1) {
				Rooted<ReferenceType> ref{
				    new ReferenceType(owner->getManager(), "@", nullptr)};
				resultCallback(ref, owner, logger);
				return true;
			}

			// Remove the '@', try to resolve the rest as an element of the
			// class "Descriptor"
			std::vector<std::string> p = path;
			p.front() = first.substr(1, first.size() - 1);
			std::string name = "@" + p.back();
			return resolve(
			    &RttiTypes::Descriptor, p, owner, logger,
			    [resultCallback, name](Handle<Node> resolved,
			                           Handle<Node> owner, Logger &logger) {
				    if (resolved != nullptr) {
					    Rooted<ReferenceType> ref{
					        new ReferenceType(owner->getManager(), name,
					                          resolved.cast<Descriptor>())};
					    resultCallback(ref, owner, logger);
				    } else {
					    resultCallback(nullptr, owner, logger);
				    }
				});
		}
	}

	// Requested type is not an array, call the usual resolve function
	return resolve(&RttiTypes::Type, path, owner, logger, resultCallback);
}

bool ParserScope::resolveType(const std::string &name, Handle<Node> owner,
                              Logger &logger,
                              ResolutionResultCallback resultCallback)
{
	return resolveType(Utils::split(name, '.'), owner, logger, resultCallback);
}

bool ParserScope::resolveValue(Variant &data, Handle<Type> type, Logger &logger)
{
	// Run the build function
	return type->build(
	    data, logger,
	    [&logger, this](const Rtti *rttiType,
	                    const std::vector<std::string> &path) mutable {
		    return ParserScopeBase::resolve(rttiType, path, logger);
		});
}

bool ParserScope::resolveTypeWithValue(const std::vector<std::string> &path,
                                       Handle<Node> owner, Variant &value,
                                       Logger &logger,
                                       ResolutionResultCallback resultCallback)
{
	// Fork the parser scope -- constants need to be resolved in the same
	// context as this resolve call
	ParserScope scope = fork();
	Variant *valuePtr = &value;

	return resolveType(
	    path, owner, logger,
	    [=](Handle<Node> resolved, Handle<Node> owner, Logger &logger) mutable {
		    if (resolved != nullptr) {
			    Rooted<Type> type = resolved.cast<Type>();
			    scope.resolveValue(*valuePtr, type, logger);
		    }

		    // Call the result callback with the type
		    resultCallback(resolved, owner, logger);
		});
}

bool ParserScope::resolveTypeWithValue(const std::string &name,
                                       Handle<Node> owner, Variant &value,
                                       Logger &logger,
                                       ResolutionResultCallback resultCallback)
{
	return resolveTypeWithValue(Utils::split(name, '.'), owner, value, logger,
	                            resultCallback);
}

bool ParserScope::performDeferredResolution(Logger &logger, bool postpone)
{
	// Repeat the resolution process as long as something has changed in the
	// last iteration (resolving a node may cause other nodes to be resolvable).
	while (true) {
		// Iterate over all deferred resolution processes,
		bool hasChange = false;
		for (auto it = deferred.begin(); it != deferred.end();) {
			if (it->resolve(awaitingResolution, logger)) {
				if (it->owner != nullptr) {
					awaitingResolution.erase(it->owner.get());
				}
				it = deferred.erase(it);
				hasChange = true;
			} else {
				it++;
			}
		}

		// Abort if nothing has changed in the last iteration
		if (!hasChange) {
			// In a last step, clear the "awaitingResolution" list to allow
			// cyclical dependencies to be resolved -- if the postpone flag
			// is set, do not do this
			if (!awaitingResolution.empty() && !postpone) {
				awaitingResolution.clear();
			} else {
				break;
			}
		}
	}

	// We were successful if there are no more deferred resolutions
	if (deferred.empty()) {
		return true;
	}

	// If the postpone flag is set to true, we'll abort here -- this function
	// will be called again later
	if (postpone) {
		return false;
	}

	// Output error messages for all elements for which resolution did not
	// succeed.
	for (auto &failed : deferred) {
		failed.fail(logger);
		logger.error(std::string("Could not resolve ") + failed.type->name +
		                 std::string(" \"") + Utils::join(failed.path, ".") +
		                 std::string("\""),
		             *failed.owner);
	}
	deferred.clear();
	return false;
}

bool ParserScope::resolveFieldDescriptor(
    const std::vector<std::string> &path, Handle<Node> owner, Logger &logger,
    ResolutionResultCallback resultCallback)
{
	// if the last element of the path is the default field name, we want to
	// resolve the parent descriptor first.
	if (path.back() == DEFAULT_FIELD_NAME) {
		std::vector<std::string> descPath;
		descPath.insert(descPath.end(), path.begin(), path.end() - 1);
		return resolve(&RttiTypes::Descriptor, descPath, owner, logger,
		               [resultCallback](Handle<Node> resolved,
		                                Handle<Node> owner, Logger &logger) {
			if (resolved != nullptr) {
				resultCallback(
				    resolved.cast<Descriptor>()->getFieldDescriptor(), owner,
				    logger);
			} else {
				resultCallback(nullptr, owner, logger);
			}
		});
	}

	// If it is not the default field, we can just forward to resolve
	return resolve(&RttiTypes::FieldDescriptor, path, owner, logger,
	               resultCallback);
}

bool ParserScope::resolveFieldDescriptor(
    const std::string &name, Handle<Node> owner, Logger &logger,
    ResolutionResultCallback resultCallback)
{
	return resolveFieldDescriptor(Utils::split(name, '.'), owner, logger,
	                              resultCallback);
}
}