#ifndef V8DOMMap_h
#define V8DOMMap_h
#include <wtf/HashMap.h>
#include <wtf/OwnPtr.h>
#include <v8.h>
namespace WebCore {
class Node;
#if ENABLE(SVG)
class SVGElementInstance;
#endif
template<class KeyType, class ValueType> class WeakReferenceMap {
public:
WeakReferenceMap(v8::WeakReferenceCallback callback) : m_weakReferenceCallback(callback) { }
#ifndef NDEBUG
virtual ~WeakReferenceMap()
{
if (m_map.size() > 0)
fprintf(stderr, "Leak %d JS wrappers.\n", m_map.size());
}
#endif
virtual v8::Persistent<ValueType> get(KeyType* obj)
{
ValueType* wrapper = m_map.get(obj);
return wrapper ? v8::Persistent<ValueType>(wrapper) : v8::Persistent<ValueType>();
}
virtual void set(KeyType* obj, v8::Persistent<ValueType> wrapper)
{
ASSERT(!m_map.contains(obj));
wrapper.MakeWeak(obj, m_weakReferenceCallback);
m_map.set(obj, *wrapper);
}
virtual void forget(KeyType* obj)
{
ASSERT(obj);
ValueType* wrapper = m_map.take(obj);
if (!wrapper)
return;
v8::Persistent<ValueType> handle(wrapper);
handle.Dispose();
handle.Clear();
}
bool contains(KeyType* obj) { return m_map.contains(obj); }
HashMap<KeyType*, ValueType*>& impl() { return m_map; }
protected:
HashMap<KeyType*, ValueType*> m_map;
v8::WeakReferenceCallback m_weakReferenceCallback;
};
template <class KeyType> class DOMWrapperMap : public WeakReferenceMap<KeyType, v8::Object> {
public:
DOMWrapperMap(v8::WeakReferenceCallback callback) : WeakReferenceMap<KeyType, v8::Object>(callback) { }
class Visitor {
public:
virtual void visitDOMWrapper(KeyType* key, v8::Persistent<v8::Object> object) = 0;
};
};
class DOMDataStore;
class DOMDataStoreHandle {
public:
DOMDataStoreHandle();
~DOMDataStoreHandle();
DOMDataStore* getStore() const { return m_store.get(); }
private:
OwnPtr<DOMDataStore> m_store;
};
void weakActiveDOMObjectCallback(v8::Persistent<v8::Value> v8Object, void* domObject);
DOMWrapperMap<Node>& getDOMNodeMap();
void visitDOMNodesInCurrentThread(DOMWrapperMap<Node>::Visitor*);
DOMWrapperMap<void>& getDOMObjectMap();
void visitDOMObjectsInCurrentThread(DOMWrapperMap<void>::Visitor*);
DOMWrapperMap<void>& getActiveDOMObjectMap();
void visitActiveDOMObjectsInCurrentThread(DOMWrapperMap<void>::Visitor*);
void removeAllDOMObjectsInCurrentThread();
#if ENABLE(SVG)
DOMWrapperMap<SVGElementInstance>& getDOMSVGElementInstanceMap();
void visitSVGElementInstancesInCurrentThread(DOMWrapperMap<SVGElementInstance>::Visitor*);
DOMWrapperMap<void>& getDOMSVGObjectWithContextMap();
void visitDOMSVGObjectsInCurrentThread(DOMWrapperMap<void>::Visitor*);
#endif
}
#endif // V8DOMMap_h