#include "config.h"
#include "WKPageGroup.h"
#include "APIArray.h"
#include "APIContentRuleList.h"
#include "APIContentWorld.h"
#include "APIUserScript.h"
#include "APIUserStyleSheet.h"
#include "InjectUserScriptImmediately.h"
#include "WKAPICast.h"
#include "WebPageGroup.h"
#include "WebPreferences.h"
#include "WebUserContentController.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 allowedURLPatterns, WKArrayRef blockedURLPatterns, WKUserContentInjectedFrames injectedFrames)
{
auto source = toWTFString(sourceRef);
if (source.isEmpty())
return;
auto baseURLString = toWTFString(baseURLRef);
auto allowlist = toImpl(allowedURLPatterns);
auto blocklist = toImpl(blockedURLPatterns);
Ref<API::UserStyleSheet> userStyleSheet = API::UserStyleSheet::create(WebCore::UserStyleSheet { source, (baseURLString.isEmpty() ? aboutBlankURL() : URL(URL(), baseURLString)), allowlist ? allowlist->toStringVector() : Vector<String>(), blocklist ? blocklist->toStringVector() : Vector<String>(), toUserContentInjectedFrames(injectedFrames), WebCore::UserStyleUserLevel }, API::ContentWorld::pageContentWorld());
toImpl(pageGroupRef)->userContentController().addUserStyleSheet(userStyleSheet.get());
}
void WKPageGroupRemoveAllUserStyleSheets(WKPageGroupRef pageGroup)
{
toImpl(pageGroup)->userContentController().removeAllUserStyleSheets();
}
void WKPageGroupAddUserScript(WKPageGroupRef pageGroupRef, WKStringRef sourceRef, WKURLRef baseURLRef, WKArrayRef allowedURLPatterns, WKArrayRef blockedURLPatterns, WKUserContentInjectedFrames injectedFrames, _WKUserScriptInjectionTime injectionTime)
{
auto source = toWTFString(sourceRef);
if (source.isEmpty())
return;
auto baseURLString = toWTFString(baseURLRef);
auto allowlist = toImpl(allowedURLPatterns);
auto blocklist = toImpl(blockedURLPatterns);
auto url = baseURLString.isEmpty() ? aboutBlankURL() : URL(URL(), baseURLString);
Ref<API::UserScript> userScript = API::UserScript::create(WebCore::UserScript { WTFMove(source), WTFMove(url), allowlist ? allowlist->toStringVector() : Vector<String>(), blocklist ? blocklist->toStringVector() : Vector<String>(), toUserScriptInjectionTime(injectionTime), toUserContentInjectedFrames(injectedFrames), WebCore::WaitForNotificationBeforeInjecting::No }, API::ContentWorld::pageContentWorld());
toImpl(pageGroupRef)->userContentController().addUserScript(userScript.get(), InjectUserScriptImmediately::No);
}
void WKPageGroupRemoveAllUserScripts(WKPageGroupRef pageGroupRef)
{
toImpl(pageGroupRef)->userContentController().removeAllUserScripts();
}