WebIconDatabase.cpp [plain text]
#include "config.h"
#include "WebKitDLL.h"
#include "WebIconDatabase.h"
#include "COMPropertyBag.h"
#include "WebNotificationCenter.h"
#include "WebPreferences.h"
#include "shlobj.h"
#include <WebCore/BString.h>
#include <WebCore/BitmapInfo.h>
#include <WebCore/COMPtr.h>
#include <WebCore/FileSystem.h>
#include <WebCore/HWndDC.h>
#include <WebCore/IconDatabase.h>
#include <WebCore/Image.h>
#include <WebCore/SharedBuffer.h>
#include <wtf/MainThread.h>
#include <wtf/text/WTFString.h>
using namespace WebCore;
using namespace WTF;
WebIconDatabase* WebIconDatabase::m_sharedWebIconDatabase = 0;
WebIconDatabase::WebIconDatabase()
: m_refCount(0)
, m_deliveryRequested(false)
{
gClassCount++;
gClassNameCount.add("WebIconDatabase");
}
WebIconDatabase::~WebIconDatabase()
{
gClassCount--;
gClassNameCount.remove("WebIconDatabase");
}
void WebIconDatabase::init()
{
WebPreferences* standardPrefs = WebPreferences::sharedStandardPreferences();
BOOL enabled = FALSE;
if (FAILED(standardPrefs->iconDatabaseEnabled(&enabled))) {
enabled = FALSE;
LOG_ERROR("Unable to get icon database enabled preference");
}
iconDatabase().setEnabled(!!enabled);
if (!(!!enabled))
return;
startUpIconDatabase();
}
void WebIconDatabase::startUpIconDatabase()
{
WebPreferences* standardPrefs = WebPreferences::sharedStandardPreferences();
iconDatabase().setClient(this);
BString prefDatabasePath;
if (FAILED(standardPrefs->iconDatabaseLocation(&prefDatabasePath)))
LOG_ERROR("Unable to get icon database location preference");
String databasePath(prefDatabasePath, SysStringLen(prefDatabasePath));
if (databasePath.isEmpty()) {
databasePath = localUserSpecificStorageDirectory();
if (databasePath.isEmpty())
LOG_ERROR("Failed to construct default icon database path");
}
if (!iconDatabase().open(databasePath, WebCore::IconDatabase::defaultDatabaseFilename()))
LOG_ERROR("Failed to open icon database path");
}
void WebIconDatabase::shutDownIconDatabase()
{
}
WebIconDatabase* WebIconDatabase::createInstance()
{
WebIconDatabase* instance = new WebIconDatabase();
instance->AddRef();
return instance;
}
WebIconDatabase* WebIconDatabase::sharedWebIconDatabase()
{
if (m_sharedWebIconDatabase) {
m_sharedWebIconDatabase->AddRef();
return m_sharedWebIconDatabase;
}
m_sharedWebIconDatabase = createInstance();
m_sharedWebIconDatabase->init();
return m_sharedWebIconDatabase;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IWebIconDatabase*>(this);
else if (IsEqualGUID(riid, IID_IWebIconDatabase))
*ppvObject = static_cast<IWebIconDatabase*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG STDMETHODCALLTYPE WebIconDatabase::AddRef(void)
{
return ++m_refCount;
}
ULONG STDMETHODCALLTYPE WebIconDatabase::Release(void)
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::sharedIconDatabase(
IWebIconDatabase** result)
{
*result = sharedWebIconDatabase();
return S_OK;
}
HRESULT WebIconDatabase::iconForURL(BSTR url, LPSIZE size, BOOL , HBITMAP* bitmap)
{
if (!size)
return E_POINTER;
IntSize intSize(*size);
Image* icon = 0;
if (url)
icon = iconDatabase().synchronousIconForPageURL(String(url, SysStringLen(url)), intSize);
if (icon && icon->width()) {
*bitmap = getOrCreateSharedBitmap(intSize);
if (!icon->getHBITMAPOfSize(*bitmap, &intSize)) {
LOG_ERROR("Failed to draw Image to HBITMAP");
*bitmap = 0;
return E_FAIL;
}
return S_OK;
}
return defaultIconWithSize(size, bitmap);
}
HRESULT WebIconDatabase::defaultIconWithSize(LPSIZE size, HBITMAP* result)
{
if (!size)
return E_POINTER;
IntSize intSize(*size);
*result = getOrCreateDefaultIconBitmap(intSize);
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::retainIconForURL(
BSTR url)
{
iconDatabase().retainIconForPageURL(String(url, SysStringLen(url)));
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::releaseIconForURL(
BSTR url)
{
iconDatabase().releaseIconForPageURL(String(url, SysStringLen(url)));
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::removeAllIcons(void)
{
iconDatabase().removeAllIcons();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::delayDatabaseCleanup(void)
{
IconDatabase::delayDatabaseCleanup();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::allowDatabaseCleanup(void)
{
IconDatabase::allowDatabaseCleanup();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::iconURLForURL(
BSTR url,
BSTR* iconURL)
{
if (!url || !iconURL)
return E_POINTER;
BString iconURLBSTR(iconDatabase().synchronousIconURLForPageURL(String(url, SysStringLen(url))));
*iconURL = iconURLBSTR.release();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::isEnabled(
BOOL *result)
{
*result = iconDatabase().isEnabled();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::setEnabled(
BOOL flag)
{
BOOL currentlyEnabled;
isEnabled(¤tlyEnabled);
if (currentlyEnabled && !flag) {
iconDatabase().setEnabled(false);
shutDownIconDatabase();
} else if (!currentlyEnabled && flag) {
iconDatabase().setEnabled(true);
startUpIconDatabase();
}
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebIconDatabase::hasIconForURL(
BSTR url,
BOOL* result)
{
if (!url || !result)
return E_POINTER;
String urlString(url, SysStringLen(url));
iconDatabase().synchronousIconForPageURL(urlString, IntSize(0, 0));
*result = !(iconDatabase().synchronousIconURLForPageURL(urlString).isEmpty());
return S_OK;
}
static HBITMAP createDIB(const IntSize& size)
{
BitmapInfo bmInfo = BitmapInfo::create(size);
HWndDC dc(0);
return CreateDIBSection(dc, &bmInfo, DIB_RGB_COLORS, 0, 0, 0);
}
HBITMAP WebIconDatabase::getOrCreateSharedBitmap(const IntSize& size)
{
HBITMAP result = m_sharedIconMap.get(size);
if (result)
return result;
result = createDIB(size);
m_sharedIconMap.set(size, result);
return result;
}
HBITMAP WebIconDatabase::getOrCreateDefaultIconBitmap(const IntSize& size)
{
HBITMAP result = m_defaultIconMap.get(size);
if (result)
return result;
result = createDIB(size);
m_defaultIconMap.set(size, result);
if (!iconDatabase().defaultIcon(size) || !iconDatabase().defaultIcon(size)->getHBITMAPOfSize(result, &size)) {
LOG_ERROR("Failed to draw Image to HBITMAP");
return 0;
}
return result;
}
void WebIconDatabase::didRemoveAllIcons()
{
MutexLocker locker(m_notificationMutex);
m_notificationQueue.append(String());
scheduleNotificationDelivery();
}
void WebIconDatabase::didImportIconURLForPageURL(const WTF::String& pageURL)
{
MutexLocker locker(m_notificationMutex);
m_notificationQueue.append(pageURL.isolatedCopy());
scheduleNotificationDelivery();
}
void WebIconDatabase::didImportIconDataForPageURL(const WTF::String& pageURL)
{
didImportIconURLForPageURL(pageURL);
}
void WebIconDatabase::didChangeIconForPageURL(const WTF::String& pageURL)
{
didImportIconURLForPageURL(pageURL);
}
void WebIconDatabase::didFinishURLImport()
{
}
void WebIconDatabase::scheduleNotificationDelivery()
{
ASSERT(!m_notificationMutex.tryLock());
if (!m_deliveryRequested) {
m_deliveryRequested = true;
callOnMainThread(deliverNotifications, 0);
}
}
BSTR WebIconDatabase::iconDatabaseDidAddIconNotification()
{
static BSTR didAddIconName = SysAllocString(WebIconDatabaseDidAddIconNotification);
return didAddIconName;
}
BSTR WebIconDatabase::iconDatabaseNotificationUserInfoURLKey()
{
static BSTR iconUserInfoURLKey = SysAllocString(WebIconNotificationUserInfoURLKey);
return iconUserInfoURLKey;
}
BSTR WebIconDatabase::iconDatabaseDidRemoveAllIconsNotification()
{
static BSTR didRemoveAllIconsName = SysAllocString(WebIconDatabaseDidRemoveAllIconsNotification);
return didRemoveAllIconsName;
}
static void postDidRemoveAllIconsNotification(WebIconDatabase* iconDB)
{
IWebNotificationCenter* notifyCenter = WebNotificationCenter::defaultCenterInternal();
notifyCenter->postNotificationName(WebIconDatabase::iconDatabaseDidRemoveAllIconsNotification(), static_cast<IWebIconDatabase*>(iconDB), 0);
}
static void postDidAddIconNotification(const String& pageURL, WebIconDatabase* iconDB)
{
HashMap<String, String> dictionary;
dictionary.set(WebIconDatabase::iconDatabaseNotificationUserInfoURLKey(), pageURL);
COMPtr<IPropertyBag> userInfo(AdoptCOM, COMPropertyBag<String>::adopt(dictionary));
IWebNotificationCenter* notifyCenter = WebNotificationCenter::defaultCenterInternal();
notifyCenter->postNotificationName(WebIconDatabase::iconDatabaseDidAddIconNotification(), static_cast<IWebIconDatabase*>(iconDB), userInfo.get());
}
void WebIconDatabase::deliverNotifications(void*)
{
ASSERT(m_sharedWebIconDatabase);
if (!m_sharedWebIconDatabase)
return;
ASSERT(m_sharedWebIconDatabase->m_deliveryRequested);
Vector<String> queue;
{
MutexLocker locker(m_sharedWebIconDatabase->m_notificationMutex);
queue.swap(m_sharedWebIconDatabase->m_notificationQueue);
m_sharedWebIconDatabase->m_deliveryRequested = false;
}
for (unsigned i = 0; i < queue.size(); ++i) {
if (queue[i].isNull())
postDidRemoveAllIconsNotification(m_sharedWebIconDatabase);
else
postDidAddIconNotification(queue[i], m_sharedWebIconDatabase);
}
}