InspectorDOMStorageAgent.cpp [plain text]
#include "config.h"
#if ENABLE(INSPECTOR)
#include "InspectorDOMStorageAgent.h"
#include "Database.h"
#include "DOMWindow.h"
#include "Document.h"
#include "ExceptionCode.h"
#include "ExceptionCodeDescription.h"
#include "Frame.h"
#include "InspectorFrontend.h"
#include "InspectorPageAgent.h"
#include "InspectorState.h"
#include "InspectorValues.h"
#include "InstrumentingAgents.h"
#include "Page.h"
#include "PageGroup.h"
#include "SecurityOrigin.h"
#include "Storage.h"
#include "StorageArea.h"
#include "StorageNamespace.h"
#include "VoidCallback.h"
#include <wtf/Vector.h>
namespace WebCore {
namespace DOMStorageAgentState {
static const char domStorageAgentEnabled[] = "domStorageAgentEnabled";
};
InspectorDOMStorageAgent::InspectorDOMStorageAgent(InstrumentingAgents* instrumentingAgents, InspectorPageAgent* pageAgent, InspectorCompositeState* state)
: InspectorBaseAgent<InspectorDOMStorageAgent>("DOMStorage", instrumentingAgents, state)
, m_pageAgent(pageAgent)
, m_frontend(0)
{
m_instrumentingAgents->setInspectorDOMStorageAgent(this);
}
InspectorDOMStorageAgent::~InspectorDOMStorageAgent()
{
m_instrumentingAgents->setInspectorDOMStorageAgent(0);
m_instrumentingAgents = 0;
}
void InspectorDOMStorageAgent::setFrontend(InspectorFrontend* frontend)
{
m_frontend = frontend;
}
void InspectorDOMStorageAgent::clearFrontend()
{
m_frontend = 0;
disable(0);
}
bool InspectorDOMStorageAgent::isEnabled() const
{
return m_state->getBoolean(DOMStorageAgentState::domStorageAgentEnabled);
}
void InspectorDOMStorageAgent::enable(ErrorString*)
{
m_state->setBoolean(DOMStorageAgentState::domStorageAgentEnabled, true);
}
void InspectorDOMStorageAgent::disable(ErrorString*)
{
m_state->setBoolean(DOMStorageAgentState::domStorageAgentEnabled, false);
}
void InspectorDOMStorageAgent::getDOMStorageItems(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, RefPtr<TypeBuilder::Array<TypeBuilder::Array<String> > >& items)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(errorString, storageId, frame);
if (!storageArea) {
if (errorString)
*errorString = "No StorageArea for given storageId";
return;
}
RefPtr<TypeBuilder::Array<TypeBuilder::Array<String> > > storageItems = TypeBuilder::Array<TypeBuilder::Array<String> >::create();
for (unsigned i = 0; i < storageArea->length(); ++i) {
String key = storageArea->key(i);
String value = storageArea->item(key);
RefPtr<TypeBuilder::Array<String> > entry = TypeBuilder::Array<String>::create();
entry->addItem(key);
entry->addItem(value);
storageItems->addItem(entry.release());
}
items = storageItems.release();
}
void InspectorDOMStorageAgent::setDOMStorageItem(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, const String& key, const String& value)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(0, storageId, frame);
if (!storageArea) {
*errorString = "Storage not found";
return;
}
bool quotaException = false;
storageArea->setItem(frame, key, value, quotaException);
if (quotaException)
*errorString = ExceptionCodeDescription(QUOTA_EXCEEDED_ERR).name;
}
void InspectorDOMStorageAgent::removeDOMStorageItem(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, const String& key)
{
Frame* frame;
RefPtr<StorageArea> storageArea = findStorageArea(0, storageId, frame);
if (!storageArea) {
*errorString = "Storage not found";
return;
}
storageArea->removeItem(frame, key);
}
String InspectorDOMStorageAgent::storageId(Storage* storage)
{
ASSERT(storage);
Document* document = storage->frame()->document();
ASSERT(document);
DOMWindow* window = document->domWindow();
ASSERT(window);
RefPtr<SecurityOrigin> securityOrigin = document->securityOrigin();
bool isLocalStorage = window->optionalLocalStorage() == storage;
return storageId(securityOrigin.get(), isLocalStorage)->toJSONString();
}
PassRefPtr<TypeBuilder::DOMStorage::StorageId> InspectorDOMStorageAgent::storageId(SecurityOrigin* securityOrigin, bool isLocalStorage)
{
return TypeBuilder::DOMStorage::StorageId::create()
.setSecurityOrigin(securityOrigin->toRawString())
.setIsLocalStorage(isLocalStorage).release();
}
void InspectorDOMStorageAgent::didDispatchDOMStorageEvent(const String& key, const String& oldValue, const String& newValue, StorageType storageType, SecurityOrigin* securityOrigin, Page*)
{
if (!m_frontend || !isEnabled())
return;
RefPtr<TypeBuilder::DOMStorage::StorageId> id = storageId(securityOrigin, storageType == LocalStorage);
if (key.isNull())
m_frontend->domstorage()->domStorageItemsCleared(id);
else if (newValue.isNull())
m_frontend->domstorage()->domStorageItemRemoved(id, key);
else if (oldValue.isNull())
m_frontend->domstorage()->domStorageItemAdded(id, key, newValue);
else
m_frontend->domstorage()->domStorageItemUpdated(id, key, oldValue, newValue);
}
PassRefPtr<StorageArea> InspectorDOMStorageAgent::findStorageArea(ErrorString* errorString, const RefPtr<InspectorObject>& storageId, Frame*& targetFrame)
{
String securityOrigin;
bool isLocalStorage = false;
bool success = storageId->getString("securityOrigin", &securityOrigin);
if (success)
success = storageId->getBoolean("isLocalStorage", &isLocalStorage);
if (!success) {
if (errorString)
*errorString = "Invalid storageId format";
targetFrame = 0;
return 0;
}
targetFrame = m_pageAgent->findFrameWithSecurityOrigin(securityOrigin);
if (!targetFrame) {
if (errorString)
*errorString = "Frame not found for the given security origin";
return 0;
}
Page* page = m_pageAgent->page();
if (isLocalStorage)
return page->group().localStorage()->storageArea(targetFrame->document()->securityOrigin());
return page->sessionStorage()->storageArea(targetFrame->document()->securityOrigin());
}
}
#endif // ENABLE(INSPECTOR)