From 82a12396eecf2c08352742b7442cbd3850af9bbe Mon Sep 17 00:00:00 2001
From: Zuoling Rong <rzlrong@outlook.com>
Date: Mon, 3 Feb 2025 00:28:02 +0800
Subject: [PATCH] Update generator.js

---
 generator.js | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/generator.js b/generator.js
index f2278e7..950447e 100644
--- a/generator.js
+++ b/generator.js
@@ -202,16 +202,37 @@ ${generateSearchResultsPage()}
 </html>`;
 }
 
+// 复制静态文件
+function copyStaticFiles(config) {
+    // 如果配置了favicon，确保文件存在并复制
+    if (config.site.favicon) {
+        try {
+            if (fs.existsSync(config.site.favicon)) {
+                fs.copyFileSync(config.site.favicon, path.basename(config.site.favicon));
+                console.log(`Copied favicon: ${config.site.favicon}`);
+            } else {
+                console.warn(`Warning: Favicon file not found: ${config.site.favicon}`);
+            }
+        } catch (e) {
+            console.error('Error copying favicon:', e);
+        }
+    }
+}
+
 // 主函数
 function main() {
     const config = loadConfig();
     const html = generateHTML(config);
     
     try {
+        // 生成HTML
         fs.writeFileSync('index.html', html);
         console.log('Successfully generated index.html');
+        
+        // 复制静态文件
+        copyStaticFiles(config);
     } catch (e) {
-        console.error('Error writing index.html:', e);
+        console.error('Error in main function:', e);
         process.exit(1);
     }
 }