From 4fff7508ba16db35ec48e635d3bf59e5ab39e67b Mon Sep 17 00:00:00 2001
From: Ever <ieverx@outlook.com>
Date: Sun, 25 Aug 2019 01:05:02 +0800
Subject: [PATCH] ADD: reload when themes change (#771)

* ADD: reload when themes change

close #713

* MOD: test can_detect_kind_of_changes
---
 src/cmd/serve.rs | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/src/cmd/serve.rs b/src/cmd/serve.rs
index c02d872c..080eaf2f 100644
--- a/src/cmd/serve.rs
+++ b/src/cmd/serve.rs
@@ -52,6 +52,7 @@ use rebuild;
 enum ChangeKind {
     Content,
     Templates,
+    Themes,
     StaticFiles,
     Sass,
     Config,
@@ -155,6 +156,7 @@ pub fn serve(
     // Setup watchers
     let mut watching_static = false;
     let mut watching_templates = false;
+    let mut watching_themes = false;
     let (tx, rx) = channel();
     let mut watcher = watcher(tx, Duration::from_secs(1)).unwrap();
     watcher
@@ -178,6 +180,13 @@ pub fn serve(
             .map_err(|e| ZolaError::chain("Can't watch the `templates` folder.", e))?;
     }
 
+    if Path::new("themes").exists() {
+        watching_themes = true;
+        watcher
+            .watch("themes/", RecursiveMode::Recursive)
+            .map_err(|e| ZolaError::chain("Can't watch the `themes` folder.", e))?;
+    }
+
     // Sass support is optional so don't make it an error to no have a sass folder
     let _ = watcher.watch("sass/", RecursiveMode::Recursive);
 
@@ -248,6 +257,9 @@ pub fn serve(
     if watching_templates {
         watchers.push("templates");
     }
+    if watching_themes {
+        watchers.push("themes");
+    }
     if site.config.compile_sass {
         watchers.push("sass");
     }
@@ -362,6 +374,19 @@ pub fn serve(
                             ChangeKind::Templates => reload_templates(&mut site, &path),
                             ChangeKind::StaticFiles => copy_static(&site, &path, &partial_path),
                             ChangeKind::Sass => reload_sass(&site, &path, &partial_path),
+                            ChangeKind::Themes => {
+                                console::info("-> Themes changed. The whole site will be reloaded.");
+                                site = create_new_site(
+                                    interface,
+                                    port,
+                                    output_dir,
+                                    base_url,
+                                    config_file,
+                                )
+                                .unwrap()
+                                .0;
+                                rebuild_done_handling(&broadcaster, Ok(()), "/x.js");
+                            },
                             ChangeKind::Config => {
                                 console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible.");
                                 site = create_new_site(
@@ -406,6 +431,19 @@ pub fn serve(
                             (ChangeKind::Templates, _) => reload_templates(&mut site, &path),
                             (ChangeKind::StaticFiles, p) => copy_static(&site, &path, &p),
                             (ChangeKind::Sass, p) => reload_sass(&site, &path, &p),
+                            (ChangeKind::Themes, _) => {
+                                console::info("-> Themes changed. The whole site will be reloaded.");
+                                site = create_new_site(
+                                    interface,
+                                    port,
+                                    output_dir,
+                                    base_url,
+                                    config_file,
+                                )
+                                .unwrap()
+                                .0;
+                                rebuild_done_handling(&broadcaster, Ok(()), "/x.js");
+                            },
                             (ChangeKind::Config, _) => {
                                 console::info("-> Config changed. The whole site will be reloaded. The browser needs to be refreshed to make the changes visible.");
                                 site = create_new_site(
@@ -471,6 +509,8 @@ fn detect_change_kind(pwd: &Path, path: &Path) -> (ChangeKind, PathBuf) {
 
     let change_kind = if partial_path.starts_with("/templates") {
         ChangeKind::Templates
+    } else if partial_path.starts_with("/themes") {
+        ChangeKind::Themes
     } else if partial_path.starts_with("/content") {
         ChangeKind::Content
     } else if partial_path.starts_with("/static") {
@@ -527,6 +567,11 @@ mod tests {
                 Path::new("/home/vincent/site"),
                 Path::new("/home/vincent/site/templates/hello.html"),
             ),
+            (
+                (ChangeKind::Themes, PathBuf::from("/themes/hello.html")),
+                Path::new("/home/vincent/site"),
+                Path::new("/home/vincent/site/themes/hello.html"),
+            ),
             (
                 (ChangeKind::StaticFiles, PathBuf::from("/static/site.css")),
                 Path::new("/home/vincent/site"),