FlameInTheDark / dtbot

api/translate/translate.go

package translate

import (
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"strings"

	"github.com/FlameInTheDark/dtbot/bot"
)

// TranslateResponse : Translate API struct
type TranslateResponse struct {

type name will be used as translate.TranslateResponse by other packages, and that stutters; consider calling this Response

	Code     int      `json:"code"`
	Language string   `json:"lang"`
	Text     []string `json:"text"`
	Message  string   `json:"message"`
}

// GetTranslation returns translated text
func GetTranslation(ctx *bot.Context) (string, error) {
	var (
		result    TranslateResponse
		translate string
	)

	if len(ctx.Args) > 1 {
		translate = strings.Join(ctx.Args[1:], "+")
	} else {
		return "", errors.New(ctx.Loc("translate_request_error"))
	}

	resp, err := http.Get(fmt.Sprintf("https://translate.yandex.net/api/v1.5/tr.json/translate?key=%v&text=%v&lang=%v&format=plain", ctx.Conf.Translate.APIKey, translate, ctx.Args[0]))
	if err != nil {
		return "", fmt.Errorf("%v: %v", ctx.Loc("translate_get_error"), err)
	}

	err = json.NewDecoder(resp.Body).Decode(&result)
	if err != nil {
		return "", fmt.Errorf("%v: %v", ctx.Loc("translate_parse_error"), err)
	}

	// Checking request status
	switch result.Code {
	case 502:
		return "", errors.New(ctx.Loc("translate_request_error"))
	case 200:
		return strings.Join(result.Text, "\n"), nil
	default:
		return "", errors.New(ctx.Loc("translate_api_error"))
	}
}