/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * 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 2
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */

//include <stdlib.h>
//include <string.h>
#include "ags/engine/script/systemimports.h"

namespace AGS3 {

extern void quit(const char *);

SystemImports simp;
SystemImports simp_for_plugin;

int SystemImports::add(const String &name, const RuntimeScriptValue &value, ccInstance *anotherscr) {
	int ixof;

	if ((ixof = get_index_of(name)) >= 0) {
		// Only allow override if not a script-exported function
		if (anotherscr == nullptr) {
			imports[ixof].Value = value;
			imports[ixof].InstancePtr = anotherscr;
		}
		return 0;
	}

	ixof = imports.size();
	for (size_t i = 0; i < imports.size(); ++i) {
		if (imports[i].Name == nullptr) {
			ixof = i;
			break;
		}
	}

	btree[name] = ixof;
	if (ixof == (int)imports.size())
		imports.push_back(ScriptImport());
	imports[ixof].Name = name; // TODO: rather make a string copy here for safety reasons
	imports[ixof].Value = value;
	imports[ixof].InstancePtr = anotherscr;
	return 0;
}

void SystemImports::remove(const String &name) {
	int idx = get_index_of(name);
	if (idx < 0)
		return;
	btree.erase(imports[idx].Name);
	imports[idx].Name = nullptr;
	imports[idx].Value.Invalidate();
	imports[idx].InstancePtr = nullptr;
}

const ScriptImport *SystemImports::getByName(const String &name) {
	int o = get_index_of(name);
	if (o < 0)
		return nullptr;

	return &imports[o];
}

const ScriptImport *SystemImports::getByIndex(int index) {
	if ((size_t)index >= imports.size())
		return nullptr;

	return &imports[index];
}

int SystemImports::get_index_of(const String &name) {
	IndexMap::const_iterator it = btree.find(name);
	if (it != btree.end())
		return it->_value;

	// CHECKME: what are "mangled names" and where do they come from?
	String mangled_name = String::FromFormat("%s$", name.GetCStr());
	// if it's a function with a mangled name, allow it
	it = btree.lower_bound(mangled_name);
	if (it != btree.end() && it->_key.CompareLeft(mangled_name) == 0)
		return it->_value;

	if (name.GetLength() > 3) {
		size_t c = name.FindCharReverse('^');
		if (c != String::npos && (c == name.GetLength() - 2 || c == name.GetLength() - 3)) {
			// Function with number of prametrs on the end
			// attempt to find it without the param count
			return get_index_of(name.Left(c));
		}
	}
	return -1;
}

void SystemImports::RemoveScriptExports(ccInstance *inst) {
	if (!inst) {
		return;
	}

	for (size_t i = 0; i < imports.size(); ++i) {
		if (imports[i].Name == nullptr)
			continue;

		if (imports[i].InstancePtr == inst) {
			btree.erase(imports[i].Name);
			imports[i].Name = nullptr;
			imports[i].Value.Invalidate();
			imports[i].InstancePtr = nullptr;
		}
	}
}

void SystemImports::clear() {
	btree.clear();
	imports.clear();
}

} // namespace AGS3