package server
import (
"bytes"
"fmt"
"html"
"io"
"io/ioutil"
"mime"
"net/http"
"os"
"strings"
"sync"
"github.com/jaschaephraim/lrserver"
"github.com/osteele/gojekyll/site"
"github.com/osteele/liquid"
"github.com/pkg/browser"
)
// Server serves the site on HTTP.
type Server struct {
sync.Mutex
Site *site.Site
lr *lrserver.Server
}
// Run runs the server.
func (s *Server) Run(open bool, logger func(label, value string)) error {
cfg := s.Site.Config()
s.Site.SetAbsoluteURL("")
address := fmt.Sprintf("%s:%d", cfg.Host, cfg.Port)
logger("Server address:", "http://"+address+"/")
if cfg.Watch {
if err := s.StartLiveReloader(); err != nil {
return err
}
if err := s.watchAndReload(); err != nil {
return err
}
}
http.HandleFunc("/", s.handler)
c := make(chan error)
go func() {
c <- http.ListenAndServe(address, nil)
}()
logger("Server running...", "press ctrl-c to stop.")
if open {
if err := browser.OpenURL("http://" + address); err != nil {
fmt.Println("Error opening page:", err)
}
}
return <-c
}
func (s *Server) handler(rw http.ResponseWriter, r *http.Request) {
s.Lock()
defer s.Unlock()
var (
site = s.Site
urlpath = r.URL.Path
p, found = site.URLPage(urlpath)
)
if !found {
rw.WriteHeader(http.StatusNotFound)
p, found = site.Routes["/404.html"]
}
if !found {
fmt.Fprintf(rw, "404 page not found: %s\n", urlpath) // nolint: gas
return
}
mimeType := mime.TypeByExtension(p.OutputExt())
if mimeType != "" {
rw.Header().Set("Content-Type", mimeType)
}
var w io.Writer = rw
if strings.HasPrefix(mimeType, "text/html;") {
w = NewLiveReloadInjector(w)
}
err := site.WriteDocument(w, p)
if err != nil {
fmt.Fprintf(os.Stderr, "Error rendering %s: %s\n", urlpath, err)
eng := liquid.NewEngine()
excerpt, path := fileErrorContext(err)
out, e := eng.ParseAndRenderString(renderErrorTemplate, liquid.Bindings{
"error": err,
"excerpt": excerpt,
"path": path,
"watch": site.Config().Watch,
})
if e != nil {
panic(e)
}
if _, err := io.WriteString(w, out); err != nil {
fmt.Fprintf(os.Stderr, "Error writing HTTP response: %s", err)
}
}
}
func fileErrorContext(e error) (s, path string) {
cause, ok := e.(liquid.SourceError)
if !ok {
return
}
path, n := cause.Path(), cause.LineNumber()
b, err := ioutil.ReadFile(path)
if err != nil {
return
}
lines := strings.Split(strings.TrimRight(string(b), "\n"), "\n")
l0, l1 := n-4, n+4
w := new(bytes.Buffer)
for i := l0; i < l1; i++ {
if i < 0 || len(lines) <= i {
continue
}
var class string
if i+1 == n {
class = "error"
}
fmt.Fprintf(w, `%4d%s
`, class, i+1, html.EscapeString(lines[i]))
}
return w.String(), path
}
// CSS theme adapted from github.com/facebookincubator/create-react-app
const renderErrorTemplate = `
Failed to render.
{{ error }}:
{{ excerpt }}
{% if watch and path != "" %}
{% endif %}
`