[Content Filtering] Determine navigation and content policy before continuing to...
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKit2Cocoa / ContentFilteringPlugIn.mm
1 /*
2  * Copyright (C) 2015 Apple 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  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27
28 #if WK_API_ENABLED
29
30 #import "MockContentFilterSettings.h"
31 #import <WebKit/WKWebProcessPlugIn.h>
32
33 using MockContentFilterSettings = WebCore::MockContentFilterSettings;
34 using Decision = MockContentFilterSettings::Decision;
35 using DecisionPoint = MockContentFilterSettings::DecisionPoint;
36
37 @interface MockContentFilterEnabler : NSObject <NSCopying, NSSecureCoding>
38 @end
39
40 @implementation MockContentFilterEnabler
41
42 + (BOOL)supportsSecureCoding
43 {
44     return YES;
45 }
46
47 - (id)copyWithZone:(NSZone *)zone
48 {
49     return [self retain];
50 }
51
52 - (instancetype)initWithCoder:(NSCoder *)decoder
53 {
54     if (!(self = [super init]))
55         return nil;
56
57     auto& settings = MockContentFilterSettings::singleton();
58     settings.setEnabled(true);
59     settings.setDecision(static_cast<Decision>([decoder decodeIntForKey:@"Decision"]));
60     settings.setDecisionPoint(static_cast<DecisionPoint>([decoder decodeIntForKey:@"DecisionPoint"]));
61     return self;
62 }
63
64 - (void)dealloc
65 {
66     MockContentFilterSettings::singleton().setEnabled(false);
67     [super dealloc];
68 }
69
70 - (void)encodeWithCoder:(NSCoder *)coder
71 {
72 }
73
74 @end
75
76 @interface ContentFilteringPlugIn : NSObject <WKWebProcessPlugIn>
77 @end
78
79 @implementation ContentFilteringPlugIn {
80     RetainPtr<MockContentFilterEnabler> _contentFilterEnabler;
81     RetainPtr<WKWebProcessPlugInController> _plugInController;
82 }
83
84 - (void)webProcessPlugIn:(WKWebProcessPlugInController *)plugInController initializeWithObject:(id)initializationObject
85 {
86     ASSERT(!_plugInController);
87     _plugInController = plugInController;
88     [plugInController.parameters addObserver:self forKeyPath:NSStringFromClass([MockContentFilterEnabler class]) options:NSKeyValueObservingOptionInitial context:NULL];
89 }
90
91 - (void)dealloc
92 {
93     [[_plugInController parameters] removeObserver:self forKeyPath:NSStringFromClass([MockContentFilterEnabler class])];
94     [super dealloc];
95 }
96
97 - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context
98 {
99     id contentFilterEnabler = [object valueForKeyPath:keyPath];
100     ASSERT([contentFilterEnabler isKindOfClass:[MockContentFilterEnabler class]]);
101     _contentFilterEnabler = contentFilterEnabler;
102 }
103
104 @end
105
106 #endif // WK_API_ENABLED