Swiftgram/Swiftgram/SGRequests/Sources/File.swift

import Foundation
import SwiftSignalKit


public func requestsDownload(url: URL) -> Signal<(Data, URLResponse?), Error?> {
    return Signal { subscriber in
        let completed = Atomic<Bool>(value: false)

        let downloadTask = URLSession.shared.downloadTask(with: url, completionHandler: { location, response, error in
            let _ = completed.swap(true)
            if let location = location, let data = try? Data(contentsOf: location) {
                subscriber.putNext((data, response))
                subscriber.putCompletion()
            } else {
                subscriber.putError(error)
            }
        })
        downloadTask.resume()

        return ActionDisposable {
            if !completed.with({ $0 }) {
                downloadTask.cancel()
            }
        }
    }
}

public func requestsGet(url: URL) -> Signal<(Data, URLResponse?), Error?> {
    return Signal { subscriber in
        let completed = Atomic<Bool>(value: false)

        let urlTask = URLSession.shared.dataTask(with: url, completionHandler: { data, response, error in
            let _ = completed.swap(true)
            if let strongData = data {
                subscriber.putNext((strongData, response))
                subscriber.putCompletion()
            } else {
                subscriber.putError(error)
            }
        })
        urlTask.resume()

        return ActionDisposable {
            if !completed.with({ $0 }) {
                urlTask.cancel()
            }
        }
    }
}


public func requestsCustom(request: URLRequest) -> Signal<(Data, URLResponse?), Error?> {
    return Signal { subscriber in
        let completed = Atomic<Bool>(value: false)
        let urlTask = URLSession.shared.dataTask(with: request, completionHandler: { data, response, error in
            _ = completed.swap(true)
            if let strongData = data {
                subscriber.putNext((strongData, response))
                subscriber.putCompletion()
            } else {
                subscriber.putError(error)
            }
        })
        urlTask.resume()

        return ActionDisposable {
            if !completed.with({ $0 }) {
                urlTask.cancel()
            }
        }
    }
}