#ifndef _EVENT_EMITTER_
#define _EVENT_EMITTER_


#include <addon-tools.hpp>

#include <map>
#include <deque>
#include <iostream>


#define THIS_EMITTER                                                          \
	EventEmitter *emitter = ObjectWrap::Unwrap<EventEmitter>(info.This());


// Use the dummy template for static-int initialization

template< class Dummy >
class EventEmitterStatics {
protected:
	static int _defaultMaxListeners;
};

template< class Dummy >
int EventEmitterStatics<Dummy>::_defaultMaxListeners = 10;


class EventEmitter : public Nan::ObjectWrap, public EventEmitterStatics<void> {
	
	typedef Nan::CopyablePersistentTraits<v8::Function>::CopyablePersistent FN_TYPE;
	typedef std::deque<FN_TYPE> VEC_TYPE;
	typedef std::map<std::string, VEC_TYPE> MAP_TYPE;
	typedef std::map<int, FN_TYPE> FNMAP_TYPE;
	typedef VEC_TYPE::iterator IT_TYPE;
	typedef MAP_TYPE::iterator MAP_IT_TYPE;
	typedef FNMAP_TYPE::iterator FNMAP_IT_TYPE;
	
	
public:
	
	// C++ side emit() method
	void emit(const std::string &name, int argc = 0, v8::Local<v8::Value> *argv = NULL) {
		
		VEC_TYPE list = _listeners[name];
		
		if (list.empty()) {
			return;
		}
		
		for (IT_TYPE it = list.begin(); it != list.end(); ++it) {
			
			Nan::Callback callback(Nan::New(*it));
			
			if ( ! callback.IsEmpty() ) {
				callback.Call(argc, argv);
			}
			
		}
		
	}
	
	// C++ side on() method
	void on(const std::string &name, v8::Local<v8::Value> that, const std::string &method) {
		
		v8::Local<v8::String> code = JS_STR(
			"((emitter, name, that, method) => emitter.on(name, that[method]))"
		);
		
		v8::Local<v8::Function> connector = v8::Local<v8::Function>::Cast(v8::Script::Compile(code)->Run());
		Nan::Callback connectorCb(connector);
		
		v8::Local<v8::Object> emitter = Nan::New<v8::Object>();
		this->Wrap(emitter);
		
		v8::Local<v8::Value> argv[] = { emitter, JS_STR(name.c_str()), that, JS_STR(method.c_str()) };
		connectorCb.Call(4, argv);
		
	}
	
	
protected:
	
	EventEmitter () {
		_maxListeners = _defaultMaxListeners;
		_freeId = 0;
	}
	~EventEmitter () {}
	
	
	static void extend(v8::Local<v8::FunctionTemplate> &ctor) {
		
		// -------- dynamic
		
		Nan::SetPrototypeMethod(ctor, "listenerCount", jsListenerCount);
		Nan::SetPrototypeMethod(ctor, "addListener", jsAddListener);
		Nan::SetPrototypeMethod(ctor, "emit", jsEmit);
		Nan::SetPrototypeMethod(ctor, "eventNames", jsEventNames);
		Nan::SetPrototypeMethod(ctor, "getMaxListeners", jsGetMaxListeners);
		Nan::SetPrototypeMethod(ctor, "listeners", jsListeners);
		Nan::SetPrototypeMethod(ctor, "on", jsOn);
		Nan::SetPrototypeMethod(ctor, "once", jsOnce);
		Nan::SetPrototypeMethod(ctor, "prependListener", jsPrependListener);
		Nan::SetPrototypeMethod(ctor, "prependOnceListener", jsPrependOnceListener);
		Nan::SetPrototypeMethod(ctor, "removeAllListeners", jsRemoveAllListeners);
		Nan::SetPrototypeMethod(ctor, "removeListener", jsRemoveListener);
		Nan::SetPrototypeMethod(ctor, "setMaxListeners", jsSetMaxListeners);
		Nan::SetPrototypeMethod(ctor, "rawListeners", jsRawListeners);
		
		
		// -------- static
		
		v8::Local<v8::Function> ctorFunc = Nan::GetFunction(ctor).ToLocalChecked();
		v8::Local<v8::Object> ctorObj = ctorFunc;
		
		Nan::SetMethod(ctorFunc, "listenerCount", jsStaticListenerCount);
		ACCESSOR_RW(ctorObj, defaultMaxListeners);
		
		//Local<ObjectTemplate> proto = ctor->PrototypeTemplate();
		// ACCESSOR_RW(proto, type);
		
	}
	
	
	// Deprecated static method
	static NAN_METHOD(jsStaticListenerCount) {
		
		REQ_OBJ_ARG(0, obj);
		EventEmitter *emitter = ObjectWrap::Unwrap<EventEmitter>(obj);
		REQ_UTF8_ARG(1, name);
		
		const VEC_TYPE &list = emitter->_listeners[*name];
		
		RET_VALUE(JS_INT(static_cast<int>(list.size())));
		
	}
	
	
	static NAN_SETTER(defaultMaxListenersSetter) { THIS_EMITTER; SETTER_INT32_ARG;
		
		EventEmitter::_defaultMaxListeners = v;
		
	}
	
	static NAN_GETTER(defaultMaxListenersGetter) { THIS_EMITTER;
		
		RET_VALUE(JS_INT(EventEmitter::_defaultMaxListeners));
		
	}
	
	
	static NAN_METHOD(jsAddListener) { THIS_EMITTER;
		
		REQ_UTF8_ARG(0, name);
		REQ_FUN_ARG(1, cb);
		
		Nan::Persistent<v8::Function> persistentCb;
		persistentCb.Reset(cb);
		
		emitter->_listeners[std::string(*name)].push_back(persistentCb);
		
	}
	
	
	static NAN_METHOD(jsEmit) { THIS_EMITTER;
		std::cout << "emit 0 " << emitter->_listeners.size() << std::endl;
		for (MAP_IT_TYPE it = emitter->_listeners.begin(); it != emitter->_listeners.end(); ++it) {
			
			std::cout << "emit 0 _listeners " << it->first << std::endl;
			
		}
		for (MAP_IT_TYPE it = emitter->_raw.begin(); it != emitter->_raw.end(); ++it) {
			
			std::cout << "emit 0 _raw " << it->first << std::endl;
			
		}
		REQ_UTF8_ARG(0, name);
		
		int length = info.Length();
		
		std::vector< v8::Local<v8::Value> > args;
		std::cout << "emit 1 " << emitter->_listeners.size() << std::endl;
		for (int i = 1; i < length; i++) {
			args.push_back(info[i]);
		}
		std::cout << "emit 2 " << emitter->_listeners.size() << std::endl;
		emitter->emit(*name, length - 1, &args[0]);
		std::cout << "emit 3 " << emitter->_listeners.size() << std::endl;
	}
	
	
	static NAN_METHOD(jsEventNames) { THIS_EMITTER;
		std::cout << "jsEventNames 0 " << emitter->_raw.size() << std::endl;
		v8::Local<v8::Array> jsNames = Nan::New<v8::Array>(emitter->_raw.size());
		
		if (emitter->_raw.empty()) {
			RET_VALUE(jsNames);
			return;
		}
		std::cout << "jsEventNames 1 " << emitter->_raw.size() << std::endl;
		int i = 0;
		for (MAP_IT_TYPE it = emitter->_raw.begin(); it != emitter->_raw.end(); ++it, i++) {
			
			jsNames->Set(JS_INT(i), JS_STR(it->first));
			
		}
		std::cout << "jsEventNames 2 " << emitter->_raw.size() << std::endl;
		RET_VALUE(jsNames);
		
	}
	
	
	static NAN_METHOD(jsGetMaxListeners) { THIS_EMITTER;
		
		RET_VALUE(JS_INT(emitter->_maxListeners));
		
	}
	
	
	static NAN_METHOD(jsListenerCount) { THIS_EMITTER;
		
		REQ_UTF8_ARG(0, name);
		
		const VEC_TYPE &list = emitter->_listeners[*name];
		
		RET_VALUE(JS_INT(static_cast<int>(list.size())));
		
	}
	
	
	static NAN_METHOD(jsListeners) { THIS_EMITTER;
		
		REQ_UTF8_ARG(0, name);
		
		VEC_TYPE &list = emitter->_listeners[*name];
		
		v8::Local<v8::Array> jsListeners = Nan::New<v8::Array>(list.size());
		
		if (list.empty()) {
			RET_VALUE(jsListeners);
			return;
		}
		
		int i = 0;
		for (IT_TYPE it = list.begin(); it != list.end(); ++it, i++) {
			
			jsListeners->Set(JS_INT(i), Nan::New(*it));
			
		}
		
		RET_VALUE(jsListeners);
		
	}
	
	
	static inline void _addListener(
		const Nan::FunctionCallbackInfo<v8::Value> &info,
		const std::string &name,
		Nan::Persistent<v8::Function> &cb,
		bool isFront
	) { THIS_EMITTER;
		
		v8::Local<v8::Value> args[] = { info[0], info[1] };
		emitter->emit("newListener", 2, args);
		
		if (isFront) {
			emitter->_listeners[name].push_front(cb);
			emitter->_raw[name].push_front(cb);
		} else {
			emitter->_listeners[name].push_back(cb);
			emitter->_raw[name].push_back(cb);
		}
		
	}
	
	
	static inline void _wrapListener(
		const Nan::FunctionCallbackInfo<v8::Value> &info,
		bool isFront = false
	) {
		
		REQ_UTF8_ARG(0, name);
		REQ_FUN_ARG(1, cb);
		
		Nan::Persistent<v8::Function> persistentCb;
		persistentCb.Reset(cb);
		
		_addListener(info, *name, persistentCb, isFront);
		
	}
	
	
	static inline void _addOnceListener(
		const Nan::FunctionCallbackInfo<v8::Value> &info,
		const std::string &name,
		Nan::Persistent<v8::Function> &raw,
		Nan::Persistent<v8::Function> &cb,
		bool isFront
	) { THIS_EMITTER;
		
		v8::Local<v8::Value> args[] = { info[0], info[1] };
		emitter->emit("newListener", 2, args);
		
		if (isFront) {
			emitter->_listeners[name].push_front(cb);
			emitter->_raw[name].push_front(raw);
		} else {
			emitter->_listeners[name].push_back(cb);
			emitter->_raw[name].push_back(raw);
		}
		
		int nextId = emitter->_freeId++;
		emitter->_wrappedIds[nextId] = cb;
		emitter->_rawIds[nextId] = raw;
		
	}
	
	
	static inline void _wrapOnceListener(
		const Nan::FunctionCallbackInfo<v8::Value> &info,
		bool isFront = false
	) {
		
		REQ_UTF8_ARG(0, name);
		REQ_FUN_ARG(1, raw);
		
		v8::Local<v8::String> code = JS_STR(
			"((emitter, name, cb) => (...args) => {\n\
				cb(...args);\n\
				emitter.removeListener(name, cb);\n\
			})"
		);
		
		v8::Local<v8::Function> decor = v8::Local<v8::Function>::Cast(v8::Script::Compile(code)->Run());
		Nan::Callback decorCb(decor);
		v8::Local<v8::Value> argv[] = { info.This(), info[0], raw };
		v8::Local<v8::Function> wrap = v8::Local<v8::Function>::Cast(decorCb.Call(3, argv));
		
		Nan::Persistent<v8::Function> persistentWrap;
		persistentWrap.Reset(wrap);
		
		Nan::Persistent<v8::Function> persistentRaw;
		persistentRaw.Reset(raw);
		
		_addOnceListener(info, *name, persistentRaw, persistentWrap, isFront);
		
	}
	
	
	static NAN_METHOD(jsOn) { _wrapListener(info); }
	
	static NAN_METHOD(jsOnce) { _wrapOnceListener(info); }
	
	static NAN_METHOD(jsPrependListener) { _wrapListener(info, true); }
	
	static NAN_METHOD(jsPrependOnceListener) { _wrapOnceListener(info, true); }
	
	
	static NAN_METHOD(jsRemoveAllListeners) { THIS_EMITTER;
		
		if (info.Length() > 0 && info[0]->IsString()) {
			
			MAP_TYPE tmpMap = emitter->_raw;
			
			emitter->_listeners.clear();
			emitter->_raw.clear();
			emitter->_wrappedIds.clear();
			emitter->_rawIds.clear();
			
			for (MAP_IT_TYPE itMap = tmpMap.begin(); itMap != tmpMap.end(); ++itMap) {
				
				const std::string &current = itMap->first;
				VEC_TYPE &list = itMap->second;
				
				for (IT_TYPE it = list.begin(); it != list.end(); ++it) {
					
					v8::Local<v8::Value> args[] = { JS_STR(current.c_str()), Nan::New(*it) };
					emitter->emit("removeListener", 2, args);
					
				}
				
			}
			
			return;
			
		}
		
		REQ_UTF8_ARG(0, n);
		
		std::string name = std::string(*n);
		VEC_TYPE &list = emitter->_raw[name];
		
		if (list.empty()) {
			return;
		}
		
		if (emitter->_rawIds.size()) {
			
			std::vector<int> removes;
			
			for (IT_TYPE it = list.begin(); it != list.end(); ++it) {
				
				FN_TYPE fn = *it;
				
				for (FNMAP_IT_TYPE itRaw = emitter->_rawIds.begin(); itRaw != emitter->_rawIds.end(); ++itRaw) {
					if (fn == itRaw->second) {
						removes.push_back(itRaw->first);
					}
				}
				
			}
			
			if (removes.size()) {
				for (std::vector<int>::const_iterator it = removes.begin(); it != removes.end(); ++it) {
					
					emitter->_wrappedIds.erase(*it);
					emitter->_rawIds.erase(*it);
					
				}
			}
			
		}
		
		
		VEC_TYPE tmpVec = emitter->_raw[name];
		
		emitter->_listeners[name].clear();
		emitter->_raw[name].clear();
		
		for (IT_TYPE it = tmpVec.begin(); it != tmpVec.end(); ++it) {
			
			v8::Local<v8::Value> args[] = { JS_STR(name.c_str()), Nan::New(*it) };
			emitter->emit("removeListener", 2, args);
			
		}
		
	}
	
	
	static NAN_METHOD(jsRemoveListener) { THIS_EMITTER;
		
		REQ_UTF8_ARG(0, n);
		REQ_FUN_ARG(1, raw);
		
		Nan::Persistent<v8::Function> persistentRaw;
		persistentRaw.Reset(raw);
		
		std::string name = std::string(*n);
		
		VEC_TYPE &rawList = emitter->_raw[name];
		
		if (rawList.empty()) {
			return;
		}
		
		v8::Local<v8::Value> args[] = { info[0], info[1] };
		
		for (IT_TYPE it = rawList.begin(); it != rawList.end(); ++it) {
			
			if (*it == persistentRaw) {
				rawList.erase(it);
				if (rawList.empty()) {
					emitter->_raw.erase(name);
				}
				break;
			}
			
		}
		
		
		VEC_TYPE &wrapList = emitter->_listeners[name];
		
		if (emitter->_wrappedIds.size() == 0) {
			
			for (IT_TYPE it = wrapList.begin(); it != wrapList.end(); ++it) {
				
				if (*it == persistentRaw) {
					wrapList.erase(it);
					if (wrapList.empty()) {
						emitter->_listeners.erase(name);
					}
					break;
				}
				
			}
			
			emitter->emit("removeListener", 2, args);
			return;
			
		}
		
		
		for (FNMAP_IT_TYPE itRaw = emitter->_rawIds.begin(); itRaw != emitter->_rawIds.end(); ++itRaw) {
			
			if (persistentRaw == itRaw->second) {
				
				FN_TYPE fn = emitter->_wrappedIds[itRaw->first];
				
				for (IT_TYPE it = wrapList.begin(); it != wrapList.end(); ++it) {
					
					if (*it == fn) {
						wrapList.erase(it);
						if (wrapList.empty()) {
							emitter->_listeners.erase(name);
						}
						break;
					}
					
				}
				
				emitter->_wrappedIds.erase(itRaw->first);
				emitter->_rawIds.erase(itRaw->first);
				
				break;
				
			}
			
		}
		
		emitter->emit("removeListener", 2, args);
		
	}
	
	
	static NAN_METHOD(jsSetMaxListeners) { THIS_EMITTER;
		
		REQ_INT32_ARG(0, value);
		
		emitter->_maxListeners = value;
		
	}
	
	
	static NAN_METHOD(jsRawListeners) { THIS_EMITTER;
		
		REQ_UTF8_ARG(0, name);
		
		VEC_TYPE &list = emitter->_raw[*name];
		
		v8::Local<v8::Array> jsListeners = Nan::New<v8::Array>(list.size());
		
		if (list.empty()) {
			RET_VALUE(jsListeners);
			return;
		}
		
		int i = 0;
		for (IT_TYPE it = list.begin(); it != list.end(); ++it, i++) {
			
			jsListeners->Set(JS_INT(i), Nan::New(*it));
			
		}
		
		RET_VALUE(jsListeners);
		
	}
	
	
private:
	
	int _maxListeners;
	
	MAP_TYPE _listeners;
	MAP_TYPE _raw;
	
	int _freeId;
	FNMAP_TYPE _wrappedIds;
	FNMAP_TYPE _rawIds;
	
};


#endif // _EVENT_EMITTER_