diff options
| -rw-r--r-- | hswg/main.go | 48 |
1 files changed, 29 insertions, 19 deletions
diff --git a/hswg/main.go b/hswg/main.go index 9972817..4626472 100644 --- a/hswg/main.go +++ b/hswg/main.go @@ -284,7 +284,8 @@ func finalizeEntries(entries *map[string]*Entry) { } } -func writeIndex(t *template.Template, entries *map[string]*Entry) error { +func writeIndex(path string, t *template.Template, + entries *map[string]*Entry) error { // Reorder entries reversely, primarily by date, secondarily by filename. ordered := []*Entry{} for _, e := range *entries { @@ -309,8 +310,13 @@ func writeIndex(t *template.Template, entries *map[string]*Entry) error { return p2.Before(*p1) }) + f, err := os.Create(path) + if err != nil { + return err + } + // TODO(p): Splitting content to categories would be nice. - return t.Execute(os.Stdout, ordered) + return t.Execute(f, ordered) } func singleFile() { @@ -330,43 +336,47 @@ func main() { singleFile() return } - if len(os.Args) < 3 { - log.Fatalf("usage: %s TEMPLATE GLOB...\n", os.Args[0]) + if len(os.Args) < 4 { + log.Fatalf("usage: %s TEMPLATE INDEX GLOB...\n", os.Args[0]) } - // Read the common page header. - header, err := ioutil.ReadFile(os.Args[1]) + argTemplate, argIndex, argGlobs := os.Args[1], os.Args[2], os.Args[3:] + + // Read a template for entries. + header, err := ioutil.ReadFile(argTemplate) if err != nil { log.Fatalln(err) } - t, err := template.New("page").Parse(string(header)) + tmplEntry, err := template.New("entry").Parse(string(header)) + if err != nil { + log.Fatalln(err) + } + + // Read a template for the index from the standard input. + index, err := ioutil.ReadAll(os.Stdin) + if err != nil { + log.Fatalln(err) + } + tmplIndex, err := template.New("index").Parse(string(index)) if err != nil { log.Fatalln(err) } // Process all entries. - entries, err := loadEntries(os.Args[2:]) + entries, err := loadEntries(argGlobs) if err != nil { log.Fatalln(err) } finalizeEntries(&entries) for _, e := range entries { - if err := writeEntry(e, t, &entries); err != nil { + if err := writeEntry(e, tmplEntry, &entries); err != nil { log.Fatalln(err) } } - // Read a template from the standard input, write an index. - var input []byte - if input, err = ioutil.ReadAll(os.Stdin); err != nil { - log.Fatalln(err) - } - t, err = template.New("-").Parse(string(input)) - if err != nil { - log.Fatalln(err) - } - if err := writeIndex(t, &entries); err != nil { + // Write an index. + if err := writeIndex(argIndex, tmplIndex, &entries); err != nil { log.Fatalln(err) } } |