FlameInTheDark / dtbot

api/fun/images.go

package fun

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"net/http"
)

type ImageResponse struct {

exported type ImageResponse should have comment or be unexported

	Error    string `json:"error"`
	Success  bool   `json:"success"`
	ImageURL string `json:"image"`
}

func GetImageURL(category string) (string, error) {

exported function GetImageURL should have comment or be unexported

	resp, err := http.Get(fmt.Sprintf("https://botimages.realpha.ru/?category=%v", category))
	if err != nil && resp.StatusCode == http.StatusOK {
		fmt.Printf("Getting image url error: %v", err)
		return "", errors.New("getting image url error")
	}

	var result ImageResponse

	err = json.NewDecoder(resp.Body).Decode(&result)
	if err != nil {
		fmt.Printf(err.Error())
		return "", err
	}

	if result.Success {
		return fmt.Sprintf("https://botimages.realpha.ru/%v", result.ImageURL), nil
	}
	return "", errors.New("wrong data")
}

func GetImage(category string) (*bytes.Buffer, error) {

exported function GetImage should have comment or be unexported

	resp, err := http.Get(fmt.Sprintf("https://botimages.realpha.ru/?category=%v", category))
	if err != nil {
		fmt.Printf("Getting image url error: %v", err)
		return nil, errors.New("getting image url error")
	}

	buf := new(bytes.Buffer)

	_, err = io.Copy(buf, resp.Body) //png.Encode(buf, resp.Body)
	if err != nil {
		fmt.Printf("Map image: %v", err.Error())
	}
	return buf, err
}