Push and in-app messages

1
2
3
4
5
6
7
8
9
10
11import UserNotifications

func registerUnplugCategories() {
    let primary = UNNotificationAction(identifier: "primary", title: "Open", options: [.foreground])
    let secondary = UNNotificationAction(identifier: "secondary", title: "Not now", options: [])
    let one = UNNotificationCategory(identifier: "UNPLUG_ONE_BUTTON",
                                     actions: [primary], intentIdentifiers: [])
    let two = UNNotificationCategory(identifier: "UNPLUG_TWO_BUTTONS",
                                     actions: [primary, secondary], intentIdentifiers: [])
    UNUserNotificationCenter.current().setNotificationCategories([one, two])
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73import UserNotifications

struct PushButton: Decodable {
    let id: String       // "primary" or "secondary"
    let label: String
    let action: String   // "open_screen", "open_url" or "close"
    let url: String?
}

final class NotificationService: UNNotificationServiceExtension {
    private var contentHandler: ((UNNotificationContent) -> Void)?
    private var content: UNMutableNotificationContent?

    override func didReceive(_ request: UNNotificationRequest,
                             withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
        guard let content = request.content.mutableCopy() as? UNMutableNotificationContent else {
            return contentHandler(request.content)
        }
        self.contentHandler = contentHandler
        self.content = content
        let info = content.userInfo
        let group = DispatchGroup()

        // Button titles: re-register the category with this push's labels.
        let buttons = Array(Self.buttons(from: info["buttons"]).prefix(2))
        if !buttons.isEmpty {
            let actions = buttons.map {
                UNNotificationAction(identifier: $0.id, title: $0.label,
                                     options: $0.action == "close" ? [] : [.foreground])
            }
            let id = actions.count == 2 ? "UNPLUG_TWO_BUTTONS" : "UNPLUG_ONE_BUTTON"
            let category = UNNotificationCategory(identifier: id, actions: actions, intentIdentifiers: [])
            content.categoryIdentifier = id
            group.enter()
            let center = UNUserNotificationCenter.current()
            center.getNotificationCategories { existing in
                center.setNotificationCategories(existing.filter { $0.identifier != id }.union([category]))
                group.leave()
            }
        }

        // Image: download it and attach it.
        if let link = info["image_url"] as? String, let url = URL(string: link) {
            group.enter()
            URLSession.shared.downloadTask(with: url) { tmp, _, _ in
                defer { group.leave() }
                guard let tmp = tmp else { return }
                let ext = url.pathExtension.isEmpty ? "jpg" : url.pathExtension
                let file = FileManager.default.temporaryDirectory
                    .appendingPathComponent(UUID().uuidString).appendingPathExtension(ext)
                try? FileManager.default.moveItem(at: tmp, to: file)
                if let image = try? UNNotificationAttachment(identifier: "image", url: file) {
                    content.attachments = [image]
                }
            }.resume()
        }

        group.notify(queue: .main) { contentHandler(content) }
    }

    override func serviceExtensionTimeWillExpire() {
        if let handler = contentHandler, let content = content { handler(content) }
    }

    // buttons arrives as a JSON array from APNs, or as a JSON string through FCM.
    private static func buttons(from raw: Any?) -> [PushButton] {
        var data: Data?
        if let text = raw as? String { data = text.data(using: .utf8) }
        else if let list = raw { data = try? JSONSerialization.data(withJSONObject: list) }
        guard let data = data else { return [] }
        return (try? JSONDecoder().decode([PushButton].self, from: data)) ?? []
    }
}

1
2
3
4
5
6
7
8
9fun createUnplugChannels(context: Context) {
    if (Build.VERSION.SDK_INT < Build.VERSION_CODES.O) return
    val manager = context.getSystemService(NotificationManager::class.java)
    val marketing = NotificationChannel("marketing", "Offers and news",
        NotificationManager.IMPORTANCE_DEFAULT).apply { setSound(null, null) }
    val transactional = NotificationChannel("transactional", "Orders and rewards",
        NotificationManager.IMPORTANCE_DEFAULT).apply { setSound(null, null) }
    manager.createNotificationChannels(listOf(marketing, transactional))
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66class AppMessagingService : FirebaseMessagingService() {

    override fun onNewToken(token: String) {
        // Re-register: PUT /v1/members/me/devices/{device_id} with the new token.
        DeviceRegistration.sync(applicationContext)
    }

    override fun onMessageReceived(message: RemoteMessage) {
        val d = message.data
        val messageId = d["message_id"] ?: return

        if (d["silent"] == "1") {
            if (d["inapp"] == "1") InAppMessages.fetchNext(applicationContext)
            return
        }

        val builder = NotificationCompat.Builder(this, d["channel_id"] ?: "marketing")
            .setSmallIcon(R.drawable.ic_notification)
            .setContentTitle(d["title"])
            .setContentText(d["body"])
            .setAutoCancel(true)
            .setContentIntent(openIntent(d, buttonId = null, url = d["deep_link"]))

        d["thumbnail_url"]?.let { loadBitmap(it) }?.let { builder.setLargeIcon(it) }
        d["image_url"]?.let { loadBitmap(it) }?.let {
            builder.setStyle(NotificationCompat.BigPictureStyle().bigPicture(it))
        }

        d["buttons"]?.let { json ->
            val buttons = JSONArray(json)
            for (i in 0 until minOf(buttons.length(), 2)) {
                val b = buttons.getJSONObject(i)
                val id = b.getString("id")
                val intent = if (b.getString("action") == "close") dismissIntent(d, id)
                             else openIntent(d, id, b.optString("url"))
                builder.addAction(0, b.getString("label"), intent)
            }
        }

        if (NotificationManagerCompat.from(this).areNotificationsEnabled()) {
            NotificationManagerCompat.from(this).notify(messageId.hashCode(), builder.build())
        }
    }

    private fun openIntent(d: Map<String, String>, buttonId: String?, url: String?): PendingIntent {
        val intent = Intent(this, MainActivity::class.java).apply {
            action = "com.example.app.PUSH_OPEN"
            putExtra("message_id", d["message_id"])
            putExtra("button", buttonId)
            putExtra("url", url)
            putExtra("inapp", d["inapp"])
            flags = Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_CLEAR_TOP
        }
        return PendingIntent.getActivity(this, (d["message_id"] + buttonId).hashCode(), intent,
            PendingIntent.FLAG_UPDATE_CURRENT or PendingIntent.FLAG_IMMUTABLE)
    }

    // dismissIntent(d, id) points at a BroadcastReceiver that cancels the
    // notification and reports the tap with POST /v1/members/me/push_opens.

    private fun loadBitmap(url: String): Bitmap? = try {
        URL(url).openStream().use { BitmapFactory.decodeStream(it) }
    } catch (e: IOException) {
        null
    }
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19{
  "aps": {
    "alert": {
      "title": "A pastry on us",
      "subtitle": "South Congress",
      "body": "Pick any pastry with your next order. It's in your wallet until November 12."
    },
    "mutable-content": 1,
    "category": "UNPLUG_TWO_BUTTONS"
  },
  "message_id": "5b81d3e0-2c4f-4a96-8e17-d0f9a3b62c58",
  "campaign_id": "e4a09b27-7d13-4c5e-9f82-31b6c0d8a4f1",
  "image_url": "https://cdn.unplugdining.com/demo/campaigns/fall-pastry.jpg",
  "deep_link": "yourapp://wallet/7e3a9c51-0b8d-4f2e-a6c4-5d1b9e7f3a28",
  "buttons": [
    { "id": "primary", "label": "See my wallet", "action": "open_screen", "url": "yourapp://wallet/7e3a9c51-0b8d-4f2e-a6c4-5d1b9e7f3a28" },
    { "id": "secondary", "label": "Not now", "action": "close", "url": null }
  ]
}

1
2
3
4
5
6
7
8
9
10
11{
  "message_id": "5b81d3e0-2c4f-4a96-8e17-d0f9a3b62c58",
  "campaign_id": "e4a09b27-7d13-4c5e-9f82-31b6c0d8a4f1",
  "title": "A pastry on us",
  "body": "Pick any pastry with your next order. It's in your wallet until November 12.",
  "image_url": "https://cdn.unplugdining.com/demo/campaigns/fall-pastry.jpg",
  "thumbnail_url": "https://cdn.unplugdining.com/demo/campaigns/fall-pastry-thumb.jpg",
  "deep_link": "yourapp://wallet/7e3a9c51-0b8d-4f2e-a6c4-5d1b9e7f3a28",
  "buttons": "[{\"id\":\"primary\",\"label\":\"See my wallet\",\"action\":\"open_screen\",\"url\":\"yourapp://wallet/7e3a9c51-0b8d-4f2e-a6c4-5d1b9e7f3a28\"},{\"id\":\"secondary\",\"label\":\"Not now\",\"action\":\"close\",\"url\":null}]",
  "channel_id": "marketing"
}