72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
"flag"
|
|
"github.com/yuin/goldmark"
|
|
"github.com/yuin/goldmark/parser"
|
|
"go.abhg.dev/goldmark/frontmatter"
|
|
"html/template"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
//go:embed frame.tmpl
|
|
var frameTmpl embed.FS
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if flag.NArg() != 2 {
|
|
log.Fatalln("usage: gendocs MARKDOWN OUTFILE")
|
|
}
|
|
|
|
srcFile := flag.Arg(0)
|
|
dstFile := flag.Arg(1)
|
|
|
|
md := goldmark.New(goldmark.WithExtensions(&frontmatter.Extender{}))
|
|
|
|
mdData, err := os.ReadFile(srcFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ctx := parser.NewContext()
|
|
|
|
var buf bytes.Buffer
|
|
if err := md.Convert(mdData, &buf, parser.WithContext(ctx)); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var frontMatter struct {
|
|
Title string `yaml:"title"`
|
|
}
|
|
if fm := frontmatter.Get(ctx); fm != nil {
|
|
if err := fm.Decode(&frontMatter); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
frameTmpls, err := template.ParseFS(frameTmpl, "*.tmpl")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
var res bytes.Buffer
|
|
if err := frameTmpls.ExecuteTemplate(&res, "frame.tmpl", map[string]interface{}{
|
|
"Title": frontMatter.Title,
|
|
"Body": template.HTML(buf.Bytes()),
|
|
}); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
destDir := filepath.Dir(dstFile)
|
|
if err := os.MkdirAll(destDir, 0755); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(dstFile, res.Bytes(), 0644); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|