blob: 33fc0130e264d0a8490fd23f47858c6cb137310a (
plain)
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
|
package net.lacolaco.smileessence.entity
import twitter4j.HashtagEntity
import twitter4j.MediaEntity
import twitter4j.URLEntity
import twitter4j.UserMentionEntity
import java.util.*
class Entities private constructor(
val mentions: List<String>,
val hashtags: List<String>,
val mediaUrls: List<String>,
val urlsExpanded: List<String>
) {
fun extractText(status: twitter4j.EntitySupport, text: String): String {
val set = TreeSet<twitter4j.URLEntity> { a, b -> a.start - b.start }
if (status.urlEntities != null)
set += status.urlEntities
if (status.mediaEntities != null)
set += status.mediaEntities
return set.fold(text) { t, entity ->
t.replaceFirst(entity.text.toRegex(), entity.displayURL)
}
}
companion object {
fun fromTwitter(es: twitter4j.EntitySupport) = Entities(
mentions = extractMentions(es.userMentionEntities),
hashtags = extractHashtags(es.hashtagEntities),
mediaUrls = extractMediaUrls(es.mediaEntities),
urlsExpanded = extractExpandedUrls(es.urlEntities)
)
private fun extractMentions(entities: Array<UserMentionEntity>?) =
entities?.map { it.screenName } ?: listOf()
private fun extractExpandedUrls(entities: Array<URLEntity>?) =
entities?.map { it.expandedURL } ?: listOf()
private fun extractMediaUrls(entities: Array<MediaEntity>?) =
entities?.map {
when (it.type) {
"photo" -> it.mediaURLHttps
else -> it.expandedURL
}
} ?: listOf()
private fun extractHashtags(entities: Array<HashtagEntity>?) =
entities?.map { it.text } ?: listOf()
val placeHolder = Entities(listOf(), listOf(), listOf(), listOf())
}
}
|