WordPress/src/main/java/org/wordpress/android/ui/stats/models/GeoviewsModel.java


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
94
95
96

package org.wordpress.android.ui.stats.models;

import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;


public class GeoviewsModel extends BaseStatsModel {
    private String mDate;
    private String mBlogID;
    private int otherViews;
    private int totalViews;
    private List<GeoviewModel> countries;

    public GeoviewsModel(String blogID, JSONObject response) throws JSONException {
        this.mBlogID = blogID;
        this.mDate = response.getString("date");

        JSONObject jDaysObject = response.getJSONObject("days");
        if (jDaysObject.length() == 0) {
            throw new JSONException("Invalid document returned from the REST API");
        }

        // Read the first day
        Iterator<String> keys = jDaysObject.keys();
        String firstDayKey = keys.next();
        JSONObject firstDayObject = jDaysObject.getJSONObject(firstDayKey);
        this.otherViews = firstDayObject.getInt("other_views");
        this.totalViews = firstDayObject.getInt("total_views");

        JSONObject countryInfoJSON = response.optJSONObject("country-info");
        JSONArray viewsJSON = firstDayObject.optJSONArray("views");

        if (viewsJSON != null && countryInfoJSON != null) {
            countries = new ArrayList<>(viewsJSON.length());
            for (int i = 0; i < viewsJSON.length(); i++) {
                JSONObject currentCountryJSON = viewsJSON.getJSONObject(i);
                String currentCountryCode = currentCountryJSON.getString("country_code");
                int currentCountryViews = currentCountryJSON.getInt("views");
                String flagIcon = null;
                String flatFlagIcon = null;
                String countryFullName = null;
                JSONObject currentCountryDetails = countryInfoJSON.optJSONObject(currentCountryCode);
                if (currentCountryDetails != null) {
                    flagIcon = currentCountryDetails.optString("flag_icon");
                    flatFlagIcon = currentCountryDetails.optString("flat_flag_icon");
                    countryFullName = currentCountryDetails.optString("country_full");
                }
                GeoviewModel m = new GeoviewModel(currentCountryCode, countryFullName, currentCountryViews, flagIcon, flatFlagIcon);
                countries.add(m);

            }

            // Sort the countries by views.
            Collections.sort(countries, new java.util.Comparator<GeoviewModel>() {
                public int compare(GeoviewModel o1, GeoviewModel o2) {
                    // descending order
                    return o2.getViews() - o1.getViews();
                }
            });
        }
    }

    public String getBlogID() {
        return mBlogID;
    }

    public void setBlogID(String blogID) {
        this.mBlogID = blogID;
    }

    public String getDate() {
        return mDate;
    }

    public void setDate(String date) {
        this.mDate = date;
    }

    public List<GeoviewModel> getCountries() {
        return this.countries;
    }

    public int getOtherViews() {
        return otherViews;
    }

    public int getTotalViews() {
        return totalViews;
    }
}