search.go
author Mikael Berthe <mikael@lilotux.net>
Sun, 16 Feb 2020 18:48:57 +0100
changeset 254 0362ce76324f
parent 240 80c81e9b77b4
permissions -rw-r--r--
Update dependencies
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
130
c450bb73f59a Update credits
Mikael Berthe <mikael@lilotux.net>
parents: 120
diff changeset
     1
/*
207
301d5b94be3f Update copyrights
Mikael Berthe <mikael@lilotux.net>
parents: 155
diff changeset
     2
Copyright 2017-2018 Mikael Berthe
130
c450bb73f59a Update credits
Mikael Berthe <mikael@lilotux.net>
parents: 120
diff changeset
     3
c450bb73f59a Update credits
Mikael Berthe <mikael@lilotux.net>
parents: 120
diff changeset
     4
Licensed under the MIT license.  Please see the LICENSE file is this directory.
c450bb73f59a Update credits
Mikael Berthe <mikael@lilotux.net>
parents: 120
diff changeset
     5
*/
c450bb73f59a Update credits
Mikael Berthe <mikael@lilotux.net>
parents: 120
diff changeset
     6
138
23d3a518d0ad Update package name in source files
Mikael Berthe <mikael@lilotux.net>
parents: 130
diff changeset
     7
package madon
113
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     8
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
     9
import (
240
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    10
	"strings"
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    11
113
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    12
	"github.com/sendgrid/rest"
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    13
)
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    14
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    15
// Search search for contents (accounts or statuses) and returns a Results
240
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    16
func (mc *Client) searchV1(params apiCallParams) (*Results, error) {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    17
	// We use a custom structure with shadowed Hashtags field,
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    18
	// since the v1 version only returns strings.
236
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    19
	var resultsV1 struct {
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    20
		Results
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    21
		Hashtags []string `json:"hashtags"`
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    22
	}
238
1c0042e76902 Do not use a global API version
Mikael Berthe <mikael@lilotux.net>
parents: 236
diff changeset
    23
	if err := mc.apiCall("v1/"+"search", rest.Get, params, nil, nil, &resultsV1); err != nil {
120
579912e9d0ef Refactor API calls
Mikael Berthe <mikael@lilotux.net>
parents: 113
diff changeset
    24
		return nil, err
113
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    25
	}
236
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    26
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    27
	var results Results
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    28
	results.Accounts = resultsV1.Accounts
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    29
	results.Statuses = resultsV1.Statuses
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    30
	for _, t := range resultsV1.Hashtags {
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    31
		results.Hashtags = append(results.Hashtags, Tag{Name: t})
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    32
	}
5b87cc73ed97 Update search result type (us rich tag structure)
Mikael Berthe <mikael@lilotux.net>
parents: 207
diff changeset
    33
113
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    34
	return &results, nil
bb9aaa5440c1 Add Search() (using endpoint /search)
Mikael Berthe <mikael@lilotux.net>
parents:
diff changeset
    35
}
240
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    36
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    37
func (mc *Client) searchV2(params apiCallParams) (*Results, error) {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    38
	var results Results
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    39
	if err := mc.apiCall("v2/"+"search", rest.Get, params, nil, nil, &results); err != nil {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    40
		return nil, err
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    41
	}
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    42
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    43
	return &results, nil
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    44
}
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    45
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    46
// Search search for contents (accounts or statuses) and returns a Results
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    47
func (mc *Client) Search(query string, resolve bool) (*Results, error) {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    48
	if query == "" {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    49
		return nil, ErrInvalidParameter
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    50
	}
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    51
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    52
	// The parameters are the same in both v1 & v2 API versions
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    53
	params := make(apiCallParams)
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    54
	params["q"] = query
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    55
	if resolve {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    56
		params["resolve"] = "true"
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    57
	}
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    58
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    59
	r, err := mc.searchV2(params)
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    60
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    61
	// This is not a very beautiful way to check the error cause, I admit.
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    62
	if err != nil && strings.Contains(err.Error(), "bad server status code (404)") {
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    63
		// Fall back to v1 API endpoint
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    64
		r, err = mc.searchV1(params)
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    65
	}
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    66
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    67
	return r, err
80c81e9b77b4 Support search v2 API
Mikael Berthe <mikael@lilotux.net>
parents: 238
diff changeset
    68
}