From 4a6157525367c748d6e76915d3b4329cf62539a2 Mon Sep 17 00:00:00 2001
From: Luke Barnard <lukeb@openmarket.com>
Date: Thu, 29 Sep 2016 13:09:22 +0100
Subject: [PATCH] Finish Guggy integration

---
 src/github.com/matrix-org/go-neb/goneb.go     |  1 +
 .../matrix-org/go-neb/services/guggy/guggy.go | 66 ++++++++-----------
 2 files changed, 27 insertions(+), 40 deletions(-)

diff --git a/src/github.com/matrix-org/go-neb/goneb.go b/src/github.com/matrix-org/go-neb/goneb.go
index e0811ad..c9ce110 100644
--- a/src/github.com/matrix-org/go-neb/goneb.go
+++ b/src/github.com/matrix-org/go-neb/goneb.go
@@ -9,6 +9,7 @@ import (
 	_ "github.com/matrix-org/go-neb/realms/jira"
 	"github.com/matrix-org/go-neb/server"
 	_ "github.com/matrix-org/go-neb/services/echo"
+	_ "github.com/matrix-org/go-neb/services/guggy"
 	_ "github.com/matrix-org/go-neb/services/giphy"
 	_ "github.com/matrix-org/go-neb/services/github"
 	_ "github.com/matrix-org/go-neb/services/jira"
diff --git a/src/github.com/matrix-org/go-neb/services/guggy/guggy.go b/src/github.com/matrix-org/go-neb/services/guggy/guggy.go
index be27cad..501a29b 100644
--- a/src/github.com/matrix-org/go-neb/services/guggy/guggy.go
+++ b/src/github.com/matrix-org/go-neb/services/guggy/guggy.go
@@ -2,30 +2,16 @@ package services
 
 import (
 	"encoding/json"
-	"errors"
+	"bytes"
+	"math"
 	log "github.com/Sirupsen/logrus"
 	"github.com/matrix-org/go-neb/matrix"
 	"github.com/matrix-org/go-neb/plugin"
 	"github.com/matrix-org/go-neb/types"
 	"net/http"
-	"net/url"
 	"strconv"
 	"strings"
 )
-
-type result struct {
-	Slug   string `json:"slug"`
-	Images struct {
-		Original struct {
-			URL string `json:"url"`
-			// guggy returns ints as strings..
-			Width  string `json:"width"`
-			Height string `json:"height"`
-			Size   string `json:"size"`
-		} `json:"original"`
-	} `json:"images"`
-}
-
 type guggyQuery struct {
 	// "mp4" of "gif"
 	Format string `json:"format"`
@@ -33,14 +19,17 @@ type guggyQuery struct {
 	Sentence  string `json:"sentence"`
 }
 
-type guggySearch struct {
-	Data []result
+type guggyGifResult struct {
+	ReqID string `json:"reqId"`
+	GIF string `json:"gif"`
+	Width float64 `json:"width"`
+	Height float64 `json:"height"`
 }
 
 type guggyService struct {
 	id            string
 	serviceUserID string
-	APIKey        string // key is Cb7aEsJIdjD37c3
+	APIKey        string
 }
 
 func (s *guggyService) ServiceUserID() string { return s.serviceUserID }
@@ -70,45 +59,43 @@ func (s *guggyService) cmdGuggy(client *matrix.Client, roomID, userID string, ar
 	if err != nil {
 		return nil, err
 	}
-	mxc, err := client.UploadLink(gifResult.Images.Original.URL)
+	mxc, err := client.UploadLink(gifResult.GIF)
 	if err != nil {
 		return nil, err
 	}
 
 	return matrix.ImageMessage{
 		MsgType: "m.image",
-		Body:    gifResult.Slug,
+		Body:    gifResult.ReqID,
 		URL:     mxc,
 		Info: matrix.ImageInfo{
-			Height:   asInt(gifResult.Images.Original.Height),
-			Width:    asInt(gifResult.Images.Original.Width),
+			Height:   uint(math.Floor(gifResult.Height)),
+			Width:    uint(math.Floor(gifResult.Width)),
 			Mimetype: "image/gif",
-			Size:     asInt(gifResult.Images.Original.Size),
 		},
 	}, nil
 }
 
 // text2gifGuggy returns info about a gif
-func (s *guggyService) text2gifGuggy(querySentence string) (*result, error) {
+func (s *guggyService) text2gifGuggy(querySentence string) (*guggyGifResult, error) {
 	log.Info("Transforming to GIF query ", querySentence)
-	u, err := url.Parse("http://text2gif.guggy.com/guggify")
-	if err != nil {
-		return nil, err
-	}
 
 	client := &http.Client{ }
 
 	var query guggyQuery
-	query.format = "gif"
-	query.sentence = querySentence
+	query.Format = "gif"
+	query.Sentence = querySentence
 
-	var reqBody bytes.Buffer
-	if json.NewEncoder(reqBody).Encode(query); err != nil {
+	reqBody, err := json.Marshal(query);
+	if err != nil {
 		return nil, err
 	}
 
-	req, err := http.NewRequest("POST", u, reqBody)
+	reader := bytes.NewReader(reqBody)
+
+	req, err := http.NewRequest("POST", "http://text2gif.guggy.com/guggify", reader)
 	if err != nil {
+		log.Error(err)
 		return nil, err
 	}
 	req.Header.Add("Content-Type", "application/json")
@@ -119,16 +106,15 @@ func (s *guggyService) text2gifGuggy(querySentence string) (*result, error) {
 		defer res.Body.Close()
 	}
 	if err != nil {
+		log.Error(err)
 		return nil, err
 	}
-	var search guggySearch
-	if err := json.NewDecoder(res.Body).Decode(&search); err != nil {
+	var result guggyGifResult
+	if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
 		return nil, err
 	}
-	if len(search.Data) == 0 {
-		return nil, errors.New("No results")
-	}
-	return &search.Data[0], nil
+
+	return &result, nil
 }
 
 func asInt(strInt string) uint {