105 lines
2.1 KiB
Go
105 lines
2.1 KiB
Go
package build
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"html/template"
|
|
"strings"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type PageData struct {
|
|
Title string
|
|
Icon string
|
|
Stylesheet string
|
|
Header string
|
|
Content template.HTML
|
|
NextPost string
|
|
PrevPost string
|
|
Footer string
|
|
}
|
|
|
|
type Metadata map[string]interface{}
|
|
|
|
func processWithYaml(f []byte) (Metadata, []byte, error) {
|
|
// Check if the file has valid metadata
|
|
if !bytes.HasPrefix(f, []byte("---\n")) {
|
|
// No valid yaml, so return the entire content
|
|
return nil, f, nil
|
|
}
|
|
// Separate YAML from rest of document
|
|
split := strings.SplitN(string(f), "---\n", 3)
|
|
if len(split) < 3 {
|
|
return nil, nil, fmt.Errorf("Invalid frontmatter format.")
|
|
}
|
|
var meta Metadata
|
|
// Parse YAML
|
|
if err := yaml.Unmarshal([]byte(split[1]), &meta); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return meta, []byte(split[2]), nil
|
|
}
|
|
|
|
func buildPageData(m Metadata, path string) *PageData {
|
|
p := &PageData{}
|
|
if title, ok := m["title"].(string); ok {
|
|
p.Title = wordsToTitle(title)
|
|
} else {
|
|
p.Title = pathToTitle(path)
|
|
}
|
|
if icon, ok := m["icon"].(string); ok {
|
|
p.Icon = icon
|
|
} else {
|
|
p.Icon = DefaultIcon
|
|
}
|
|
if style, ok := m["style"].(string); ok {
|
|
p.Stylesheet = style
|
|
} else {
|
|
p.Stylesheet = DefaultStylesheet
|
|
}
|
|
if header, ok := m["header"].(string); ok {
|
|
p.Header = header
|
|
} else {
|
|
p.Header = DefaultHeader
|
|
}
|
|
if footer, ok := m["footer"].(string); ok {
|
|
p.Footer = footer
|
|
} else {
|
|
p.Footer = DefaultFooter
|
|
}
|
|
return p
|
|
}
|
|
|
|
func ConvertFile(in string, out string) error {
|
|
mdPre, err := ReadFile(in)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
metadata, md, err := processWithYaml(mdPre)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pd := buildPageData(metadata, in)
|
|
fmt.Println("Title: ", pd.Title)
|
|
|
|
// build according to template here
|
|
html, err := MdToHTML(md)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
pd.Content = template.HTML(html)
|
|
|
|
tmpl, err := template.New("webpage").Parse(DefaultTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var output bytes.Buffer
|
|
if err := tmpl.Execute(&output, pd); err != nil {
|
|
return err
|
|
}
|
|
|
|
err = WriteFile(output.Bytes(), out)
|
|
return err
|
|
}
|