You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

150 lines
3.7 KiB

// Package guggy implements a Service which adds !commands for Guggy.
package guggy
import (
"bytes"
"encoding/json"
"fmt"
"io/ioutil"
"math"
"net/http"
"strings"
"github.com/matrix-org/go-neb/types"
"github.com/matrix-org/gomatrix"
log "github.com/sirupsen/logrus"
)
// ServiceType of the Guggy service
const ServiceType = "guggy"
var httpClient = &http.Client{}
type guggyQuery struct {
// "mp4" or "gif"
Format string `json:"format"`
// Query sentence
Sentence string `json:"sentence"`
}
type guggyGifResult struct {
ReqID string `json:"reqId"`
GIF string `json:"gif"`
Width float64 `json:"width"`
Height float64 `json:"height"`
}
// Service contains the Config fields for the Guggy service.
//
// Example request:
// {
// "api_key": "fkweugfyuwegfweyg"
// }
type Service struct {
types.DefaultService
// The Guggy API key to use when making HTTP requests to Guggy.
APIKey string `json:"api_key"`
}
// Commands supported:
// !guggy some search query without quotes
// Responds with a suitable GIF into the same room as the command.
func (s *Service) Commands(client *gomatrix.Client) []types.Command {
return []types.Command{
types.Command{
Path: []string{"guggy"},
Command: func(roomID, userID string, args []string) (interface{}, error) {
return s.cmdGuggy(client, roomID, userID, args)
},
},
}
}
func (s *Service) cmdGuggy(client *gomatrix.Client, roomID, userID string, args []string) (interface{}, error) {
// only 1 arg which is the text to search for.
querySentence := strings.Join(args, " ")
gifResult, err := s.text2gifGuggy(querySentence)
if err != nil {
return nil, fmt.Errorf("Failed to query Guggy: %s", err.Error())
}
if gifResult.GIF == "" {
return gomatrix.TextMessage{
MsgType: "m.notice",
Body: "No GIF found!",
}, nil
}
resUpload, err := client.UploadLink(gifResult.GIF)
if err != nil {
return nil, fmt.Errorf("Failed to upload Guggy image to matrix: %s", err.Error())
}
return gomatrix.ImageMessage{
MsgType: "m.image",
Body: querySentence,
URL: resUpload.ContentURI,
Info: gomatrix.ImageInfo{
Height: uint(math.Floor(gifResult.Height)),
Width: uint(math.Floor(gifResult.Width)),
Mimetype: "image/gif",
},
}, nil
}
// text2gifGuggy returns info about a gif
func (s *Service) text2gifGuggy(querySentence string) (*guggyGifResult, error) {
log.Info("Transforming to GIF query ", querySentence)
var query guggyQuery
query.Format = "gif"
query.Sentence = querySentence
reqBody, err := json.Marshal(query)
if err != nil {
return nil, err
}
reader := bytes.NewReader(reqBody)
req, err := http.NewRequest("POST", "https://text2gif.guggy.com/guggify", reader)
if err != nil {
log.Error(err)
return nil, err
}
req.Header.Add("Content-Type", "application/json")
req.Header.Add("apiKey", s.APIKey)
res, err := httpClient.Do(req)
if res != nil {
defer res.Body.Close()
}
if err != nil {
log.Error(err)
return nil, err
}
if res.StatusCode < 200 || res.StatusCode >= 300 {
resBytes, err := ioutil.ReadAll(res.Body)
if err != nil {
log.WithError(err).Error("Failed to decode Guggy response body")
}
log.WithFields(log.Fields{
"code": res.StatusCode,
"body": string(resBytes),
}).Error("Failed to query Guggy")
return nil, fmt.Errorf("Failed to decode response (HTTP %d)", res.StatusCode)
}
var result guggyGifResult
if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
return nil, fmt.Errorf("Failed to decode response (HTTP %d): %s", res.StatusCode, err.Error())
}
return &result, nil
}
func init() {
types.RegisterService(func(serviceID, serviceUserID, webhookEndpointURL string) types.Service {
return &Service{
DefaultService: types.NewDefaultService(serviceID, serviceUserID, ServiceType),
}
})
}