From 566f4e69195ceaf5d06b1097bb1a276363eeb937 Mon Sep 17 00:00:00 2001
From: Vincent Prouillet <vincent@wearewizards.io>
Date: Mon, 1 May 2017 16:31:23 +0900
Subject: [PATCH] Fix rebuild content logic to include sections

---
 src/site.rs | 34 +++++++++++++++++++++++++---------
 1 file changed, 25 insertions(+), 9 deletions(-)

diff --git a/src/site.rs b/src/site.rs
index 23e6552b..70fe7ad4 100644
--- a/src/site.rs
+++ b/src/site.rs
@@ -284,24 +284,40 @@ impl Site {
     }
 
     pub fn rebuild_after_content_change(&mut self, path: &Path) -> Result<()> {
+        let is_section = path.ends_with("_index.md");
+        let is_index_section = if is_section {
+            path.parent().unwrap() == self.base_path.join("content")
+        } else {
+            false
+        };
+
         if path.exists() {
             // file exists, either a new one or updating content
-            if self.pages.contains_key(path) {
-                if path.ends_with("_index.md") {
-                    self.add_section(path)?;
+            if is_section {
+                if is_index_section {
+                    self.index = Some(Section::from_file(path, &self.config)?);
                 } else {
-                    // probably just an update so just re-parse that page
-                    self.add_page_and_render(path)?;
+                    self.add_section(path)?;
                 }
             } else {
-                // new file?
+                // probably just an update so just re-parse that page
                 self.add_page_and_render(path)?;
             }
         } else {
             // File doesn't exist -> a deletion so we remove it from everything
-            let relative_path = self.pages[path].relative_path.clone();
-            self.pages.remove(path);
-            self.permalinks.remove(&relative_path);
+            if is_section {
+                if !is_index_section {
+                    let relative_path = self.sections[path].relative_path.clone();
+                    self.sections.remove(path);
+                    self.permalinks.remove(&relative_path);
+                } else {
+                    self.index = None;
+                }
+            } else {
+                let relative_path = self.pages[path].relative_path.clone();
+                self.pages.remove(path);
+                self.permalinks.remove(&relative_path);
+            }
         }
         self.populate_sections();
         self.populate_tags_and_categories();