111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"github.com/labstack/echo/v4"
|
|
client "github.com/siteworxpro/top-wallpaper/reddit"
|
|
"github.com/siteworxpro/top-wallpaper/resize"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
func Get(c echo.Context) error {
|
|
cc := c.(*CustomContext)
|
|
|
|
var size int
|
|
var err error
|
|
|
|
if sizeS, ok := os.LookupEnv("MAX_SIZE"); ok {
|
|
size, err = strconv.Atoi(sizeS)
|
|
if err != nil {
|
|
size = 1200
|
|
}
|
|
} else {
|
|
size = 1200
|
|
}
|
|
|
|
if size < 100 {
|
|
size = 1200
|
|
}
|
|
|
|
var latestImageVal string
|
|
if cc.redis != nil {
|
|
latestImage := cc.redis.Get(context.TODO(), "latestImage")
|
|
latestImageVal, err = latestImage.Result()
|
|
}
|
|
|
|
if err != nil || latestImageVal == "" {
|
|
c.Logger().Info("Fetching latest image")
|
|
latestImageVal, err = client.GetLatestImage(func(url string) (*http.Response, error) {
|
|
return http.Get(url)
|
|
})
|
|
if err != nil {
|
|
return c.String(http.StatusInternalServerError, "Error fetching latest image")
|
|
}
|
|
|
|
if cc.redis != nil {
|
|
cmd := cc.redis.Set(context.TODO(), "latestImage", latestImageVal, 600*time.Second)
|
|
if cmd.Err() != nil {
|
|
c.Logger().Warn("could not cache image")
|
|
}
|
|
}
|
|
} else {
|
|
c.Logger().Info("Image name fetched from cache")
|
|
}
|
|
|
|
var imageData string
|
|
if cc.redis != nil {
|
|
latestImageBin := cc.redis.Get(context.TODO(), "latestImage:bin:"+latestImageVal)
|
|
imageData = latestImageBin.Val()
|
|
}
|
|
|
|
if imageData != "" {
|
|
c.Logger().Info("Image data fetched from cache")
|
|
|
|
return c.Blob(http.StatusOK, "image/jpeg", []byte(imageData))
|
|
}
|
|
|
|
response, err := http.Get(latestImageVal)
|
|
if err != nil {
|
|
return c.String(http.StatusInternalServerError, "Error fetching image")
|
|
}
|
|
|
|
if response.StatusCode != http.StatusOK {
|
|
return c.String(http.StatusInternalServerError, "Error fetching image")
|
|
}
|
|
|
|
defer func(Body io.ReadCloser) {
|
|
_ = Body.Close()
|
|
}(response.Body)
|
|
|
|
imageDataBytes, err := io.ReadAll(response.Body)
|
|
if err != nil {
|
|
return c.String(http.StatusInternalServerError, "Error fetching image")
|
|
}
|
|
|
|
imageData = string(imageDataBytes)
|
|
resized, err := resize.Shrink(imageData, uint(size), 70)
|
|
|
|
if err != nil {
|
|
return c.String(http.StatusInternalServerError, "Error resizing image")
|
|
}
|
|
|
|
go func(data string) {
|
|
if cc.redis == nil {
|
|
return
|
|
}
|
|
|
|
_, err = cc.redis.Set(context.TODO(), "latestImage:bin:"+latestImageVal, data, 600*time.Second).Result()
|
|
if err != nil {
|
|
c.Logger().Warn("could not cache image")
|
|
}
|
|
}(resized)
|
|
|
|
c.Response().Header().Set("Cache-Control", "public, max-age=600")
|
|
|
|
return c.Blob(http.StatusOK, "image/jpeg", []byte(resized))
|
|
}
|