1 files changed, 18 insertions, 0 deletions
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)
+	}
 }