Alamofire.swift 43 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969
  1. // Alamofire.swift
  2. //
  3. // Copyright (c) 2014 Alamofire (http://alamofire.org)
  4. //
  5. // Permission is hereby granted, free of charge, to any person obtaining a copy
  6. // of this software and associated documentation files (the "Software"), to deal
  7. // in the Software without restriction, including without limitation the rights
  8. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. // copies of the Software, and to permit persons to whom the Software is
  10. // furnished to do so, subject to the following conditions:
  11. //
  12. // The above copyright notice and this permission notice shall be included in
  13. // all copies or substantial portions of the Software.
  14. //
  15. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21. // THE SOFTWARE.
  22. import Foundation
  23. public struct Alamofire {
  24. // HTTP Method Definitions; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec9.html
  25. public enum Method: String {
  26. case OPTIONS = "OPTIONS"
  27. case GET = "GET"
  28. case HEAD = "HEAD"
  29. case POST = "POST"
  30. case PUT = "PUT"
  31. case PATCH = "PATCH"
  32. case DELETE = "DELETE"
  33. case TRACE = "TRACE"
  34. case CONNECT = "CONNECT"
  35. }
  36. public enum ParameterEncoding {
  37. case URL
  38. case JSON(options: NSJSONWritingOptions)
  39. case PropertyList(format: NSPropertyListFormat, options: NSPropertyListWriteOptions)
  40. func encode(request: NSURLRequest, parameters: [String: AnyObject]?) -> (NSURLRequest, NSError?) {
  41. if !parameters {
  42. return (request, nil)
  43. }
  44. var mutableRequest: NSMutableURLRequest! = request.mutableCopy() as NSMutableURLRequest
  45. var error: NSError? = nil
  46. switch self {
  47. case .URL:
  48. func query(parameters: [String: AnyObject]) -> String! {
  49. func queryComponents(key: String, value: AnyObject) -> [(String, String)] {
  50. func dictionaryQueryComponents(key: String, dictionary: [String: AnyObject]) -> [(String, String)] {
  51. var components: [(String, String)] = []
  52. for (nestedKey, value) in dictionary {
  53. components += queryComponents("\(key)[\(nestedKey)]", value)
  54. }
  55. return components
  56. }
  57. func arrayQueryComponents(key: String, array: [AnyObject]) -> [(String, String)] {
  58. var components: [(String, String)] = []
  59. for value in array {
  60. components += queryComponents("\(key)[]", value)
  61. }
  62. return components
  63. }
  64. var components: [(String, String)] = []
  65. if let dictionary = value as? [String: AnyObject] {
  66. components += dictionaryQueryComponents(key, dictionary)
  67. } else if let array = value as? [AnyObject] {
  68. components += arrayQueryComponents(key, array)
  69. } else {
  70. components += (key, "\(value)")
  71. }
  72. return components
  73. }
  74. var components: [(String, String)] = []
  75. for key in sorted(Array(parameters.keys), <) {
  76. let value: AnyObject! = parameters[key]
  77. components += queryComponents(key, value)
  78. }
  79. return join("&", components.map{"\($0)=\($1)"} as [String])
  80. }
  81. func encodesParametersInURL(method: Method) -> Bool {
  82. switch method {
  83. case .GET, .HEAD, .DELETE:
  84. return true
  85. default:
  86. return false
  87. }
  88. }
  89. if encodesParametersInURL(Method.fromRaw(request.HTTPMethod)!) {
  90. let URLComponents = NSURLComponents(URL: mutableRequest.URL, resolvingAgainstBaseURL: false)
  91. URLComponents.query = (URLComponents.query ? URLComponents.query + "&" : "") + query(parameters!)
  92. mutableRequest.URL = URLComponents.URL
  93. } else {
  94. if !mutableRequest.valueForHTTPHeaderField("Content-Type") {
  95. mutableRequest.setValue("application/x-www-form-urlencoded", forHTTPHeaderField: "Content-Type")
  96. }
  97. mutableRequest.HTTPBody = query(parameters!).dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: false)
  98. }
  99. case .JSON(let options):
  100. let data = NSJSONSerialization.dataWithJSONObject(parameters, options: options, error: &error)
  101. if data {
  102. let charset = CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(NSUTF8StringEncoding))
  103. mutableRequest.setValue("application/json; charset=\(charset)", forHTTPHeaderField: "Content-Type")
  104. mutableRequest.HTTPBody = data
  105. }
  106. case .PropertyList(let (format, options)):
  107. let data = NSPropertyListSerialization.dataWithPropertyList(parameters, format: format, options: options, error: &error)
  108. if data {
  109. let charset = CFStringConvertEncodingToIANACharSetName(CFStringConvertNSStringEncodingToEncoding(NSUTF8StringEncoding));
  110. mutableRequest.setValue("application/x-plist; charset=\(charset)", forHTTPHeaderField: "Content-Type")
  111. mutableRequest.HTTPBody = data
  112. }
  113. }
  114. return (mutableRequest, error)
  115. }
  116. }
  117. // MARK: -
  118. class Manager {
  119. class var sharedInstance: Manager {
  120. struct Singleton {
  121. static let instance = Manager()
  122. }
  123. return Singleton.instance
  124. }
  125. let delegate: SessionDelegate
  126. let session: NSURLSession!
  127. let operationQueue: NSOperationQueue = NSOperationQueue()
  128. var automaticallyStartsRequests: Bool = true
  129. lazy var defaultHeaders: [String: String] = {
  130. // Accept-Language HTTP Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.3
  131. let acceptEncoding: String = "gzip;q=1.0,compress;q=0.5"
  132. // Accept-Language HTTP Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.4
  133. let acceptLanguage: String = {
  134. var components: [String] = []
  135. for (index, languageCode) in enumerate(NSLocale.preferredLanguages() as [String]) {
  136. let q = 1.0 - (Double(index) * 0.1)
  137. components += "\(languageCode);q=\(q)"
  138. if q <= 0.5 {
  139. break
  140. }
  141. }
  142. return components.reduce("", {$0 == "" ? $1 : "\($0),\($1)"})
  143. }()
  144. // User-Agent Header; see http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.43
  145. let userAgent: String = {
  146. if let info = NSBundle.mainBundle().infoDictionary {
  147. let executable: AnyObject? = info[kCFBundleExecutableKey]
  148. let bundle: AnyObject? = info[kCFBundleIdentifierKey]
  149. let version: AnyObject? = info[kCFBundleVersionKey]
  150. let os: AnyObject? = NSProcessInfo.processInfo()?.operatingSystemVersionString
  151. var mutableUserAgent = NSMutableString(string: "\(executable!)/\(bundle!) (\(version!); OS \(os!))") as CFMutableString
  152. let transform = NSString(string: "Any-Latin; Latin-ASCII; [:^ASCII:] Remove") as CFString
  153. if CFStringTransform(mutableUserAgent, nil, transform, 0) == 1 {
  154. return mutableUserAgent as NSString
  155. }
  156. }
  157. return "Alamofire"
  158. }()
  159. return ["Accept-Encoding": acceptEncoding,
  160. "Accept-Language": acceptLanguage,
  161. "User-Agent": userAgent]
  162. }()
  163. required init(configuration: NSURLSessionConfiguration! = nil) {
  164. self.delegate = SessionDelegate()
  165. self.session = NSURLSession(configuration: configuration, delegate: self.delegate, delegateQueue: self.operationQueue)
  166. }
  167. deinit {
  168. self.session.invalidateAndCancel()
  169. }
  170. // MARK: -
  171. func request(request: NSURLRequest) -> Request {
  172. var mutableRequest: NSMutableURLRequest! = request.mutableCopy() as NSMutableURLRequest
  173. for (field, value) in self.defaultHeaders {
  174. if !mutableRequest.valueForHTTPHeaderField(field){
  175. mutableRequest.setValue(value, forHTTPHeaderField: field)
  176. }
  177. }
  178. var dataTask: NSURLSessionDataTask?
  179. dispatch_sync(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0)) {
  180. dataTask = self.session.dataTaskWithRequest(mutableRequest)
  181. }
  182. let request = Request(session: self.session, task: dataTask!)
  183. self.delegate[request.delegate.task] = request.delegate
  184. request.resume()
  185. return request
  186. }
  187. class SessionDelegate: NSObject, NSURLSessionDelegate, NSURLSessionTaskDelegate, NSURLSessionDataDelegate, NSURLSessionDownloadDelegate {
  188. private var subdelegates: [Int: Request.TaskDelegate]
  189. private subscript(task: NSURLSessionTask) -> Request.TaskDelegate? {
  190. get {
  191. return self.subdelegates[task.taskIdentifier]
  192. }
  193. set(newValue) {
  194. self.subdelegates[task.taskIdentifier] = newValue
  195. }
  196. }
  197. var sessionDidBecomeInvalidWithError: ((NSURLSession!, NSError!) -> Void)?
  198. var sessionDidFinishEventsForBackgroundURLSession: ((NSURLSession!) -> Void)?
  199. var sessionDidReceiveChallenge: ((NSURLSession!, NSURLAuthenticationChallenge) -> (NSURLSessionAuthChallengeDisposition, NSURLCredential!))?
  200. var taskWillPerformHTTPRedirection: ((NSURLSession!, NSURLSessionTask!, NSHTTPURLResponse!, NSURLRequest!) -> (NSURLRequest!))?
  201. var taskDidReceiveChallenge: ((NSURLSession!, NSURLSessionTask!, NSURLAuthenticationChallenge) -> (NSURLSessionAuthChallengeDisposition, NSURLCredential?))?
  202. var taskDidSendBodyData: ((NSURLSession!, NSURLSessionTask!, Int64, Int64, Int64) -> Void)?
  203. var taskNeedNewBodyStream: ((NSURLSession!, NSURLSessionTask!) -> (NSInputStream!))?
  204. var dataTaskDidReceiveResponse: ((NSURLSession!, NSURLSessionDataTask!, NSURLResponse!) -> (NSURLSessionResponseDisposition))?
  205. var dataTaskDidBecomeDownloadTask: ((NSURLSession!, NSURLSessionDataTask!) -> Void)?
  206. var dataTaskDidReceiveData: ((NSURLSession!, NSURLSessionDataTask!, NSData!) -> Void)?
  207. var dataTaskWillCacheResponse: ((NSURLSession!, NSURLSessionDataTask!, NSCachedURLResponse!) -> (NSCachedURLResponse))?
  208. var downloadTaskDidFinishDownloadingToURL: ((NSURLSession!, NSURLSessionDownloadTask!, NSURL) -> (NSURL))?
  209. var downloadTaskDidWriteData: ((NSURLSession!, NSURLSessionDownloadTask!, Int64, Int64, Int64) -> Void)?
  210. var downloadTaskDidResumeAtOffset: ((NSURLSession!, NSURLSessionDownloadTask!, Int64, Int64) -> Void)?
  211. required init() {
  212. self.subdelegates = Dictionary()
  213. super.init()
  214. }
  215. // MARK: NSURLSessionDelegate
  216. func URLSession(session: NSURLSession!, didBecomeInvalidWithError error: NSError!) {
  217. self.sessionDidBecomeInvalidWithError?(session, error)
  218. }
  219. func URLSession(session: NSURLSession!, didReceiveChallenge challenge: NSURLAuthenticationChallenge!, completionHandler: ((NSURLSessionAuthChallengeDisposition, NSURLCredential!) -> Void)!) {
  220. if self.sessionDidReceiveChallenge {
  221. completionHandler(self.sessionDidReceiveChallenge!(session, challenge))
  222. } else {
  223. completionHandler(.PerformDefaultHandling, nil)
  224. }
  225. }
  226. func URLSessionDidFinishEventsForBackgroundURLSession(session: NSURLSession!) {
  227. self.sessionDidFinishEventsForBackgroundURLSession?(session)
  228. }
  229. // MARK: NSURLSessionTaskDelegate
  230. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, willPerformHTTPRedirection response: NSHTTPURLResponse!, newRequest request: NSURLRequest!, completionHandler: ((NSURLRequest!) -> Void)!) {
  231. var redirectRequest = request
  232. if self.taskWillPerformHTTPRedirection {
  233. redirectRequest = self.taskWillPerformHTTPRedirection!(session, task, response, request)
  234. }
  235. completionHandler(redirectRequest)
  236. }
  237. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didReceiveChallenge challenge: NSURLAuthenticationChallenge!, completionHandler: ((NSURLSessionAuthChallengeDisposition, NSURLCredential!) -> Void)!) {
  238. if let delegate = self[task] {
  239. delegate.URLSession(session, task: task, didReceiveChallenge: challenge, completionHandler: completionHandler)
  240. } else {
  241. self.URLSession(session, didReceiveChallenge: challenge, completionHandler: completionHandler)
  242. }
  243. }
  244. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, needNewBodyStream completionHandler: ((NSInputStream!) -> Void)!) {
  245. if let delegate = self[task] {
  246. delegate.URLSession(session, task: task, needNewBodyStream: completionHandler)
  247. }
  248. }
  249. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didSendBodyData bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) {
  250. if let delegate = self[task] as? Request.UploadTaskDelegate {
  251. delegate.URLSession(session, task: task, didSendBodyData: bytesSent, totalBytesSent: totalBytesSent, totalBytesExpectedToSend: totalBytesExpectedToSend)
  252. }
  253. }
  254. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didCompleteWithError error: NSError!) {
  255. if let delegate = self[task] {
  256. delegate.URLSession(session, task: task, didCompleteWithError: error)
  257. }
  258. }
  259. // MARK: NSURLSessionDataDelegate
  260. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didReceiveResponse response: NSURLResponse!, completionHandler: ((NSURLSessionResponseDisposition) -> Void)!) {
  261. var disposition: NSURLSessionResponseDisposition = .Allow
  262. if self.dataTaskDidReceiveResponse {
  263. disposition = self.dataTaskDidReceiveResponse!(session, dataTask, response)
  264. }
  265. completionHandler(disposition)
  266. }
  267. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didBecomeDownloadTask downloadTask: NSURLSessionDownloadTask!) {
  268. let downloadDelegate = Request.DownloadTaskDelegate(task: downloadTask)
  269. self[downloadTask] = downloadDelegate
  270. }
  271. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didReceiveData data: NSData!) {
  272. if let delegate = self[dataTask] as? Request.DataTaskDelegate {
  273. delegate.URLSession(session, dataTask: dataTask, didReceiveData: data)
  274. }
  275. self.dataTaskDidReceiveData?(session, dataTask, data)
  276. }
  277. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, willCacheResponse proposedResponse: NSCachedURLResponse!, completionHandler: ((NSCachedURLResponse!) -> Void)!) {
  278. var cachedResponse = proposedResponse
  279. if self.dataTaskWillCacheResponse {
  280. cachedResponse = self.dataTaskWillCacheResponse!(session, dataTask, proposedResponse)
  281. }
  282. completionHandler(cachedResponse)
  283. }
  284. // MARK: NSURLSessionDownloadDelegate
  285. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didFinishDownloadingToURL location: NSURL!) {
  286. if let delegate = self[downloadTask] as? Request.DownloadTaskDelegate {
  287. delegate.URLSession(session, downloadTask: downloadTask, didFinishDownloadingToURL: location)
  288. }
  289. self.downloadTaskDidFinishDownloadingToURL?(session, downloadTask, location)
  290. }
  291. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didWriteData bytesWritten: Int64, totalBytesWritten: Int64, totalBytesExpectedToWrite: Int64) {
  292. if let delegate = self[downloadTask] as? Request.DownloadTaskDelegate {
  293. delegate.URLSession(session, downloadTask: downloadTask, didWriteData: bytesWritten, totalBytesWritten: totalBytesWritten, totalBytesExpectedToWrite: totalBytesExpectedToWrite)
  294. }
  295. self.downloadTaskDidWriteData?(session, downloadTask, bytesWritten, totalBytesWritten, totalBytesExpectedToWrite)
  296. }
  297. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didResumeAtOffset fileOffset: Int64, expectedTotalBytes: Int64) {
  298. if let delegate = self[downloadTask] as? Request.DownloadTaskDelegate {
  299. delegate.URLSession(session, downloadTask: downloadTask, didResumeAtOffset: fileOffset, expectedTotalBytes: expectedTotalBytes)
  300. }
  301. self.downloadTaskDidResumeAtOffset?(session, downloadTask, fileOffset, expectedTotalBytes)
  302. }
  303. // MARK: NSObject
  304. override func respondsToSelector(selector: Selector) -> Bool {
  305. switch selector {
  306. case "URLSession:didBecomeInvalidWithError:":
  307. return self.sessionDidBecomeInvalidWithError ? true : false
  308. case "URLSession:didReceiveChallenge:completionHandler:":
  309. return self.sessionDidReceiveChallenge ? true : false
  310. case "URLSessionDidFinishEventsForBackgroundURLSession:":
  311. return self.sessionDidFinishEventsForBackgroundURLSession ? true : false
  312. case "URLSession:task:willPerformHTTPRedirection:newRequest:completionHandler:":
  313. return self.taskWillPerformHTTPRedirection ? true : false
  314. case "URLSession:dataTask:didReceiveResponse:completionHandler:":
  315. return self.dataTaskDidReceiveResponse ? true : false
  316. case "URLSession:dataTask:willCacheResponse:completionHandler:":
  317. return self.dataTaskWillCacheResponse ? true : false
  318. default:
  319. return self.dynamicType.instancesRespondToSelector(selector)
  320. }
  321. }
  322. }
  323. }
  324. // MARK: -
  325. class Request {
  326. private let delegate: TaskDelegate
  327. private var session: NSURLSession
  328. private var task: NSURLSessionTask { return self.delegate.task }
  329. var request: NSURLRequest! { return self.task.originalRequest }
  330. var response: NSHTTPURLResponse! { return self.task.response as? NSHTTPURLResponse }
  331. var progress: NSProgress? { return self.delegate.progress }
  332. private init(session: NSURLSession, task: NSURLSessionTask) {
  333. self.session = session
  334. if task is NSURLSessionUploadTask {
  335. self.delegate = UploadTaskDelegate(task: task)
  336. } else if task is NSURLSessionDownloadTask {
  337. self.delegate = DownloadTaskDelegate(task: task)
  338. } else if task is NSURLSessionDataTask {
  339. self.delegate = DataTaskDelegate(task: task)
  340. } else {
  341. self.delegate = TaskDelegate(task: task)
  342. }
  343. }
  344. // MARK: Authentication
  345. func authenticate(HTTPBasic user: String, password: String) -> Self {
  346. let credential = NSURLCredential(user: user, password: password, persistence: .ForSession)
  347. let protectionSpace = NSURLProtectionSpace(host: self.request.URL.host, port: 0, `protocol`: self.request.URL.scheme, realm: nil, authenticationMethod: NSURLAuthenticationMethodHTTPBasic)
  348. return authenticate(usingCredential: credential, forProtectionSpace: protectionSpace)
  349. }
  350. func authenticate(usingCredential credential: NSURLCredential, forProtectionSpace protectionSpace: NSURLProtectionSpace) -> Self {
  351. self.session.configuration.URLCredentialStorage.setCredential(credential, forProtectionSpace: protectionSpace)
  352. return self
  353. }
  354. // MARK: Progress
  355. func progress(closure: ((Int64, Int64, Int64) -> Void)? = nil) -> Self {
  356. if let uploadDelegate = self.delegate as? UploadTaskDelegate {
  357. uploadDelegate.uploadProgress = closure
  358. } else if let downloadDelegate = self.delegate as? DownloadTaskDelegate {
  359. downloadDelegate.downloadProgress = closure
  360. }
  361. return self
  362. }
  363. // MARK: Response
  364. func response(completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  365. return response({ (request, response, data, error) in
  366. return (data, error)
  367. }, completionHandler: completionHandler)
  368. }
  369. func response(priority: Int = DISPATCH_QUEUE_PRIORITY_DEFAULT, queue: dispatch_queue_t? = nil, serializer: (NSURLRequest, NSHTTPURLResponse?, NSData?, NSError?) -> (AnyObject?, NSError?), completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  370. dispatch_async(self.delegate.queue, {
  371. dispatch_async(dispatch_get_global_queue(priority, 0), {
  372. let (responseObject: AnyObject?, error: NSError?) = serializer(self.request, self.response, self.delegate.data, self.delegate.error)
  373. dispatch_async(queue ? queue : dispatch_get_main_queue(), {
  374. completionHandler(self.request, self.response, responseObject, error)
  375. })
  376. })
  377. })
  378. return self
  379. }
  380. func suspend() {
  381. self.task.suspend()
  382. }
  383. func resume() {
  384. self.task.resume()
  385. }
  386. func cancel() {
  387. if let downloadDelegate = self.delegate as? DownloadTaskDelegate {
  388. downloadDelegate.downloadTask.cancelByProducingResumeData { (data) in
  389. downloadDelegate.resumeData = data
  390. }
  391. } else {
  392. self.task.cancel()
  393. }
  394. }
  395. private class TaskDelegate: NSObject, NSURLSessionTaskDelegate {
  396. let task: NSURLSessionTask
  397. let queue: dispatch_queue_t?
  398. let progress: NSProgress
  399. var data: NSData! { return nil }
  400. private(set) var error: NSError?
  401. var taskWillPerformHTTPRedirection: ((NSURLSession!, NSURLSessionTask!, NSHTTPURLResponse!, NSURLRequest!) -> (NSURLRequest!))?
  402. var taskDidReceiveChallenge: ((NSURLSession!, NSURLSessionTask!, NSURLAuthenticationChallenge) -> (NSURLSessionAuthChallengeDisposition, NSURLCredential?))?
  403. var taskDidSendBodyData: ((NSURLSession!, NSURLSessionTask!, Int64, Int64, Int64) -> Void)?
  404. var taskNeedNewBodyStream: ((NSURLSession!, NSURLSessionTask!) -> (NSInputStream!))?
  405. init(task: NSURLSessionTask) {
  406. self.task = task
  407. self.progress = NSProgress(totalUnitCount: 0)
  408. let label: String = "com.alamofire.task-\(task.taskIdentifier)"
  409. let queue = dispatch_queue_create(label.bridgeToObjectiveC().UTF8String, DISPATCH_QUEUE_SERIAL)
  410. dispatch_suspend(queue)
  411. self.queue = queue
  412. }
  413. // MARK: NSURLSessionTaskDelegate
  414. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, willPerformHTTPRedirection response: NSHTTPURLResponse!, newRequest request: NSURLRequest!, completionHandler: ((NSURLRequest!) -> Void)!) {
  415. var redirectRequest = request
  416. if self.taskWillPerformHTTPRedirection {
  417. redirectRequest = self.taskWillPerformHTTPRedirection!(session, task, response, request)
  418. }
  419. completionHandler(redirectRequest)
  420. }
  421. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didReceiveChallenge challenge: NSURLAuthenticationChallenge!, completionHandler: ((NSURLSessionAuthChallengeDisposition, NSURLCredential!) -> Void)!) {
  422. var disposition: NSURLSessionAuthChallengeDisposition = .PerformDefaultHandling
  423. var credential: NSURLCredential?
  424. if self.taskDidReceiveChallenge {
  425. (disposition, credential) = self.taskDidReceiveChallenge!(session, task, challenge)
  426. } else {
  427. if challenge.protectionSpace.authenticationMethod == NSURLAuthenticationMethodServerTrust {
  428. // TODO: Incorporate Trust Evaluation & TLS Chain Validation
  429. credential = NSURLCredential(forTrust: challenge.protectionSpace.serverTrust)
  430. disposition = .UseCredential
  431. }
  432. }
  433. completionHandler(disposition, credential)
  434. }
  435. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, needNewBodyStream completionHandler: ((NSInputStream!) -> Void)!) {
  436. var bodyStream: NSInputStream?
  437. if self.taskNeedNewBodyStream {
  438. bodyStream = self.taskNeedNewBodyStream!(session, task)
  439. }
  440. completionHandler(bodyStream)
  441. }
  442. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didCompleteWithError error: NSError!) {
  443. dispatch_resume(self.queue)
  444. }
  445. }
  446. private class DataTaskDelegate: TaskDelegate, NSURLSessionDataDelegate {
  447. var dataTask: NSURLSessionDataTask! { return self.task as NSURLSessionDataTask }
  448. private var mutableData: NSMutableData
  449. override var data: NSData! {
  450. return self.mutableData
  451. }
  452. var dataTaskDidReceiveResponse: ((NSURLSession!, NSURLSessionDataTask!, NSURLResponse!) -> (NSURLSessionResponseDisposition))?
  453. var dataTaskDidBecomeDownloadTask: ((NSURLSession!, NSURLSessionDataTask!) -> Void)?
  454. var dataTaskDidReceiveData: ((NSURLSession!, NSURLSessionDataTask!, NSData!) -> Void)?
  455. var dataTaskWillCacheResponse: ((NSURLSession!, NSURLSessionDataTask!, NSCachedURLResponse!) -> (NSCachedURLResponse))?
  456. init(task: NSURLSessionTask) {
  457. self.mutableData = NSMutableData()
  458. super.init(task: task)
  459. }
  460. // MARK: NSURLSessionDataDelegate
  461. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didReceiveResponse response: NSURLResponse!, completionHandler: ((NSURLSessionResponseDisposition) -> Void)!) {
  462. var disposition: NSURLSessionResponseDisposition = .Allow
  463. if self.dataTaskDidReceiveResponse {
  464. disposition = self.dataTaskDidReceiveResponse!(session, dataTask, response)
  465. }
  466. completionHandler(disposition)
  467. }
  468. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didBecomeDownloadTask downloadTask: NSURLSessionDownloadTask!) {
  469. self.dataTaskDidBecomeDownloadTask?(session, dataTask)
  470. }
  471. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, didReceiveData data: NSData!) {
  472. self.dataTaskDidReceiveData?(session, dataTask, data)
  473. self.mutableData.appendData(data)
  474. }
  475. func URLSession(session: NSURLSession!, dataTask: NSURLSessionDataTask!, willCacheResponse proposedResponse: NSCachedURLResponse!, completionHandler: ((NSCachedURLResponse!) -> Void)!) {
  476. var cachedResponse = proposedResponse
  477. if self.dataTaskWillCacheResponse {
  478. cachedResponse = self.dataTaskWillCacheResponse!(session, dataTask, proposedResponse)
  479. }
  480. completionHandler(cachedResponse)
  481. }
  482. }
  483. }
  484. }
  485. // MARK: - Upload
  486. extension Alamofire.Manager {
  487. private enum Uploadable {
  488. case Data(NSURLRequest, NSData)
  489. case File(NSURLRequest, NSURL)
  490. case Stream(NSURLRequest, NSInputStream)
  491. }
  492. private func upload(uploadable: Uploadable) -> Alamofire.Request {
  493. var uploadTask: NSURLSessionUploadTask!
  494. var stream: NSInputStream?
  495. switch uploadable {
  496. case .Data(let request, let data):
  497. uploadTask = self.session.uploadTaskWithRequest(request, fromData: data)
  498. case .File(let request, let fileURL):
  499. uploadTask = self.session.uploadTaskWithRequest(request, fromFile: fileURL)
  500. case .Stream(let request, var stream):
  501. uploadTask = self.session.uploadTaskWithStreamedRequest(request)
  502. }
  503. let request = Alamofire.Request(session: self.session, task: uploadTask)
  504. if stream {
  505. request.delegate.taskNeedNewBodyStream = { _, _ in
  506. return stream
  507. }
  508. }
  509. self.delegate[request.delegate.task] = request.delegate
  510. if self.automaticallyStartsRequests {
  511. request.resume()
  512. }
  513. return request
  514. }
  515. // MARK: File
  516. func upload(request: NSURLRequest, file: NSURL) -> Alamofire.Request {
  517. return upload(.File(request, file))
  518. }
  519. // MARK: Data
  520. func upload(request: NSURLRequest, data: NSData) -> Alamofire.Request {
  521. return upload(.Data(request, data))
  522. }
  523. // MARK: Stream
  524. func upload(request: NSURLRequest, stream: NSInputStream) -> Alamofire.Request {
  525. return upload(.Stream(request, stream))
  526. }
  527. }
  528. extension Alamofire.Request {
  529. private class UploadTaskDelegate: DataTaskDelegate {
  530. var uploadTask: NSURLSessionUploadTask! { return self.task as NSURLSessionUploadTask }
  531. var uploadProgress: ((Int64, Int64, Int64) -> Void)!
  532. // MARK: NSURLSessionTaskDelegate
  533. func URLSession(session: NSURLSession!, task: NSURLSessionTask!, didSendBodyData bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) {
  534. if self.uploadProgress {
  535. self.uploadProgress(bytesSent, totalBytesSent, totalBytesExpectedToSend)
  536. }
  537. self.progress.totalUnitCount = totalBytesExpectedToSend
  538. self.progress.completedUnitCount = totalBytesSent
  539. }
  540. }
  541. }
  542. // MARK: - Download
  543. extension Alamofire.Manager {
  544. private enum Downloadable {
  545. case Request(NSURLRequest)
  546. case ResumeData(NSData)
  547. }
  548. private func download(downloadable: Downloadable, destination: (NSURL, NSHTTPURLResponse) -> (NSURL)) -> Alamofire.Request {
  549. var downloadTask: NSURLSessionDownloadTask!
  550. switch downloadable {
  551. case .Request(let request):
  552. downloadTask = self.session.downloadTaskWithRequest(request)
  553. case .ResumeData(let resumeData):
  554. downloadTask = self.session.downloadTaskWithResumeData(resumeData)
  555. }
  556. let request = Alamofire.Request(session: self.session, task: downloadTask)
  557. if let downloadDelegate = request.delegate as? Alamofire.Request.DownloadTaskDelegate {
  558. downloadDelegate.downloadTaskDidFinishDownloadingToURL = { (session, downloadTask, URL) in
  559. return destination(URL, downloadTask.response as NSHTTPURLResponse)
  560. }
  561. }
  562. self.delegate[request.delegate.task] = request.delegate
  563. if self.automaticallyStartsRequests {
  564. request.resume()
  565. }
  566. return request
  567. }
  568. // MARK: Request
  569. func download(request: NSURLRequest, destination: (NSURL, NSHTTPURLResponse) -> (NSURL)) -> Alamofire.Request {
  570. return download(.Request(request), destination: destination)
  571. }
  572. // MARK: Resume Data
  573. func download(resumeData: NSData, destination: (NSURL, NSHTTPURLResponse) -> (NSURL)) -> Alamofire.Request {
  574. return download(.ResumeData(resumeData), destination: destination)
  575. }
  576. }
  577. extension Alamofire.Request {
  578. class func suggestedDownloadDestination(directory: NSSearchPathDirectory = .DocumentDirectory, domain: NSSearchPathDomainMask = .UserDomainMask) -> (NSURL, NSHTTPURLResponse) -> (NSURL) {
  579. return { (temporaryURL, response) -> (NSURL) in
  580. if let directoryURL = NSFileManager.defaultManager().URLsForDirectory(.DocumentDirectory, inDomains: .UserDomainMask)[0] as? NSURL {
  581. return directoryURL.URLByAppendingPathComponent(response.suggestedFilename)
  582. }
  583. return temporaryURL
  584. }
  585. }
  586. private class DownloadTaskDelegate: TaskDelegate, NSURLSessionDownloadDelegate {
  587. var downloadTask: NSURLSessionDownloadTask! { return self.task as NSURLSessionDownloadTask }
  588. var downloadProgress: ((Int64, Int64, Int64) -> Void)?
  589. var resumeData: NSData?
  590. override var data: NSData { return self.resumeData! }
  591. var downloadTaskDidFinishDownloadingToURL: ((NSURLSession!, NSURLSessionDownloadTask!, NSURL) -> (NSURL))?
  592. var downloadTaskDidWriteData: ((NSURLSession!, NSURLSessionDownloadTask!, Int64, Int64, Int64) -> Void)?
  593. var downloadTaskDidResumeAtOffset: ((NSURLSession!, NSURLSessionDownloadTask!, Int64, Int64) -> Void)?
  594. // MARK: NSURLSessionDownloadDelegate
  595. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didFinishDownloadingToURL location: NSURL!) {
  596. if self.downloadTaskDidFinishDownloadingToURL {
  597. let destination = self.downloadTaskDidFinishDownloadingToURL!(session, downloadTask, location)
  598. var fileManagerError: NSError?
  599. NSFileManager.defaultManager().moveItemAtURL(location, toURL: destination, error: &fileManagerError)
  600. // TODO: NSNotification on failure
  601. }
  602. }
  603. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didWriteData bytesWritten: Int64, totalBytesWritten: Int64, totalBytesExpectedToWrite: Int64) {
  604. self.downloadTaskDidWriteData?(session, downloadTask, bytesWritten, totalBytesWritten, totalBytesExpectedToWrite)
  605. self.downloadProgress?(bytesWritten, totalBytesWritten, totalBytesExpectedToWrite)
  606. self.progress.totalUnitCount = totalBytesExpectedToWrite
  607. self.progress.completedUnitCount = totalBytesWritten
  608. }
  609. func URLSession(session: NSURLSession!, downloadTask: NSURLSessionDownloadTask!, didResumeAtOffset fileOffset: Int64, expectedTotalBytes: Int64) {
  610. self.downloadTaskDidResumeAtOffset?(session, downloadTask, fileOffset, expectedTotalBytes)
  611. self.progress.totalUnitCount = expectedTotalBytes
  612. self.progress.completedUnitCount = fileOffset
  613. }
  614. }
  615. }
  616. // MARK: - Printable
  617. extension Alamofire.Request: Printable {
  618. var description: String {
  619. var description = "\(self.request.HTTPMethod) \(self.request.URL)"
  620. if self.response {
  621. description += " (\(self.response?.statusCode))"
  622. }
  623. return description
  624. }
  625. }
  626. extension Alamofire.Request: DebugPrintable {
  627. func cURLRepresentation() -> String {
  628. var components: [String] = ["$ curl -i"]
  629. let URL = self.request.URL!
  630. if self.request.HTTPMethod != "GET" {
  631. components += "-X \(self.request.HTTPMethod)"
  632. }
  633. if let credentialStorage = self.session.configuration.URLCredentialStorage {
  634. let protectionSpace = NSURLProtectionSpace(host: URL.host, port: URL.port ? URL.port : 0, `protocol`: URL.scheme, realm: URL.host, authenticationMethod: NSURLAuthenticationMethodHTTPBasic)
  635. if let credentials = credentialStorage.credentialsForProtectionSpace(protectionSpace)?.values.array {
  636. if !credentials.isEmpty {
  637. if let credential = credentials[0] as? NSURLCredential {
  638. components += "-u \(credential.user):\(credential.password)"
  639. }
  640. }
  641. }
  642. }
  643. if let cookieStorage = self.session.configuration.HTTPCookieStorage {
  644. if let cookies = cookieStorage.cookiesForURL(URL) as? [NSHTTPCookie] {
  645. if !cookies.isEmpty {
  646. let string = cookies.reduce(""){ $0 + "\($1.name)=\($1.value);" }
  647. components += "-b \"\(string.substringToIndex(string.endIndex.predecessor()))\""
  648. }
  649. }
  650. }
  651. for (field, value) in self.request.allHTTPHeaderFields {
  652. switch field {
  653. case "Cookie":
  654. continue
  655. default:
  656. components += "-H \"\(field): \(value)\""
  657. }
  658. }
  659. if let HTTPBody = self.request.HTTPBody {
  660. components += "-d \"\(NSString(data: HTTPBody, encoding: NSUTF8StringEncoding))\""
  661. }
  662. // TODO: -T arguments for files
  663. components += "\"\(URL.absoluteString)\""
  664. return join(" \\\n\t", components)
  665. }
  666. var debugDescription: String {
  667. return self.cURLRepresentation()
  668. }
  669. }
  670. // MARK: - Response Serializers
  671. // MARK: String
  672. extension Alamofire.Request {
  673. class func stringResponseSerializer(encoding: NSStringEncoding = NSUTF8StringEncoding) -> (NSURLRequest, NSHTTPURLResponse?, NSData?, NSError?) -> (AnyObject?, NSError?) {
  674. return { (_, _, data, error) in
  675. let string = NSString(data: data, encoding: encoding)
  676. return (string, error)
  677. }
  678. }
  679. func responseString(completionHandler: (NSURLRequest, NSHTTPURLResponse?, String?, NSError?) -> Void) -> Self {
  680. return responseString(completionHandler: completionHandler)
  681. }
  682. func responseString(encoding: NSStringEncoding = NSUTF8StringEncoding, completionHandler: (NSURLRequest, NSHTTPURLResponse?, String?, NSError?) -> Void) -> Self {
  683. return response(serializer: Alamofire.Request.stringResponseSerializer(encoding: encoding), completionHandler: { request, response, string, error in
  684. completionHandler(request, response, string as? String, error)
  685. })
  686. }
  687. }
  688. // MARK: JSON
  689. extension Alamofire.Request {
  690. class func JSONResponseSerializer(options: NSJSONReadingOptions = .AllowFragments) -> (NSURLRequest, NSHTTPURLResponse?, NSData?, NSError?) -> (AnyObject?, NSError?) {
  691. return { (request, response, data, error) in
  692. var serializationError: NSError?
  693. let JSON: AnyObject! = NSJSONSerialization.JSONObjectWithData(data as NSData, options: options, error: &serializationError)
  694. return (JSON, serializationError)
  695. }
  696. }
  697. func responseJSON(completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  698. return responseJSON(completionHandler: completionHandler)
  699. }
  700. func responseJSON(options: NSJSONReadingOptions = .AllowFragments, completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  701. return response(serializer: Alamofire.Request.JSONResponseSerializer(options: options), completionHandler: { (request, response, JSON, error) in
  702. completionHandler(request, response, JSON, error)
  703. })
  704. }
  705. }
  706. // MARK: Property List
  707. extension Alamofire.Request {
  708. class func propertyListResponseSerializer(options: NSPropertyListReadOptions = 0) -> (NSURLRequest, NSHTTPURLResponse?, NSData?, NSError?) -> (AnyObject?, NSError?) {
  709. return { (request, response, data, error) in
  710. var propertyListSerializationError: NSError?
  711. let plist: AnyObject! = NSPropertyListSerialization.propertyListWithData(data as NSData, options: options, format: nil, error: &propertyListSerializationError)
  712. return (plist, propertyListSerializationError)
  713. }
  714. }
  715. func responsePropertyList(completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  716. return responsePropertyList(completionHandler: completionHandler)
  717. }
  718. func responsePropertyList(options: NSPropertyListReadOptions = 0, completionHandler: (NSURLRequest, NSHTTPURLResponse?, AnyObject?, NSError?) -> Void) -> Self {
  719. return response(serializer: Alamofire.Request.propertyListResponseSerializer(options: options), completionHandler: { (request, response, plist, error) in
  720. completionHandler(request, response, plist, error)
  721. })
  722. }
  723. }
  724. // MARK: - Convenience
  725. extension Alamofire {
  726. private static func URLRequest(method: Method, _ URL: String) -> NSURLRequest {
  727. let mutableURLRequest = NSMutableURLRequest(URL: NSURL(string: URL))
  728. mutableURLRequest.HTTPMethod = method.toRaw()
  729. return mutableURLRequest
  730. }
  731. // MARK: Request
  732. static func request(method: Method, _ URL: String, parameters: [String: AnyObject]? = nil, encoding: ParameterEncoding = .URL) -> Request {
  733. var mutableRequest = NSMutableURLRequest(URL: NSURL(string: URL))
  734. mutableRequest.HTTPMethod = method.toRaw()
  735. return Manager.sharedInstance.request(encoding.encode(URLRequest(method, URL), parameters: parameters).0)
  736. }
  737. // MARK: Upload
  738. static func upload(method: Method, _ URL: String, file: NSURL) -> Alamofire.Request {
  739. return Manager.sharedInstance.upload(URLRequest(method, URL), file: file)
  740. }
  741. static func upload(method: Method, _ URL: String, data: NSData) -> Alamofire.Request {
  742. return Manager.sharedInstance.upload(URLRequest(method, URL), data: data)
  743. }
  744. static func upload(method: Method, _ URL: String, stream: NSInputStream) -> Alamofire.Request {
  745. return Manager.sharedInstance.upload(URLRequest(method, URL), stream: stream)
  746. }
  747. // MARK: Download
  748. static func download(method: Method, _ URL: String, destination: (NSURL, NSHTTPURLResponse) -> (NSURL)) -> Alamofire.Request {
  749. return Manager.sharedInstance.download(URLRequest(method, URL), destination: destination)
  750. }
  751. static func download(resumeData data: NSData, destination: (NSURL, NSHTTPURLResponse) -> (NSURL)) -> Alamofire.Request {
  752. return Manager.sharedInstance.download(data, destination: destination)
  753. }
  754. }
  755. typealias AF = Alamofire