mirror of
https://github.com/danog/gojekyll.git
synced 2024-11-30 08:39:01 +01:00
149 lines
3.7 KiB
Go
149 lines
3.7 KiB
Go
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, `<span class="line %s"><span class="gutter"></span><span class="lineno">%4d</span>%s<br /></span>`, class, i+1, html.EscapeString(lines[i]))
|
|
}
|
|
return w.String(), path
|
|
}
|
|
|
|
// CSS theme adapted from github.com/facebookincubator/create-react-app
|
|
const renderErrorTemplate = `<html><head>
|
|
<style type="text/css">
|
|
body { background-color: black; color: rgb(232, 232, 232); font-family: Menlo, Consolas, monospace; padding: 2rem; line-height: 1.2; }
|
|
h1 { color: #E36049 }
|
|
div { margin: 20px 0; }
|
|
code { font-size: xx-large; }
|
|
.line.error .gutter::before { content: "⚠️"; width: 0; float:left; }
|
|
.line.error, .line.error .lineno { color: red; }
|
|
.lineno { color: #6D7891; border-right: 1px solid #6D7891; padding-right: 10px; margin: 0 10px 0 55px; display: inline-block; text-align: right; }
|
|
footer { border-top: 1px solid #6D7891; margin-top: 5ex; padding-top: 5px; }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<h1>Failed to render.</h1>
|
|
<div>{{ error }}:</div>
|
|
<code>{{ excerpt }}</code>
|
|
{% if watch and path != "" %}
|
|
<footer>Edit and save “{{ path }}” to reload this page.</footer>
|
|
{% endif %}
|
|
</body>
|
|
</html>`
|