CyteKit/CyteWebView.mm

   1 #include "CyteKit/CyteWebView.h"
   2
   3 #include "iPhonePrivate.h"
   4 #include "substrate.h"
   5
   6 // CYWebPolicyDecision* {{{
   7 enum CYWebPolicyDecision {
   8     CYWebPolicyDecisionUnknown,
   9     CYWebPolicyDecisionDownload,
  10     CYWebPolicyDecisionIgnore,
  11     CYWebPolicyDecisionUse,
  12 };
  13
  14 @interface CYWebPolicyDecisionMediator : NSObject <
  15     WebPolicyDecisionListener
  16 > {
  17     id<WebPolicyDecisionListener> listener_;
  18     CYWebPolicyDecision decision_;
  19 }
  20
  21 - (id) initWithListener:(id<WebPolicyDecisionListener>)listener;
  22
  23 - (CYWebPolicyDecision) decision;
  24 - (bool) decided;
  25 - (bool) decide;
  26
  27 @end
  28
  29 @implementation CYWebPolicyDecisionMediator
  30
  31 - (id) initWithListener:(id<WebPolicyDecisionListener>)listener {
  32     if ((self = [super init]) != nil) {
  33         listener_ = listener;
  34     } return self;
  35 }
  36
  37 - (CYWebPolicyDecision) decision {
  38     return decision_;
  39 }
  40
  41 - (bool) decided {
  42     return decision_ != CYWebPolicyDecisionUnknown;
  43 }
  44
  45 - (bool) decide {
  46     switch (decision_) {
  47         case CYWebPolicyDecisionUnknown:
  48         default:
  49             NSLog(@"CYWebPolicyDecisionUnknown");
  50             return false;
  51
  52         case CYWebPolicyDecisionDownload: [listener_ download]; break;
  53         case CYWebPolicyDecisionIgnore: [listener_ ignore]; break;
  54         case CYWebPolicyDecisionUse: [listener_ use]; break;
  55     }
  56
  57     return true;
  58 }
  59
  60 - (void) download {
  61     decision_ = CYWebPolicyDecisionDownload;
  62 }
  63
  64 - (void) ignore {
  65     decision_ = CYWebPolicyDecisionIgnore;
  66 }
  67
  68 - (void) use {
  69     decision_ = CYWebPolicyDecisionUse;
  70 }
  71
  72 @end
  73 // }}}
  74
  75 @implementation CYWebView : UIWebView
  76
  77 #if ShowInternals
  78 #include "CyteKit/UCInternal.h"
  79 #endif
  80
  81 - (id) initWithFrame:(CGRect)frame {
  82     if ((self = [super initWithFrame:frame]) != nil) {
  83     } return self;
  84 }
  85
  86 - (void) dealloc {
  87     [super dealloc];
  88 }
  89
  90 - (id<CYWebViewDelegate>) delegate {
  91     return (id<CYWebViewDelegate>) [super delegate];
  92 }
  93
  94 /*- (WebView *) webView:(WebView *)view createWebViewWithRequest:(NSURLRequest *)request {
  95     id<CYWebViewDelegate> delegate([self delegate]);
  96     WebView *created(nil);
  97     if (created == nil && [delegate respondsToSelector:@selector(webView:createWebViewWithRequest:)])
  98         created = [delegate webView:view createWebViewWithRequest:request];
  99     if (created == nil && [UIWebView instancesRespondToSelector:@selector(webView:createWebViewWithRequest:)])
 100         created = [super webView:view createWebViewWithRequest:request];
 101     return created;
 102 }*/
 103
 104 // webView:addMessageToConsole: (X.Xx) {{{
 105 static void $UIWebViewWebViewDelegate$webView$addMessageToConsole$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *message) {
 106     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 107     if ([uiWebView respondsToSelector:@selector(webView:addMessageToConsole:)])
 108         [uiWebView webView:view addMessageToConsole:message];
 109 }
 110
 111 - (void) webView:(WebView *)view addMessageToConsole:(NSDictionary *)message {
 112     id<CYWebViewDelegate> delegate([self delegate]);
 113     if ([delegate respondsToSelector:@selector(webView:addMessageToConsole:)])
 114         [delegate webView:view addMessageToConsole:message];
 115     if ([UIWebView instancesRespondToSelector:@selector(webView:addMessageToConsole:)])
 116         [super webView:view addMessageToConsole:message];
 117 }
 118 // }}}
 119 // webView:decidePolicyForNavigationAction:request:frame:decisionListener: (2.0+) {{{
 120 - (void) webView:(WebView *)view decidePolicyForNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
 121     id<CYWebViewDelegate> delegate([self delegate]);
 122     CYWebPolicyDecisionMediator *mediator([[[CYWebPolicyDecisionMediator alloc] initWithListener:listener] autorelease]);
 123     if (![mediator decided] && [delegate respondsToSelector:@selector(webView:decidePolicyForNavigationAction:request:frame:decisionListener:)])
 124         [delegate webView:view decidePolicyForNavigationAction:action request:request frame:frame decisionListener:mediator];
 125     if (![mediator decided] && [UIWebView instancesRespondToSelector:@selector(webView:decidePolicyForNavigationAction:request:frame:decisionListener:)])
 126         [super webView:view decidePolicyForNavigationAction:action request:request frame:frame decisionListener:mediator];
 127     [mediator decide];
 128 }
 129 // }}}
 130 // webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener: (3.0+) {{{
 131 static void $UIWebViewWebViewDelegate$webView$decidePolicyForNewWindowAction$request$newFrameName$decisionListener$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSDictionary *action, NSURLRequest *request, NSString *frame, id<WebPolicyDecisionListener> listener) {
 132     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 133     if ([uiWebView respondsToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
 134         [uiWebView webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:listener];
 135 }
 136
 137 - (void) webView:(WebView *)view decidePolicyForNewWindowAction:(NSDictionary *)action request:(NSURLRequest *)request newFrameName:(NSString *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
 138     id<CYWebViewDelegate> delegate([self delegate]);
 139     CYWebPolicyDecisionMediator *mediator([[[CYWebPolicyDecisionMediator alloc] initWithListener:listener] autorelease]);
 140     if (![mediator decided] && [delegate respondsToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
 141         [delegate webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:mediator];
 142     if (![mediator decided] && [UIWebView instancesRespondToSelector:@selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:)])
 143         [super webView:view decidePolicyForNewWindowAction:action request:request newFrameName:frame decisionListener:mediator];
 144     [mediator decide];
 145 }
 146 // }}}
 147 // webView:didClearWindowObject:forFrame: (3.2+) {{{
 148 static void $UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, WebScriptObject *window, WebFrame *frame) {
 149     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 150     if ([uiWebView respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
 151         [uiWebView webView:view didClearWindowObject:window forFrame:frame];
 152 }
 153
 154 - (void) webView:(WebView *)view didClearWindowObject:(WebScriptObject *)window forFrame:(WebFrame *)frame {
 155     id<CYWebViewDelegate> delegate([self delegate]);
 156     if ([delegate respondsToSelector:@selector(webView:didClearWindowObject:forFrame:)])
 157         [delegate webView:view didClearWindowObject:window forFrame:frame];
 158     if ([UIWebView instancesRespondToSelector:@selector(webView:didClearWindowObject:forFrame:)])
 159         [super webView:view didClearWindowObject:window forFrame:frame];
 160 }
 161 // }}}
 162 // webView:didFailLoadWithError:forFrame: (2.0+) {{{
 163 - (void) webView:(WebView *)view didFailLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 164     id<CYWebViewDelegate> delegate([self delegate]);
 165     if ([delegate respondsToSelector:@selector(webView:didFailLoadWithError:forFrame:)])
 166         [delegate webView:view didFailLoadWithError:error forFrame:frame];
 167     if ([UIWebView instancesRespondToSelector:@selector(webView:didFailLoadWithError:forFrame:)])
 168         [super webView:view didFailLoadWithError:error forFrame:frame];
 169 }
 170 // }}}
 171 // webView:didFailProvisionalLoadWithError:forFrame: (2.0+) {{{
 172 - (void) webView:(WebView *)view didFailProvisionalLoadWithError:(NSError *)error forFrame:(WebFrame *)frame {
 173     id<CYWebViewDelegate> delegate([self delegate]);
 174     if ([delegate respondsToSelector:@selector(webView:didFailProvisionalLoadWithError:forFrame:)])
 175         [delegate webView:view didFailProvisionalLoadWithError:error forFrame:frame];
 176     if ([UIWebView instancesRespondToSelector:@selector(webView:didFailProvisionalLoadWithError:forFrame:)])
 177         [super webView:view didFailProvisionalLoadWithError:error forFrame:frame];
 178 }
 179 // }}}
 180 // webView:didFinishLoadForFrame: (2.0+) {{{
 181 - (void) webView:(WebView *)view didFinishLoadForFrame:(WebFrame *)frame {
 182     id<CYWebViewDelegate> delegate([self delegate]);
 183     if ([delegate respondsToSelector:@selector(webView:didFinishLoadForFrame:)])
 184         [delegate webView:view didFinishLoadForFrame:frame];
 185     if ([UIWebView instancesRespondToSelector:@selector(webView:didFinishLoadForFrame:)])
 186         [super webView:view didFinishLoadForFrame:frame];
 187 }
 188 // }}}
 189 // webView:didReceiveTitle:forFrame: (3.2+) {{{
 190 static void $UIWebViewWebViewDelegate$webView$didReceiveTitle$forFrame$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, NSString *title, WebFrame *frame) {
 191     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 192     if ([uiWebView respondsToSelector:@selector(webView:didReceiveTitle:forFrame:)])
 193         [uiWebView webView:view didReceiveTitle:title forFrame:frame];
 194 }
 195
 196 - (void) webView:(WebView *)view didReceiveTitle:(NSString *)title forFrame:(WebFrame *)frame {
 197     id<CYWebViewDelegate> delegate([self delegate]);
 198     if ([delegate respondsToSelector:@selector(webView:didReceiveTitle:forFrame:)])
 199         [delegate webView:view didReceiveTitle:title forFrame:frame];
 200     if ([UIWebView instancesRespondToSelector:@selector(webView:didReceiveTitle:forFrame:)])
 201         [super webView:view didReceiveTitle:title forFrame:frame];
 202 }
 203 // }}}
 204 // webView:didStartProvisionalLoadForFrame: (2.0+) {{{
 205 - (void) webView:(WebView *)view didStartProvisionalLoadForFrame:(WebFrame *)frame {
 206     id<CYWebViewDelegate> delegate([self delegate]);
 207     if ([delegate respondsToSelector:@selector(webView:didStartProvisionalLoadForFrame:)])
 208         [delegate webView:view didStartProvisionalLoadForFrame:frame];
 209     if ([UIWebView instancesRespondToSelector:@selector(webView:didStartProvisionalLoadForFrame:)])
 210         [super webView:view didStartProvisionalLoadForFrame:frame];
 211 }
 212 // }}}
 213 // webView:resource:willSendRequest:redirectResponse:fromDataSource: (3.2+) {{{
 214 static NSURLRequest *$UIWebViewWebViewDelegate$webView$resource$willSendRequest$redirectResponse$fromDataSource$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view, id identifier, NSURLRequest *request, NSURLResponse *response, WebDataSource *source) {
 215     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 216     if ([uiWebView respondsToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
 217         request = [uiWebView webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
 218     return request;
 219 }
 220
 221 - (NSURLRequest *) webView:(WebView *)view resource:(id)identifier willSendRequest:(NSURLRequest *)request redirectResponse:(NSURLResponse *)response fromDataSource:(WebDataSource *)source {
 222     id<CYWebViewDelegate> delegate([self delegate]);
 223     if ([UIWebView instancesRespondToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
 224         request = [super webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
 225     if ([delegate respondsToSelector:@selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:)])
 226         request = [delegate webView:view resource:identifier willSendRequest:request redirectResponse:response fromDataSource:source];
 227     return request;
 228 }
 229 // }}}
 230 // webView:runJavaScriptAlertPanelWithMessage:initiatedByFrame: (2.1+) {{{
 231 - (void) webView:(WebView *)view runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
 232     id<CYWebViewDelegate> delegate([self delegate]);
 233     if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptAlertPanelWithMessage:initiatedByFrame:)])
 234         if (
 235             ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptAlertPanelWithMessage:initiatedByFrame:)] ||
 236             [delegate webView:view shouldRunJavaScriptAlertPanelWithMessage:message initiatedByFrame:frame]
 237         )
 238             [super webView:view runJavaScriptAlertPanelWithMessage:message initiatedByFrame:frame];
 239 }
 240 // }}}
 241 // webView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame: (2.1+) {{{
 242 - (BOOL) webView:(WebView *)view runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(WebFrame *)frame {
 243     id<CYWebViewDelegate> delegate([self delegate]);
 244     if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame:)])
 245         if (
 246             ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptConfirmPanelWithMessage:initiatedByFrame:)] ||
 247             [delegate webView:view shouldRunJavaScriptConfirmPanelWithMessage:message initiatedByFrame:frame]
 248         )
 249             return [super webView:view runJavaScriptConfirmPanelWithMessage:message initiatedByFrame:frame];
 250     return NO;
 251 }
 252 // }}}
 253 // webView:runJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame: (2.1+) {{{
 254 - (NSString *) webView:(WebView *)view runJavaScriptTextInputPanelWithPrompt:(NSString *)prompt defaultText:(NSString *)text initiatedByFrame:(WebFrame *)frame {
 255     id<CYWebViewDelegate> delegate([self delegate]);
 256     if ([UIWebView instancesRespondToSelector:@selector(webView:runJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame:)])
 257         if (
 258             ![delegate respondsToSelector:@selector(webView:shouldRunJavaScriptTextInputPanelWithPrompt:defaultText:initiatedByFrame:)] ||
 259             [delegate webView:view shouldRunJavaScriptTextInputPanelWithPrompt:prompt defaultText:text initiatedByFrame:frame]
 260         )
 261             return [super webView:view runJavaScriptTextInputPanelWithPrompt:prompt defaultText:text initiatedByFrame:frame];
 262     return nil;
 263 }
 264 // }}}
 265 // webViewClose: (3.2+) {{{
 266 static void $UIWebViewWebViewDelegate$webViewClose$(UIWebViewWebViewDelegate *self, SEL sel, WebView *view) {
 267     UIWebView *uiWebView(MSHookIvar<UIWebView *>(self, "uiWebView"));
 268     if ([uiWebView respondsToSelector:@selector(webViewClose:)])
 269         [uiWebView webViewClose:view];
 270 }
 271
 272 - (void) webViewClose:(WebView *)view {
 273     id<CYWebViewDelegate> delegate([self delegate]);
 274     if ([delegate respondsToSelector:@selector(webViewClose:)])
 275         [delegate webViewClose:view];
 276     if ([UIWebView instancesRespondToSelector:@selector(webViewClose:)])
 277         [super webViewClose:view];
 278 }
 279 // }}}
 280
 281 - (void) _updateViewSettings {
 282     [super _updateViewSettings];
 283
 284     id<CYWebViewDelegate> delegate([self delegate]);
 285     if ([delegate respondsToSelector:@selector(webViewUpdateViewSettings:)])
 286         [delegate webViewUpdateViewSettings:self];
 287 }
 288
 289 + (void) initialize {
 290     if (Class $UIWebViewWebViewDelegate = objc_getClass("UIWebViewWebViewDelegate")) {
 291         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:addMessageToConsole:), (IMP) &$UIWebViewWebViewDelegate$webView$addMessageToConsole$, "v16@0:4@8@12");
 292         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:decidePolicyForNewWindowAction:request:newFrameName:decisionListener:), (IMP) &$UIWebViewWebViewDelegate$webView$decidePolicyForNewWindowAction$request$newFrameName$decisionListener$, "v28@0:4@8@12@16@20@24");
 293         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didClearWindowObject:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didClearWindowObject$forFrame$, "v20@0:4@8@12@16");
 294         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:didReceiveTitle:forFrame:), (IMP) &$UIWebViewWebViewDelegate$webView$didReceiveTitle$forFrame$, "v20@0:4@8@12@16");
 295         class_addMethod($UIWebViewWebViewDelegate, @selector(webView:resource:willSendRequest:redirectResponse:fromDataSource:), (IMP) &$UIWebViewWebViewDelegate$webView$resource$willSendRequest$redirectResponse$fromDataSource$, "@28@0:4@8@12@16@20@24");
 296         class_addMethod($UIWebViewWebViewDelegate, @selector(webViewClose:), (IMP) &$UIWebViewWebViewDelegate$webViewClose$, "v12@0:4@8");
 297     }
 298 }
 299
 300 @end