import UserNotifications

class NotificationService: UNNotificationServiceExtension {

    var contentHandler: ((UNNotificationContent) -> Void)?
    var bestAttemptContent: UNMutableNotificationContent?

    override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
        // APNsから届く通知のペイロードに、 `"mutable-content": 1` が入っていると、
        // アプリが foreground, backgroundであっても、キルされていても、必ずここを通過します。
        // このメソッドの実行にかけられる時間は、30秒以内です。
        // 30秒を超えてしまうと、`serviceExtensionTimeWillExpire()`の処理が実行されます。

        // 引数で受け取った contentHandler と request.content を、
        // インスタンスプロパティに保持しています
        // request.content は mutableCopy() を行い、その内容を複製しつつ変更できるようにしています
        self.contentHandler = contentHandler
        bestAttemptContent = (request.content.mutableCopy() as? UNMutableNotificationContent)

        // ここからが、内容の変更処理です。
        // この例ではタイトルを変更しているだけです。
        if let bestAttemptContent = bestAttemptContent {
            bestAttemptContent.title = "\(bestAttemptContent.title) [modified]"

            // 処理が終わったら、変更済みのコンテンツを contentHandler に渡します
            contentHandler(bestAttemptContent)
        }
    }

    override func serviceExtensionTimeWillExpire() {
        // 上の `didReceive(_:withContentHandler:)` の処理がタイムアウト（30秒）した場合に、
        // このメソッドの処理が即実行される。
        // ここを実装しない場合は、オリジナルの通知がユーザーに通知される
        if let contentHandler = contentHandler, let bestAttemptContent = bestAttemptContent {
            contentHandler(bestAttemptContent)
        }
    }
}

{
    "aps":{
      "alert" : {
         "title" : "Game Request",
         "subtitle" : "Five Card Draw",
         "body" : "Bob wants to play poker"
      },
      "badge" : 0,
      "sound" : "default",
      "mutable-content" : 1
    }
}

override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
    self.contentHandler = contentHandler
    bestAttemptContent = request.content.mutableCopy() as? UNMutableNotificationContent
    
    if let bestAttemptContent = bestAttemptContent {
        // Modify the notification content here...
        let modifiedTitle = "\(bestAttemptContent.title) [modified]"
        bestAttemptContent.title = modifiedTitle

        if let urlString = request.content.userInfo["attachement-url"] as? String,
           let url = URL(string: urlString) {
            let session = URLSession(configuration: URLSessionConfiguration.default)
            let task = session.dataTask(with: url) { data, response, error in
                let logger = Logger(subsystem: "PushNotificationPractice", category: "NotificationServiceExtension")

                if let e = error {
                    bestAttemptContent.body = "\(e.localizedDescription) \(bestAttemptContent.body)"
                    logger.error("\(e.localizedDescription)")
                    contentHandler(bestAttemptContent)
                    return
                } else {
                    logger.info("No Error.")
                }

                let path = URL(fileURLWithPath: NSTemporaryDirectory().appending("attachement.jpg"))
                // おそらくは、ちゃんとしたファイルパスにしたほうが良いでしょう
                // パス名もプッシュ通知で届けるか、UUIDをつけたりすることで対応できそうです
                // ※ 拡張子は必須です。拡張子がないと、Attachmentのインスタンスができません

                do {
                    try data?.write(to: path)
                } catch {
                    bestAttemptContent.body = "\(error.localizedDescription) \(bestAttemptContent.body)"
                    logger.error("\(error.localizedDescription)")
                    contentHandler(bestAttemptContent)
                    return
                }

                if let attachement = try? UNNotificationAttachment(identifier: "",
                                                                   url: path,
                                                                   options: nil) {
                    // identifierは、ファイル名やUNIXTIMEなど、一意なものにしておくと良さそうです
                    // UNNotificationAttachmentの詳しい情報はドキュメントサイトと見るとよいです
                    bestAttemptContent.attachments = [attachement]
                } else {
                    bestAttemptContent.body = "Creating attachement is failed. \(bestAttemptContent.body)"
                }
                contentHandler(bestAttemptContent)
            }
            task.resume()
        } else {
            bestAttemptContent.body = "Attachement couldn't loaded. \(bestAttemptContent.body)"
            contentHandler(bestAttemptContent)
        }
    }

{
    "aps":{
      "alert" : {
         "title" : "Game Request",
         "subtitle" : "Five Card Draw",
         "body" : "Bob wants to play poker"
      },
      "badge" : 0,
      "sound" : "default",
      "mutable-content" : 1
	},
	"attachement-url": "https://pbs.twimg.com/media/FEwWGJKWQAUh4IR?format=jpg&name=large"
}

// アプリ本体側の例
func updateLabel() {
    let defaults = UserDefaults(suiteName: "group.com.example.PushNotificationPractice")
    if let title = defaults?.string(forKey: "PushTitle") {
        tokenLabel.text = title
    }
}

// Notification Service Extension側の例
override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
    self.contentHandler = contentHandler
    bestAttemptContent = request.content.mutableCopy() as? UNMutableNotificationContent

    if let bestAttemptContent = bestAttemptContent {
        let modifiedTitle = "\(bestAttemptContent.title) [modified]"
        bestAttemptContent.title = modifiedTitle

        let defaults = UserDefaults(suiteName: "group.com.example.PushNotificationPractice")
        defaults?.setValue(modifiedTitle, forKeyPath: "PushTitle")
    }
}