#include "angelscript.h"
#include "scriptstdstring.h"
#include "scripthelper.h"
#include <stdio.h>
#include <iostream>
#include <stdexcept>
using namespace std;

class EventSource
{
public:
	EventSource() {refCount = 1; value = 42;};
	virtual ~EventSource() {}
	virtual void AddRef() {refCount++;}
	virtual void Release() {if( --refCount == 0 ) delete this;}
	int refCount;

	int value;
};

class ASConsole : public EventSource
{
public:
	static ASConsole *factory() { return new ASConsole(); }
};

template<class A, class B> B* ASRefCast(A* a)
{
	// If the handle already is a null handle, then just return the null handle
	if (a==NULL) return NULL;
	// Now try to dynamically cast the pointer to the wanted type
	B* b = dynamic_cast<B*>(a);
	if (b!=NULL) {
		// Since the cast was made, we need to increase the ref counter for the returned handle
		b->AddRef();
	}
	cout << "ASRefCast: returning " << hex << b << endl;
	return b;
}

void Assert(bool condition)
{
	if (!condition) {
		int line;
		const char* section;
		asGetActiveContext()->GetLineNumber(0, &line, &section);
		cerr << "Assertion failed; Section " << section << " line " << line << cout;
	}
}


void MessageCallback(const asSMessageInfo *msg, void *param)
{
  const char *type = "ERR ";
  if( msg->type == asMSGTYPE_WARNING )
    type = "WARN";
  else if( msg->type == asMSGTYPE_INFORMATION )
    type = "INFO";
  printf("%s (%d, %d) : %s : %s\n", msg->section, msg->row, msg->col, type, msg->message);
}

void addListener(EventSource* source, int mask) {
	cout << "addListener: source = " << hex << source << endl;
	cout << "addListener: source.value = " << (int)source->value << endl;
	cout << endl;
	cout << endl;
}

bool Test()
{
	bool fail = false;
	int r;
	asIScriptEngine *engine;

	// http://www.gamedev.n...sting-directly/
	{

		ASConsole* c = new ASConsole();

		engine = asCreateScriptEngine(ANGELSCRIPT_VERSION);
		engine->SetMessageCallback(asFUNCTION(MessageCallback), 0, asCALL_CDECL);
		engine->RegisterGlobalFunction("void assert(bool)", asFUNCTION(Assert), asCALL_CDECL);
		RegisterStdString(engine);

		engine->RegisterObjectType("EventSource", 0, asOBJ_REF);
		engine->RegisterObjectBehaviour("EventSource", asBEHAVE_ADDREF, "void f()", asMETHOD(EventSource, AddRef), asCALL_THISCALL);
		engine->RegisterObjectBehaviour("EventSource", asBEHAVE_RELEASE, "void f()", asMETHOD(EventSource, Release), asCALL_THISCALL);
		engine->RegisterObjectProperty("EventSource", "int value", asOFFSET(EventSource, value));

		engine->RegisterObjectType("ASConsole", 0, asOBJ_REF);
		engine->RegisterObjectBehaviour("ASConsole", asBEHAVE_FACTORY, "ASConsole @f()", asFUNCTION(ASConsole::factory), asCALL_CDECL);
		engine->RegisterObjectBehaviour("ASConsole", asBEHAVE_ADDREF, "void f()", asMETHOD(ASConsole, AddRef), asCALL_THISCALL);
		engine->RegisterObjectBehaviour("ASConsole", asBEHAVE_RELEASE, "void f()", asMETHOD(ASConsole, Release), asCALL_THISCALL);
		engine->RegisterObjectBehaviour("ASConsole", asBEHAVE_IMPLICIT_REF_CAST, "EventSource@ f()",
				asFUNCTION((ASRefCast<ASConsole, EventSource>)), asCALL_CDECL_OBJLAST);

		engine->RegisterGlobalFunction("void addListener(EventSource &inout, const int &in)", asFUNCTION(addListener), asCALL_CDECL);

		asIScriptModule *mod = engine->GetModule("test", asGM_ALWAYS_CREATE);
		mod->AddScriptSection("test",
			"enum E { ET_READLINE = 24 } \n"
			"ASConsole @console;\n"
			"void main() \n"
			"{ \n"
			"  EventSource@ x = cast<EventSource>(console); \n"
			"  addListener(x, ET_READLINE); \n"
			"  addListener(cast<EventSource>(console), ET_READLINE); \n"
			"} \n");
		r = mod->Build();

		void** consoleVarAddr = (void**)mod->GetAddressOfGlobalVar(mod->GetGlobalVarIndexByName("console"));
		*consoleVarAddr = c;

		if( r < 0 )
			throw runtime_error("Compilation failed");

		r = ExecuteString(engine, "main()", mod);
		if( r != asEXECUTION_FINISHED )
			throw runtime_error("Execution failed");

		engine->Release();
	}

 	return fail;
}


int main(int argc, char** argv) {
	Test();
}