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