Rork Max の通知を長押しすると自前の画面が出る — Notification Content Extension の実装

import UIKit
import UserNotifications
import UserNotificationsUI
 
final class NotificationViewController: UIViewController, UNNotificationContentExtension {
 
    private let imageView = UIImageView()
    private let titleLabel = UILabel()
    private let statusLabel = UILabel()
 
    override func viewDidLoad() {
        super.viewDidLoad()
 
        imageView.contentMode = .scaleAspectFill
        imageView.clipsToBounds = true
        imageView.layer.cornerRadius = 12
        titleLabel.font = .preferredFont(forTextStyle: .headline)
        statusLabel.font = .preferredFont(forTextStyle: .footnote)
        statusLabel.textColor = .secondaryLabel
 
        let stack = UIStackView(arrangedSubviews: [imageView, titleLabel, statusLabel])
        stack.axis = .vertical
        stack.spacing = 8
        stack.translatesAutoresizingMaskIntoConstraints = false
        view.addSubview(stack)
 
        NSLayoutConstraint.activate([
            stack.leadingAnchor.constraint(equalTo: view.leadingAnchor, constant: 16),
            stack.trailingAnchor.constraint(equalTo: view.trailingAnchor, constant: -16),
            stack.topAnchor.constraint(equalTo: view.topAnchor, constant: 16),
            stack.bottomAnchor.constraint(equalTo: view.bottomAnchor, constant: -16),
            imageView.heightAnchor.constraint(equalTo: imageView.widthAnchor, multiplier: 0.66)
        ])
    }
 
    func didReceive(_ notification: UNNotification) {
        let content = notification.request.content
        titleLabel.text = content.title
 
        // Service Extension が添付した画像を読む
        if let attachment = content.attachments.first,
           attachment.url.startAccessingSecurityScopedResource() {
            defer { attachment.url.stopAccessingSecurityScopedResource() }
            imageView.image = UIImage(contentsOfFile: attachment.url.path)
        }
 
        // App Group を見て、すでに保存済みかどうかを反映
        statusLabel.text = SharedStore.isSaved(id: content.threadIdentifier)
            ? "保存済み" : "通知から保存できます"
    }
}

extension NotificationViewController {
    func didReceive(
        _ response: UNNotificationResponse,
        completionHandler completion: @escaping (UNNotificationContentExtensionResponseOption) -> Void
    ) {
        let id = response.notification.request.content.threadIdentifier
 
        switch response.actionIdentifier {
        case "SAVE_PICK":
            SharedStore.markSaved(id: id)        // App Group に書く
            statusLabel.text = "保存済み"
            completion(.doNotDismiss)             // ← 閉じずに UI を更新
 
        case "SHUFFLE_PICK":
            completion(.dismissAndForwardAction)  // 本体アプリへ引き継ぐ
 
        default:
            completion(.dismiss)
        }
    }
}

enum SharedStore {
    private static let suite = UserDefaults(suiteName: "group.net.rorklab.dailypick")!
 
    static func markSaved(id: String) {
        var ids = suite.stringArray(forKey: "savedIDs") ?? []
        if !ids.contains(id) { ids.append(id) }
        suite.set(ids, forKey: "savedIDs")
    }
 
    static func isSaved(id: String) -> Bool {
        (suite.stringArray(forKey: "savedIDs") ?? []).contains(id)
    }
}