feat: add extension to file if missing

internal/http.go

@@ -6,6 +6,7 @@ import (
 	"math/rand"
 	"net/http"
 	"os"
+	"path/filepath"
 	"strings"
 	"time"
 
@@ -74,6 +75,18 @@ func validateContentType(contentType string, imageTypes []string) error {
 	return fmt.Errorf("image type '%s' is not allowed", imageType)
 }
 
+func addExtensionIfMissing(filePath, contentType string) string {
+	extension := filepath.Ext(filePath)
+
+	if extension != "" {
+		return filePath
+	}
+
+	extension = "." + strings.Split(contentType, "/")[1]
+
+	return filePath + extension
+}
+
 func DownloadImageFromUrl(url, filePath string, parameters Parameters) error {
 	response, err := request(url, parameters.UserAgent)
 	if err != nil {
@@ -87,7 +100,9 @@ func DownloadImageFromUrl(url, filePath string, parameters Parameters) error {
 		return err
 	}
 
-	file, err := os.Create(filePath)
+	correctFilePath := addExtensionIfMissing(filePath, contentType)
+
+	file, err := os.Create(correctFilePath)
 	if err != nil {
 		return err
 	}