From e838cf94aa507f6a3ea93c65451b902cc1085b34 Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <eddy@0xf7.com>
Date: Mon, 3 Feb 2020 22:26:00 +0100
Subject: Added cleanup, basic specs are implemented

---
 main.go | 18 ++++++++++++++++++
 page.go |  1 -
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 870e0f9..5b444a8 100644
--- a/main.go
+++ b/main.go
@@ -19,6 +19,7 @@ func main() {
 
 	// create temporary target directory
 	tmp := createTempDir("godocs")
+	defer os.RemoveAll(tmp)
 
 	// gather all sites in docsRoot
 	sites := getSites(cfg)
@@ -42,4 +43,21 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
+
+	// if we are here, generation succeeded, so we move the generated content to the target directory
+	err = os.RemoveAll(cfg.TargetDir)
+	if err != nil {
+		panic(err)
+	}
+
+	err = os.Rename(tmp, cfg.TargetDir)
+	if err != nil {
+		panic(err)
+	}
+
+	// fix because for some reason the generated directory ends up with the wrong permissions
+	err = os.Chmod(cfg.TargetDir, 0755)
+	if err != nil {
+		panic(err)
+	}
 }
diff --git a/page.go b/page.go
index a88f437..be52e3c 100644
--- a/page.go
+++ b/page.go
@@ -55,7 +55,6 @@ func createPageMap(site *siteData, entries []Renderer) {
 		if err != nil {
 			panic(err)
 		}
-
 	}
 
 	// create landing page
-- 
cgit v1.2.3