initial import
[vuplus_webkit] / Tools / DumpRenderTree / mac / NavigationController.m
1 /*
2  * Copyright (C) 2006 Apple Computer, 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  *
8  * 1.  Redistributions of source code must retain the above copyright
9  *     notice, this list of conditions and the following disclaimer. 
10  * 2.  Redistributions in binary form must reproduce the above copyright
11  *     notice, this list of conditions and the following disclaimer in the
12  *     documentation and/or other materials provided with the distribution. 
13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14  *     its contributors may be used to endorse or promote products derived
15  *     from this software without specific prior written permission. 
16  *
17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23     * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 #import "config.h"
30 #import "NavigationController.h"
31
32 #import <WebKit/WebFrame.h>
33 #import <WebKit/WebScriptObject.h>
34
35
36 @implementation NavigationController
37 + (BOOL)isSelectorExcludedFromWebScript:(SEL)selector
38 {
39     if (selector == @selector(evaluateWebScript:afterBackForwardNavigation:))
40         return NO;
41     return YES;
42 }
43
44 + (NSString *)webScriptNameForSelector:(SEL)selector
45 {
46     if (selector == @selector(evaluateWebScript:afterBackForwardNavigation:))
47         return @"evalAfterBackForwardNavigation";
48     return nil;
49 }
50
51 - (void)setPendingScript:(NSString *)script
52 {
53     if (script != pendingScript) {
54         [pendingScript release];
55         pendingScript = [script copy];
56     }
57 }
58
59 - (void)setPendingRequest:(NSURLRequest *)request
60 {
61     if (request != pendingRequest) {
62         [pendingRequest release];
63         pendingRequest = [request copy];
64     }
65 }
66
67 - (void)evaluateWebScript:(NSString *)script afterBackForwardNavigation:(NSString *)navigation
68 {
69     // Allow both arguments to be optional
70     if (![script isKindOfClass:[NSString class]])
71         script = @"";
72     if (![navigation isKindOfClass:[NSString class]])
73         navigation = @"about:blank";
74     
75     [self setPendingScript:script];
76     [self setPendingRequest:[NSURLRequest requestWithURL:[NSURL URLWithString:navigation]]];
77     pendingAction = Load;
78 }
79
80 - (void)webView:(WebView *)sender didFinishLoadForFrame:(WebFrame *)frame
81 {
82     if (frame == [[frame webView] mainFrame]) {
83         switch (pendingAction) {
84             case Load:
85                 pendingAction = GoBack;
86                 [frame loadRequest:pendingRequest];
87                 [self setPendingRequest:nil];
88                 break;
89             case GoBack:
90                 pendingAction = ExecuteScript;
91                 [[frame webView] goBack];
92                 break;
93             case ExecuteScript:
94                 pendingAction = None;
95                 [[[frame webView] windowScriptObject] evaluateWebScript:pendingScript];
96                 [self setPendingScript:nil];
97                 break;
98             case None:
99             default:
100                 break;
101         }
102     }
103 }
104
105 - (void)dealloc
106 {
107     [self setPendingScript:nil];
108     [self setPendingRequest:nil];
109     [super dealloc];
110 }
111 @end
112