1/* 2 * Copyright (C) 2010 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of Apple Inc. ("Apple") nor the names of 14 * its contributors may be used to endorse or promote products derived 15 * from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY 18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29#ifndef InspectorDatabaseAgent_h 30#define InspectorDatabaseAgent_h 31 32#if ENABLE(INSPECTOR) && ENABLE(SQL_DATABASE) 33 34#include "InspectorWebAgentBase.h" 35#include "InspectorWebBackendDispatchers.h" 36#include "InspectorWebFrontendDispatchers.h" 37#include <wtf/HashMap.h> 38#include <wtf/text/WTFString.h> 39 40namespace Inspector { 41class InspectorArray; 42} 43 44namespace WebCore { 45 46class Database; 47class InspectorDatabaseResource; 48class InstrumentingAgents; 49 50typedef String ErrorString; 51 52class InspectorDatabaseAgent : public InspectorAgentBase, public Inspector::InspectorDatabaseBackendDispatcherHandler { 53 WTF_MAKE_FAST_ALLOCATED; 54public: 55 explicit InspectorDatabaseAgent(InstrumentingAgents*); 56 ~InspectorDatabaseAgent(); 57 58 virtual void didCreateFrontendAndBackend(Inspector::InspectorFrontendChannel*, Inspector::InspectorBackendDispatcher*) override; 59 virtual void willDestroyFrontendAndBackend(Inspector::InspectorDisconnectReason) override; 60 61 void clearResources(); 62 63 // Called from the front-end. 64 virtual void enable(ErrorString*) override; 65 virtual void disable(ErrorString*) override; 66 virtual void getDatabaseTableNames(ErrorString*, const String& databaseId, RefPtr<Inspector::TypeBuilder::Array<String>>& names) override; 67 virtual void executeSQL(ErrorString*, const String& databaseId, const String& query, PassRefPtr<ExecuteSQLCallback>) override; 68 69 // Called from the injected script. 70 String databaseId(Database*); 71 72 void didOpenDatabase(PassRefPtr<Database>, const String& domain, const String& name, const String& version); 73private: 74 Database* databaseForId(const String& databaseId); 75 InspectorDatabaseResource* findByFileName(const String& fileName); 76 77 std::unique_ptr<Inspector::InspectorDatabaseFrontendDispatcher> m_frontendDispatcher; 78 RefPtr<Inspector::InspectorDatabaseBackendDispatcher> m_backendDispatcher; 79 typedef HashMap<String, RefPtr<InspectorDatabaseResource>> DatabaseResourcesMap; 80 DatabaseResourcesMap m_resources; 81 bool m_enabled; 82}; 83 84} // namespace WebCore 85 86#endif // ENABLE(INSPECTOR) && ENABLE(SQL_DATABASE) 87 88#endif // !defined(InspectorDatabaseAgent_h) 89