blob: c8f77ed0258ff179385360ef0b459af29ca51277 [file] [log] [blame]
// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "ios/web_view/public/cwv_user_content_controller.h"
#import "ios/web_view/internal/cwv_user_content_controller_internal.h"
#import "ios/web_view/internal/cwv_web_view_configuration_internal.h"
#include "ios/web_view/internal/web_view_browser_state.h"
#import "ios/web_view/internal/web_view_early_page_script_provider.h"
#import "ios/web_view/public/cwv_user_script.h"
#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
@interface CWVUserContentController ()
@property(weak, nonatomic) CWVWebViewConfiguration* configuration;
@implementation CWVUserContentController {
NSMutableArray<CWVUserScript*>* _userScripts;
@synthesize configuration = _configuration;
- (nonnull instancetype)initWithConfiguration:
(nonnull __weak CWVWebViewConfiguration*)configuration {
self = [super init];
if (self) {
_configuration = configuration;
_userScripts = [[NSMutableArray alloc] init];
return self;
- (void)addUserScript:(nonnull CWVUserScript*)userScript {
[_userScripts addObject:userScript];
[self updateEarlyPageScript];
- (void)removeAllUserScripts {
[_userScripts removeAllObjects];
[self updateEarlyPageScript];
- (nonnull NSArray<CWVUserScript*>*)userScripts {
return _userScripts;
// Updates the early page script associated with the BrowserState with the
// content of _userScripts.
- (void)updateEarlyPageScript {
NSMutableString* joinedScript = [[NSMutableString alloc] init];
for (CWVUserScript* script in _userScripts) {
[joinedScript appendString:script.source];
// Inserts "\n" between scripts to make it safer to join multiple scripts,
// in case the first script doesn't end with ";" or "\n".
[joinedScript appendString:@"\n"];