source: code/trunk/zs.go@ 31

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

added check for fs walk errors
File size: 6.9 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 := Funcs{}
71 for k, v := range funcs {
72 f[k] = v
73 }
74 for k, v := range vars {
75 f[k] = varFunc(v)
76 }
77 // Plugin functions
78 files, _ := ioutil.ReadDir(ZSDIR)
79 for _, file := range files {
80 if !file.IsDir() {
81 name := file.Name()
82 if !strings.HasSuffix(name, ".html") && !strings.HasSuffix(name, ".amber") {
83 f[strings.TrimSuffix(name, filepath.Ext(name))] = pluginFunc(name, vars)
84 }
85 }
86 }
87
88 tmpl, err := template.New("").Funcs(template.FuncMap(f)).Parse(s)
89 if err != nil {
90 return "", err
91 }
92 out := &bytes.Buffer{}
93 if err := tmpl.Execute(out, vars); err != nil {
94 return "", err
95 }
96 return string(out.Bytes()), nil
97}
98
99// Renders markdown with the given layout into html expanding all the macros
100func buildMarkdown(path string, w io.Writer, funcs Funcs, vars Vars) error {
101 v, body, err := md(path, vars)
102 if err != nil {
103 return err
104 }
105 content, err := render(body, funcs, v)
106 if err != nil {
107 return err
108 }
109 v["content"] = string(blackfriday.MarkdownBasic([]byte(content)))
110 if w == nil {
111 out, err := os.Create(filepath.Join(PUBDIR, renameExt(path, "", ".html")))
112 if err != nil {
113 return err
114 }
115 defer out.Close()
116 w = out
117 }
118 if strings.HasSuffix(v["layout"], ".amber") {
119 return buildAmber(filepath.Join(ZSDIR, v["layout"]), w, funcs, v)
120 } else {
121 return buildHTML(filepath.Join(ZSDIR, v["layout"]), w, funcs, v)
122 }
123}
124
125// Renders text file expanding all variable macros inside it
126func buildHTML(path string, w io.Writer, funcs Funcs, vars Vars) error {
127 b, err := ioutil.ReadFile(path)
128 if err != nil {
129 return err
130 }
131 content, err := render(string(b), funcs, vars)
132 if err != nil {
133 return err
134 }
135 if w == nil {
136 f, err := os.Create(filepath.Join(PUBDIR, path))
137 if err != nil {
138 return err
139 }
140 defer f.Close()
141 w = f
142 }
143 _, err = io.WriteString(w, content)
144 return err
145}
146
147// Renders .amber file into .html
148func buildAmber(path string, w io.Writer, funcs Funcs, vars Vars) error {
149 a := amber.New()
150 err := a.ParseFile(path)
151 if err != nil {
152 return err
153 }
154
155 data := map[string]interface{}{}
156 for k, v := range vars {
157 data[k] = v
158 }
159 for k, v := range funcs {
160 data[k] = v
161 }
162
163 t, err := a.Compile()
164 if err != nil {
165 return err
166 }
167 if w == nil {
168 f, err := os.Create(filepath.Join(PUBDIR, renameExt(path, ".amber", ".html")))
169 if err != nil {
170 return err
171 }
172 defer f.Close()
173 w = f
174 }
175 return t.Execute(w, data)
176}
177
178// Compiles .gcss into .css
179func buildGCSS(path string, w io.Writer) error {
180 f, err := os.Open(path)
181 if err != nil {
182 return err
183 }
184 defer f.Close()
185
186 if w == nil {
187 s := strings.TrimSuffix(path, ".gcss") + ".css"
188 css, err := os.Create(filepath.Join(PUBDIR, s))
189 if err != nil {
190 return err
191 }
192 defer css.Close()
193 w = css
194 }
195 _, err = gcss.Compile(w, f)
196 return err
197}
198
199// Copies file as is from path to writer
200func buildRaw(path string, w io.Writer) error {
201 in, err := os.Open(path)
202 if err != nil {
203 return err
204 }
205 defer in.Close()
206 if w == nil {
207 if out, err := os.Create(filepath.Join(PUBDIR, path)); err != nil {
208 return err
209 } else {
210 defer out.Close()
211 w = out
212 }
213 }
214 _, err = io.Copy(w, in)
215 return err
216}
217
218func build(path string, w io.Writer, funcs Funcs, vars Vars) error {
219 ext := filepath.Ext(path)
220 if ext == ".md" || ext == ".mkd" {
221 return buildMarkdown(path, w, funcs, vars)
222 } else if ext == ".html" || ext == ".xml" {
223 return buildHTML(path, w, funcs, vars)
224 } else if ext == ".amber" {
225 return buildAmber(path, w, funcs, vars)
226 } else if ext == ".gcss" {
227 return buildGCSS(path, w)
228 } else {
229 return buildRaw(path, w)
230 }
231}
232
233func buildAll(watch bool) {
234 lastModified := time.Unix(0, 0)
235 modified := false
236
237 vars := globals()
238 for {
239 os.Mkdir(PUBDIR, 0755)
240 funcs := builtins()
241 err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
242 // ignore hidden files and directories
243 if filepath.Base(path)[0] == '.' || strings.HasPrefix(path, ".") {
244 return nil
245 }
246 // inform user about fs walk errors, but continue iteration
247 if err != nil {
248 log.Println("ERROR:", err)
249 return nil
250 }
251
252 if info.IsDir() {
253 os.Mkdir(filepath.Join(PUBDIR, path), 0755)
254 return nil
255 } else if info.ModTime().After(lastModified) {
256 if !modified {
257 // About to be modified, so run pre-build hook
258 // FIXME on windows it might not work well
259 run(filepath.Join(ZSDIR, "pre"), []string{}, nil, nil)
260 modified = true
261 }
262 log.Println("build: ", path)
263 return build(path, nil, funcs, vars)
264 }
265 return nil
266 })
267 if err != nil {
268 log.Println("ERROR:", err)
269 }
270 if modified {
271 // Something was modified, so post-build hook
272 // FIXME on windows it might not work well
273 run(filepath.Join(ZSDIR, "post"), []string{}, nil, nil)
274 modified = false
275 }
276 if !watch {
277 break
278 }
279 lastModified = time.Now()
280 time.Sleep(1 * time.Second)
281 }
282}
283
284func main() {
285 if len(os.Args) == 1 {
286 fmt.Println(os.Args[0], "<command> [args]")
287 return
288 }
289 cmd := os.Args[1]
290 args := os.Args[2:]
291 switch cmd {
292 case "build":
293 if len(args) == 0 {
294 buildAll(false)
295 } else if len(args) == 1 {
296 if err := build(args[0], os.Stdout, builtins(), globals()); err != nil {
297 fmt.Println("ERROR: " + err.Error())
298 }
299 } else {
300 fmt.Println("ERROR: too many arguments")
301 }
302 case "watch":
303 buildAll(true)
304 case "var":
305 fmt.Println(Var(args))
306 case "lorem":
307 fmt.Println(Lorem(args))
308 case "dateparse":
309 fmt.Println(DateParse(args))
310 case "datefmt":
311 fmt.Println(DateFmt(args))
312 case "wc":
313 fmt.Println(WordCount(args))
314 case "timetoread":
315 fmt.Println(TimeToRead(args))
316 default:
317 err := run(path.Join(ZSDIR, cmd), args, globals(), os.Stdout)
318 if err != nil {
319 log.Println("ERROR:", err)
320 }
321 }
322}
Note: See TracBrowser for help on using the repository browser.