1package main
  2
  3import (
  4	"github.com/cross-cpm/go-shutil"
  5	"io/ioutil"
  6	"os"
  7	"os/exec"
  8	"path/filepath"
  9	"strings"
 10	"text/template"
 11)
 12
 13var cfg = parseConfig()
 14
 15func execute(cmds []string) {
 16	for _, cmd := range cmds {
 17		_, err := exec.Command(cmd).Output()
 18		printMsg("running:", cmd)
 19		if err != nil {
 20			printErr(err)
 21		}
 22	}
 23}
 24
 25func processTemplate(tmplPath string) *template.Template {
 26	tmplFile := filepath.Join("templates", tmplPath)
 27	tmpl, err := template.ParseFiles(tmplFile)
 28	if err != nil {
 29		printErr(err)
 30	}
 31
 32	return tmpl
 33}
 34
 35func handleMd(mdPath string) {
 36	content, err := ioutil.ReadFile(mdPath)
 37	if err != nil {
 38		printErr(err)
 39	}
 40
 41	restContent, fm := parseFrontmatter(content)
 42	bodyHtml := mdRender(restContent)
 43	relPath, _ := filepath.Rel("pages/", mdPath)
 44
 45	var buildPath string
 46	if strings.HasSuffix(relPath, "_index.md") {
 47		dir, _ := filepath.Split(relPath)
 48		buildPath = filepath.Join("build", dir)
 49	} else {
 50		buildPath = filepath.Join(
 51			"build",
 52			strings.TrimSuffix(relPath, filepath.Ext(relPath)),
 53		)
 54	}
 55
 56	os.MkdirAll(buildPath, 0755)
 57
 58	fm.Body = string(bodyHtml)
 59
 60	// combine config and matter structs
 61	combined := struct {
 62		Cfg Config
 63		Fm  Matter
 64	}{cfg, fm}
 65
 66	htmlFile, err := os.Create(filepath.Join(buildPath, "index.html"))
 67	if err != nil {
 68		printErr(err)
 69		return
 70	}
 71	if fm.Template == "" {
 72		fm.Template = "text.html"
 73	}
 74	tmpl := processTemplate(fm.Template)
 75	err = tmpl.Execute(htmlFile, combined)
 76	if err != nil {
 77		printErr(err)
 78		return
 79	}
 80	htmlFile.Close()
 81}
 82
 83func viteBuild() {
 84	printMsg("executing pre-build actions...")
 85	execute(cfg.Prebuild)
 86	err := filepath.Walk("./pages", func(path string, info os.FileInfo, err error) error {
 87		if err != nil {
 88			printErr(err)
 89			return err
 90		}
 91		if filepath.Ext(path) == ".md" {
 92			handleMd(path)
 93		} else {
 94			f, err := os.Stat(path)
 95			if err != nil {
 96				printErr(err)
 97			}
 98			mode := f.Mode()
 99			if mode.IsRegular() {
100				options := shutil.CopyOptions{}
101				relPath, _ := filepath.Rel("pages/", path)
102				options.FollowSymlinks = true
103				shutil.CopyFile(
104					path,
105					filepath.Join("build", relPath),
106					&options,
107				)
108			}
109		}
110		return nil
111	})
112	if err != nil {
113		printErr(err)
114	}
115
116	_, err = shutil.CopyTree("static", filepath.Join("build", "static"), nil)
117	if err != nil {
118		printErr(err)
119	}
120	printMsg("site build complete")
121	printMsg("executing post-build actions...")
122	execute(cfg.Postbuild)
123}