Add back the original X-Unique-ID behavior for Add Source.

[cydia.git] / CyteKit / WebViewController.mm

diff --git a/CyteKit/WebViewController.mm b/CyteKit/WebViewController.mm
index 2298d75c3a3c451ed38d3d6c7811223fb206d22d..2d489b3507360ca9a5765e3019b4b7e9ecb50b97 100644 (file)
--- a/CyteKit/WebViewController.mm
+++ b/CyteKit/WebViewController.mm
@@ -5,6 +5,7 @@
 
 #include "iPhonePrivate.h"
 
 
 #include "iPhonePrivate.h"
 

+#include "CyteKit/IndirectDelegate.h"

 #include "CyteKit/Localize.h"
 #include "CyteKit/WebViewController.h"
 #include "CyteKit/PerlCompatibleRegEx.hpp"
 #include "CyteKit/Localize.h"
 #include "CyteKit/WebViewController.h"
 #include "CyteKit/PerlCompatibleRegEx.hpp"


@@ -16,8 +17,8 @@ extern NSString * const kCAFilterNearest;
 
 #include <WebCore/WebCoreThread.h>
 
 
 #include <WebCore/WebCoreThread.h>
 

-#include <WebKit/WebKitErrors.h>
-#include <WebKit/WebPreferences.h>
+#import <WebKit/WebKitErrors.h>
+#import <WebKit/WebPreferences.h>

 
 #include <WebKit/DOMCSSPrimitiveValue.h>
 #include <WebKit/DOMCSSStyleDeclaration.h>
 
 #include <WebKit/DOMCSSPrimitiveValue.h>
 #include <WebKit/DOMCSSStyleDeclaration.h>


@@ -47,12 +48,6 @@ float CYScrollViewDecelerationRateNormal;
 - (void) _setAllowsMessaging:(BOOL)allows;
 @end
 
 - (void) _setAllowsMessaging:(BOOL)allows;
 @end
 

-@interface WebPreferences (Apple)
-+ (void) _setInitialDefaultTextEncodingToSystemEncoding;
-- (void) _setLayoutInterval:(NSInteger)interval;
-- (void) setOfflineWebApplicationCacheEnabled:(BOOL)enabled;
-@end
-

 @implementation WebFrame (Cydia)
 
 - (NSString *) description {
 @implementation WebFrame (Cydia)
 
 - (NSString *) description {


@@ -62,14 +57,6 @@ float CYScrollViewDecelerationRateNormal;
 @end
 
 /* Indirect Delegate {{{ */
 @end
 
 /* Indirect Delegate {{{ */

-@interface IndirectDelegate : NSObject {
-    _transient volatile id delegate_;
-}
-
-- (void) setDelegate:(id)delegate;
-- (id) initWithDelegate:(id)delegate;
-@end
-

 @implementation IndirectDelegate
 
 - (void) setDelegate:(id)delegate {
 @implementation IndirectDelegate
 
 - (void) setDelegate:(id)delegate {


@@ -148,14 +135,21 @@ float CYScrollViewDecelerationRateNormal;
     return true;
 }
 
     return true;
 }
 

+- (void) releaseNetworkActivityIndicator {
+    if ([loading_ count] != 0) {
+        [loading_ removeAllObjects];
+
+        if ([self retainsNetworkActivityIndicator])
+            [delegate_ releaseNetworkActivityIndicator];
+    }
+}
+

 - (void) dealloc {
 #if LogBrowser
     NSLog(@"[CyteWebViewController dealloc]");
 #endif
 
 - (void) dealloc {
 #if LogBrowser
     NSLog(@"[CyteWebViewController dealloc]");
 #endif
 

-    if ([loading_ count] != 0)
-        if ([self retainsNetworkActivityIndicator])
-            [delegate_ releaseNetworkActivityIndicator];
+    [self releaseNetworkActivityIndicator];

 
     [super dealloc];
 }
 
     [super dealloc];
 }


@@ -172,12 +166,16 @@ float CYScrollViewDecelerationRateNormal;
     return url;
 }
 
     return url;
 }
 

-- (NSURLRequest *) requestWithURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {
-    return [NSURLRequest
+- (NSURLRequest *) requestWithURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy referrer:(NSString *)referrer {
+    NSMutableURLRequest *request([NSMutableURLRequest

         requestWithURL:[self URLWithURL:url]
         cachePolicy:policy
         timeoutInterval:DefaultTimeout_
         requestWithURL:[self URLWithURL:url]
         cachePolicy:policy
         timeoutInterval:DefaultTimeout_

-    ];
+    ]);
+
+    [request setValue:referrer forHTTPHeaderField:@"Referer"];
+
+    return request;

 }
 
 - (void) setRequest:(NSURLRequest *)request {
 }
 
 - (void) setRequest:(NSURLRequest *)request {


@@ -186,11 +184,15 @@ float CYScrollViewDecelerationRateNormal;
 }
 
 - (void) setURL:(NSURL *)url {
 }
 
 - (void) setURL:(NSURL *)url {

-    [self setRequest:[self requestWithURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy]];
+    [self setURL:url withReferrer:nil];
+}
+
+- (void) setURL:(NSURL *)url withReferrer:(NSString *)referrer {
+    [self setRequest:[self requestWithURL:url cachePolicy:NSURLRequestUseProtocolCachePolicy referrer:referrer]];

 }
 
 - (void) loadURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {
 }
 
 - (void) loadURL:(NSURL *)url cachePolicy:(NSURLRequestCachePolicy)policy {

-    [self loadRequest:[self requestWithURL:url cachePolicy:policy]];
+    [self loadRequest:[self requestWithURL:url cachePolicy:policy referrer:nil]];

 }
 
 - (void) loadURL:(NSURL *)url {
 }
 
 - (void) loadURL:(NSURL *)url {


@@ -218,7 +220,7 @@ float CYScrollViewDecelerationRateNormal;
 
     request_ = request;
 
 
     request_ = request;
 

-    if ([request_ HTTPBody] == nil && [request_ HTTPBodyStream] == nil)
+    if (cache || [request_ HTTPBody] == nil && [request_ HTTPBodyStream] == nil)

         [self loadRequest:request_];
     else {
         UIAlertView *alert = [[[UIAlertView alloc]
         [self loadRequest:request_];
     else {
         UIAlertView *alert = [[[UIAlertView alloc]


@@ -236,10 +238,6 @@ float CYScrollViewDecelerationRateNormal;
     }
 }
 
     }
 }
 

-- (void) reloadURL {
-    [self reloadURLWithCache:YES];
-}
-

 - (void) reloadData {
     [super reloadData];
 
 - (void) reloadData {
     [super reloadData];
 


@@ -352,7 +350,11 @@ float CYScrollViewDecelerationRateNormal;
 }
 
 - (void) _didFailWithError:(NSError *)error forFrame:(WebFrame *)frame {
 }
 
 - (void) _didFailWithError:(NSError *)error forFrame:(WebFrame *)frame {

-    [loading_ removeObject:[NSValue valueWithNonretainedObject:frame]];
+    NSValue *object([NSValue valueWithNonretainedObject:frame]);
+    if (![loading_ containsObject:object])
+        return;
+    [loading_ removeObject:object];
+

     [self _didFinishLoading];
 
     if ([[error domain] isEqualToString:NSURLErrorDomain] && [error code] == NSURLErrorCancelled)
     [self _didFinishLoading];
 
     if ([[error domain] isEqualToString:NSURLErrorDomain] && [error code] == NSURLErrorCancelled)


@@ -373,11 +375,20 @@ float CYScrollViewDecelerationRateNormal;
     }
 }
 
     }
 }
 

-- (void) pushRequest:(NSURLRequest *)request asPop:(bool)pop {
+- (void) pushRequest:(NSURLRequest *)request forAction:(NSDictionary *)action asPop:(bool)pop {
+    WebFrame *frame(nil);
+    if (NSDictionary *WebActionElement = [action objectForKey:@"WebActionElementKey"])
+        frame = [WebActionElement objectForKey:@"WebElementFrame"];
+    if (frame == nil)
+        frame = [[[[self webView] _documentView] webView] mainFrame];
+
+    WebDataSource *source([frame provisionalDataSource] ?: [frame dataSource]);
+    NSString *referrer([request valueForHTTPHeaderField:@"Referer"] ?: [[[source request] URL] absoluteString]);
+

     NSURL *url([request URL]);
 
     // XXX: filter to internal usage?
     NSURL *url([request URL]);
 
     // XXX: filter to internal usage?

-    CyteViewController *page([delegate_ pageForURL:url forExternal:NO]);
+    CyteViewController *page([delegate_ pageForURL:url forExternal:NO withReferrer:referrer]);

 
     if (page == nil) {
         CyteWebViewController *browser([[[class_ alloc] init] autorelease]);
 
     if (page == nil) {
         CyteWebViewController *browser([[[class_ alloc] init] autorelease]);


@@ -425,7 +436,7 @@ float CYScrollViewDecelerationRateNormal;
 
 - (void) webView:(WebView *)view decidePolicyForNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
 #if LogBrowser
 
 - (void) webView:(WebView *)view decidePolicyForNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
 #if LogBrowser

-    NSLog(@"decidePolicyForNavigationAction:%@ request:%@ frame:%@", action, request, frame);
+    NSLog(@"decidePolicyForNavigationAction:%@ request:%@ %@ frame:%@", action, request, [request allHTTPHeaderFields], frame);

 #endif
 
     if ([frame parentFrame] == nil) {
 #endif
 
     if ([frame parentFrame] == nil) {


@@ -434,7 +445,7 @@ float CYScrollViewDecelerationRateNormal;
 
             if (request_ != nil && ![[request_ URL] isEqual:url] && ![self allowsNavigationAction]) {
                 if (url != nil)
 
             if (request_ != nil && ![[request_ URL] isEqual:url] && ![self allowsNavigationAction]) {
                 if (url != nil)

-                    [self pushRequest:request asPop:NO];
+                    [self pushRequest:request forAction:action asPop:NO];

                 [listener ignore];
             }
         }
                 [listener ignore];
             }
         }


@@ -442,29 +453,45 @@ float CYScrollViewDecelerationRateNormal;
 }
 
 - (void) webView:(WebView *)view didDecidePolicy:(CYWebPolicyDecision)decision forNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame {
 }
 
 - (void) webView:(WebView *)view didDecidePolicy:(CYWebPolicyDecision)decision forNavigationAction:(NSDictionary *)action request:(NSURLRequest *)request frame:(WebFrame *)frame {

-    if ([frame parentFrame] == nil)
-        if (decision == CYWebPolicyDecisionUse)
-            if (!error_)
-                request_ = request;
+#if LogBrowser
+    NSLog(@"didDecidePolicy:%u forNavigationAction:%@ request:%@ frame:%@", decision, action, request, [request allHTTPHeaderFields], frame);
+#endif
+
+    if ([frame parentFrame] == nil) {
+        switch (decision) {
+            case CYWebPolicyDecisionIgnore:
+                if ([[request_ URL] isEqual:[request URL]])
+                    request_ = nil;
+            break;
+
+            case CYWebPolicyDecisionUse:
+                if (!error_)
+                    request_ = request;
+            break;
+
+            default:
+            break;
+        }
+    }

 }
 
 }
 

-- (void) webView:(WebView *)view decidePolicyForNewWindowAction:(NSDictionary *)action request:(NSURLRequest *)request newFrameName:(NSString *)frame decisionListener:(id<WebPolicyDecisionListener>)listener {
+- (void) webView:(WebView *)view decidePolicyForNewWindowAction:(NSDictionary *)action request:(NSURLRequest *)request newFrameName:(NSString *)name decisionListener:(id<WebPolicyDecisionListener>)listener {

 #if LogBrowser
 #if LogBrowser

-    NSLog(@"decidePolicyForNewWindowAction:%@ request:%@ newFrameName:%@", action, request, frame);
+    NSLog(@"decidePolicyForNewWindowAction:%@ request:%@ %@ newFrameName:%@", action, request, [request allHTTPHeaderFields], name);

 #endif
 
     NSURL *url([request URL]);
     if (url == nil)
         return;
 
 #endif
 
     NSURL *url([request URL]);
     if (url == nil)
         return;
 

-    if ([frame isEqualToString:@"_open"])
+    if ([name isEqualToString:@"_open"])

         [delegate_ openURL:url];
     else {
         NSString *scheme([[url scheme] lowercaseString]);
         if ([scheme isEqualToString:@"mailto"])
             [self _openMailToURL:url];
         else
         [delegate_ openURL:url];
     else {
         NSString *scheme([[url scheme] lowercaseString]);
         if ([scheme isEqualToString:@"mailto"])
             [self _openMailToURL:url];
         else

-            [self pushRequest:request asPop:[frame isEqualToString:@"_popup"]];
+            [self pushRequest:request forAction:action asPop:[name isEqualToString:@"_popup"]];

     }
 
     [listener ignore];
     }
 
     [listener ignore];


@@ -500,7 +527,10 @@ float CYScrollViewDecelerationRateNormal;
 }
 
 - (void) webView:(WebView *)view didFinishLoadForFrame:(WebFrame *)frame {
 }
 
 - (void) webView:(WebView *)view didFinishLoadForFrame:(WebFrame *)frame {

-    [loading_ removeObject:[NSValue valueWithNonretainedObject:frame]];
+    NSValue *object([NSValue valueWithNonretainedObject:frame]);
+    if (![loading_ containsObject:object])
+        return;
+    [loading_ removeObject:object];

 
     if ([frame parentFrame] == nil) {
         if (DOMDocument *document = [frame DOMDocument])
 
     if ([frame parentFrame] == nil) {
         if (DOMDocument *document = [frame DOMDocument])


@@ -906,6 +936,8 @@ float CYScrollViewDecelerationRateNormal;
     webview_ = nil;
     scroller_ = nil;
 
     webview_ = nil;
     scroller_ = nil;
 

+    [self releaseNetworkActivityIndicator];
+

     [super releaseSubviews];
 }
 
     [super releaseSubviews];
 }
 


@@ -923,6 +955,12 @@ float CYScrollViewDecelerationRateNormal;
     } return self;
 }
 
     } return self;
 }
 

+- (id) initWithRequest:(NSURLRequest *)request {
+    if ((self = [self init]) != nil) {
+        [self setRequest:request];
+    } return self;
+}
+

 - (void) callFunction:(WebScriptObject *)function {
     WebThreadLocked lock;
 
 - (void) callFunction:(WebScriptObject *)function {
     WebThreadLocked lock;
 


@@ -935,7 +973,7 @@ float CYScrollViewDecelerationRateNormal;
 }
 
 - (void) reloadButtonClicked {
 }
 
 - (void) reloadButtonClicked {

-    [self reloadURLWithCache:YES];
+    [self reloadURLWithCache:NO];

 }
 
 - (void) _customButtonClicked {
 }
 
 - (void) _customButtonClicked {


@@ -1051,6 +1089,9 @@ float CYScrollViewDecelerationRateNormal;
     if ([self hidesNavigationBar])
         [self _setHidesNavigationBar:YES animated:animated];
 
     if ([self hidesNavigationBar])
         [self _setHidesNavigationBar:YES animated:animated];
 

+    // XXX: why isn't this evern called automatically?
+    [[self webView] setNeedsLayout];
+

     [self dispatchEvent:@"CydiaViewWillAppear"];
     [super viewWillAppear:animated];
 }
     [self dispatchEvent:@"CydiaViewWillAppear"];
     [super viewWillAppear:animated];
 }