ContactViewModel.swift

import HUD
import UIKit
import Models
import Combine
import XXModels
import Defaults
import CombineSchedulers
import DependencyInjection
import XXMessengerClient

import XXClient

struct ContactViewState: Equatable {
    var title: String?
    var email: String?
    var phone: String?
    var photo: UIImage?
    var username: String?
    var nickname: String?
}

final class ContactViewModel {
    @Dependency var database: Database
    @Dependency var messenger: Messenger
    @Dependency var getFactsFromContact: GetFactsFromContact

    @KeyObject(.username, defaultValue: nil) var username: String?
    @KeyObject(.sharingEmail, defaultValue: false) var sharingEmail: Bool
    @KeyObject(.sharingPhone, defaultValue: false) var sharingPhone: Bool

    var contact: XXModels.Contact

    var popToRootPublisher: AnyPublisher<Void, Never> { popToRootRelay.eraseToAnyPublisher() }
    var popPublisher: AnyPublisher<Void, Never> { popRelay.eraseToAnyPublisher() }
    var hudPublisher: AnyPublisher<HUDStatus, Never> { hudRelay.eraseToAnyPublisher() }
    var successPublisher: AnyPublisher<Void, Never> { successRelay.eraseToAnyPublisher() }
    var statePublisher: AnyPublisher<ContactViewState, Never> { stateRelay.eraseToAnyPublisher() }

    private let popRelay = PassthroughSubject<Void, Never>()
    private let popToRootRelay = PassthroughSubject<Void, Never>()
    private let successRelay = PassthroughSubject<Void, Never>()
    private let hudRelay = CurrentValueSubject<HUDStatus, Never>(.none)
    private let stateRelay = CurrentValueSubject<ContactViewState, Never>(.init())

    var myId: Data {
        try! messenger.e2e.get()!.getContact().getId()
    }

    var backgroundScheduler: AnySchedulerOf<DispatchQueue> = DispatchQueue.global().eraseToAnyScheduler()

    init(_ contact: XXModels.Contact) {
        self.contact = contact

        let facts = try? getFactsFromContact(contact.marshaled!)
        let email = facts?.first(where: { $0.type == .email })?.value
        let phone = facts?.first(where: { $0.type == .phone })?.value

        stateRelay.value = .init(
            title: contact.nickname ?? contact.username,
            email: email,
            phone: phone,
            photo: contact.photo != nil ? UIImage(data: contact.photo!) : nil,
            username: contact.username,
            nickname: contact.nickname
        )
    }

    func didChoosePhoto(_ photo: UIImage) {
        stateRelay.value.photo = photo
        contact.photo = photo.jpegData(compressionQuality: 0.0)
        _ = try? database.saveContact(contact)
    }

    func didTapDelete() {
        hudRelay.send(.on)

        do {
            try messenger.e2e.get()!.deleteRequest.partnerId(contact.id)
            try database.deleteContact(contact)

            hudRelay.send(.none)
            popToRootRelay.send()
        } catch {
            hudRelay.send(.error(.init(with: error)))
        }
    }

    func didTapReject() {
        // TODO: Reject function on the API?
        _ = try? database.deleteContact(contact)
        popRelay.send()
    }

    func didTapClear() {
        _ = try? database.deleteMessages(.init(chat: .direct(myId, contact.id)))
    }

    func didUpdateNickname(_ string: String) {
        contact.nickname = string.isEmpty ? nil : string
        stateRelay.value.title = string.isEmpty ? contact.username : string
        _ = try? database.saveContact(contact)

        stateRelay.value.nickname = contact.nickname
    }

    func didTapResend() {
        hudRelay.send(.on)
        contact.authStatus = .requesting

        backgroundScheduler.schedule { [weak self] in
            guard let self = self else { return }

            do {
                try self.database.saveContact(self.contact)

                var includedFacts: [Fact] = []
                let myFacts = try self.messenger.ud.get()!.getFacts()

                if let fact = myFacts.get(.username) {
                    includedFacts.append(fact)
                }

                if self.sharingEmail, let fact = myFacts.get(.email) {
                    includedFacts.append(fact)
                }

                if self.sharingPhone, let fact = myFacts.get(.phone) {
                    includedFacts.append(fact)
                }

                let _ = try self.messenger.e2e.get()!.requestAuthenticatedChannel(
                    partner: .live(self.contact.marshaled!),
                    myFacts: includedFacts
                )

                self.contact.authStatus = .requested
                try self.database.saveContact(self.contact)

                self.hudRelay.send(.none)
                self.popRelay.send()
            } catch {
                self.contact.authStatus = .requestFailed
                _ = try? self.database.saveContact(self.contact)
                self.hudRelay.send(.error(.init(with: error)))
            }
        }
    }

    func didTapRequest(with nickname: String) {
        hudRelay.send(.on)
        contact.nickname = nickname
        contact.authStatus = .requesting

        backgroundScheduler.schedule { [weak self] in
            guard let self = self else { return }

            do {
                try self.database.saveContact(self.contact)

                var includedFacts: [Fact] = []
                let myFacts = try self.messenger.ud.get()!.getFacts()

                if let fact = myFacts.get(.username) {
                    includedFacts.append(fact)
                }

                if self.sharingEmail, let fact = myFacts.get(.email) {
                    includedFacts.append(fact)
                }

                if self.sharingPhone, let fact = myFacts.get(.phone) {
                    includedFacts.append(fact)
                }

                let _ = try self.messenger.e2e.get()!.requestAuthenticatedChannel(
                    partner: .live(self.contact.marshaled!),
                    myFacts: includedFacts
                )

                self.contact.authStatus = .requested
                try self.database.saveContact(self.contact)

                self.hudRelay.send(.none)
                self.successRelay.send()
            } catch {
                self.contact.authStatus = .requestFailed
                _ = try? self.database.saveContact(self.contact)
                self.hudRelay.send(.error(.init(with: error)))
            }
        }
    }

    func didTapAccept(_ nickname: String) {
        hudRelay.send(.on)
        contact.nickname = nickname
        contact.authStatus = .confirming

        backgroundScheduler.schedule { [weak self] in
            guard let self = self else { return }

            do {
                try self.database.saveContact(self.contact)

                let _ = try self.messenger.e2e.get()!.confirmReceivedRequest(partner: XXClient.Contact.live(self.contact.marshaled!))

                self.contact.authStatus = .friend
                try self.database.saveContact(self.contact)

                self.hudRelay.send(.none)
                self.popRelay.send()
            } catch {
                self.contact.authStatus = .confirmationFailed
                _ = try? self.database.saveContact(self.contact)
                self.hudRelay.send(.error(.init(with: error)))
            }
        }
    }
}