Marvin Preuss
1d4ae27878
All checks were successful
continuous-integration/drone/push Build is passing
117 lines
4.9 KiB
Go
117 lines
4.9 KiB
Go
package twitter
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/dghubble/sling"
|
|
)
|
|
|
|
// PremiumSearch represents the result of a Tweet search.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search
|
|
type PremiumSearch struct {
|
|
Results []Tweet `json:"results"`
|
|
Next string `json:"next"`
|
|
RequestParameters *RequestParameters `json:"requestParameters"`
|
|
}
|
|
|
|
// RequestParameters describes a request parameter that was passed to a Premium search API.
|
|
type RequestParameters struct {
|
|
MaxResults int `json:"maxResults"`
|
|
FromDate string `json:"fromDate"`
|
|
ToDate string `json:"toDate"`
|
|
}
|
|
|
|
// PremiumSearchCount describes a response of Premium search API's count endpoint.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search#CountsEndpoint
|
|
type PremiumSearchCount struct {
|
|
Results []TweetCount `json:"results"`
|
|
TotalCount int64 `json:"totalCount"`
|
|
RequestParameters *RequestCountParameters `json:"requestParameters"`
|
|
}
|
|
|
|
// RequestCountParameters describes a request parameter that was passed to a Premium search API.
|
|
type RequestCountParameters struct {
|
|
Bucket string `json:"bucket"`
|
|
FromDate string `json:"fromDate"`
|
|
ToDate string `json:"toDate"`
|
|
}
|
|
|
|
// TweetCount represents a count of Tweets in the TimePeriod matching a search query.
|
|
type TweetCount struct {
|
|
TimePeriod string `json:"timePeriod"`
|
|
Count int64 `json:"count"`
|
|
}
|
|
|
|
// PremiumSearchService provides methods for accessing Twitter premium search API endpoints.
|
|
type PremiumSearchService struct {
|
|
sling *sling.Sling
|
|
}
|
|
|
|
// newSearchService returns a new SearchService.
|
|
func newPremiumSearchService(sling *sling.Sling) *PremiumSearchService {
|
|
return &PremiumSearchService{
|
|
sling: sling.Path("tweets/search/"),
|
|
}
|
|
}
|
|
|
|
// PremiumSearchTweetParams are the parameters for PremiumSearchService.SearchFullArchive and Search30Days
|
|
type PremiumSearchTweetParams struct {
|
|
Query string `url:"query,omitempty"`
|
|
Tag string `url:"tag,omitempty"`
|
|
FromDate string `url:"fromDate,omitempty"`
|
|
ToDate string `url:"toDate,omitempty"`
|
|
MaxResults int `url:"maxResults,omitempty"`
|
|
Next string `url:"next,omitempty"`
|
|
}
|
|
|
|
// PremiumSearchCountTweetParams are the parameters for PremiumSearchService.CountFullArchive and Count30Days
|
|
type PremiumSearchCountTweetParams struct {
|
|
Query string `url:"query,omitempty"`
|
|
Tag string `url:"tag,omitempty"`
|
|
FromDate string `url:"fromDate,omitempty"`
|
|
ToDate string `url:"toDate,omitempty"`
|
|
Bucket string `url:"bucket,omitempty"`
|
|
Next string `url:"next,omitempty"`
|
|
}
|
|
|
|
// SearchFullArchive returns a collection of Tweets matching a search query from tweets back to the very first tweet.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search
|
|
func (s *PremiumSearchService) SearchFullArchive(params *PremiumSearchTweetParams, label string) (*PremiumSearch, *http.Response, error) {
|
|
search := new(PremiumSearch)
|
|
apiError := new(APIError)
|
|
path := fmt.Sprintf("fullarchive/%s.json", label)
|
|
resp, err := s.sling.New().Get(path).QueryStruct(params).Receive(search, apiError)
|
|
return search, resp, relevantError(err, *apiError)
|
|
}
|
|
|
|
// Search30Days returns a collection of Tweets matching a search query from Tweets posted within the last 30 days.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search
|
|
func (s *PremiumSearchService) Search30Days(params *PremiumSearchTweetParams, label string) (*PremiumSearch, *http.Response, error) {
|
|
search := new(PremiumSearch)
|
|
apiError := new(APIError)
|
|
path := fmt.Sprintf("30day/%s.json", label)
|
|
resp, err := s.sling.New().Get(path).QueryStruct(params).Receive(search, apiError)
|
|
return search, resp, relevantError(err, *apiError)
|
|
}
|
|
|
|
// CountFullArchive returns a counts of Tweets matching a search query from tweets back to the very first tweet.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search#CountsEndpoint
|
|
func (s *PremiumSearchService) CountFullArchive(params *PremiumSearchCountTweetParams, label string) (*PremiumSearchCount, *http.Response, error) {
|
|
counts := new(PremiumSearchCount)
|
|
apiError := new(APIError)
|
|
path := fmt.Sprintf("fullarchive/%s/counts.json", label)
|
|
resp, err := s.sling.New().Get(path).QueryStruct(params).Receive(counts, apiError)
|
|
return counts, resp, relevantError(err, *apiError)
|
|
}
|
|
|
|
// Count30Days returns a counts of Tweets matching a search query from Tweets posted within the last 30 days.
|
|
// https://developer.twitter.com/en/docs/tweets/search/api-reference/premium-search#CountsEndpoint
|
|
func (s *PremiumSearchService) Count30Days(params *PremiumSearchCountTweetParams, label string) (*PremiumSearchCount, *http.Response, error) {
|
|
counts := new(PremiumSearchCount)
|
|
apiError := new(APIError)
|
|
path := fmt.Sprintf("30day/%s/counts.json", label)
|
|
resp, err := s.sling.New().Get(path).QueryStruct(params).Receive(counts, apiError)
|
|
return counts, resp, relevantError(err, *apiError)
|
|
}
|