comply/internal/render/site.go

package render

import (
	"fmt"
	"net/http"
	"os"
	"path/filepath"
	"sync"
	"time"

	"github.com/gorilla/websocket"
	"github.com/pkg/errors"
	"github.com/yosssi/ace"
)

const BindAddress = "0.0.0.0"

var ServePort int

var upgrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

var aceOpts = &ace.Options{
	DynamicReload: true,
	Indent:        "  ",
}

var watchChMu sync.Mutex
var watchCh chan struct{}

func subscribe() chan struct{} {
	watchChMu.Lock()
	defer watchChMu.Unlock()
	if watchCh == nil {
		watchCh = make(chan struct{})
	}
	return watchCh
}

func broadcast() {
	watchChMu.Lock()
	defer watchChMu.Unlock()
	close(watchCh)
	watchCh = nil
}

var lastModifiedMu sync.Mutex
var lastModified = make(map[string]time.Time)

func recordModified(path string, t time.Time) {
	lastModifiedMu.Lock()
	defer lastModifiedMu.Unlock()

	previous, ok := lastModified[path]
	if !ok || t.After(previous) {
		lastModified[path] = t
	}
}

func isNewer(path string, t time.Time) bool {
	lastModifiedMu.Lock()
	defer lastModifiedMu.Unlock()

	previous, ok := lastModified[path]
	if !ok {
		return true
	}

	// is tested after previous? Then isNewer is true.
	return t.After(previous)
}

// Build generates all PDF and HTML output to the target directory with optional live reload.
func Build(output string, live bool) error {
	err := os.RemoveAll(output)
	if err != nil {
		errors.Wrap(err, "unable to remove files from output directory")
	}

	err = os.MkdirAll(output, os.FileMode(0755))
	if err != nil {
		errors.Wrap(err, "unable to create output directory")
	}

	var wg sync.WaitGroup
	errCh := make(chan error, 0)
	wgCh := make(chan struct{})

	if live {
		watch(errCh)

		go func() {
			http.Handle("/", http.FileServer(http.Dir(filepath.Join(".", "output"))))
			err := http.ListenAndServe(fmt.Sprintf("%s:%d", BindAddress, ServePort), nil)
			if err != nil {
				panic(err)
			}
		}()

		fmt.Printf("Serving content of output/ at http://%s:%d (ctrl-c to quit)\n", BindAddress, ServePort)
	}
	// PDF
	wg.Add(1)
	go pdf(output, live, errCh, &wg)

	// HTML
	wg.Add(1)
	go html(output, live, errCh, &wg)

	// WG monitor
	go func() {
		wg.Wait()
		close(wgCh)
	}()

	select {
	case <-wgCh:
		// success
	case err := <-errCh:
		return errors.Wrap(err, "error during build")
	}

	return nil
}