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

67 lines
1.4 KiB
Swift

//
// Author+Database.swift
// Database
//
// Created by Brent Simmons on 7/8/17.
// Copyright © 2017 Ranchero Software. All rights reserved.
//
import Foundation
import Data
import RSDatabase
import RSParser
// MARK: - DatabaseObject
extension Author {
init?(row: FMResultSet) {
let authorID = row.string(forColumn: DatabaseKey.authorID)
let name = row.string(forColumn: DatabaseKey.name)
let url = row.string(forColumn: DatabaseKey.url)
let avatarURL = row.string(forColumn: DatabaseKey.avatarURL)
let emailAddress = row.string(forColumn: DatabaseKey.emailAddress)
self.init(authorID: authorID, name: name, url: url, avatarURL: avatarURL, emailAddress: emailAddress)
}
init?(parsedAuthor: ParsedAuthor) {
self.init(authorID: nil, name: parsedAuthor.name, url: parsedAuthor.url, avatarURL: parsedAuthor.avatarURL, emailAddress: parsedAuthor.emailAddress)
}
}
extension Author: DatabaseObject {
public var databaseID: String {
get {
return authorID
}
}
public func databaseDictionary() -> NSDictionary? {
var d = NSMutableDictionary()
// TODO
if d.count < 1 {
return nil
}
return (d.copy() as! NSDictionary)
}
}
// MARK: Author creation from Set<ParsedAuthor>
extension Set where Element == ParsedAuthor {
func authors() -> Set<Author>? {
let createdAuthors = Set(self.flatMap { Author(parsedAuthor: $0) })
return createdAuthors.isEmpty ? nil: createdAuthors
}
}