session-ios/SessionMessagingKit/Database/Models/RecipientState.swift

// Copyright © 2022 Rangeproof Pty Ltd. All rights reserved.

import Foundation
import GRDB
import SessionUtilitiesKit
import SessionUIKit

public struct RecipientState: Codable, Equatable, FetchableRecord, PersistableRecord, TableRecord, ColumnExpressible {
    public static var databaseTableName: String { "recipientState" }
    internal static let profileForeignKey = ForeignKey([Columns.recipientId], to: [Profile.Columns.id])
    internal static let interactionForeignKey = ForeignKey([Columns.interactionId], to: [Interaction.Columns.id])
    private static let profile = hasOne(Profile.self, using: profileForeignKey)
    internal static let interaction = belongsTo(Interaction.self, using: interactionForeignKey)
    
    public typealias Columns = CodingKeys
    public enum CodingKeys: String, CodingKey, ColumnExpression {
        case interactionId
        case recipientId
        case state
        case readTimestampMs
        case mostRecentFailureText
    }
    
    public enum State: Int, Codable, Hashable, DatabaseValueConvertible {
        /// These cases **MUST** remain in this order (even though having `failed` as `0` would be more logical) as the order
        /// is optimised for the desired "interactionState" grouping behaviour we want which makes the query to retrieve the interaction
        /// state run ~16 times than the alternate approach which required a sub-query (check git history to see the old approach at the
        /// bottom of this file if desired)
        ///
        /// The expected behaviour of the grouped "interactionState" that both the `SessionThreadViewModel` and
        /// `MessageViewModel` should use is `IFNULL(MIN("recipientState"."state"), 'sending')` (joining on the
        /// `interaction.id` and `state != 'skipped'`):
        ///  - The 'skipped' state should be ignored entirely
        ///  - If there is no state (ie. interaction recipient records not yet created) then the interaction state should be 'sending'
        ///  - If there is a single 'sending' state then the interaction state should be 'sending'
        ///  - If there is a single 'failed' state and no 'sending' state then the interaction state should be 'failed'
        ///  - If there are neither 'sending' or 'failed' states then the interaction state should be 'sent'
        case sending
        case failed
        case skipped
        case sent
        case failedToSync   // One-to-one Only
        case syncing        // One-to-one Only
        
        public func statusIconInfo(
            variant: Interaction.Variant,
            hasAtLeastOneReadReceipt: Bool,
            hasAttachments: Bool
        ) -> (image: UIImage?, text: String?, themeTintColor: ThemeValue) {
            guard variant == .standardOutgoing else {
                return (nil, "read".localized(), .messageBubble_deliveryStatus)
            }

            switch (self, hasAtLeastOneReadReceipt, hasAttachments) {
                case (.sending, _, true):
                    return (
                        UIImage(systemName: "ellipsis.circle"),
                        "uploading".localized(),
                        .messageBubble_deliveryStatus
                    )
                    
                case (.sending, _, _):
                    return (
                        UIImage(systemName: "ellipsis.circle"),
                        "sending".localized(),
                        .messageBubble_deliveryStatus
                    )

                case (.sent, false, _), (.skipped, _, _):
                    return (
                        UIImage(systemName: "checkmark.circle"),
                        "disappearingMessagesSent".localized(),
                        .messageBubble_deliveryStatus
                    )

                case (.sent, true, _):
                    return (
                        UIImage(systemName: "eye.fill"),
                        "read".localized(),
                        .messageBubble_deliveryStatus
                    )
                    
                case (.failed, _, _):
                    return (
                        UIImage(systemName: "exclamationmark.triangle"),
                        "messageStatusFailedToSend".localized(),
                        .danger
                    )
                    
                case (.failedToSync, _, _):
                    return (
                        UIImage(systemName: "exclamationmark.triangle"),
                        "messageStatusFailedToSync".localized(),
                        .warning
                    )
                    
                case (.syncing, _, _):
                    return (
                        UIImage(systemName: "ellipsis.circle"),
                        "messageStatusSyncing".localized(),
                        .warning
                    )

            }
        }
    }
    
    /// The id for the interaction this state belongs to
    public let interactionId: Int64
    
    /// The id for the recipient that has this state
    ///
    /// **Note:** For contact and closedGroup threads this can be used as a lookup for a contact/profile but in an
    /// openGroup thread this will be the threadId so won’t resolve to a contact/profile
    public let recipientId: String
    
    /// The current state for the recipient
    public let state: State
    
    /// When the interaction was read in milliseconds since epoch
    ///
    /// This value will be null for outgoing messages
    ///
    /// **Note:** This currently will be set when opening the thread for the first time after receiving this interaction
    /// rather than when the interaction actually appears on the screen
    public let readTimestampMs: Int64?
    
    public let mostRecentFailureText: String?
    
    // MARK: - Relationships
         
    public var interaction: QueryInterfaceRequest<Interaction> {
        request(for: RecipientState.interaction)
    }
    
    public var profile: QueryInterfaceRequest<Profile> {
        request(for: RecipientState.profile)
    }
    
    // MARK: - Initialization
    
    public init(
        interactionId: Int64,
        recipientId: String,
        state: State,
        readTimestampMs: Int64? = nil,
        mostRecentFailureText: String? = nil
    ) {
        self.interactionId = interactionId
        self.recipientId = recipientId
        self.state = state
        self.readTimestampMs = readTimestampMs
        self.mostRecentFailureText = mostRecentFailureText
    }
}