ZcashLightClientKit/ZcashLightClientKit/ZcashRustBackend.swift

//
//  ZcashRustBackend.swift
//  ZcashLightClientKit
//
//  Created by Jack Grigg on 5/8/19.
//  Copyright © 2019 Electric Coin Company. All rights reserved.
//

import Foundation


public class ZcashRustBackend {
    static func osStrFromURL(_ url: URL) -> (String, UInt) {
        let path = url.absoluteString
        return (path, UInt(path.lengthOfBytes(using: .utf8)))
    }

    public static func getLastError() -> String? {
        let errorLen = zcashlc_last_error_length()
        if errorLen > 0 {
            let error = UnsafeMutablePointer<Int8>.allocate(capacity: Int(errorLen))
            zcashlc_error_message_utf8(error, errorLen)
            zcashlc_clear_last_error()
            return String(validatingUTF8: error)!
        } else {
            return nil
        }
    }

    public static func initDataDb(dbData: URL) -> Bool {
        let dbData = osStrFromURL(dbData)
        return zcashlc_init_data_database(dbData.0, dbData.1) != 0
    }

    public static func initAccountsTable(dbData: URL, seed: [UInt8], accounts: Int32) -> [String]? {
        let dbData = osStrFromURL(dbData)
        let extsksCStr = zcashlc_init_accounts_table(dbData.0, dbData.1, seed, UInt(seed.count), accounts)
        if extsksCStr == nil {
            return nil
        }

        let extsks = UnsafeBufferPointer(start: extsksCStr, count: Int(accounts)).map {
            String(cString: $0!)
        }
        zcashlc_vec_string_free(extsksCStr, UInt(accounts))
        return extsks
    }

    public static func initBlocksTable(dbData: URL, height: Int32, hash: String, time: UInt32, saplingTree: String) -> Bool {
        let dbData = osStrFromURL(dbData)
        return zcashlc_init_blocks_table(dbData.0, dbData.1, height, [CChar](hash.utf8CString), time, [CChar](saplingTree.utf8CString)) != 0
    }

    public static func getAddress(dbData: URL, account: Int32) -> String? {
        let dbData = osStrFromURL(dbData)

        let addressCStr = zcashlc_get_address(dbData.0, dbData.1, account)
        if addressCStr == nil {
            return nil
        }

        let address = String(validatingUTF8: addressCStr!)
        zcashlc_string_free(addressCStr)
        return address
    }

    public static func getBalance(dbData: URL, account: Int32) -> Int64 {
        let dbData = osStrFromURL(dbData)
        return zcashlc_get_balance(dbData.0, dbData.1, account)
    }

    public static func getVerifiedBalance(dbData: URL, account: Int32) -> Int64 {
        let dbData = osStrFromURL(dbData)
        return zcashlc_get_verified_balance(dbData.0, dbData.1, account)
    }

    public static func getReceivedMemoAsUTF8(dbData: URL, idNote: Int64) -> String? {
        let dbData = osStrFromURL(dbData)

        let memoCStr = zcashlc_get_received_memo_as_utf8(dbData.0, dbData.1, idNote)
        if memoCStr == nil {
            return nil
        }

        let memo = String(validatingUTF8: memoCStr!)
        zcashlc_string_free(memoCStr)
        return memo
    }

    public static func getSentMemoAsUTF8(dbData: URL, idNote: Int64) -> String? {
        let dbData = osStrFromURL(dbData)

        let memoCStr = zcashlc_get_sent_memo_as_utf8(dbData.0, dbData.1, idNote)
        if memoCStr == nil {
            return nil
        }

        let memo = String(validatingUTF8: memoCStr!)
        zcashlc_string_free(memoCStr)
        return memo
    }

    public static func validateCombinedChain(dbCache: URL, dbData: URL) -> Int32 {
        let dbCache = osStrFromURL(dbCache)
        let dbData = osStrFromURL(dbData)
        return zcashlc_validate_combined_chain(dbCache.0, dbCache.1, dbData.0, dbData.1)
    }

    public static func rewindToHeight(dbData: URL, height: Int32) -> Bool {
        let dbData = osStrFromURL(dbData)
        return zcashlc_rewind_to_height(dbData.0, dbData.1, height) != 0
    }

    public static func scanBlocks(dbCache: URL, dbData: URL) -> Bool {
        let dbCache = osStrFromURL(dbCache)
        let dbData = osStrFromURL(dbData)
        return zcashlc_scan_blocks(dbCache.0, dbCache.1, dbData.0, dbData.1) != 0
    }

    public static func sendToAddress(dbData: URL, account: Int32, extsk: String, to: String, value: Int64, memo: String?, spendParams: URL, outputParams: URL) -> Int64 {
        let dbData = osStrFromURL(dbData)
        let spendParams = osStrFromURL(spendParams)
        let outputParams = osStrFromURL(outputParams)
        return zcashlc_send_to_address(dbData.0, dbData.1, account, extsk, to, value, memo, spendParams.0, spendParams.1, outputParams.0, outputParams.1)
    }
}