import Foundation
@testable import ThreemaFramework

final class ContactStoreMock: NSObject, ContactStoreProtocol {
    private let callOnCompletion: Bool
    private var contact: ContactEntity?
    private let errorHandler: NSError?

    private(set) var numberOfSynchronizeAddressBookCalls = 0
    private(set) var numberOfUpdateStatusCalls = 0
    var markContactAsDeletedCalls = [String]()
    var addContactCalls = [String]()
    var addAsWorkCalls = [String]()

    required init(callOnCompletion: Bool, _ contact: ContactEntity? = nil, errorHandler: NSError? = nil) {
        self.callOnCompletion = callOnCompletion
        self.contact = contact
        self.errorHandler = errorHandler
    }
    
    override convenience init() {
        self.init(callOnCompletion: false)
    }
    
    func contact(for identity: String?) -> NSObject? {
        contact?.identity == identity ? contact : nil
    }
    
    func prefetchIdentityInfo(_ identities: Set<String>, onCompletion: () -> Void, onError: (Error) -> Void) {
        if callOnCompletion {
            onCompletion()
        }
    }
    
    func fetchWorkIdentities(
        _ identities: [Any],
        onCompletion: @escaping ([Any]) -> Void,
        onError: @escaping (Error) -> Void
    ) {
        if callOnCompletion {
            onCompletion([])
        }
    }
    
    func fetchPublicKey(
        for identity: String,
        acquaintanceLevel: ContactAcquaintanceLevel,
        onCompletion: @escaping (Data) -> Void,
        onError: @escaping (Error) -> Void
    ) {
        if callOnCompletion {
            onCompletion(Data())
        }
    }

    func fetchPublicKey(
        for identity: String?,
        acquaintanceLevel: ContactAcquaintanceLevel,
        entityManager entityManagerObject: NSObject,
        ignoreBlockUnknown: Bool,
        onCompletion: @escaping (Data?) -> Void,
        onError: ((Error?) -> Void)? = nil
    ) {
        if let errorHandler, let onError {
            onError(errorHandler)
        }
        else if callOnCompletion {
            onCompletion(Data())
        }
    }

    func removeProfilePictureFlagForAllContacts() {
        // no-op
    }
    
    func removeProfilePictureRequest(_ identity: String) {
        // no-op
    }
    
    func synchronizeAddressBook(
        forceFullSync: Bool,
        ignoreMinimumInterval: Bool,
        onCompletion: ((Bool) -> Void)?,
        onError: ((Error?) -> Void)?
    ) {
        numberOfSynchronizeAddressBookCalls += 1
        
        if let errorHandler, let onError {
            onError(errorHandler)
        }
        else if callOnCompletion {
            onCompletion?(true)
        }
    }
    
    func updateFeatureMasks(
        forIdentities Identities: [String],
        onCompletion: @escaping () -> Void,
        onError: @escaping (Error) -> Void
    ) {
        if let errorHandler {
            onError(errorHandler)
        }
        else if callOnCompletion {
            onCompletion()
        }
    }
    
    func reflect(_ identity: String?) {
        // no-op
    }
    
    func updateProfilePicture(
        _ identity: String?,
        imageData: Data,
        shouldReflect: Bool,
        blobID: Data?,
        encryptionKey: Data?,
        didFailWithError error: NSErrorPointer
    ) {
        // no-op
    }
    
    func deleteProfilePicture(_ identity: String?, shouldReflect: Bool) {
        // no-op
    }
    
    func removeProfilePictureFlag(for identity: String) {
        // no-op
    }

    func resetEntityManager() {
        // no-op
    }

    func addContact(
        with identity: String,
        verificationLevel: Int32,
        onCompletion: @escaping (NSObject?, Bool) -> Void,
        onError: ((any Error) -> Void)? = nil
    ) {
        if let errorHandler, let onError {
            onError(errorHandler)
        }
        else {
            addContactCalls.append(identity)
            if callOnCompletion {
                onCompletion(contact, true)
            }
        }
    }
    
    func addWorkContact(
        with identity: String,
        publicKey: Data,
        firstname: String?,
        lastname: String?,
        csi: String?,
        jobTitle: String?,
        department: String?,
        acquaintanceLevel: ContactAcquaintanceLevel,
        workAvailabilityStatus: WorkAvailabilityStatus?,
        onCompletion: @escaping (NSObject) -> Void,
        onError: @escaping (any Error) -> Void
    ) {
        // nop-op
    }

    func addAsWork(identities: NSOrderedSet, contactSyncer mediatorSyncableContacts: MediatorSyncableContacts?) {
        for item in identities {
            addAsWorkCalls.append(item as! String)
        }
    }

    func updateContact(withIdentity identity: String, avatar: Data?, firstName: String?, lastName: String?) {
        // no-op
    }
    
    func updateStatus(
        forAllContactsIgnoreInterval ignoreInterval: Bool,
        onCompletion: @escaping () -> Void,
        onError: @escaping (Error) -> Void
    ) {
        numberOfUpdateStatusCalls += 1
        
        if let errorHandler {
            onError(errorHandler)
        }
        else {
            onCompletion()
        }
    }

    func updateAllContacts() {
        // no-op
    }

    func markContactAsDeleted(identity: String, entityManagerObject: NSObject) {
        markContactAsDeletedCalls.append(identity)
    }
    
    func resetCustomReadReceipts() {
        // no-op
    }
}