WebNotificationCenter.cpp [plain text]
#include "WebKitDLL.h"
#include "WebNotificationCenter.h"
#include "WebNotification.h"
#include <WebCore/BString.h>
#include <WebCore/COMPtr.h>
#include <utility>
#include <wchar.h>
#include <wtf/HashMap.h>
#include <wtf/HashTraits.h>
#include <wtf/Vector.h>
#include <wtf/text/StringHash.h>
#include <wtf/text/WTFString.h>
using namespace WebCore;
typedef std::pair<COMPtr<IUnknown>, COMPtr<IWebNotificationObserver> > ObjectObserverPair;
typedef Vector<ObjectObserverPair> ObjectObserverList;
typedef ObjectObserverList::iterator ObserverListIterator;
typedef HashMap<String, ObjectObserverList> MappedObservers;
struct WebNotificationCenterPrivate {
MappedObservers m_mappedObservers;
};
IWebNotificationCenter* WebNotificationCenter::m_defaultCenter = 0;
WebNotificationCenter::WebNotificationCenter()
: m_refCount(0)
, d(std::make_unique<WebNotificationCenterPrivate>())
{
gClassCount++;
gClassNameCount().add("WebNotificationCenter");
}
WebNotificationCenter::~WebNotificationCenter()
{
gClassCount--;
gClassNameCount().remove("WebNotificationCenter");
}
WebNotificationCenter* WebNotificationCenter::createInstance()
{
WebNotificationCenter* instance = new WebNotificationCenter();
instance->AddRef();
return instance;
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IWebNotificationCenter*>(this);
else if (IsEqualGUID(riid, IID_IWebNotificationCenter))
*ppvObject = static_cast<IWebNotificationCenter*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG STDMETHODCALLTYPE WebNotificationCenter::AddRef(void)
{
return ++m_refCount;
}
ULONG STDMETHODCALLTYPE WebNotificationCenter::Release(void)
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
IWebNotificationCenter* WebNotificationCenter::defaultCenterInternal()
{
if (!m_defaultCenter)
m_defaultCenter = WebNotificationCenter::createInstance();
return m_defaultCenter;
}
void WebNotificationCenter::postNotificationInternal(IWebNotification* notification, BSTR notificationName, IUnknown* anObject)
{
String name(notificationName, SysStringLen(notificationName));
MappedObservers::iterator it = d->m_mappedObservers.find(name);
if (it == d->m_mappedObservers.end())
return;
ObjectObserverList list = it->value;
ObserverListIterator end = list.end();
for (ObserverListIterator it2 = list.begin(); it2 != end; ++it2) {
IUnknown* observedObject = it2->first.get();
IWebNotificationObserver* observer = it2->second.get();
if (!observedObject || !anObject || observedObject == anObject)
observer->onNotify(notification);
}
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::defaultCenter(
IWebNotificationCenter** center)
{
*center = defaultCenterInternal();
(*center)->AddRef();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::addObserver(
IWebNotificationObserver* observer,
BSTR notificationName,
IUnknown* anObject)
{
String name(notificationName, SysStringLen(notificationName));
MappedObservers::iterator it = d->m_mappedObservers.find(name);
if (it != d->m_mappedObservers.end())
it->value.append(ObjectObserverPair(anObject, observer));
else {
ObjectObserverList list;
list.append(ObjectObserverPair(anObject, observer));
d->m_mappedObservers.add(name, list);
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::postNotification(
IWebNotification* notification)
{
BString name;
HRESULT hr = notification->name(&name);
if (FAILED(hr))
return hr;
COMPtr<IUnknown> obj;
hr = notification->getObject(&obj);
if (FAILED(hr))
return hr;
postNotificationInternal(notification, name, obj.get());
return hr;
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::postNotificationName(
BSTR notificationName,
IUnknown* anObject,
IPropertyBag* userInfo)
{
COMPtr<WebNotification> notification(AdoptCOM, WebNotification::createInstance(notificationName, anObject, userInfo));
postNotificationInternal(notification.get(), notificationName, anObject);
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebNotificationCenter::removeObserver(
IWebNotificationObserver* anObserver,
BSTR notificationName,
IUnknown* anObject)
{
String name(notificationName, SysStringLen(notificationName));
MappedObservers::iterator it = d->m_mappedObservers.find(name);
if (it == d->m_mappedObservers.end())
return E_FAIL;
ObjectObserverList& observerList = it->value;
ObserverListIterator end = observerList.end();
observerList.removeFirstMatching([anObject, anObserver] (const ObjectObserverPair& pair) {
IUnknown* observedObject = pair.first.get();
IWebNotificationObserver* observer = pair.second.get();
return observer == anObserver && (!anObject || anObject == observedObject);
});
if (observerList.isEmpty())
d->m_mappedObservers.remove(name);
return S_OK;
}