NetNewsWire/Frameworks/Database/Extensions/Article+Database.swift

//
//  Article+Database.swift
//  Database
//
//  Created by Brent Simmons on 7/3/17.
//  Copyright © 2017 Ranchero Software. All rights reserved.
//

import Foundation
import RSDatabase
import Data

extension Article {
	
	convenience init?(row: FMResultSet, account: Account) {
		
		guard let feedID = row.string(forColumn: DatabaseKey.feedID) else {
			return nil
		}
		guard let uniqueID = row.string(forColumn: DatabaseKey.uniqueID) else {
			return nil
		}
		
		let articleID = row.string(forColumn: DatabaseKey.articleID)!
		let title = row.string(forColumn: DatabaseKey.title)
		let contentHTML = row.string(forColumn: DatabaseKey.contentHTML)
		let contentText = row.string(forColumn: DatabaseKey.contentText)
		let url = row.string(forColumn: DatabaseKey.url)
		let externalURL = row.string(forColumn: DatabaseKey.externalURL)
		let summary = row.string(forColumn: DatabaseKey.summary)
		let imageURL = row.string(forColumn: DatabaseKey.imageURL)
		let bannerImageURL = row.string(forColumn: DatabaseKey.bannerImageURL)
		let datePublished = row.date(forColumn: DatabaseKey.datePublished)
		let dateModified = row.date(forColumn: DatabaseKey.dateModified)
		let accountInfo: [String: Any]? = nil // TODO

		// authors, tags, and attachments are fetched from related tables, after init.
		let authors: [Author]? = nil
		let tags: Set<String>? = nil
		let attachments: [Attachment]? = nil
		
		self.init(account: account, articleID: articleID, feedID: feedID, uniqueID: uniqueID, title: title, contentHTML: contentHTML, contentText: contentText, url: url, externalURL: externalURL, summary: summary, imageURL: imageURL, bannerImageURL: bannerImageURL, datePublished: datePublished, dateModified: dateModified, authors: authors, tags: tags, attachments: attachments, accountInfo: accountInfo)
	}

	func databaseDictionary() -> NSDictionary {
		
		let d = NSMutableDictionary()
		
		
		return d.copy() as! NSDictionary
	}
}

extension Article: DatabaseObject {
	
	public var databaseID: String {
		get {
			return articleID
		}
	}
}

extension Set where Element == Article {

	func withNilProperty<T>(_ keyPath: KeyPath<Article,T?>) -> Set<Article> {

		return Set(filter{ $0[keyPath: keyPath] == nil })
	}

	func articleIDs() -> Set<String> {

		return Set<String>(map { $0.databaseID })
	}

	func eachHasAStatus() -> Bool {

		for article in self {
			if article.status == nil {
				return false
			}
		}
		return true
	}

	func missingStatuses() -> Set<Article> {

		return withNilProperty(\Article.status)
	}
	
	func statuses() -> Set<ArticleStatus> {
		
		return Set<ArticleStatus>(self.flatMap { $0.status })
	}

	func dictionary() -> [String: Article] {

		var d = [String: Article]()
		for article in self {
			d[article.articleID] = article
		}
		return d
	}
}