Bark/NotificationServiceExtension/Processor/CallProcessor.swift

//
//  CallProcessor.swift
//  NotificationServiceExtension
//
//  Created by huangfeng on 2024/6/6.
//  Copyright © 2024 Fin. All rights reserved.
//

import AudioToolbox
import Foundation

class CallProcessor: NotificationContentProcessor {
    /// 循环播放的铃声
    var soundID: SystemSoundID = 0
    /// 播放完毕后，返回的 content
    var content: UNMutableNotificationContent? = nil
    /// 是否需要停止播放，由主APP发出停止通知赋值
    var needsStop = false
    
    func process(identifier: String, content bestAttemptContent: UNMutableNotificationContent) async throws -> UNMutableNotificationContent {
        guard let call = bestAttemptContent.userInfo["call"] as? String, call == "1" else {
            return bestAttemptContent
        }
        self.content = bestAttemptContent
        
        self.registerObserver()
        self.sendLocalNotification(identifier: identifier, content: bestAttemptContent)
        self.cancelRemoteNotification(content: bestAttemptContent)
        await startAudioWork()
        return bestAttemptContent
    }
    
    func serviceExtensionTimeWillExpire(contentHandler: (UNNotificationContent) -> Void) {
        stopAudioWork()
        if let content {
            contentHandler(content)
        }
    }
    
    /// 生成一个本地推送
    private func sendLocalNotification(identifier: String, content: UNMutableNotificationContent) {
        // 推送id和推送的内容都使用远程APNS的
        guard let content = content.mutableCopy() as? UNMutableNotificationContent else {
            return
        }
        content.sound = nil
        let request = UNNotificationRequest(identifier: identifier, content: content, trigger: nil)
        UNUserNotificationCenter.current().add(request)
    }
    
    /// 响铃结束时取消显示远程推送，因为已经用本地推送显示了一遍
    private func cancelRemoteNotification(content: UNMutableNotificationContent) {
        // 远程推送在响铃结束后静默不显示
        // 至于iOS15以下的设备，因不支持这个特性会在响铃结束后再展示一次, 但会取消声音
        // 如果设置了 level 参数，就还是以 level 参数为准不做修改
        if #available(iOSApplicationExtension 15.0, *), self.content?.userInfo["level"] == nil {
            self.content?.interruptionLevel = .passive
        } else {
            content.sound = nil
        }
    }
    
    // 开始播放铃声，startAudioWork(completion:) 方法的异步包装
    private func startAudioWork() async {
        return await withCheckedContinuation { continuation in
            self.startAudioWork {
                continuation.resume()
            }
        }
    }

    /// 铃声播放结束时的回调
    var startAudioWorkCompletion: (() -> Void)? = nil
    /// 播放铃声
    private func startAudioWork(completion: @escaping () -> Void) {
        guard let content else {
            completion()
            return
        }
        self.startAudioWorkCompletion = completion
        
        let sound = ((content.userInfo["aps"] as? [String: Any])?["sound"] as? String)?.split(separator: ".")
        let soundName: String
        let soundType: String
        if sound?.count == 2, let first = sound?.first, let last = sound?.last {
            soundName = String(first)
            soundType = String(last)
        } else {
            soundName = "multiwayinvitation"
            soundType = "caf"
        }
        
        // 先找自定义上传的铃声，再找内置铃声
        guard let audioPath = getSoundInCustomSoundsDirectory(soundName: "\(soundName).\(soundType)") ??
            Bundle.main.path(forResource: soundName, ofType: soundType)
        else {
            completion()
            return
        }
        
        let fileUrl = URL(string: audioPath)
        // 创建响铃任务
        AudioServicesCreateSystemSoundID(fileUrl! as CFURL, &soundID)
        // 播放震动、响铃
        AudioServicesPlayAlertSound(soundID)
        // 监听响铃完成状态
        let selfPointer = unsafeBitCast(self, to: UnsafeMutableRawPointer.self)
        AudioServicesAddSystemSoundCompletion(soundID, nil, nil, { sound, clientData in
            guard let pointer = clientData else { return }
            let processor = unsafeBitCast(pointer, to: CallProcessor.self)
            if processor.needsStop {
                processor.startAudioWorkCompletion?()
                return
            }
            // 音频文件一次播放完成，再次播放
            AudioServicesPlayAlertSound(sound)
        }, selfPointer)
    }

    /// 停止播放
    private func stopAudioWork() {
        AudioServicesRemoveSystemSoundCompletion(soundID)
        AudioServicesDisposeSystemSoundID(soundID)
    }
    
    /// 注册停止通知
    func registerObserver() {
        let notification = CFNotificationCenterGetDarwinNotifyCenter()
        let observer = Unmanaged.passUnretained(self).toOpaque()
        CFNotificationCenterAddObserver(notification, observer, { _, pointer, _, _, _ in
            guard let observer = pointer else { return }
            let processor = Unmanaged<CallProcessor>.fromOpaque(observer).takeUnretainedValue()
            processor.needsStop = true
        }, kStopCallProcessorKey as CFString, nil, .deliverImmediately)
    }
    
    func getSoundInCustomSoundsDirectory(soundName: String) -> String? {
        // 扩展访问不到主APP中的铃声，需要先共享铃声文件，再实现自定义铃声响铃
        guard let soundsDirectoryUrl = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: "group.bark")?.appendingPathComponent("Sounds") else {
            return nil
        }
        let path = soundsDirectoryUrl.appendingPathComponent(soundName).absoluteString
        if FileManager.default.fileExists(atPath: path) {
            return path
        }
        return nil
    }

    deinit {
        let observer = Unmanaged.passUnretained(self).toOpaque()
        let name = CFNotificationName(kStopCallProcessorKey as CFString)
        CFNotificationCenterRemoveObserver(CFNotificationCenterGetDarwinNotifyCenter(), observer, name, nil)
    }
}