cleaning up code that runs hugo

client.go

@@ -61,18 +61,21 @@ func (ch *CaddyHugo) renderDraft(es *editSession) error {
 		return fmt.Errorf("error starting hugo: %v", err)
 	}
 
+	renderer := HugoCmdRenderer{
+		TmpDir:   es.tmpdir,
+		SrcDir:   ch.Dir,
+		Filename: es.filename,
+	}
+
 	go func() {
 		ticker := time.NewTicker(WebsocketFileTicker)
 		idleTicks := 0
+		closeTicks := int(IdleWebsocketTimeout / WebsocketFileTicker)
 
 		defer func() {
-			err := hugoCmd.Process.Signal(os.Interrupt)
+			err = renderer.Stop()
 			if err != nil {
-				fmt.Println("error signaling to hugo:", err)
-			}
-			err = hugoCmd.Wait()
-			if err != nil {
-				fmt.Println("error waiting for hugo:", err)
+				fmt.Println("error stopping hugo:", err)
 			}
 		}()
 
@@ -80,15 +83,21 @@ func (ch *CaddyHugo) renderDraft(es *editSession) error {
 			<-ticker.C
 			ch.mtx.Lock()
 
-			err := ioutil.WriteFile(es.filename, []byte(es.doc.Contents()), 0644)
-			if err != nil {
-				fmt.Println("error saving document contents:", err)
+			// render if there are connected clients OR if we haven't
+			// been idle for very long.
+			if es.clients > 0 || idleTicks < 2 {
+				err := renderer.Render(es.doc.Contents())
+				if err != nil {
+					fmt.Println("error saving document contents:", err)
+				}
 			}
 
+			// count up idle time and possibly close down
 			if es.clients == 0 {
 				idleTicks++
-				idleTime := time.Duration(idleTicks) * WebsocketFileTicker
-				if idleTime >= IdleWebsocketTimeout {
+
+				if idleTicks >= closeTicks {
+					idleTime := time.Duration(idleTicks) * WebsocketFileTicker
 					err := ch.Publish()
 					fmt.Printf("idle for %v, quitting\n", idleTime)
 					if err != nil {
@@ -100,8 +109,6 @@ func (ch *CaddyHugo) renderDraft(es *editSession) error {
 					delete(ch.docs, es.filename)
 					ch.mtx.Unlock()
 					return
-				} else {
-					fmt.Println("idle for", idleTime)
 				}
 			} else {
 				idleTicks = 0