#include "config.h"
#include "WKPageGroup.h"
#include "APIArray.h"
#include "APIContentRuleList.h"
#include "APIUserContentWorld.h"
#include "APIUserScript.h"
#include "APIUserStyleSheet.h"
#include "WKAPICast.h"
#include "WebPageGroup.h"
#include "WebPreferences.h"
#include "WebUserContentControllerProxy.h"
using namespace WebKit;
WKTypeID WKPageGroupGetTypeID()
{
return toAPI(WebPageGroup::APIType);
}
WKPageGroupRef WKPageGroupCreateWithIdentifier(WKStringRef identifier)
{
auto pageGroup = WebPageGroup::create(toWTFString(identifier));
return toAPI(&pageGroup.leakRef());
}
void WKPageGroupSetPreferences(WKPageGroupRef pageGroupRef, WKPreferencesRef preferencesRef)
{
toImpl(pageGroupRef)->setPreferences(toImpl(preferencesRef));
}
WKPreferencesRef WKPageGroupGetPreferences(WKPageGroupRef pageGroupRef)
{
return toAPI(&toImpl(pageGroupRef)->preferences());
}
WKUserContentControllerRef WKPageGroupGetUserContentController(WKPageGroupRef pageGroupRef)
{
return toAPI(&toImpl(pageGroupRef)->userContentController());
}
void WKPageGroupAddUserStyleSheet(WKPageGroupRef pageGroupRef, WKStringRef sourceRef, WKURLRef baseURLRef, WKArrayRef whitelistedURLPatterns, WKArrayRef blacklistedURLPatterns, WKUserContentInjectedFrames injectedFrames)
{
auto source = toWTFString(sourceRef);
if (source.isEmpty())
return;
auto baseURLString = toWTFString(baseURLRef);
auto whitelist = toImpl(whitelistedURLPatterns);
auto blacklist = toImpl(blacklistedURLPatterns);
Ref<API::UserStyleSheet> userStyleSheet = API::UserStyleSheet::create(WebCore::UserStyleSheet { source, (baseURLString.isEmpty() ? WebCore::blankURL() : WebCore::URL(WebCore::URL(), baseURLString)), whitelist ? whitelist->toStringVector() : Vector<String>(), blacklist ? blacklist->toStringVector() : Vector<String>(), toUserContentInjectedFrames(injectedFrames), WebCore::UserStyleUserLevel }, API::UserContentWorld::normalWorld());
toImpl(pageGroupRef)->userContentController().addUserStyleSheet(userStyleSheet.get());
}
void WKPageGroupRemoveAllUserStyleSheets(WKPageGroupRef pageGroup)
{
toImpl(pageGroup)->userContentController().removeAllUserStyleSheets();
}
void WKPageGroupAddUserScript(WKPageGroupRef pageGroupRef, WKStringRef sourceRef, WKURLRef baseURLRef, WKArrayRef whitelistedURLPatterns, WKArrayRef blacklistedURLPatterns, WKUserContentInjectedFrames injectedFrames, _WKUserScriptInjectionTime injectionTime)
{
auto source = toWTFString(sourceRef);
if (source.isEmpty())
return;
auto baseURLString = toWTFString(baseURLRef);
auto whitelist = toImpl(whitelistedURLPatterns);
auto blacklist = toImpl(blacklistedURLPatterns);
auto url = baseURLString.isEmpty() ? WebCore::blankURL() : WebCore::URL(WebCore::URL(), baseURLString);
Ref<API::UserScript> userScript = API::UserScript::create(WebCore::UserScript { WTFMove(source), WTFMove(url), whitelist ? whitelist->toStringVector() : Vector<String>(), blacklist ? blacklist->toStringVector() : Vector<String>(), toUserScriptInjectionTime(injectionTime), toUserContentInjectedFrames(injectedFrames) }, API::UserContentWorld::normalWorld());
toImpl(pageGroupRef)->userContentController().addUserScript(userScript.get());
}
void WKPageGroupRemoveAllUserScripts(WKPageGroupRef pageGroupRef)
{
toImpl(pageGroupRef)->userContentController().removeAllUserScripts();
}