app/src/main/kotlin/com/pitchedapps/frost/facebook/parsers/BadgeParser.kt


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

package com.pitchedapps.frost.facebook.parsers

import com.pitchedapps.frost.R
import com.pitchedapps.frost.facebook.FB_URL_BASE
import org.jsoup.nodes.Document

object BadgeParser : FrostParser<FrostBadges> by BadgeParserImpl()

data class FrostBadges(
    val feed: String?,
    val friends: String?,
    val messages: String?,
    val notifications: String?
) : ParseData {
    override val isEmpty: Boolean
        get() = feed.isNullOrEmpty()
            && friends.isNullOrEmpty()
            && messages.isNullOrEmpty()
            && notifications.isNullOrEmpty()
}

private class BadgeParserImpl : FrostParserBase<FrostBadges>(false) {
    // Not actually displayed
    override var nameRes: Int = R.string.frost_name

    override val url: String = FB_URL_BASE

    override fun parseImpl(doc: Document): FrostBadges? {
        val header = doc.getElementById("header") ?: return null
        if (header.select("[data-sigil=count]").isEmpty())
            return null
        val (feed, requests, messages, notifications) = listOf(
            "feed",
            "requests",
            "messages",
            "notifications"
        )
            .map { "[data-sigil*=$it] [data-sigil=count]" }
            .map { doc.select(it) }
            .map { e -> e?.getOrNull(0)?.ownText() }
        return FrostBadges(
            feed = feed,
            friends = requests,
            messages = messages,
            notifications = notifications
        )
    }
}