NetNewsWire/Frameworks/SyncDatabase/SyncStatusTable.swift

//
//  SyncStatusTable.swift
//  NetNewsWire
//
//  Created by Maurice Parker on 5/14/19.
//  Copyright © 2019 Ranchero Software. All rights reserved.
//

import Foundation
import Articles
import RSDatabase

final class SyncStatusTable: DatabaseTable {
	
	let name = DatabaseTableName.syncStatus
	private let queue: RSDatabaseQueue
	
	init(queue: RSDatabaseQueue) {
		self.queue = queue
	}

	func selectForProcessing() -> [SyncStatus] {

		var statuses: Set<SyncStatus>? = nil
		
		self.queue.updateSync { database in
			
			let updateSQL = "update syncStatus set selected = true"
			database.executeUpdate(updateSQL, withArgumentsIn: nil)
			
			let selectSQL = "select * from syncStatus where selected == true"
			if let resultSet = database.executeQuery(selectSQL, withArgumentsIn: nil) {
				statuses = resultSet.mapToSet(self.statusWithRow)
			}
			
		}

		return statuses != nil ? Array(statuses!) : [SyncStatus]()
		
	}
	
	func selectPendingCount() -> Int {
		
		var count: Int = 0
		
		self.queue.fetchSync { (database) in
			let sql = "select count(*) from syncStatus"
			if let resultSet = database.executeQuery(sql, withArgumentsIn: nil) {
				resultSet.next()
				count = Int(resultSet.int(forColumnIndex: 0))
			}
			
		}
		
		return count
		
	}
	
	func resetSelectedForProcessing(_ articleIDs: [String]) {
		self.queue.update { database in
			let parameters = articleIDs.map { $0 as AnyObject }
			let placeholders = NSString.rs_SQLValueList(withPlaceholders: UInt(articleIDs.count))!
			let updateSQL = "update syncStatus set selected = false where articleID in \(placeholders)"
			database.executeUpdate(updateSQL, withArgumentsIn: parameters)
		}
	}
	
	func deleteSelectedForProcessing(_ articleIDs: [String]) {
		self.queue.update { database in
			let parameters = articleIDs.map { $0 as AnyObject }
			let placeholders = NSString.rs_SQLValueList(withPlaceholders: UInt(articleIDs.count))!
			let deleteSQL = "delete from syncStatus where articleID in \(placeholders)"
			database.executeUpdate(deleteSQL, withArgumentsIn: parameters)
		}
	}
	
	func insertStatuses(_ statuses: [SyncStatus]) {
		self.queue.update { database in
			let statusArray = statuses.map { $0.databaseDictionary() }
			self.insertRows(statusArray, insertType: .orReplace, in: database)
		}
	}
	
}

private extension SyncStatusTable {

	func statusWithRow(_ row: FMResultSet) -> SyncStatus? {
		
		guard let articleID = row.string(forColumn: DatabaseKey.articleID),
			let rawKey = row.string(forColumn: DatabaseKey.key),
			let key = ArticleStatus.Key(rawValue: rawKey) else {
				return nil
		}
		
		let flag = row.bool(forColumn: DatabaseKey.flag)
		let selected = row.bool(forColumn: DatabaseKey.selected)
		
		return SyncStatus(articleID: articleID, key: key, flag: flag, selected: selected)
		
	}
}