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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
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>?): List<String> {
val names = ArrayList<String>()
if (entities != null) {
for (entity in entities) {
names.add(entity.screenName)
}
}
return names
}
private fun extractSymbols(entities: Array<SymbolEntity>?): List<String> {
val names = ArrayList<String>()
if (entities != null) {
for (entity in entities) {
names.add(entity.text)
}
}
return names
}
private fun extractExpandedUrls(entities: Array<URLEntity>?): List<String> {
val names = ArrayList<String>()
if (entities != null) {
for (entity in entities) {
names.add(entity.expandedURL)
}
}
return names
}
private fun extractMediaUrls(entities: Array<MediaEntity>?): List<String> {
val names = ArrayList<String>()
if (entities != null) {
for (entity in entities) {
if ("photo" == entity.type) {
names.add(entity.mediaURLHttps)
} else {
names.add(entity.expandedURL)
}
}
}
return names
}
private fun extractHashtags(entities: Array<HashtagEntity>?): List<String> {
val names = ArrayList<String>()
if (entities != null) {
for (entity in entities) {
names.add(entity.text)
}
}
return names
}
protected fun extractText(status: twitter4j.Status, expand: Boolean): String {
return extractText(status, status.text, expand)
}
protected fun extractText(status: twitter4j.DirectMessage, expand: Boolean): String {
return 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) {
Collections.addAll(set, *status.urlEntities)
}
if (status.mediaEntities != null) {
Collections.addAll(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
}
}
|
