session-ios/SessionMessagingKit/Sending & Receiving/Typing Indicators/TypingIndicators.swift

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

import Foundation
import GRDB
import SessionUtilitiesKit

public class TypingIndicators {
    // MARK: - Direction
    
    public enum Direction {
        case outgoing
        case incoming
    }
    
    private class Indicator {
        fileprivate let threadId: String
        fileprivate let direction: Direction
        fileprivate let timestampMs: Int64
        
        fileprivate var refreshTimer: Timer?
        fileprivate var stopTimer: Timer?
        
        init?(
            threadId: String,
            threadVariant: SessionThread.Variant,
            threadIsMessageRequest: Bool,
            direction: Direction,
            timestampMs: Int64?
        ) {
            // The `typingIndicatorsEnabled` flag reflects the user-facing setting in the app
            // preferences, if it's disabled we don't want to emit "typing indicator" messages
            // or show typing indicators for other users
            //
            // We also don't want to show/send typing indicators for message requests
            guard GRDBStorage.shared[.typingIndicatorsEnabled] && !threadIsMessageRequest else {
                return nil
            }
            
            // Don't send typing indicators in group threads
            guard threadVariant != .closedGroup && threadVariant != .openGroup else { return nil }
            
            self.threadId = threadId
            self.direction = direction
            self.timestampMs = (timestampMs ?? Int64(floor(Date().timeIntervalSince1970 * 1000)))
        }
        
        fileprivate func starting(_ db: Database) -> Indicator {
            let direction: Direction = self.direction
            let timestampMs: Int64 = self.timestampMs
            
            // Start the typing indicator
            switch direction {
                case .outgoing:
                    scheduleRefreshCallback(db, shouldSend: (refreshTimer == nil))
                    
                case .incoming:
                    try? ThreadTypingIndicator(
                        threadId: self.threadId,
                        timestampMs: timestampMs
                    )
                    .save(db)
            }
            
            // Schedule the 'stopCallback' to cancel the typing indicator
            stopTimer?.invalidate()
            stopTimer = Timer.scheduledTimerOnMainThread(
                withTimeInterval: (direction == .outgoing ? 3 : 5),
                repeats: false
            ) { [weak self] _ in
                GRDBStorage.shared.write { db in
                    self?.stoping(db)
                }
            }
            
            return self
        }
        
        @discardableResult fileprivate func stoping(_ db: Database) -> Indicator? {
            self.refreshTimer?.invalidate()
            self.refreshTimer = nil
            self.stopTimer?.invalidate()
            self.stopTimer = nil
            
            switch direction {
                case .outgoing:
                    guard let thread: SessionThread = try? SessionThread.fetchOne(db, id: self.threadId) else {
                        return nil
                    }
                    
                    try? MessageSender.send(
                        db,
                        message: TypingIndicator(kind: .stopped),
                        interactionId: nil,
                        in: thread
                    )
                    
                case .incoming:
                    _ = try? ThreadTypingIndicator
                        .filter(ThreadTypingIndicator.Columns.threadId == self.threadId)
                        .deleteAll(db)
            }
            
            return nil
        }
        
        private func scheduleRefreshCallback(_ db: Database, shouldSend: Bool = true) {
            if shouldSend {
                guard let thread: SessionThread = try? SessionThread.fetchOne(db, id: self.threadId) else {
                    return
                }
                
                try? MessageSender.send(
                    db,
                    message: TypingIndicator(kind: .started),
                    interactionId: nil,
                    in: thread
                )
            }
            
            refreshTimer?.invalidate()
            refreshTimer = Timer.scheduledTimerOnMainThread(
                withTimeInterval: 10,
                repeats: false
            ) { [weak self] _ in
                GRDBStorage.shared.write { db in
                    self?.scheduleRefreshCallback(db)
                }
            }
        }
    }
    
    // MARK: - Variables
    
    public static let shared: TypingIndicators = TypingIndicators()
    
    private static var outgoing: Atomic<[String: Indicator]> = Atomic([:])
    private static var incoming: Atomic<[String: Indicator]> = Atomic([:])
    
    // MARK: - Functions
    
    public static func didStartTyping(
        _ db: Database,
        threadId: String,
        threadVariant: SessionThread.Variant,
        threadIsMessageRequest: Bool,
        direction: Direction,
        timestampMs: Int64?
    ) {
        switch direction {
            case .outgoing:
                let updatedIndicator: Indicator? = (
                    outgoing.wrappedValue[threadId] ??
                    Indicator(
                        threadId: threadId,
                        threadVariant: threadVariant,
                        threadIsMessageRequest: threadIsMessageRequest,
                        direction: direction,
                        timestampMs: timestampMs
                    )
                )?.starting(db)
                
                outgoing.mutate { $0[threadId] = updatedIndicator }
                
            case .incoming:
                let updatedIndicator: Indicator? = (
                    incoming.wrappedValue[threadId] ??
                    Indicator(
                        threadId: threadId,
                        threadVariant: threadVariant,
                        threadIsMessageRequest: threadIsMessageRequest,
                        direction: direction,
                        timestampMs: timestampMs
                    )
                )?.starting(db)
                
                incoming.mutate { $0[threadId] = updatedIndicator }
        }
    }
    
    public static func didStopTyping(_ db: Database, threadId: String, direction: Direction) {
        switch direction {
            case .outgoing:
                let updatedIndicator: Indicator? = outgoing.wrappedValue[threadId]?.stoping(db)
                
                outgoing.mutate { $0[threadId] = updatedIndicator }
                
            case .incoming:
                let updatedIndicator: Indicator? = incoming.wrappedValue[threadId]?.stoping(db)
                
                incoming.mutate { $0[threadId] = updatedIndicator }
        }
    }
}