NetNewsWire/Frameworks/Account/LocalAccount/LocalAccountRefresher.swift

116 lines
3.0 KiB
Swift
Raw Normal View History

//
// LocalAccountRefresher.swift
// NetNewsWire
//
// Created by Brent Simmons on 9/6/16.
// Copyright © 2016 Ranchero Software, LLC. All rights reserved.
//
import Foundation
import RSCore
import RSParser
import RSWeb
import Articles
final class LocalAccountRefresher {
private lazy var downloadSession: DownloadSession = {
return DownloadSession(delegate: self)
}()
var progress: DownloadProgress {
return downloadSession.progress
}
public func refreshFeeds(_ feeds: Set<Feed>) {
downloadSession.downloadObjects(feeds as NSSet)
}
}
// MARK: - DownloadSessionDelegate
extension LocalAccountRefresher: DownloadSessionDelegate {
func downloadSession(_ downloadSession: DownloadSession, requestForRepresentedObject representedObject: AnyObject) -> URLRequest? {
2017-09-17 21:08:50 +02:00
guard let feed = representedObject as? Feed else {
return nil
}
guard let url = URL(string: feed.url) else {
return nil
}
let request = NSMutableURLRequest(url: url)
2017-09-27 06:43:40 +02:00
if let conditionalGetInfo = feed.conditionalGetInfo {
conditionalGetInfo.addRequestHeadersToURLRequest(request)
}
return request as URLRequest
}
func downloadSession(_ downloadSession: DownloadSession, downloadDidCompleteForRepresentedObject representedObject: AnyObject, response: URLResponse?, data: Data, error: NSError?) {
2017-09-17 21:08:50 +02:00
guard let feed = representedObject as? Feed, !data.isEmpty else {
return
}
if let error = error {
print("Error downloading \(feed.url) - \(error)")
return
}
let dataHash = (data as NSData).rs_md5HashString()
if dataHash == feed.contentHash {
return
}
let parserData = ParserData(url: feed.url, data: data)
FeedParser.parse(parserData) { (parsedFeed, error) in
2017-09-17 21:08:50 +02:00
guard let account = feed.account, let parsedFeed = parsedFeed, error == nil else {
return
}
2017-09-17 21:08:50 +02:00
account.update(feed, with: parsedFeed) {
if let httpResponse = response as? HTTPURLResponse {
2017-09-27 06:43:40 +02:00
feed.conditionalGetInfo = HTTPConditionalGetInfo(urlResponse: httpResponse)
}
feed.contentHash = dataHash
}
}
}
func downloadSession(_ downloadSession: DownloadSession, shouldContinueAfterReceivingData data: Data, representedObject: AnyObject) -> Bool {
2017-09-17 21:08:50 +02:00
guard let feed = representedObject as? Feed else {
return false
}
if data.isEmpty {
return true
}
if data.isDefinitelyNotFeed() {
return false
}
if data.count > 4096 {
let parserData = ParserData(url: feed.url, data: data)
return FeedParser.mightBeAbleToParseBasedOnPartialData(parserData)
}
return true
}
func downloadSession(_ downloadSession: DownloadSession, didReceiveUnexpectedResponse response: URLResponse, representedObject: AnyObject) {
2019-04-25 14:13:14 +02:00
}
2019-04-25 14:13:14 +02:00
func downloadSession(_ downloadSession: DownloadSession, didReceiveNotModifiedResponse: URLResponse, representedObject: AnyObject) {
}
}
// MARK: - Utility
private extension Data {
func isDefinitelyNotFeed() -> Bool {
// We only detect a few image types for now. This should get fleshed-out at some later date.
return (self as NSData).rs_dataIsImage()
}
}