NetNewsWire/Frameworks/RSDatabase/DatabaseTable.swift

//
//  DatabaseTable.swift
//  RSDatabase
//
//  Created by Brent Simmons on 7/16/17.
//  Copyright © 2017 Ranchero Software, LLC. All rights reserved.
//

import Foundation

public protocol DatabaseTable {
	
	var name: String {get}
	var queue: RSDatabaseQueue {get}
	
	init(name: String, queue: RSDatabaseQueue)
}

extension DatabaseTable {
	
	public func selectRowsWhere(key: String, equals value: Any, in database: FMDatabase) -> FMResultSet? {
		
		return database.rs_selectRowsWhereKey(key, equalsValue: value, tableName: self.name)
	}
	
	public func deleteRowsWhere(key: String, equalsAnyValue values: [Any], in database: FMDatabase) {
		
		if values.isEmpty {
			return
		}
		
		database.rs_deleteRowsWhereKey(key, inValues: values, tableName: name)
	}

	// MARK: Counts

	func numberWithCountResultSet(_ resultSet: FMResultSet?) -> Int {

		if let resultSet = resultSet, resultSet.next() {
			return Int(resultSet.int(forColumnIndex: 0))
		}
		return 0
	}

	func numberWithSQLAndParameters(_ sql: String, _ parameters: [Any], in database: FMDatabase) -> Int {

		let resultSet = database.executeQuery(sql, withArgumentsIn: parameters)
		return numberWithCountResultSet(resultSet)
	}
}