A set of extensions to PodioKit to make it easier to work with in Swift.

PodioKitExtensions.swift

extension PKTAsyncTask {

  func onSuccess<ResultType>(successBlock: ResultType! -> Void) {
    onSuccess { obj in
      let typedObj = obj as? ResultType
      successBlock(typedObj)
    }
  }
  
  func onComplete<ResultType>(completeBlock: (ResultType?, NSError?) -> Void) {
    onComplete { (obj, error) in
      let typedObj = obj as? ResultType
      completeBlock(typedObj, error)
    }
  }
  
  func flattenMap<ResultType>(mapBlock: (ResultType!) -> PKTAsyncTask) {
    flattenMap { obj in
      let typedObj = obj as? ResultType
      return mapBlock(typedObj)
    }
  }
}