diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index d34ceea..7fa3932 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -2,7 +2,6 @@ stages: - Testing - build - include: - component: $CI_SERVER_FQDN/shared/blueprints/golang-tests@v1.2.0 rules: diff --git a/http/handler.go b/http/handler.go index 0545a65..143eded 100644 --- a/http/handler.go +++ b/http/handler.go @@ -7,14 +7,31 @@ import ( "github.com/siteworxpro/top-wallpaper/resize" "io" "net/http" + "os" + "strconv" "time" ) func Get(c echo.Context) error { cc := c.(*CustomContext) - var latestImageVal string + 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() @@ -70,7 +87,7 @@ func Get(c echo.Context) error { } imageData = string(imageDataBytes) - resized, err := resize.Shrink(imageData, 1200, 70) + resized, err := resize.Shrink(imageData, uint(size), 70) if err != nil { return c.String(http.StatusInternalServerError, "Error resizing image")