Export MainSite var

Fixes hacdias/caddy-hugo#46

diff --git a/commands/benchmark.go b/commands/benchmark.go
index c90f64516421278b6051fda24cac96f4f9cbc0d2..d10289be4ec53cf12496351724edb03912ff1b77 100644 (file)
--- a/commands/benchmark.go
+++ b/commands/benchmark.go
@@ -14,9 +14,10 @@
 package commands
 
 import (
-       "github.com/spf13/cobra"
        "os"
        "runtime/pprof"
+
+       "github.com/spf13/cobra"
 )
 
 var (
@@ -56,7 +57,7 @@ func benchmark(cmd *cobra.Command, args []string) error {
                        return err
                }
                for i := 0; i < benchmarkTimes; i++ {
-                       mainSite = nil
+                       MainSite = nil
                        _ = buildSite()
                }
                pprof.WriteHeapProfile(f)
@@ -75,7 +76,7 @@ func benchmark(cmd *cobra.Command, args []string) error {
                pprof.StartCPUProfile(f)
                defer pprof.StopCPUProfile()
                for i := 0; i < benchmarkTimes; i++ {
-                       mainSite = nil
+                       MainSite = nil
                        _ = buildSite()
                }
        }

diff --git a/commands/hugo.go b/commands/hugo.go
index ce96e21c92fbbb75a3e52f0afb60f398771f7b76..f363cc7a0127eb0b2dec2b194202aea633ece323 100644 (file)
--- a/commands/hugo.go
+++ b/commands/hugo.go
@@ -46,7 +46,7 @@ import (
        "gopkg.in/fsnotify.v1"
 )
 
-var mainSite *hugolib.Site
+var MainSite *hugolib.Site
 
 // userError is an error used to signal different error situations in command handling.
 type commandError struct {
@@ -648,18 +648,19 @@ func getDirList() []string {
 }
 
 func buildSite(watching ...bool) (err error) {
+       fmt.Println("Started building site")
        startTime := time.Now()
-       if mainSite == nil {
-               mainSite = new(hugolib.Site)
+       if MainSite == nil {
+               MainSite = new(hugolib.Site)
        }
        if len(watching) > 0 && watching[0] {
-               mainSite.RunMode.Watching = true
+               MainSite.RunMode.Watching = true
        }
-       err = mainSite.Build()
+       err = MainSite.Build()
        if err != nil {
                return err
        }
-       mainSite.Stats()
+       MainSite.Stats()
        jww.FEEDBACK.Printf("in %v ms\n", int(1000*time.Since(startTime).Seconds()))
 
        return nil
@@ -667,11 +668,11 @@ func buildSite(watching ...bool) (err error) {
 
 func rebuildSite(events []fsnotify.Event) error {
        startTime := time.Now()
-       err := mainSite.ReBuild(events)
+       err := MainSite.ReBuild(events)
        if err != nil {
                return err
        }
-       mainSite.Stats()
+       MainSite.Stats()
        jww.FEEDBACK.Printf("in %v ms\n", int(1000*time.Since(startTime).Seconds()))
 
        return nil