#include "data/parser/java/JavaParser.h"

#include <jni.h>

#include "data/parser/java/JavaEnvironmentFactory.h"
#include "data/parser/ParseLocation.h"
#include "data/parser/ReferenceKind.h"
#include "data/parser/ParserClient.h"
#include "settings/ApplicationSettings.h"
#include "utility/file/FileSystem.h"
#include "utility/text/TextAccess.h"
#include "utility/utilityString.h"

JavaParser::JavaParser(ParserClient* client)
	: Parser(client)
	, m_id(s_nextParserId++)
	, m_currentFilePath("")
{
	std::shared_ptr<JavaEnvironmentFactory> factory = JavaEnvironmentFactory::getInstance();
	if (factory)
	{
		m_javaEnvironment = factory->createEnvironment();

		std::vector<JavaEnvironment::NativeMethod> methods;

		methods.push_back({"logInfo", "(ILjava/lang/String;)V", (void*)&JavaParser::LogInfo});
		methods.push_back({"logWarning", "(ILjava/lang/String;)V", (void*)&JavaParser::LogWarning});
		methods.push_back({"logError", "(ILjava/lang/String;)V", (void*)&JavaParser::LogError});
		methods.push_back({"recordSymbol", "(ILjava/lang/String;III)V", (void*)&JavaParser::RecordSymbol});
		methods.push_back({"recordSymbolWithLocation", "(ILjava/lang/String;IIIIIII)V", (void*)&JavaParser::RecordSymbolWithLocation});
		methods.push_back({"recordSymbolWithLocationAndScope", "(ILjava/lang/String;IIIIIIIIIII)V", (void*)&JavaParser::RecordSymbolWithLocationAndScope});
		methods.push_back({"recordReference", "(IILjava/lang/String;Ljava/lang/String;IIII)V", (void*)&JavaParser::RecordReference});
		methods.push_back({"recordLocalSymbol", "(ILjava/lang/String;IIII)V", (void*)&JavaParser::RecordLocalSymbol});
		methods.push_back({"recordComment", "(IIIII)V", (void*)&JavaParser::RecordComment});
		methods.push_back({"recordError", "(ILjava/lang/String;IIIIII)V", (void*)&JavaParser::RecordError});

		m_javaEnvironment->registerNativeMethods("io/coati/JavaIndexer", methods);
	}
	{
		std::lock_guard<std::mutex> lock(s_parsersMutex);
		s_parsers[m_id] = this;
	}
}

JavaParser::~JavaParser()
{
	s_parsers.erase(m_id);
}

void JavaParser::parseFiles(const std::vector<FilePath>& filePaths, const Arguments& arguments)
{
	//m_fileRegister->setFilePaths(filePaths);
	//setupParsing(arguments);

	//std::vector<std::string> sourcePaths;
	//for (const FilePath& path : m_fileRegister->getUnparsedSourceFilePaths()) // filter headers
	//{
	//	sourcePaths.push_back(path.absolute().str());
	//}

	//runTool(sourcePaths);
}

void JavaParser::parseFile(const FilePath& filePath, std::shared_ptr<TextAccess> textAccess, const Arguments& arguments)
{
	if (m_javaEnvironment)
	{
		m_currentFilePath = filePath.str();
		m_client->onFileParsed(FileSystem::getFileInfoForPath(filePath));
		std::string classPath = "";
		for (const FilePath& path: arguments.javaClassPaths)
		{
			// the separator used here should be the same as the one used in JavaIndexer.java
			classPath += path.str() + ";";
		}

		// remove tabs because they screw with javaparser's location resolver
		std::string fileContent = utility::replace(textAccess->getText(), "\t", " ");

		m_javaEnvironment->callStaticVoidMethod(
			"io/coati/JavaIndexer",
			"processFile",
			m_id,
			filePath.str(),
			fileContent,
			classPath,
			ApplicationSettings::getInstance()->getVerboseIndexerLoggingEnabled() ? 1 : 0
		);
	}
}








int JavaParser::s_nextParserId = 0;

std::map<int, JavaParser*> JavaParser::s_parsers;

std::mutex JavaParser::s_parsersMutex;










// definition of native methods

void JavaParser::doLogInfo(jstring jInfo)
{
	LOG_INFO_STREAM_BARE(<< "Indexer - " << m_javaEnvironment->toStdString(jInfo));
}

void JavaParser::doLogWarning(jstring jWarning)
{
	LOG_WARNING_STREAM_BARE(<< "Indexer - " << m_javaEnvironment->toStdString(jWarning));
}

void JavaParser::doLogError(jstring jError)
{
	LOG_ERROR_STREAM_BARE(<< "Indexer - " << m_javaEnvironment->toStdString(jError));
}

void JavaParser::doRecordSymbol(
	jstring jSymbolName, jint jSymbolType,
	jint jAccess, jint jIsImplicit
)
{
	AccessKind access = intToAccessKind(jAccess);
	bool isImplicit = jIsImplicit;

	m_client->recordSymbol(
		NameHierarchy::deserialize(m_javaEnvironment->toStdString(jSymbolName)),
		intToSymbolKind(jSymbolType),
		access,
		isImplicit
	);
}

void JavaParser::doRecordSymbolWithLocation(
	jstring jSymbolName, jint jSymbolType,
	jint beginLine, jint beginColumn, jint endLine, jint endColumn,
	jint jAccess, jint jIsImplicit
)
{
	AccessKind access = intToAccessKind(jAccess);
	bool isImplicit = jIsImplicit;

	m_client->recordSymbol(
		NameHierarchy::deserialize(m_javaEnvironment->toStdString(jSymbolName)),
		intToSymbolKind(jSymbolType),
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn),
		access,
		isImplicit
	);
}

void JavaParser::doRecordSymbolWithLocationAndScope(
	jstring jSymbolName, jint jSymbolType,
	jint beginLine, jint beginColumn, jint endLine, jint endColumn,
	jint scopeBeginLine, jint scopeBeginColumn, jint scopeEndLine, jint scopeEndColumn,
	jint jAccess, jint jIsImplicit
)
{
	AccessKind access = intToAccessKind(jAccess);
	bool isImplicit = jIsImplicit;

	m_client->recordSymbol(
		NameHierarchy::deserialize(m_javaEnvironment->toStdString(jSymbolName)),
		intToSymbolKind(jSymbolType),
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn),
		ParseLocation(m_currentFilePath, scopeBeginLine, scopeBeginColumn, scopeEndLine, scopeEndColumn),
		access,
		isImplicit
	);
}

void JavaParser::doRecordReference(
	jint jReferenceKind, jstring jReferencedName, jstring jContextName,
	jint beginLine, jint beginColumn, jint endLine, jint endColumn
)
{
	m_client->recordReference(
		intToReferenceKind(jReferenceKind),
		NameHierarchy::deserialize(m_javaEnvironment->toStdString(jReferencedName)),
		NameHierarchy::deserialize(m_javaEnvironment->toStdString(jContextName)),
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn)
	);
}

void JavaParser::doRecordLocalSymbol(jstring jSymbolName, jint beginLine, jint beginColumn, jint endLine, jint endColumn)
{
	m_client->onLocalSymbolParsed(
		m_javaEnvironment->toStdString(jSymbolName),
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn)
	);
}

void JavaParser::doRecordComment(
	jint beginLine, jint beginColumn, jint endLine, jint endColumn
)
{
	m_client->onCommentParsed(
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn)
	);
}

void JavaParser::doRecordError(
	jstring jMessage, jint jFatal, jint jIndexed,
	jint beginLine, jint beginColumn, jint endLine, jint endColumn
)
{
	bool fatal = jFatal;
	bool indexed = jIndexed;

	m_client->onError(
		ParseLocation(m_currentFilePath, beginLine, beginColumn, endLine, endColumn),
		m_javaEnvironment->toStdString(jMessage),
		fatal, indexed
	);
}