package build import ( "bytes" "fmt" "html/template" "path/filepath" "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 } // this function converts a file path to its title form func pathToTitle(path string) string { stripped := ChangeExtension(filepath.Base(path), "") replaced := strings.NewReplacer("-", " ", "_", " ", `\ `, " ").Replace(stripped) return strings.ToTitle(replaced) } func buildPageData(m Metadata, path string) *PageData { p := &PageData{} if title, ok := m["title"].(string); ok { p.Title = 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) // 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 }