Merge pull request sparkle-project#489 from mgunneras/http-headers

kornelski · kornelski · commit f216ef7a1387 · 2015-02-26T01:10:55.000Z
Ability to add custom http headers to appcast request
diff --git a/Sparkle/SUAppcast.h b/Sparkle/SUAppcast.h
@@ -19,6 +19,7 @@ SU_EXPORT @interface SUAppcast : NSObject <NSURLDownloadDelegate>
 
 @property (weak) id<SUAppcastDelegate> delegate;
 @property (copy) NSString *userAgentString;
+@property (copy) NSDictionary *httpHeaders;
 
 - (void)fetchAppcastFromURL:(NSURL *)url;
 
diff --git a/Sparkle/SUAppcast.m b/Sparkle/SUAppcast.m
@@ -45,6 +45,7 @@ @implementation SUAppcast
 @synthesize downloadFilename;
 @synthesize delegate;
 @synthesize userAgentString;
+@synthesize httpHeaders;
 @synthesize download;
 @synthesize items;
 
@@ -55,6 +56,13 @@ - (void)fetchAppcastFromURL:(NSURL *)url
         [request setValue:self.userAgentString forHTTPHeaderField:@"User-Agent"];
     }
 
+    if (self.httpHeaders) {
+        for (NSString *key in self.httpHeaders) {
+            id value = [self.httpHeaders objectForKey:key];
+            [request setValue:value forHTTPHeaderField:key];
+        }
+    }
+
     [request setValue:@"application/rss+xml,*/*;q=0.1" forHTTPHeaderField:@"Accept"];
 
     self.download = [[NSURLDownload alloc] initWithRequest:request delegate:self];
diff --git a/Sparkle/SUBasicUpdateDriver.m b/Sparkle/SUBasicUpdateDriver.m
@@ -56,6 +56,7 @@ - (void)checkForUpdatesAtURL:(NSURL *)URL host:(SUHost *)aHost
 
     [appcast setDelegate:self];
     [appcast setUserAgentString:[self.updater userAgentString]];
+    [appcast setHttpHeaders:[self.updater httpHeaders]];
     [appcast fetchAppcastFromURL:URL];
 }
 
diff --git a/Sparkle/SUUpdater.h b/Sparkle/SUUpdater.h
@@ -46,6 +46,8 @@ SU_EXPORT @interface SUUpdater : NSObject
 
 @property (nonatomic, copy) NSString *userAgentString;
 
+@property (copy) NSDictionary *httpHeaders;
+
 @property BOOL sendsSystemProfile;
 
 @property BOOL automaticallyDownloadsUpdates;
diff --git a/Sparkle/SUUpdater.m b/Sparkle/SUUpdater.m
@@ -51,7 +51,7 @@ @implementation SUUpdater
 @synthesize delegate;
 @synthesize checkTimer;
 @synthesize userAgentString = customUserAgentString;
-
+@synthesize httpHeaders;
 @synthesize driver;
 @synthesize host;
 

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,7 @@ - (void)checkForUpdatesAtURL:(NSURL )URL host:(SUHost )aHost`
`56`	`56`
`57`	`57`	`[appcast setDelegate:self];`
`58`	`58`	`[appcast setUserAgentString:[self.updater userAgentString]];`
	`59`	`+ [appcast setHttpHeaders:[self.updater httpHeaders]];`
`59`	`60`	`[appcast fetchAppcastFromURL:URL];`
`60`	`61`	`}`
`61`	`62`