import ImSDK_Plus

class AppDelegate: UIResponder, UIApplicationDelegate, V2TIMAPNSListener, V2TIMConversationListener {
    
    var unreadNumber: UInt64 = 0
    
    // 1. Set the listener
    func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
        // Listen for push notifications
        V2TIMManager.sharedInstance()?.setAPNSListener(self)
        // Listen for unread conversation counts
        V2TIMManager.sharedInstance()?.setConversationListener(self)
        return true
    }
    
    // 2. Save the unread count after it changes
    func onTotalUnreadMessageCountChanged(_ totalUnreadCount: UInt64) {
        self.unreadNumber = totalUnreadCount
    }
    
    // 3. Report custom-defined unread count after the app is pushed to the background
    /// After the application enters the background, customize the app's unread count. If not handled, the default app unread count is the sum of all conversation unread counts
    /// - Returns: Custom-defined unread count
    func onSetAPPUnreadCount() -> UInt32 {
        // 1. Get the custom-defined badge
        var customBadgeNumber: UInt32 = 0
        
        // 2. Add the IM message unread count
        customBadgeNumber += UInt32(self.unreadNumber)
        
        // 3. Report to the IM server via IMSDK
        return customBadgeNumber
    }
}