package gif import ( "encoding/json" "fmt" "io/ioutil" "net/http" "time" "go.uber.org/zap" "github.com/status-im/status-go/logutils" "github.com/status-im/status-go/multiaccounts/accounts" "github.com/status-im/status-go/multiaccounts/settings" ) type Gif struct { ID string `json:"id"` Title string `json:"title"` URL string `json:"url"` TinyURL string `json:"tinyUrl"` Height int `json:"height"` IsFavorite bool `json:"isFavorite"` } type Container struct { Items []Gif `json:"items"` } var tenorAPIKey = "" var defaultParams = "&media_filter=minimal&limit=50&key=" const maxRetry = 3 const baseURL = "https://g.tenor.com/v1/" func NewGifAPI(db *accounts.Database) *API { return &API{db} } // API is class with methods available over RPC. type API struct { db *accounts.Database } func (api *API) SetTenorAPIKey(key string) (err error) { logutils.ZapLogger().Info("[GifAPI::SetTenorAPIKey]") err = api.db.SaveSettingField(settings.GifAPIKey, key) if err != nil { return err } tenorAPIKey = key return nil } func (api *API) GetContentWithRetry(path string) (value string, err error) { var currentRetry = 0 var response *http.Response for currentRetry < maxRetry { transport := &http.Transport{ Proxy: http.ProxyFromEnvironment, ResponseHeaderTimeout: time.Second * 1, } client := http.Client{ Timeout: 1 * time.Second, Transport: transport, } response, err = client.Get(baseURL + path + defaultParams + tenorAPIKey) if err != nil { logutils.ZapLogger().Error("can't get content", zap.String("path", path), zap.Error(err)) currentRetry++ time.Sleep(100 * time.Millisecond) } else { break } } if response == nil { return "", fmt.Errorf("Could not reach Tenor API") } defer response.Body.Close() if response.StatusCode != http.StatusOK { return "", fmt.Errorf("Status error: %v", response.StatusCode) } data, err := ioutil.ReadAll(response.Body) if err != nil { return "", fmt.Errorf("Read body: %v", err) } return string(data), nil } func (api *API) FetchGifs(path string) (value string, err error) { return api.GetContentWithRetry(path) } func (api *API) UpdateRecentGifs(updatedGifs json.RawMessage) (err error) { recentGifsContainer := Container{} err = json.Unmarshal(updatedGifs, &recentGifsContainer) if err != nil { return err } err = api.db.SaveSettingField(settings.GifRecents, recentGifsContainer.Items) if err != nil { return err } return nil } func (api *API) UpdateFavoriteGifs(updatedGifs json.RawMessage) (err error) { favsGifsContainer := Container{} err = json.Unmarshal(updatedGifs, &favsGifsContainer) if err != nil { return err } err = api.db.SaveSettingField(settings.GifFavourites, favsGifsContainer.Items) if err != nil { return err } return nil } func (api *API) GetRecentGifs() (recentGifs []Gif, err error) { gifs, err := api.db.GifRecents() if err != nil { return nil, err } recentGifs = make([]Gif, 0) savedRecentGifs := []Gif{} if len(gifs) > 0 { err = json.Unmarshal(gifs, &savedRecentGifs) if err != nil { return nil, err } recentGifs = savedRecentGifs } return recentGifs, nil } func (api *API) GetFavoriteGifs() (favoriteGifs []Gif, err error) { gifs, err := api.db.GifFavorites() if err != nil { return nil, err } favoriteGifs = make([]Gif, 0) savedFavGifs := []Gif{} if len(gifs) > 0 { err = json.Unmarshal(gifs, &savedFavGifs) if err != nil { return nil, err } favoriteGifs = savedFavGifs } return favoriteGifs, nil }