From b266262c7a22b8f113bae3ddcdd064b45e40c27a Mon Sep 17 00:00:00 2001
From: Henrique Dias <hacdias@gmail.com>
Date: Thu, 11 Feb 2016 20:19:22 +0000
Subject: [PATCH] simplify a little bit

---
 hugo.go        | 23 +++++++++++++++++++++++
 utils/utils.go | 44 ++------------------------------------------
 2 files changed, 25 insertions(+), 42 deletions(-)

diff --git a/hugo.go b/hugo.go
index 77c1cd4e..75c35882 100644
--- a/hugo.go
+++ b/hugo.go
@@ -18,11 +18,34 @@ import (
 	"github.com/hacdias/caddy-hugo/utils"
 	"github.com/mholt/caddy/caddy/setup"
 	"github.com/mholt/caddy/middleware"
+	"github.com/spf13/cobra"
+	"github.com/spf13/hugo/commands"
 )
 
 // Setup configures the middleware
 func Setup(c *setup.Controller) (middleware.Middleware, error) {
 	config, _ := config.ParseHugo(c)
+
+	create := false
+
+	if _, err := os.Stat(config.Path + "config.yaml"); os.IsNotExist(err) {
+		create = true
+	}
+
+	if _, err := os.Stat(config.Path + "config.json"); os.IsNotExist(err) {
+		create = true
+	}
+
+	if _, err := os.Stat(config.Path + "config.toml"); os.IsNotExist(err) {
+		create = true
+	}
+
+	if create {
+		cmd := &cobra.Command{}
+		cmd.Flags().Bool("force", true, "")
+		commands.NewSite(cmd, []string{config.Path})
+	}
+
 	utils.Run(config)
 
 	return func(next middleware.Handler) middleware.Handler {
diff --git a/utils/utils.go b/utils/utils.go
index d122b08a..c96aeddf 100644
--- a/utils/utils.go
+++ b/utils/utils.go
@@ -13,7 +13,6 @@ import (
 
 	"github.com/hacdias/caddy-hugo/assets"
 	"github.com/hacdias/caddy-hugo/config"
-	"github.com/spf13/cobra"
 	"github.com/spf13/hugo/commands"
 	"github.com/spf13/viper"
 )
@@ -167,51 +166,12 @@ func ParseComponents(r *http.Request) []string {
 
 // Run is used to run the static website generator
 func Run(c *config.Config) {
-	create := false
-
-	if _, err := os.Stat(c.Path + "config.yaml"); os.IsNotExist(err) {
-		create = true
-	}
-
-	if _, err := os.Stat(c.Path + "config.json"); os.IsNotExist(err) {
-		create = true
-	}
-
-	if _, err := os.Stat(c.Path + "config.toml"); os.IsNotExist(err) {
-		create = true
-	}
-
-	if create {
-		cmd := &cobra.Command{}
-		cmd.Flags().Bool("force", true, "")
-		commands.NewSite(cmd, []string{c.Path})
-	}
-
-	cwd, err := os.Getwd()
-
-	if err != nil {
-		log.Print("Can't get working directory.")
-	}
-
-	err = os.Chdir(c.Path)
-
-	if err != nil {
-		log.Print("Can't get working directory.")
-	}
-
 	viper.Reset()
+	c.Args = append([]string{"--source", c.Path, "--destination", "public"}, c.Args...)
 	commands.HugoCmd.ParseFlags(c.Args)
-	err = commands.HugoCmd.RunE(nil, nil)
-
-	if err != nil {
+	if err := commands.HugoCmd.RunE(nil, nil); err != nil {
 		log.Print(err)
 	}
-
-	err = os.Chdir(cwd)
-
-	if err != nil {
-		log.Print("Can't get working directory.")
-	}
 }
 
 var splitCapitalizeExceptions = map[string]string{