#include "config.h"
#include "WebPageGroup.h"
#include "APIArray.h"
#include "WebPageGroupProxyMessages.h"
#include "WebPageProxy.h"
#include "WebPreferences.h"
#include <wtf/HashMap.h>
#include <wtf/NeverDestroyed.h>
#include <wtf/text/StringConcatenate.h>
namespace WebKit {
static uint64_t generatePageGroupID()
{
static uint64_t uniquePageGroupID = 1;
return uniquePageGroupID++;
}
typedef HashMap<uint64_t, WebPageGroup*> WebPageGroupMap;
static WebPageGroupMap& webPageGroupMap()
{
static NeverDestroyed<WebPageGroupMap> map;
return map;
}
PassRefPtr<WebPageGroup> WebPageGroup::create(const String& identifier, bool visibleToInjectedBundle, bool visibleToHistoryClient)
{
return adoptRef(new WebPageGroup(identifier, visibleToInjectedBundle, visibleToHistoryClient));
}
PassRef<WebPageGroup> WebPageGroup::createNonNull(const String& identifier, bool visibleToInjectedBundle, bool visibleToHistoryClient)
{
return adoptRef(*new WebPageGroup(identifier, visibleToInjectedBundle, visibleToHistoryClient));
}
WebPageGroup* WebPageGroup::get(uint64_t pageGroupID)
{
return webPageGroupMap().get(pageGroupID);
}
static WebPageGroupData pageGroupData(const String& identifier, bool visibleToInjectedBundle, bool visibleToHistoryClient)
{
WebPageGroupData data;
data.pageGroupID = generatePageGroupID();
if (!identifier.isEmpty())
data.identifer = identifier;
else
data.identifer = makeString("__uniquePageGroupID-", String::number(data.pageGroupID));
data.visibleToInjectedBundle = visibleToInjectedBundle;
data.visibleToHistoryClient = visibleToHistoryClient;
return data;
}
WebPageGroup::WebPageGroup(const String& identifier, bool visibleToInjectedBundle, bool visibleToHistoryClient)
: m_data(pageGroupData(identifier, visibleToInjectedBundle, visibleToHistoryClient))
, m_preferences(WebPreferences::createWithLegacyDefaults(m_data.identifer, ".WebKit2", "WebKit2."))
{
webPageGroupMap().set(m_data.pageGroupID, this);
}
WebPageGroup::~WebPageGroup()
{
webPageGroupMap().remove(pageGroupID());
}
void WebPageGroup::addPage(WebPageProxy* page)
{
m_pages.add(page);
}
void WebPageGroup::removePage(WebPageProxy* page)
{
m_pages.remove(page);
}
void WebPageGroup::setPreferences(WebPreferences* preferences)
{
if (preferences == m_preferences)
return;
m_preferences = preferences;
for (auto& webPageProxy : m_pages)
webPageProxy->setPreferences(*m_preferences);
}
WebPreferences& WebPageGroup::preferences() const
{
return *m_preferences;
}
void WebPageGroup::preferencesDidChange()
{
for (HashSet<WebPageProxy*>::iterator it = m_pages.begin(), end = m_pages.end(); it != end; ++it) {
WebPageProxy* page = *it;
page->preferencesDidChange();
}
}
void WebPageGroup::addUserStyleSheet(const String& source, const String& baseURL, API::Array* whitelist, API::Array* blacklist, WebCore::UserContentInjectedFrames injectedFrames, WebCore::UserStyleLevel level)
{
if (source.isEmpty())
return;
WebCore::UserStyleSheet userStyleSheet = WebCore::UserStyleSheet(source, (baseURL.isEmpty() ? WebCore::blankURL() : WebCore::URL(WebCore::URL(), baseURL)), whitelist ? whitelist->toStringVector() : Vector<String>(), blacklist ? blacklist->toStringVector() : Vector<String>(), injectedFrames, level);
m_data.userStyleSheets.append(userStyleSheet);
sendToAllProcessesInGroup(Messages::WebPageGroupProxy::AddUserStyleSheet(userStyleSheet), m_data.pageGroupID);
}
void WebPageGroup::addUserScript(const String& source, const String& baseURL, API::Array* whitelist, API::Array* blacklist, WebCore::UserContentInjectedFrames injectedFrames, WebCore::UserScriptInjectionTime injectionTime)
{
if (source.isEmpty())
return;
WebCore::UserScript userScript = WebCore::UserScript(source, (baseURL.isEmpty() ? WebCore::blankURL() : WebCore::URL(WebCore::URL(), baseURL)), whitelist ? whitelist->toStringVector() : Vector<String>(), blacklist ? blacklist->toStringVector() : Vector<String>(), injectionTime, injectedFrames);
m_data.userScripts.append(userScript);
sendToAllProcessesInGroup(Messages::WebPageGroupProxy::AddUserScript(userScript), m_data.pageGroupID);
}
void WebPageGroup::removeAllUserStyleSheets()
{
m_data.userStyleSheets.clear();
sendToAllProcessesInGroup(Messages::WebPageGroupProxy::RemoveAllUserStyleSheets(), m_data.pageGroupID);
}
void WebPageGroup::removeAllUserScripts()
{
m_data.userScripts.clear();
sendToAllProcessesInGroup(Messages::WebPageGroupProxy::RemoveAllUserScripts(), m_data.pageGroupID);
}
void WebPageGroup::removeAllUserContent()
{
m_data.userStyleSheets.clear();
m_data.userScripts.clear();
sendToAllProcessesInGroup(Messages::WebPageGroupProxy::RemoveAllUserContent(), m_data.pageGroupID);
}
}