Session+Chat.swift

import Models
import Foundation
import UIKit
import Shared

extension Session {
    public func readAll(from contact: Contact) {
        do {
            try dbManager.updateAll(
                Message.self,
                Message.Request.unreadsFromContactId(contact.userId),
                with: [Message.Column.unread.set(to: false)]
            )
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    public func readAll(from group: Group) {
        do {
            try dbManager.updateAll(
                GroupMessage.self,
                GroupMessage.Request.unreadsFromGroup(group.groupId),
                with: [GroupMessage.Column.unread.set(to: false)]
            )
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    public func deleteAll(from contact: Contact) {
        do {
            try dbManager.delete(Message.self, .withContact(contact.userId))
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    public func deleteAll(from group: Group) {
        do {
            try dbManager.delete(GroupMessage.self, .fromGroup(group.groupId))
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    public func send(imageData: Data, to contact: Contact, completion: @escaping (Result<Void, Error>) -> Void) {
        client.bindings.compress(image: imageData) { [weak self] in
            guard let self = self else {
                completion(.success(()))
                return
            }

            switch $0 {
            case .success(let compressed):
                let name = "image_\(Date.asTimestamp)"
                try! FileManager.store(data: compressed, name: name, type: Attachment.Extension.image.written)
                let attachment = Attachment(name: name, data: compressed, _extension: .image)
                self.send(Payload(text: "You sent an image", reply: nil, attachment: attachment), toContact: contact)
                completion(.success(()))
            case .failure(let error):
                completion(.failure(error))
                log(string: "Error when compressing image: \(error.localizedDescription)", type: .error)
            }
        }
    }

    public func send(_ payload: Payload, toContact contact: Contact) {
        var message = Message(
            sender: client.bindings.meMarshalled,
            receiver: contact.userId,
            payload: payload,
            unread: false,
            timestamp: Date.asTimestamp,
            uniqueId: nil,
            status: payload.attachment == nil ? .sending : .sendingAttachment
        )

        do {
            message = try dbManager.save(message)
            send(message: message)
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    public func delete(messages: [Int64]) {
        messages.forEach {
            do {
                try dbManager.delete(Message.self, .withId($0))
            } catch {
                log(string: error.localizedDescription, type: .error)
            }
        }
    }

    public func retryMessage(_ id: Int64) {
        guard var message: Message = try? dbManager.fetch(withId: id) else { return }
        message.timestamp = Date.asTimestamp
        message.status = message.payload.attachment == nil ? .sending : .sendingAttachment

        do {
            message = try dbManager.save(message)
            send(message: message)
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    private func send(message: Message) {
        var message = message

        if let _ = message.payload.attachment {
            sendAttachment(message: message)
            return
        }

        DispatchQueue.global().async { [weak self] in
            guard let self = self else { return }
            switch self.client.bindings.send(message.payload.asData(), to: message.receiver) {
            case .success(let report):
                message.roundURL = report.roundURL

                self.client.bindings.listen(report: report.marshalled) { result in
                    switch result {
                    case .success(let status):
                        switch status {
                        case .failed:
                            message.status = .failedToSend
                        case .sent:
                            message.status = .sent
                        case .timedout:
                            message.status = .timedOut
                        }
                    case .failure:
                        message.status = .failedToSend
                    }

                    message.uniqueId = report.uniqueId
                    message.timestamp = Int(report.timestamp)
                    DispatchQueue.main.async {
                        do {
                            _ = try self.dbManager.save(message)
                        } catch {
                            log(string: error.localizedDescription, type: .error)
                        }
                    }
                }
            case .failure(let error):
                message.status = .failedToSend
                log(string: error.localizedDescription, type: .error)
            }

            DispatchQueue.main.async {
                do {
                    _ = try self.dbManager.save(message)
                } catch {
                    log(string: error.localizedDescription, type: .error)
                }
            }
        }
    }

    private func sendAttachment(message: Message) {
        guard let manager = client.transferManager else { fatalError("A transfer manager was not created") }

        var message = message
        let attachment = message.payload.attachment!

        DispatchQueue.global().async { [weak self] in
            guard let self = self else { return }

            do {
                let tid = try manager.uploadFile(attachment, to: message.receiver) { completed, send, arrived, total, error in
                    if completed {
                        self.endTransferFrom(message: message)
                        message.status = .sent
                        message.payload.attachment?.progress = 1.0
                        log(string: "FT Up finished", type: .info)
                    } else {
                        if let error = error {
                            log(string: error.localizedDescription, type: .error)
                            message.status = .failedToSend
                        } else {
                            let progress = Float(arrived)/Float(total)
                            message.payload.attachment?.progress = progress
                            log(string: "FT Up: \(progress)", type: .crumbs)
                        }
                    }

                    do {
                        _ = try self.dbManager.save(message) // If it fails here, means the chat was cleared.
                    } catch {
                        log(string: error.localizedDescription, type: .error)
                    }
                }

                let transfer = FileTransfer(
                    tid: tid,
                    contact: message.receiver,
                    fileName: attachment.name,
                    fileType: attachment._extension.written,
                    isIncoming: false
                )

                message.payload.attachment?.transferId = tid
                message.status = .sendingAttachment

                do {
                    _ = try self.dbManager.save(message)
                    _ = try self.dbManager.save(transfer)
                } catch {
                    log(string: error.localizedDescription, type: .error)
                }
            } catch {
                message.status = .failedToSend
                log(string: error.localizedDescription, type: .error)

                do {
                    _ = try self.dbManager.save(message)
                } catch let otherError {
                    log(string: otherError.localizedDescription, type: .error)
                }
            }
        }
    }

    private func endTransferFrom(message: Message) {
        guard let manager = client.transferManager else { fatalError("A transfer manager was not created") }
        guard let tid = message.payload.attachment?.transferId else { fatalError("Tried to finish a transfer that had no TID") }

        do {
            try manager.endTransferUpload(with: tid)

            if let transfer: FileTransfer = try? dbManager.fetch(.withTID(tid)).first {
                try dbManager.delete(transfer)
            }
        } catch {
            log(string: error.localizedDescription, type: .error)
        }
    }

    func handle(incomingTransfer transfer: FileTransfer) {
        guard let manager = client.transferManager else { fatalError("A transfer manager was not created") }

        let fileExtension: Attachment.Extension = transfer.fileType == "m4a" ? .audio : .image
        let name = "\(Date.asTimestamp)_\(transfer.fileName)"

        var fakeContent: Data

        if fileExtension == .image {
            fakeContent = Asset.transferImagePlaceholder.image.jpegData(compressionQuality: 0.1)!
        } else {
            fakeContent = FileManager.dummyAudio()
        }

        let attachment = Attachment(name: name, data: fakeContent, transferId: transfer.tid, _extension: fileExtension)

        var message = Message(
            sender: transfer.contact,
            receiver: client.bindings.meMarshalled,
            payload: .init(text: "Sent you a \(transfer.fileType)", reply: nil, attachment: attachment),
            unread: true,
            timestamp: Date.asTimestamp,
            uniqueId: nil,
            status: .receivingAttachment
        )

        do {
            message = try self.dbManager.save(message)
            try self.dbManager.save(transfer)
        } catch {
            log(string: "Failed to save message/transfer to the database. Will not start listening to transfer... \(error.localizedDescription)", type: .info)
            return
        }

        log(string: "FT Down starting", type: .info)

        try! manager.listenDownloadFromTransfer(with: transfer.tid) { completed, arrived, total, error in
            if let error = error {
                fatalError(error.localizedDescription)
            }

            if completed {
                log(string: "FT Down finished", type: .info)

                guard let rawFile = try? manager.downloadFileFromTransfer(with: transfer.tid) else {
                    log(string: "Received finalized transfer, file was nil. Ignoring...", type: .error)
                    return
                }

                try! FileManager.store(data: rawFile, name: name, type: fileExtension.written)
                var realAttachment = Attachment(name: name, data: rawFile, transferId: transfer.tid, _extension: fileExtension)
                realAttachment.progress = 1.0
                message.payload = .init(text: "Sent you a \(transfer.fileType)", reply: nil, attachment: realAttachment)
                message.status = .received

                if let toDelete: FileTransfer = try? self.dbManager.fetch(.withTID(transfer.tid)).first {
                    do {
                        try self.dbManager.delete(toDelete)
                    } catch {
                        log(string: error.localizedDescription, type: .error)
                    }
                }
            } else {
                let progress = Float(arrived)/Float(total)
                log(string: "FT Down: \(progress)", type: .crumbs)
                message.payload.attachment?.progress = progress
            }

            do {
                try self.dbManager.save(message) // If it fails here, means the chat was cleared.
            } catch {
                log(string: "Failed to update message model from an incoming transfer. Probably chat was cleared: \(error.localizedDescription)", type: .error)
            }
        }
    }
}