From 6ff23f639ebc9143e11b6a6bcea29657292ebabd Mon Sep 17 00:00:00 2001
From: wjqserver <114663932+WJQSERVER@users.noreply.github.com>
Date: Wed, 21 May 2025 11:54:43 +0800
Subject: [PATCH] add hub theme & add more check for wcache close

---
 main.go | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 2236e2a..6b84afe 100644
--- a/main.go
+++ b/main.go
@@ -213,6 +213,8 @@ func loadEmbeddedPages(cfg *config.Config) (fs.FS, fs.FS, error) {
 		pages, err = fs.Sub(pagesFS, "pages/classic")
 	case "mino":
 		pages, err = fs.Sub(pagesFS, "pages/mino")
+	case "hub":
+		pages, err = fs.Sub(pagesFS, "pages/hub")
 	default:
 		pages, err = fs.Sub(pagesFS, "pages/design") // 默认主题
 		logWarning("Invalid Pages Theme: %s, using default theme 'design'", cfg.Pages.Theme)
@@ -294,7 +296,7 @@ func setInternalRoute(cfg *config.Config, r *server.Hertz) error {
 		staticServer.ServeHTTP(adaptor.GetCompatResponseWriter(&c.Response), req)
 	})
 	r.GET("/favicon.ico", func(ctx context.Context, c *app.RequestContext) {
-		staticServer := http.FileServer(http.FS(pages))
+		staticServer := http.FileServer(http.FS(assets))
 		req, err := adaptor.GetCompatRequest(&c.Request)
 		if err != nil {
 			logError("%s", err)
@@ -510,8 +512,9 @@ func main() {
 			http.ListenAndServe("localhost:6060", nil)
 		}()
 	}
-
-	defer wcache.StopCleanup()
+	if wcache != nil {
+		defer wcache.StopCleanup()
+	}
 	defer logger.Close()
 	defer func() {
 		if hertZfile != nil {