You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
144 lines
2.8 KiB
144 lines
2.8 KiB
package caddyhugo
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"time"
|
|
|
|
"git.stephensearles.com/stephen/acedoc"
|
|
)
|
|
|
|
type editSession struct {
|
|
clients uint
|
|
docname string
|
|
filename string
|
|
doc *acedoc.Document
|
|
tmpdir string
|
|
}
|
|
|
|
func (ch *CaddyHugo) newEditSession(docName string) (*editSession, error) {
|
|
filename := ch.docFilename(docName)
|
|
|
|
fmt.Println("opening", filename)
|
|
contents, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
draftPrefix := fmt.Sprintf("draft-%s", base64.RawURLEncoding.EncodeToString([]byte(docName)))
|
|
tmpdir := path.Join(os.TempDir(), draftPrefix)
|
|
|
|
es := &editSession{
|
|
docname: docName,
|
|
filename: filename,
|
|
doc: acedoc.NewString(string(contents)),
|
|
tmpdir: tmpdir,
|
|
}
|
|
|
|
err = es.doc.LogToFile(path.Join(ch.Dir, "logs", docName))
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return nil, err
|
|
}
|
|
|
|
ch.docs[filename] = es
|
|
|
|
ch.renderDraft(es)
|
|
|
|
return es, nil
|
|
}
|
|
|
|
func (ch *CaddyHugo) renderDraft(es *editSession) error {
|
|
hugoCmd := exec.Command("hugo", "--watch", "-D", "-d", es.tmpdir)
|
|
hugoCmd.Dir = ch.Dir
|
|
err := hugoCmd.Start()
|
|
if err != nil {
|
|
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 = renderer.Stop()
|
|
if err != nil {
|
|
fmt.Println("error stopping hugo:", err)
|
|
}
|
|
}()
|
|
|
|
for {
|
|
<-ticker.C
|
|
ch.mtx.Lock()
|
|
|
|
// 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++
|
|
|
|
if idleTicks >= closeTicks {
|
|
idleTime := time.Duration(idleTicks) * WebsocketFileTicker
|
|
err := ch.Publish()
|
|
fmt.Printf("idle for %v, quitting\n", idleTime)
|
|
if err != nil {
|
|
fmt.Printf(", error publishing: %v\n", err)
|
|
}
|
|
|
|
es.doc.Close()
|
|
os.RemoveAll(es.tmpdir)
|
|
delete(ch.docs, es.filename)
|
|
ch.mtx.Unlock()
|
|
return
|
|
}
|
|
} else {
|
|
idleTicks = 0
|
|
}
|
|
ch.mtx.Unlock()
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (ch *CaddyHugo) hasEditSession(docName string) (*editSession, bool) {
|
|
dr, ok := ch.docs[ch.docFilename(docName)]
|
|
return dr, ok
|
|
}
|
|
|
|
func (ch *CaddyHugo) editSession(docName string) (*editSession, error) {
|
|
ch.mtx.Lock()
|
|
defer ch.mtx.Unlock()
|
|
|
|
var err error
|
|
|
|
dr, ok := ch.hasEditSession(docName)
|
|
if !ok {
|
|
dr, err = ch.newEditSession(docName)
|
|
}
|
|
|
|
return dr, err
|
|
}
|
|
|
|
func docNameFromEditRequest(r *http.Request) string {
|
|
return r.URL.Path[len("/hugo/edit/"):]
|
|
}
|
|
|