danog/gojekyll
1
0
Fork 0
mirror of https://github.com/danog/gojekyll.git synced 2024-11-30 04:09:00 +01:00
Code Issues Projects Releases Wiki Activity

De-clutter livereloader output

Browse Source
This commit is contained in:
Oliver Steele 2017-07-26 08:01:09 -04:00
parent 5917e94fe7
commit dd2980d2c9
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
server/livereload.go
View File

@ -2,6 +2,8 @@ package server
import (
"io"
"os"
"regexp"
"github.com/jaschaephraim/lrserver"
)
@ -9,11 +11,13 @@ import (
// liveReloadScriptTag is inserted into the HTML page.
var liveReloadScriptTag = []byte(`<script src="http://localhost:35729/livereload.js"></script>`)
// StartLiveReloader starts the Live Reload server as a go routine, and returns immediately
func (s *Server) StartLiveReloader() error {
s.lr = lrserver.New(lrserver.DefaultName, lrserver.DefaultPort)
s.lr.SetStatusLog(nil)
go s.lr.ListenAndServe() // nolint: errcheck
// startLiveReloader starts the Live Reload server as a go routine, and returns immediately
func (s *Server) startLiveReloader() error {
lr := lrserver.New(lrserver.DefaultName, lrserver.DefaultPort)
s.lr = lr
lr.SetStatusLog(nil)
lr.ErrorLog().SetOutput(outputFilter{os.Stdout})
go lr.ListenAndServe() // nolint: errcheck
return nil
}
@ -22,3 +26,16 @@ func (s *Server) StartLiveReloader() error {
func NewLiveReloadInjector(w io.Writer) io.Writer {
return TagInjector{w, liveReloadScriptTag}
}
// Remove the lines that match the exclusion pattern.
// TODO submit an upstream PR to make this unnecessary
type outputFilter struct{ w *os.File }
var excludeRE = regexp.MustCompile(`websocket: close 1006 \(abnormal closure\): unexpected EOF`)
func (w outputFilter) Write(b []byte) (int, error) {
if excludeRE.Match(b) {
return len(b), nil
}
return w.w.Write(b)
}

2
server/server.go
View File

@ -32,7 +32,7 @@ func (s *Server) Run(open bool, logger func(label, value string)) error {
address := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
logger("Server address:", "http://"+address+"/")
if cfg.Watch {
if err := s.StartLiveReloader(); err != nil {
if err := s.startLiveReloader(); err != nil {
return err
}
if err := s.watchReload(); err != nil {