aboutsummaryrefslogtreecommitdiffstats
path: root/app/src/main/java/net/lacolaco/smileessence/entity/EntitySupport.java
blob: 1df7cbd113fe684b9dd4ba31855da8855945f1d0 (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
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
93
package net.lacolaco.smileessence.entity;

import twitter4j.*;

import java.util.ArrayList;
import java.util.List;

public abstract class EntitySupport {
    private List<String> mentions;
    private List<String> hashtags;
    private List<String> mediaUrls;
    private List<String> urlsExpanded;
    private List<String> symbols;

    protected void updateEntities(twitter4j.EntitySupport status) {
        mentions = extractMentions(status.getUserMentionEntities());
        hashtags = extractHashtags(status.getHashtagEntities());
        mediaUrls = extractMediaUrls(status.getExtendedMediaEntities().length > 0 ? status.getExtendedMediaEntities() : status.getMediaEntities());
        urlsExpanded = extractExpandedUrls(status.getURLEntities());
        symbols = extractSymbols(status.getSymbolEntities());
    }

    public List<String> getMentions() {
        return mentions;
    }

    public List<String> getHashtags() {
        return hashtags;
    }

    public List<String> getMediaUrls() {
        return mediaUrls;
    }

    public List<String> getUrlsExpanded() {
        return urlsExpanded;
    }

    public List<String> getSymbols() {
        return symbols;
    }

    private List<String> extractMentions(UserMentionEntity[] entities) {
        List<String> names = new ArrayList<>();
        if (entities != null) {
            for (UserMentionEntity entity : entities) {
                names.add(entity.getScreenName());
            }
        }
        return names;
    }

    private List<String> extractSymbols(SymbolEntity[] entities) {
        List<String> names = new ArrayList<>();
        if (entities != null) {
            for (SymbolEntity entity : entities) {
                names.add(entity.getText());
            }
        }
        return names;
    }

    private List<String> extractExpandedUrls(URLEntity[] entities) {
        List<String> names = new ArrayList<>();
        if (entities != null) {
            for (URLEntity entity : entities) {
                names.add(entity.getExpandedURL());
            }
        }
        return names;
    }

    private List<String> extractMediaUrls(MediaEntity[] entities) {
        List<String> names = new ArrayList<>();
        if (entities != null) {
            for (MediaEntity entity : entities) {
                names.add(entity.getMediaURLHttps());
            }
        }
        return names;
    }

    private List<String> extractHashtags(HashtagEntity[] entities) {
        List<String> names = new ArrayList<>();
        if (entities != null) {
            for (HashtagEntity entity : entities) {
                names.add(entity.getText());
            }
        }
        return names;
    }

}