ZcashLightClientKit/Sources/ZcashLightClientKit/DAO/UnspentTransactionOutputDao.swift

//
//  UnspentTransactionOutputDAO.swift
//  ZcashLightClientKit
//
//  Created by Francisco Gindre on 12/9/20.
//

import Foundation

struct UTXO: Decodable, Encodable {
    enum CodingKeys: String, CodingKey {
        case id = "id_utxo"
        case address
        case prevoutTxId = "prevout_txid"
        case prevoutIndex = "prevout_idx"
        case script
        case valueZat = "value_zat"
        case height
        case spentInTx = "spent_in_tx"
    }
    
    var id: Int?
    var address: String
    var prevoutTxId: Data
    var prevoutIndex: Int
    var script: Data
    var valueZat: Int
    var height: Int
    var spentInTx: Int?
}

extension UTXO: UnspentTransactionOutputEntity {
    var txid: Data {
        get {
            prevoutTxId
        }
        set {
            prevoutTxId = newValue
        }
    }
    
    var index: Int {
        get {
            prevoutIndex
        }
        set {
            prevoutIndex = newValue
        }
    }
}

extension UnspentTransactionOutputEntity {
    /**
    As UTXO, with id and spentIntTx set to __nil__
    */
    func asUTXO() -> UTXO {
        UTXO(
            id: nil,
            address: address,
            prevoutTxId: txid,
            prevoutIndex: index,
            script: script,
            valueZat: valueZat,
            height: height,
            spentInTx: nil
        )
    }
}
import SQLite
class UnspentTransactionOutputSQLDAO: UnspentTransactionOutputRepository {
    enum TableColumns {
        static var id = Expression<Int>("id_utxo")
        static var address = Expression<String>("address")
        static var txid = Expression<Blob>("prevout_txid")
        static var index = Expression<Int>("prevout_idx")
        static var script = Expression<Blob>("script")
        static var valueZat = Expression<Int>("value_zat")
        static var height = Expression<Int>("height")
        static var spentInTx = Expression<Int?>("spent_in_tx")
    }

    let table = Table("utxos")

    var dbProvider: ConnectionProvider
    
    init (dbProvider: ConnectionProvider) {
        self.dbProvider = dbProvider
    }

    func initialise() throws {
        try createTableIfNeeded()
    }
    
    func createTableIfNeeded() throws {
        let stringStatement =
            """
            CREATE TABLE IF NOT EXISTS utxos (
                id_utxo INTEGER PRIMARY KEY,
                address TEXT NOT NULL,
                prevout_txid BLOB NOT NULL,
                prevout_idx INTEGER NOT NULL,
                script BLOB NOT NULL,
                value_zat INTEGER NOT NULL,
                height INTEGER NOT NULL,
                spent_in_tx INTEGER,
                FOREIGN KEY (spent_in_tx) REFERENCES transactions(id_tx),
                CONSTRAINT tx_outpoint UNIQUE (prevout_txid, prevout_idx)
            )
            """
        
        try dbProvider.connection().run(stringStatement)
    }

    func store(utxos: [UnspentTransactionOutputEntity]) throws {
        do {
            let db = try dbProvider.connection()
            try dbProvider.connection().transaction {
                for utxo in utxos.map({ $0 as? UTXO ?? $0.asUTXO() }) {
                    try db.run(table.insert(utxo))
                }
            }
        } catch {
            throw DatabaseStorageError.transactionFailed(underlyingError: error)
        }
    }

    func clearAll(address: String?) throws {
        if let tAddr = address {
            do {
                try dbProvider.connection().run(table.filter(TableColumns.address == tAddr).delete())
            } catch {
                throw DatabaseStorageError.operationFailed
            }
        } else {
            do {
                try dbProvider.connection().run(table.delete())
            } catch {
                throw DatabaseStorageError.operationFailed
            }
        }
    }
    
    func getAll(address: String?) throws -> [UnspentTransactionOutputEntity] {
        if let tAddress = address {
            let allTxs: [UTXO] = try dbProvider.connection()
                .prepare(table.filter(TableColumns.address == tAddress))
                .map { row in
                    try row.decode()
                }
            return allTxs
        } else {
            let allTxs: [UTXO] = try dbProvider.connection()
                .prepare(table)
                .map { row in
                    try row.decode()
                }
            return allTxs
        }
    }
    
    func balance(address: String, latestHeight: BlockHeight) throws -> WalletBalance {
        do {
            let verified = try dbProvider.connection().scalar(
                table.select(TableColumns.valueZat.sum)
                    .filter(TableColumns.address == address)
                    .filter(TableColumns.height <= latestHeight - ZcashSDK.defaultStaleTolerance)
            ) ?? 0
            let total = try dbProvider.connection().scalar(
                table.select(TableColumns.valueZat.sum)
                    .filter(TableColumns.address == address)
            ) ?? 0
            
            return WalletBalance(
                verified: Zatoshi(Int64(verified)),
                total: Zatoshi(Int64(total))
            )
        } catch {
            throw DatabaseStorageError.operationFailed
        }
    }
}

struct TransparentBalance {
    var balance: WalletBalance
    var address: String
}

enum UTXORepositoryBuilder {
    static func build(initializer: Initializer) -> UnspentTransactionOutputRepository {
        return UnspentTransactionOutputSQLDAO(dbProvider: SimpleConnectionProvider(path: initializer.dataDbURL.path))
    }
}