1
0
mirror of https://github.com/danog/gojekyll.git synced 2024-11-27 12:44:54 +01:00
gojekyll/collection.go
2017-06-16 20:19:46 -04:00

105 lines
2.6 KiB
Go

package main
import (
"fmt"
"os"
"path/filepath"
)
// Collection is a Jekyll collection.
type Collection struct {
Site *Site
Name string
Data VariableMap
Output bool
Pages []Page
}
// NewCollection creates a new Collection with defaults d
func NewCollection(s *Site, name string, d VariableMap) *Collection {
return &Collection{
Site: s,
Name: name,
Data: d,
Output: d.Bool("output", false),
}
}
// ReadCollections reads the pages of the collections named in the site configuration.
// It adds each collection's pages to the site map, and creates a template site variable for each collection.
func (s *Site) ReadCollections() error {
for name, d := range s.config.Collections {
c := NewCollection(s, name, d)
s.Collections = append(s.Collections, c)
if c.Output { // TODO always read the pages; just don't build them / include them in routes
if err := c.ReadPages(); err != nil {
return err
}
}
}
return nil
}
// PageTemplateObjects returns an array of page objects, for use as the template variable
// value of the collection.
func (c *Collection) PageTemplateObjects() (d []VariableMap) {
for _, p := range c.Pages {
d = append(d, p.TemplateObject())
}
return
}
// IsPosts returns true if the collection is the special "posts" collection.
func (c *Collection) IsPosts() bool {
return c.Name == "posts"
}
func (c *Collection) PathPrefix() string {
return "_" + c.Name + "/"
}
// Source returns the source directory for pages in the collection.
func (c *Collection) Source() string {
return filepath.Join(c.Site.Source, "_"+c.Name)
}
// ReadPages scans the file system for collection pages, and adds them to c.Pages.
func (c *Collection) ReadPages() error {
basePath := c.Site.Source
defaults := MergeVariableMaps(c.Data, VariableMap{
"collection": c.Name,
})
walkFn := func(name string, info os.FileInfo, err error) error {
if err != nil {
// if the issue is simply that the directory doesn't exist, warn instead of error
if os.IsNotExist(err) {
if !c.IsPosts() {
fmt.Println("Missing directory for collection", c.Name)
}
return nil
}
return err
}
rel, err := filepath.Rel(basePath, name)
switch {
case err != nil:
return err
case info.IsDir():
return nil
}
p, err := ReadPage(c.Site, c, rel, defaults)
switch {
case err != nil:
return err
case p.Static():
fmt.Printf("skipping static file inside collection: %s\n", name)
case p.Published():
c.Site.Paths[p.Permalink()] = p
c.Pages = append(c.Pages, p)
}
return nil
}
return filepath.Walk(c.Source(), walkFn)
}