caddyhugo.go

@@ -1,8 +1,10 @@
 package caddyhugo
 
 import (
+	"fmt"
 	"html/template"
 	"net/http"
+	"os/exec"
 	"path/filepath"
 	"strings"
 	"sync"
@@ -48,7 +50,12 @@ type CaddyHugo struct {
 
 // Build rebuilds the cached state of the site. TODO: determine if this republishes
 func (ch *CaddyHugo) Build() error {
-	return buildSite(ch.HugoSites)
+	err := ch.HugoSites.Build(hugolib.BuildCfg{ResetState: true})
+	if err != nil {
+		return fmt.Errorf("error building hugo sites: %v", err)
+	}
+
+	return nil
 }
 
 // BasePath returns the directory that the CaddyHugo internal/author pages are under
@@ -65,6 +72,22 @@ func (ch *CaddyHugo) docFilename(orig string) string {
 	return filepath.Join(ch.Dir, docname(orig))
 }
 
+// Publish really renders new content into the public directory
+func (ch *CaddyHugo) Publish() error {
+	err := ch.persistAllEdits()
+	if err != nil {
+		return err
+	}
+	cmd := exec.Command("hugo")
+	cmd.Dir = ch.Dir
+	_, err = cmd.CombinedOutput()
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // TmplData collects data for template execution
 func (ch *CaddyHugo) TmplData(r *http.Request, docref *editSession) interface{} {
 	var doc *acedoc.Document

hugo.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/gohugoio/hugo/deps"
+	"github.com/gohugoio/hugo/hugofs"
 	"github.com/gohugoio/hugo/hugolib"
 	"github.com/spf13/afero"
 
@@ -17,54 +18,35 @@ const (
 	WebsocketFileTicker  = 1 * time.Second
 )
 
-// Publish really renders new content into the public directory
+type HugoInteractor interface {
-func (ch *CaddyHugo) Publish() error {
+	Render(srcdir, workdir string) HugoRenderer
-	err := ch.persistAllEdits()
-	if err != nil {
-		return err
-	}
-
-	err = buildSite(ch.HugoSites)
-	if err != nil {
-		return err
-	}
-
-	return nil
 }
 
-func buildSite(sites *hugolib.HugoSites) error {
+type HugoRenderer interface {
-	err := sites.Build(hugolib.BuildCfg{ResetState: true})
+	WriteContent(contents string) error
-	if err != nil {
+	Start() error
-		return fmt.Errorf("caddy-hugo: building site: %v", err)
+	Stop() error
-	}
-
-	return err
 }
 
-func (ch *CaddyHugo) configWithFs(fs afero.Fs) (*hugolib.HugoSites, *deps.DepsCfg, error) {
+func HugoInternalProcessConfig(ch *CaddyHugo, es *editSession, touchFn func()) (idleshut.Config, error) {
+
 	var err error
-	cfg := &deps.DepsCfg{}
+	hugoCfg := &deps.DepsCfg{Fs: hugofs.NewFrom(es.tmpfs, ch.HugoCfg.Cfg)}
-	cfgPath := path.Join(ch.Dir, "config.toml")
+	fmt.Println(ch.Dir)
-	cfg.Cfg, err = hugolib.LoadConfig(fs, "", cfgPath)
+	hugoCfg.Cfg, err = hugolib.LoadConfig(es.tmpfs, "", path.Join(ch.Dir, "config.toml"))
 	if err != nil {
-		return nil, cfg, fmt.Errorf("caddy-hugo: loading site configuration: %v", err)
+		return idleshut.Config{}, fmt.Errorf("caddy-hugo: loading site configuration: %v", err)
 	}
-	cfg.Cfg.Set("workingDir", ch.Dir)
 
-	sites, err := hugolib.NewHugoSites(*cfg)
+	hugoCfg.Cfg.Set("workingDir", ch.Dir)
+	hugoSites, err := hugolib.NewHugoSites(*hugoCfg)
 	if err != nil {
-		return nil, cfg, fmt.Errorf("caddy-hugo: initializing site: %v", err)
+		return idleshut.Config{}, fmt.Errorf("caddy-hugo: initializing site: %v", err)
 	}
 
-	err = buildSite(sites)
+	err = hugoSites.Build(hugolib.BuildCfg{ResetState: true})
-
-	return sites, cfg, err
-}
-
-func HugoInternalProcessConfig(ch *CaddyHugo, es *editSession, touchFn func()) (idleshut.Config, error) {
-	hugoSites, _, err := ch.configWithFs(es.tmpfs)
 	if err != nil {
-		return idleshut.Config{}, fmt.Errorf("caddy-hugo: loading site configuration: %v", err)
+		return idleshut.Config{}, fmt.Errorf("caddy-hugo: building site: %v", err)
 	}
 
 	return idleshut.Config{

setup.go

@@ -9,7 +9,9 @@ import (
 	"git.stephensearles.com/stephen/caddy-hugo2/comments"
 	"git.stephensearles.com/stephen/caddy-hugo2/media"
 
+	"github.com/gohugoio/hugo/deps"
 	"github.com/gohugoio/hugo/hugofs"
+	"github.com/gohugoio/hugo/hugolib"
 	"github.com/mholt/caddy"
 	"github.com/mholt/caddy/caddyhttp/httpserver"
 )
@@ -54,9 +56,17 @@ func (ch *CaddyHugo) Setup(dir string) error {
 	ch.docs = make(map[string]*editSession)
 	ch.confirmingToClient = make(map[uint64]struct{})
 
-	ch.HugoSites, ch.HugoCfg, err = ch.configWithFs(hugofs.Os)
+	ch.HugoCfg = &deps.DepsCfg{}
+
+	ch.HugoCfg.Cfg, err = hugolib.LoadConfig(hugofs.Os, dir, "")
+	if err != nil {
+		return fmt.Errorf("error loading hugo config: %v", err)
+	}
+
+	ch.HugoCfg.Cfg.Set("workingdir", dir)
+	ch.HugoSites, err = hugolib.NewHugoSites(*ch.HugoCfg)
 	if err != nil {
-		return fmt.Errorf("error setting up hugo : %v", err)
+		return fmt.Errorf("error intializing hugo: %v", err)
 	}
 
 	err = ch.Build()