From 8d0f11e681bd8c1f20533bab443efcad01af2336 Mon Sep 17 00:00:00 2001
From: Ron Rise <ron@siteworxpro.com>
Date: Thu, 24 Apr 2025 20:52:11 -0400
Subject: [PATCH] [no message]

---
 .gitlab-ci.yml  |  1 -
 http/handler.go | 21 +++++++++++++++++++--
 2 files changed, 19 insertions(+), 3 deletions(-)

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")