mirror of
https://github.com/strongdm/comply
synced 2024-08-27 17:56:48 +00:00
114 lines
1.9 KiB
Go
114 lines
1.9 KiB
Go
|
package render
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"sync"
|
||
|
"time"
|
||
|
|
||
|
"github.com/gorilla/websocket"
|
||
|
"github.com/pkg/errors"
|
||
|
"github.com/yosssi/ace"
|
||
|
)
|
||
|
|
||
|
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)
|
||
|
}
|
||
|
// 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
|
||
|
}
|