160 lines
3.5 KiB
Go
160 lines
3.5 KiB
Go
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
|
|
}
|