source: code/trunk/zs.go@ 32

Last change on this file since 32 was 32, checked in by zaitsev.serge, 10 years ago

dead end with template functions
File size: 6.7 KB
Line 
1package main
2
3import (
4 "bytes"
5 "fmt"
6 "io"
7 "io/ioutil"
8 "log"
9 "os"
10 "path"
11 "path/filepath"
12 "strings"
13 "text/template"
14 "time"
15
16 "github.com/eknkc/amber"
17 "github.com/russross/blackfriday"
18 "github.com/yosssi/gcss"
19)
20
21const (
22 ZSDIR = ".zs"
23 PUBDIR = ".pub"
24)
25
26type Vars map[string]string
27type Funcs template.FuncMap
28
29// Parses markdown content. Returns parsed header variables and content
30func md(path string, globals Vars) (Vars, string, error) {
31 b, err := ioutil.ReadFile(path)
32 if err != nil {
33 return nil, "", err
34 }
35 s := string(b)
36 url := path[:len(path)-len(filepath.Ext(path))] + ".html"
37 v := Vars{
38 "title": "",
39 "description": "",
40 "keywords": "",
41 }
42 for name, value := range globals {
43 v[name] = value
44 }
45 if _, err := os.Stat(filepath.Join(ZSDIR, "layout.amber")); err == nil {
46 v["layout"] = "layout.amber"
47 } else {
48 v["layout"] = "layout.html"
49 }
50 v["file"] = path
51 v["url"] = url
52 v["output"] = filepath.Join(PUBDIR, url)
53
54 if strings.Index(s, "\n\n") == -1 {
55 return v, s, nil
56 }
57 header, body := split2(s, "\n\n")
58 for _, line := range strings.Split(header, "\n") {
59 key, value := split2(line, ":")
60 v[strings.ToLower(strings.TrimSpace(key))] = strings.TrimSpace(value)
61 }
62 if strings.HasPrefix(v["url"], "./") {
63 v["url"] = v["url"][2:]
64 }
65 return v, body, nil
66}
67
68// Use standard Go templates
69func render(s string, funcs Funcs, vars Vars) (string, error) {
70 f := makeFuncs(funcs, vars)
71 tmpl, err := template.New("").Funcs(template.FuncMap(f)).Parse(s)
72 if err != nil {
73 return "", err
74 }
75 out := &bytes.Buffer{}
76 if err := tmpl.Execute(out, vars); err != nil {
77 return "", err
78 }
79 return string(out.Bytes()), nil
80}
81
82// Renders markdown with the given layout into html expanding all the macros
83func buildMarkdown(path string, w io.Writer, funcs Funcs, vars Vars) error {
84 v, body, err := md(path, vars)
85 if err != nil {
86 return err
87 }
88 content, err := render(body, funcs, v)
89 if err != nil {
90 return err
91 }
92 v["content"] = string(blackfriday.MarkdownCommon([]byte(content)))
93 if w == nil {
94 out, err := os.Create(filepath.Join(PUBDIR, renameExt(path, "", ".html")))
95 if err != nil {
96 return err
97 }
98 defer out.Close()
99 w = out
100 }
101 if strings.HasSuffix(v["layout"], ".amber") {
102 return buildAmber(filepath.Join(ZSDIR, v["layout"]), w, funcs, v)
103 } else {
104 return buildHTML(filepath.Join(ZSDIR, v["layout"]), w, funcs, v)
105 }
106}
107
108// Renders text file expanding all variable macros inside it
109func buildHTML(path string, w io.Writer, funcs Funcs, vars Vars) error {
110 b, err := ioutil.ReadFile(path)
111 if err != nil {
112 return err
113 }
114 content, err := render(string(b), funcs, vars)
115 if err != nil {
116 return err
117 }
118 if w == nil {
119 f, err := os.Create(filepath.Join(PUBDIR, path))
120 if err != nil {
121 return err
122 }
123 defer f.Close()
124 w = f
125 }
126 _, err = io.WriteString(w, content)
127 return err
128}
129
130// Renders .amber file into .html
131func buildAmber(path string, w io.Writer, funcs Funcs, vars Vars) error {
132 a := amber.New()
133 err := a.ParseFile(path)
134 if err != nil {
135 return err
136 }
137
138 data := map[string]interface{}{}
139 for k, v := range vars {
140 data[k] = v
141 }
142 for k, v := range makeFuncs(funcs, Vars{}) {
143 data[k] = v
144 }
145
146 t, err := a.Compile()
147 if err != nil {
148 return err
149 }
150 if w == nil {
151 f, err := os.Create(filepath.Join(PUBDIR, renameExt(path, ".amber", ".html")))
152 if err != nil {
153 return err
154 }
155 defer f.Close()
156 w = f
157 }
158 return t.Execute(w, data)
159}
160
161// Compiles .gcss into .css
162func buildGCSS(path string, w io.Writer) error {
163 f, err := os.Open(path)
164 if err != nil {
165 return err
166 }
167 defer f.Close()
168
169 if w == nil {
170 s := strings.TrimSuffix(path, ".gcss") + ".css"
171 css, err := os.Create(filepath.Join(PUBDIR, s))
172 if err != nil {
173 return err
174 }
175 defer css.Close()
176 w = css
177 }
178 _, err = gcss.Compile(w, f)
179 return err
180}
181
182// Copies file as is from path to writer
183func buildRaw(path string, w io.Writer) error {
184 in, err := os.Open(path)
185 if err != nil {
186 return err
187 }
188 defer in.Close()
189 if w == nil {
190 if out, err := os.Create(filepath.Join(PUBDIR, path)); err != nil {
191 return err
192 } else {
193 defer out.Close()
194 w = out
195 }
196 }
197 _, err = io.Copy(w, in)
198 return err
199}
200
201func build(path string, w io.Writer, funcs Funcs, vars Vars) error {
202 ext := filepath.Ext(path)
203 if ext == ".md" || ext == ".mkd" {
204 return buildMarkdown(path, w, funcs, vars)
205 } else if ext == ".html" || ext == ".xml" {
206 return buildHTML(path, w, funcs, vars)
207 } else if ext == ".amber" {
208 return buildAmber(path, w, funcs, vars)
209 } else if ext == ".gcss" {
210 return buildGCSS(path, w)
211 } else {
212 return buildRaw(path, w)
213 }
214}
215
216func buildAll(watch bool) {
217 lastModified := time.Unix(0, 0)
218 modified := false
219
220 vars := globals()
221 for {
222 os.Mkdir(PUBDIR, 0755)
223 funcs := builtins()
224 err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
225 // ignore hidden files and directories
226 if filepath.Base(path)[0] == '.' || strings.HasPrefix(path, ".") {
227 return nil
228 }
229 // inform user about fs walk errors, but continue iteration
230 if err != nil {
231 log.Println("ERROR:", err)
232 return nil
233 }
234
235 if info.IsDir() {
236 os.Mkdir(filepath.Join(PUBDIR, path), 0755)
237 return nil
238 } else if info.ModTime().After(lastModified) {
239 if !modified {
240 // About to be modified, so run pre-build hook
241 // FIXME on windows it might not work well
242 run(filepath.Join(ZSDIR, "pre"), []string{}, nil, nil)
243 modified = true
244 }
245 log.Println("build: ", path)
246 return build(path, nil, funcs, vars)
247 }
248 return nil
249 })
250 if err != nil {
251 log.Println("ERROR:", err)
252 }
253 if modified {
254 // Something was modified, so post-build hook
255 // FIXME on windows it might not work well
256 run(filepath.Join(ZSDIR, "post"), []string{}, nil, nil)
257 modified = false
258 }
259 if !watch {
260 break
261 }
262 lastModified = time.Now()
263 time.Sleep(1 * time.Second)
264 }
265}
266
267func main() {
268 if len(os.Args) == 1 {
269 fmt.Println(os.Args[0], "<command> [args]")
270 return
271 }
272 cmd := os.Args[1]
273 args := os.Args[2:]
274 switch cmd {
275 case "build":
276 if len(args) == 0 {
277 buildAll(false)
278 } else if len(args) == 1 {
279 if err := build(args[0], os.Stdout, builtins(), globals()); err != nil {
280 fmt.Println("ERROR: " + err.Error())
281 }
282 } else {
283 fmt.Println("ERROR: too many arguments")
284 }
285 case "watch":
286 buildAll(true)
287 case "var":
288 fmt.Println(Var(args...))
289 case "lorem":
290 fmt.Println(Lorem(args...))
291 case "dateparse":
292 fmt.Println(DateParse(args...))
293 case "datefmt":
294 fmt.Println(DateFmt(args...))
295 case "wc":
296 fmt.Println(WordCount(args...))
297 case "ttr":
298 fmt.Println(TimeToRead(args...))
299 case "ls":
300 fmt.Println(strings.Join(List(args...), "\n"))
301 case "sort":
302 fmt.Println(strings.Join(Sort(args...), "\n"))
303 case "exec":
304 // TODO
305 default:
306 err := run(path.Join(ZSDIR, cmd), args, globals(), os.Stdout)
307 if err != nil {
308 log.Println("ERROR:", err)
309 }
310 }
311}
Note: See TracBrowser for help on using the repository browser.