blob: 121fa17df210272687bacfbc13143fe21181701a (
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.*
import java.util.*
abstract class EntitySupport(status: twitter4j.EntitySupport) {
open val mentions: List<String> = extractMentions(status.userMentionEntities)
open val hashtags: List<String> = extractHashtags(status.hashtagEntities)
open val mediaUrls: List<String> = extractMediaUrls(status.mediaEntities)
open val urlsExpanded: List<String> = extractExpandedUrls(status.urlEntities)
open val symbols: List<String> = extractSymbols(status.symbolEntities)
private fun extractMentions(entities: Array<UserMentionEntity>?) =
entities?.map { it.screenName } ?: listOf()
private fun extractSymbols(entities: Array<SymbolEntity>?) =
entities?.map { it.text } ?: 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()
protected fun extractText(status: twitter4j.Status, expand: Boolean) =
extractText(status, status.text, expand)
protected fun extractText(status: twitter4j.DirectMessage, expand: Boolean) =
extractText(status, status.text, expand)
private fun extractText(status: twitter4j.EntitySupport, text: String, expand: Boolean): 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
var t = text
for (entity in set) {
val newString = if (expand) entity.expandedURL else entity.displayURL
t = t.replaceFirst(entity.text.toRegex(), newString)
}
return t
}
}
|
