impr: Compress PNG image with save

Author: canaria-computer

cmd/lite.go

@@ -14,6 +14,7 @@ import (
 	"image"
 	"image/color"
 	"image/draw"
+	"image/png"
 	"net"
 	"net/netip"
 	"net/url"
@@ -418,10 +419,20 @@ func saveScreenshot(data []byte, targetURL string, userAgent string, outputDir s
 		if err != nil {
 			return fmt.Errorf("failed to add header: %w", err)
 		}
+
 	}
 
 	outputPath := filepath.Join(outputDir, "screenshot.png")
-	if err := imaging.Save(finalImage, outputPath); err != nil {
+	file, err := os.Create(outputPath)
+	if err != nil {
+		return fmt.Errorf("failed to create file: %w", err)
+	}
+	defer file.Close()
+
+	encoder := &png.Encoder{
+		CompressionLevel: png.BestCompression,
+	}
+	if err := encoder.Encode(file, finalImage); err != nil {
 		return fmt.Errorf("failed to save image: %w", err)
 	}