2015-05-06 10:37:29 +08:00
|
|
|
package markdown
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"bufio"
|
|
|
|
|
"bytes"
|
|
|
|
|
"fmt"
|
|
|
|
|
"io/ioutil"
|
2015-05-07 07:19:02 +08:00
|
|
|
"path/filepath"
|
2015-05-06 10:37:29 +08:00
|
|
|
"text/template"
|
2015-05-07 07:19:02 +08:00
|
|
|
|
|
|
|
|
"github.com/russross/blackfriday"
|
|
|
|
|
"strings"
|
2015-05-06 10:37:29 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
// Process the contents of a page.
|
|
|
|
|
// It parses the metadata if any and uses the template if found
|
2015-05-07 07:19:02 +08:00
|
|
|
func Process(c Config, fpath string, b []byte) ([]byte, error) {
|
2015-05-06 10:37:29 +08:00
|
|
|
metadata, markdown, err := extractMetadata(b)
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
// if template is not specified, check if Default template is set
|
|
|
|
|
if metadata.Template == "" {
|
|
|
|
|
if _, ok := c.Templates[DefaultTemplate]; ok {
|
|
|
|
|
metadata.Template = DefaultTemplate
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// if template is set, load it
|
2015-05-06 10:37:29 +08:00
|
|
|
var tmpl []byte
|
|
|
|
|
if metadata.Template != "" {
|
|
|
|
|
if t, ok := c.Templates[metadata.Template]; ok {
|
2015-05-07 07:19:02 +08:00
|
|
|
tmpl, err = ioutil.ReadFile(t)
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// process markdown
|
2015-05-07 07:19:02 +08:00
|
|
|
markdown = blackfriday.Markdown(markdown, c.Renderer, 0)
|
|
|
|
|
// set it as body for template
|
|
|
|
|
metadata.Variables["body"] = string(markdown)
|
2015-05-06 10:37:29 +08:00
|
|
|
|
2015-05-07 07:19:02 +08:00
|
|
|
return processTemplate(c, fpath, tmpl, metadata)
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// extractMetadata extracts metadata content from a page.
|
|
|
|
|
// it returns the metadata, the remaining bytes (markdown),
|
|
|
|
|
// and an error if any
|
|
|
|
|
func extractMetadata(b []byte) (metadata Metadata, markdown []byte, err error) {
|
|
|
|
|
b = bytes.TrimSpace(b)
|
|
|
|
|
reader := bytes.NewBuffer(b)
|
|
|
|
|
scanner := bufio.NewScanner(reader)
|
|
|
|
|
var parser MetadataParser
|
|
|
|
|
// if scanner.Scan() &&
|
|
|
|
|
// Read first line
|
|
|
|
|
if scanner.Scan() {
|
|
|
|
|
line := scanner.Bytes()
|
|
|
|
|
parser = findParser(line)
|
|
|
|
|
// if no parser found
|
|
|
|
|
// assume metadata not present
|
|
|
|
|
if parser == nil {
|
|
|
|
|
return metadata, b, nil
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// buffer for metadata contents
|
|
|
|
|
buf := bytes.Buffer{}
|
|
|
|
|
|
|
|
|
|
// Read remaining lines until closing identifier is found
|
|
|
|
|
for scanner.Scan() {
|
|
|
|
|
line := scanner.Bytes()
|
|
|
|
|
// closing identifier found
|
|
|
|
|
if bytes.Equal(bytes.TrimSpace(line), parser.Closing()) {
|
2015-05-07 07:19:02 +08:00
|
|
|
// parse the metadata
|
|
|
|
|
err := parser.Parse(buf.Bytes())
|
|
|
|
|
if err != nil {
|
2015-05-06 10:37:29 +08:00
|
|
|
return metadata, nil, err
|
|
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
// get the scanner to return remaining bytes
|
|
|
|
|
scanner.Split(func(data []byte, atEOF bool) (int, []byte, error) {
|
|
|
|
|
return len(data), data, nil
|
|
|
|
|
})
|
|
|
|
|
// scan the remaining bytes
|
|
|
|
|
scanner.Scan()
|
|
|
|
|
|
|
|
|
|
return parser.Metadata(), scanner.Bytes(), nil
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
|
|
|
|
buf.Write(line)
|
2015-05-07 07:19:02 +08:00
|
|
|
buf.WriteString("\r\n")
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
|
|
|
|
return metadata, nil, fmt.Errorf("Metadata not closed. '%v' not found", string(parser.Closing()))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// findParser locates the parser for an opening identifier
|
|
|
|
|
func findParser(line []byte) MetadataParser {
|
|
|
|
|
line = bytes.TrimSpace(line)
|
|
|
|
|
for _, parser := range parsers {
|
|
|
|
|
if bytes.Equal(parser.Opening(), line) {
|
|
|
|
|
return parser
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
2015-05-07 07:19:02 +08:00
|
|
|
func processTemplate(c Config, fpath string, tmpl []byte, metadata Metadata) ([]byte, error) {
|
|
|
|
|
// if template is specified
|
|
|
|
|
// replace parse the template
|
|
|
|
|
if tmpl != nil {
|
|
|
|
|
tmpl = defaultTemplate(c, metadata, fpath)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
b := &bytes.Buffer{}
|
|
|
|
|
t, err := template.New("").Parse(string(tmpl))
|
2015-05-06 10:37:29 +08:00
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
if err = t.Execute(b, metadata.Variables); err != nil {
|
|
|
|
|
return nil, err
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
return b.Bytes(), nil
|
|
|
|
|
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
|
|
|
|
|
2015-05-07 07:19:02 +08:00
|
|
|
func defaultTemplate(c Config, metadata Metadata, fpath string) []byte {
|
|
|
|
|
// else, use default template
|
|
|
|
|
var scripts, styles bytes.Buffer
|
|
|
|
|
for _, style := range c.Styles {
|
|
|
|
|
styles.WriteString(strings.Replace(cssTemplate, "{{url}}", style, 1))
|
|
|
|
|
styles.WriteString("\r\n")
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
for _, script := range c.Scripts {
|
|
|
|
|
scripts.WriteString(strings.Replace(jsTemplate, "{{url}}", script, 1))
|
|
|
|
|
scripts.WriteString("\r\n")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Title is first line (length-limited), otherwise filename
|
|
|
|
|
title := metadata.Title
|
|
|
|
|
if title == "" {
|
|
|
|
|
title = filepath.Base(fpath)
|
|
|
|
|
if body, _ := metadata.Variables["body"].([]byte); len(body) > 128 {
|
|
|
|
|
title = string(body[:128])
|
|
|
|
|
} else if len(body) > 0 {
|
|
|
|
|
title = string(body)
|
|
|
|
|
}
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
|
|
|
|
|
html := []byte(htmlTemplate)
|
|
|
|
|
html = bytes.Replace(html, []byte("{{title}}"), []byte(title), 1)
|
|
|
|
|
html = bytes.Replace(html, []byte("{{css}}"), styles.Bytes(), 1)
|
|
|
|
|
html = bytes.Replace(html, []byte("{{js}}"), scripts.Bytes(), 1)
|
|
|
|
|
|
|
|
|
|
return html
|
2015-05-06 10:37:29 +08:00
|
|
|
}
|
2015-05-07 07:19:02 +08:00
|
|
|
|
|
|
|
|
const (
|
|
|
|
|
htmlTemplate = `<!DOCTYPE html>
|
|
|
|
|
<html>
|
|
|
|
|
<head>
|
|
|
|
|
<title>{{title}}</title>
|
|
|
|
|
<meta charset="utf-8">
|
|
|
|
|
{{css}}
|
|
|
|
|
{{js}}
|
|
|
|
|
</head>
|
|
|
|
|
<body>
|
|
|
|
|
{{.body}}
|
|
|
|
|
</body>
|
|
|
|
|
</html>`
|
|
|
|
|
cssTemplate = `<link rel="stylesheet" href="{{url}}">`
|
|
|
|
|
jsTemplate = `<script src="{{url}}"></script>`
|
|
|
|
|
|
|
|
|
|
DefaultTemplate = "defaultTemplate"
|
|
|
|
|
)
|
