zs/zs.go

273 lines
5.6 KiB
Go
Raw Normal View History

2014-12-05 16:59:56 +00:00
package main
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"log"
"os"
"os/exec"
"path"
"path/filepath"
"strings"
"time"
"github.com/russross/blackfriday"
)
const (
ZSDIR = ".zs"
PUBDIR = ".pub"
)
type EvalFn func(args []string, vars map[string]string) (string, error)
2014-12-05 16:59:56 +00:00
func split2(s, delim string) (string, string) {
parts := strings.SplitN(s, delim, 2)
if len(parts) == 2 {
return parts[0], parts[1]
} else {
return parts[0], ""
}
}
func md(s string) (map[string]string, string) {
v := map[string]string{}
2014-12-05 17:09:10 +00:00
if strings.Index(s, "\n\n") == -1 {
return map[string]string{}, s
}
2014-12-05 16:59:56 +00:00
header, body := split2(s, "\n\n")
for _, line := range strings.Split(header, "\n") {
key, value := split2(line, ":")
v[strings.ToLower(strings.TrimSpace(key))] = strings.TrimSpace(value)
}
return v, body
}
func render(s string, vars map[string]string, eval EvalFn) (string, error) {
delim_open := "{{"
delim_close := "}}"
2014-12-05 16:59:56 +00:00
out := bytes.NewBuffer(nil)
for {
if from := strings.Index(s, delim_open); from == -1 {
out.WriteString(s)
2014-12-05 16:59:56 +00:00
return out.String(), nil
} else {
if to := strings.Index(s, delim_close); to == -1 {
2014-12-05 16:59:56 +00:00
return "", fmt.Errorf("Close delim not found")
} else {
out.WriteString(s[:from])
cmd := s[from+len(delim_open) : to]
s = s[to+len(delim_close):]
m := strings.Fields(cmd)
2014-12-05 16:59:56 +00:00
if len(m) == 1 {
if v, ok := vars[m[0]]; ok {
out.WriteString(v)
2014-12-05 16:59:56 +00:00
continue
}
}
if res, err := eval(m, vars); err == nil {
out.WriteString(res)
2014-12-05 16:59:56 +00:00
} else {
log.Println(err) // silent
}
}
}
}
}
func env(vars map[string]string) []string {
2014-12-05 16:59:56 +00:00
env := []string{"ZS=" + os.Args[0]}
env = append(env, os.Environ()...)
2014-12-05 16:59:56 +00:00
for k, v := range vars {
env = append(env, "ZS_"+strings.ToUpper(k)+"="+v)
}
return env
}
func run(cmd string, args []string, vars map[string]string, output io.Writer) error {
var errbuf bytes.Buffer
c := exec.Command(cmd, args...)
c.Env = env(vars)
c.Stdout = output
2014-12-05 16:59:56 +00:00
c.Stderr = &errbuf
err := c.Run()
if errbuf.Len() > 0 {
log.Println(errbuf.String())
}
if err != nil {
return err
}
return nil
}
func eval(cmd []string, vars map[string]string) (string, error) {
outbuf := bytes.NewBuffer(nil)
err := run(path.Join(ZSDIR, cmd[0]), cmd[1:], vars, outbuf)
if err != nil {
2014-12-05 18:21:42 +00:00
if _, ok := err.(*exec.ExitError); ok {
return "", err
}
outbuf = bytes.NewBuffer(nil)
err := run(cmd[0], cmd[1:], vars, outbuf)
2014-12-05 18:21:42 +00:00
// Return exit errors, but ignore if the command was not found
if _, ok := err.(*exec.ExitError); ok {
2014-12-05 16:59:56 +00:00
return "", err
}
}
return outbuf.String(), nil
}
func buildMarkdown(path string) error {
b, err := ioutil.ReadFile(path)
if err != nil {
return err
}
v, body := md(string(b))
defaultVars(v, path)
content, err := render(body, v, eval)
if err != nil {
return err
}
v["content"] = string(blackfriday.MarkdownBasic([]byte(content)))
b, err = ioutil.ReadFile(filepath.Join(ZSDIR, v["layout"]))
if err != nil {
return err
}
content, err = render(string(b), v, eval)
if err != nil {
return err
}
err = ioutil.WriteFile(v["output"], []byte(content), 0666)
if err != nil {
return err
}
return nil
}
func defaultVars(vars map[string]string, path string) {
if _, ok := vars["file"]; !ok {
vars["file"] = path
}
if _, ok := vars["url"]; !ok {
vars["url"] = path[:len(path)-len(filepath.Ext(path))] + ".html"
if strings.HasPrefix(vars["url"], "./") {
vars["url"] = vars["url"][2:]
}
}
if _, ok := vars["outdir"]; !ok {
vars["outdir"] = PUBDIR
}
if _, ok := vars["output"]; !ok {
vars["output"] = filepath.Join(PUBDIR, vars["url"])
}
if _, ok := vars["layout"]; !ok {
vars["layout"] = "index.html"
}
}
func copyFile(path string) error {
if in, err := os.Open(path); err != nil {
return err
} else {
defer in.Close()
if stat, err := in.Stat(); err != nil {
return err
} else {
// Directory?
if stat.Mode().IsDir() {
os.Mkdir(filepath.Join(PUBDIR, path), 0755)
return nil
}
if !stat.Mode().IsRegular() {
return nil
}
}
if out, err := os.Create(filepath.Join(PUBDIR, path)); err != nil {
return err
} else {
defer out.Close()
_, err = io.Copy(out, in)
return err
}
}
}
func buildAll(once bool) {
lastModified := time.Unix(0, 0)
for {
os.Mkdir(PUBDIR, 0755)
err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
// ignore hidden files and directories
if filepath.Base(path)[0] == '.' || strings.HasPrefix(path, ".") {
return nil
}
if info.ModTime().After(lastModified) {
ext := filepath.Ext(path)
if ext == ".md" || ext == "mkd" {
log.Println("mkd: ", path)
return buildMarkdown(path)
} else {
log.Println("raw: ", path)
return copyFile(path)
}
}
return nil
})
if err != nil {
log.Println("ERROR:", err)
}
lastModified = time.Now()
if once {
break
}
time.Sleep(1 * time.Second)
}
}
func main() {
if len(os.Args) == 1 {
fmt.Println(os.Args[0], "<command> [args]")
return
}
cmd := os.Args[1]
args := os.Args[2:]
switch cmd {
case "build":
buildAll(true)
case "watch":
buildAll(false) // pass duration
case "var":
if len(args) == 0 {
log.Println("ERROR: filename expected")
2014-12-05 16:59:56 +00:00
return
}
if b, err := ioutil.ReadFile(args[0]); err == nil {
vars, _ := md(string(b))
defaultVars(vars, args[0])
if len(args) > 1 {
for _, a := range args[1:] {
fmt.Println(vars[a])
}
} else {
for k, v := range vars {
fmt.Println(k + ":" + v)
}
}
} else {
log.Println(err)
2014-12-05 16:59:56 +00:00
}
default:
err := run(path.Join(ZSDIR, cmd), args, map[string]string{}, os.Stdout)
if err != nil {
2014-12-05 16:59:56 +00:00
log.Println(err)
}
}
}