initial commit

diff --git a/.gitignore b/.gitignore
new file mode 100644 (file)
index 0000000..470733c
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+zs

diff --git a/zs.go b/zs.go
new file mode 100644 (file)
index 0000000..eb241f3
--- /dev/null
+++ b/zs.go
@@ -0,0 +1,256 @@
+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"
+)
+
+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{}
+       // FIXME: if no header?
+       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
+}
+
+type EvalFn func(args []string, vars map[string]string) (string, error)
+
+func render(s string, vars map[string]string, eval EvalFn) (string, error) {
+       b := []byte(s)
+       delim_open := []byte("{{")
+       delim_close := []byte("}}")
+
+       out := bytes.NewBuffer(nil)
+       for {
+               if from := bytes.Index(b, delim_open); from == -1 {
+                       out.Write(b)
+                       return out.String(), nil
+               } else {
+                       to := bytes.Index(b, delim_close)
+                       if to == -1 {
+                               return "", fmt.Errorf("Close delim not found")
+                       } else {
+                               out.Write(b[:from])
+                               cmd := b[from+len(delim_open) : to]
+                               b = b[to+len(delim_close):]
+                               m := strings.Fields(string(cmd))
+                               if len(m) == 1 {
+                                       if v, ok := vars[m[0]]; ok {
+                                               out.Write([]byte(v))
+                                               continue
+                                       }
+                               }
+                               if res, err := eval(m, vars); err == nil {
+                                       out.Write([]byte(res))
+                               } else {
+                                       log.Println(err) // silent
+                               }
+                       }
+               }
+       }
+}
+
+func eval(cmd []string, vars map[string]string) (string, error) {
+       var outbuf, errbuf bytes.Buffer
+       c := exec.Command(path.Join(ZSDIR, cmd[0]), cmd[1:]...)
+       env := []string{"ZS=" + os.Args[0]}
+       for k, v := range vars {
+               env = append(env, "ZS_"+strings.ToUpper(k)+"="+v)
+       }
+       c.Env = append(c.Env, env...)
+       c.Stdout = &outbuf
+       c.Stderr = &errbuf
+       if err := c.Run(); err != nil {
+               log.Println(err)
+               c := exec.Command(path.Join(cmd[0]), cmd[1:]...)
+               c.Env = append(c.Env, env...)
+               c.Stdout = &outbuf
+               c.Stderr = &errbuf
+               if err := c.Run(); err != nil {
+                       return "", err
+               }
+       }
+       if errbuf.Len() > 0 {
+               log.Println(errbuf.String())
+       }
+       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 {
+                       // print error
+                       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 {
+                       // print error
+               }
+       default:
+               cmd := exec.Command(path.Join(ZSDIR, cmd), args...)
+               cmd.Env = append(cmd.Env, "ZS="+os.Args[0])
+               cmd.Stdout = os.Stdout
+               cmd.Stderr = os.Stderr
+               if err := cmd.Run(); err != nil {
+                       log.Println(err)
+               }
+       }
+}

diff --git a/zs_test.go b/zs_test.go
new file mode 100644 (file)
index 0000000..0dbd68a
--- /dev/null
+++ b/zs_test.go
@@ -0,0 +1,66 @@
+package main
+
+import "testing"
+
+func TestSplit2(t *testing.T) {
+       if a, b := split2("a:b", ":"); a != "a" || b != "b" {
+               t.Fail()
+       }
+       if a, b := split2(":b", ":"); a != "" || b != "b" {
+               t.Fail()
+       }
+       if a, b := split2("a:", ":"); a != "a" || b != "" {
+               t.Fail()
+       }
+       if a, b := split2(":", ":"); a != "" || b != "" {
+               t.Fail()
+       }
+       if a, b := split2("a", ":"); a != "a" || b != "" {
+               t.Fail()
+       }
+       if a, b := split2("", ":"); a != "" || b != "" {
+               t.Fail()
+       }
+}
+
+func TestMD(t *testing.T) {
+       v, body := md(`
+       title: Hello, world!
+       keywords: foo, bar, baz
+       empty:
+       bayan: [:|||:]
+
+this: is a content`)
+       if v["title"] != "Hello, world!" {
+               t.Error()
+       }
+       if v["keywords"] != "foo, bar, baz" {
+               t.Error()
+       }
+       if s, ok := v["empty"]; !ok || len(s) != 0 {
+               t.Error()
+       }
+       if v["bayan"] != "[:|||:]" {
+               t.Error()
+       }
+       if body != "this: is a content" {
+               t.Error(body)
+       }
+}
+
+func TestRender(t *testing.T) {
+       eval := func(a []string) ([]byte, error) {
+               return []byte("hello"), nil
+       }
+       vars := map[string]string{"foo": "bar"}
+
+       if s, err := render("plain text", vars, eval); err != nil || s != "plain text" {
+               t.Error()
+       }
+       if s, err := render("a {{greet}} text", vars, eval); err != nil || s != "a hello text" {
+               t.Error()
+       }
+       if s, err := render("{{greet}} x{{foo}}z", vars, eval); err != nil || s != "hello xbarz" {
+               t.Error()
+       }
+}