mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
d82eda1933
* Unmarshal directly into a `TwitterStats` struct by using json struct annotations * Pull stats fetching for a single user out into its own function so that closing the request body is done after each request and the HTTP request can be re-used for multiple usernames' stats * Improve type casting code used in settings parsing logic
92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package twitterstats
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
|
|
"golang.org/x/oauth2"
|
|
"golang.org/x/oauth2/clientcredentials"
|
|
)
|
|
|
|
// Client contains state that allows stats to be fetched about a list of Twitter users
|
|
type Client struct {
|
|
httpClient *http.Client
|
|
screenNames []string
|
|
}
|
|
|
|
// TwitterStats Represents a stats snapshot for a single Twitter user at a point in time
|
|
type TwitterStats struct {
|
|
FollowerCount int64 `json:"followers_count"`
|
|
TweetCount int64 `json:"statuses_count"`
|
|
}
|
|
|
|
const (
|
|
userTimelineURL = "https://api.twitter.com/1.1/users/show.json"
|
|
)
|
|
|
|
// NewClient creates a new twitterstats client that contains an OAuth2 HTTP client which can be used
|
|
func NewClient(settings *Settings) *Client {
|
|
usernames := make([]string, len(settings.screenNames))
|
|
for i, username := range settings.screenNames {
|
|
var ok bool
|
|
if usernames[i], ok = username.(string); !ok {
|
|
log.Fatalf("All `screenName`s in twitterstats config must be of type string")
|
|
}
|
|
}
|
|
|
|
conf := &clientcredentials.Config{
|
|
ClientID: settings.consumerKey,
|
|
ClientSecret: settings.consumerSecret,
|
|
TokenURL: "https://api.twitter.com/oauth2/token",
|
|
}
|
|
httpClient := conf.Client(oauth2.NoContext)
|
|
|
|
client := Client{
|
|
httpClient: httpClient,
|
|
screenNames: usernames,
|
|
}
|
|
|
|
return &client
|
|
}
|
|
|
|
// GetStatsForUser Fetches stats for a single user. If there is an error fetching or parsing the response
|
|
// from the Twitter API, an empty stats struct will be returned.
|
|
func (client *Client) GetStatsForUser(username string) TwitterStats {
|
|
stats := TwitterStats{
|
|
FollowerCount: 0,
|
|
TweetCount: 0,
|
|
}
|
|
|
|
url := fmt.Sprintf("%s?screen_name=%s", userTimelineURL, username)
|
|
res, err := client.httpClient.Get(url)
|
|
if err != nil {
|
|
return stats
|
|
}
|
|
defer res.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(res.Body)
|
|
if err != nil {
|
|
return stats
|
|
}
|
|
|
|
// If there is an error while parsing, just discard the error and return the empty stats
|
|
json.Unmarshal(body, &stats)
|
|
|
|
return stats
|
|
}
|
|
|
|
// GetStats Returns a slice of `TwitterStats` structs for each username in `client.screenNames` in the same
|
|
// order of `client.screenNames`
|
|
func (client *Client) GetStats() []TwitterStats {
|
|
stats := make([]TwitterStats, len(client.screenNames))
|
|
|
|
for i, username := range client.screenNames {
|
|
stats[i] = client.GetStatsForUser(username)
|
|
}
|
|
|
|
return stats
|
|
}
|