1 files changed, 98 insertions, 0 deletions
diff --git a/lib/image_cache.go b/lib/image_cache.go
new file mode 100644
index 0000000..58cd09b
--- /dev/null
+++ b/lib/image_cache.go
@@ -0,0 +1,98 @@
+package lib
+
+import (
+	"fmt"
+	"image"
+	"image/jpeg"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"yato/config"
+)
+
+// ImageCache handles caching and retrieving images
+type ImageCache struct {
+	cacheDir string
+}
+
+// NewImageCache creates a new ImageCache
+func NewImageCache() *ImageCache {
+	cacheDir := filepath.Join(config.ConfigDir, config.AppName, "cache")
+	return &ImageCache{cacheDir: cacheDir}
+}
+
+// GetImage retrieves an image, either from cache or by downloading it
+func (c *ImageCache) GetImage(mediaType string, malID int, size string, url string) (image.Image, error) {
+	cachePath := c.getCachePath(mediaType, malID, size)
+
+	// Check if the image is already cached
+	if img, err := c.loadFromCache(cachePath); err == nil {
+		return img, nil
+	}
+
+	// If not cached, download and cache the image
+	return c.downloadAndCache(url, cachePath)
+}
+
+func (c *ImageCache) getCachePath(mediaType string, malID int, size string) string {
+	return filepath.Join(c.cacheDir, mediaType, fmt.Sprintf("%d", malID), size+".jpg")
+}
+
+func (c *ImageCache) loadFromCache(path string) (image.Image, error) {
+	file, err := os.Open(path)
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+
+	img, err := jpeg.Decode(file)
+	if err != nil {
+		return nil, err
+	}
+
+	return img, nil
+}
+
+func (c *ImageCache) downloadAndCache(url, cachePath string) (image.Image, error) {
+	resp, err := http.Get(url)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("failed to download image: %s", resp.Status)
+	}
+
+	// Ensure the cache directory exists
+	if err := os.MkdirAll(filepath.Dir(cachePath), 0755); err != nil {
+		return nil, err
+	}
+
+	// Create the cache file
+	cacheFile, err := os.Create(cachePath)
+	if err != nil {
+		return nil, err
+	}
+	defer cacheFile.Close()
+
+	// Download and write to cache file
+	_, err = io.Copy(cacheFile, resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	// Reset file pointer and decode the image
+	_, err = cacheFile.Seek(0, 0)
+	if err != nil {
+		return nil, err
+	}
+
+	img, err := jpeg.Decode(cacheFile)
+	if err != nil {
+		return nil, err
+	}
+
+	return img, nil
+}