From 808ce89dfcc59198f5c3b9c8fb7f7b446a797029 Mon Sep 17 00:00:00 2001
From: Hazel Noack <hazel@webcontact.de>
Date: Tue, 15 Jul 2025 13:08:23 +0200
Subject: [PATCH] caching icons automatically

---
 internal/rendering/config.go        |  9 ++++
 internal/rendering/website_cards.go | 75 +++++++++++++++++++++++++++--
 main.go                             | 13 +++++
 tmp/build-errors.log                |  2 +-
 4 files changed, 95 insertions(+), 4 deletions(-)

diff --git a/internal/rendering/config.go b/internal/rendering/config.go
index 55bae00..dbd4e9f 100644
--- a/internal/rendering/config.go
+++ b/internal/rendering/config.go
@@ -136,3 +136,12 @@ func (rc *Config) LoadConfigFile(file string) error {
 
 	return toml.Unmarshal(content, rc)
 }
+
+func (c *Config) Init() error {
+	for i, w := range c.Template.Websites {
+		c.Template.Websites[i].Cache()
+		fmt.Println(w.ImageUrl)
+	}
+
+	return nil
+}
diff --git a/internal/rendering/website_cards.go b/internal/rendering/website_cards.go
index 4f496f4..96299c5 100644
--- a/internal/rendering/website_cards.go
+++ b/internal/rendering/website_cards.go
@@ -1,7 +1,76 @@
 package rendering
 
+import (
+	"crypto/sha1"
+	"encoding/hex"
+	"fmt"
+	"io"
+	"net/http"
+	"net/url"
+	"os"
+	"path/filepath"
+)
+
 type Website struct {
-	Url      string
-	Name     string
-	ImageUrl string
+	Url       string
+	Name      string
+	ImageUrl  string
+	IsFetched bool
+}
+
+const CacheUrl = "cache"
+
+func GetCacheDir() (string, error) {
+	baseDir, err := os.UserCacheDir()
+	if err != nil {
+		return "", err
+	}
+	cacheDir := filepath.Join(baseDir, "startpage")
+	err = os.MkdirAll(cacheDir, 0o755)
+	if err != nil {
+		return "", err
+	}
+	return cacheDir, nil
+}
+
+func hashUrl(url string) string {
+	h := sha1.New()
+	io.WriteString(h, url)
+	return hex.EncodeToString(h.Sum(nil))
+}
+
+func (w *Website) Cache() error {
+	if w.IsFetched {
+		return nil
+	}
+
+	cacheDir, err := GetCacheDir()
+	if err != nil {
+		return err
+	}
+
+	filename := hashUrl(w.ImageUrl)
+	fmt.Println(w.ImageUrl + " => " + filename)
+	targetPath := filepath.Join(cacheDir, filename)
+	resp, err := http.Get(w.ImageUrl)
+	if err != nil {
+		return err
+	}
+	defer resp.Body.Close()
+
+	file, err := os.Create(targetPath)
+	if err != nil {
+		return err
+	}
+	defer file.Close()
+
+	_, err = io.Copy(file, resp.Body)
+
+	if err != nil {
+		return err
+	}
+
+	w.ImageUrl, _ = url.JoinPath(CacheUrl, filename)
+	w.IsFetched = true
+	return nil
 }
diff --git a/main.go b/main.go
index db33ecf..21638c7 100644
--- a/main.go
+++ b/main.go
@@ -74,6 +74,11 @@ func main() {
 		fmt.Println(err)
 	}
 
+	err = CurrentConfig.Init()
+	if err != nil {
+		fmt.Println(err)
+	}
+
 	err = FetchDiyHrt()
 	if err != nil {
 		fmt.Println(err)
@@ -81,6 +86,14 @@ func main() {
 
 	e := echo.New()
 
+	// statically serve the file
+	cacheDir, err := rendering.GetCacheDir()
+	if err == nil {
+		e.Static("/cache", cacheDir)
+	} else {
+		fmt.Println(err)
+	}
+
 	// https://echo.labstack.com/docs/cookbook/embed-resources
 	staticHandler := http.FileServer(getFileSystem())
 	e.GET("/assets/*", echo.WrapHandler(http.StripPrefix("/", staticHandler)))
diff --git a/tmp/build-errors.log b/tmp/build-errors.log
index 4660a1c..f50898d 100644
--- a/tmp/build-errors.log
+++ b/tmp/build-errors.log
@@ -1 +1 @@
-exit status 1exit status 1
\ No newline at end of file
+exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1exit status 1
\ No newline at end of file