gojekyll/commands.go

package main

import (
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"sort"
	"strings"
	"time"

	"github.com/urfave/cli"
	yaml "gopkg.in/yaml.v2"
)

// main sets this
var commandStartTime = time.Now()

func buildCommand(c *cli.Context, site *Site) error {
	printPathSetting("Destination:", site.Destination)
	printSetting("Generating...", "")
	count, err := site.Build()
	if err != nil {
		return err
	}
	elapsed := time.Since(commandStartTime)
	printSetting("", fmt.Sprintf("created %d files in %.2fs.", count, elapsed.Seconds()))
	return nil
}

func serveCommand(c *cli.Context, site *Site) error {
	return server(site)
}

func dataCommand(c *cli.Context, site *Site) error {
	p, err := cliPage(c, site)
	if err != nil {
		return err
	}

	printSetting("Variables:", "")
	// The YAML representation including collections is impractically large for debugging.
	// (Actually it's circular, which the yaml package can't handle.)
	// Neuter it. This destroys it as Liquid data, but that's okay in this context.
	siteData := site.Variables
	for _, c := range site.Collections {
		siteData[c.Name] = fmt.Sprintf("<elided page data for %d items>", len(siteData[c.Name].([]VariableMap)))
	}
	b, _ := yaml.Marshal(p.DebugVariables())
	fmt.Println(string(b))
	return nil
}

func routesCommand(c *cli.Context, site *Site) error {
	printSetting("Routes:", "")
	urls := []string{}
	for u, p := range site.Paths {
		if !(c.Bool("dynamic") && p.Static()) {
			urls = append(urls, u)
		}
	}
	sort.Strings(urls)
	for _, u := range urls {
		fmt.Printf("  %s -> %s\n", u, site.Paths[u].Path())
	}
	return nil
}

func renderCommand(c *cli.Context, site *Site) error {
	page, err := cliPage(c, site)
	if err != nil {
		return err
	}
	printPathSetting("Render:", filepath.Join(site.Source, page.Path()))
	printSetting("URL:", page.Permalink())
	printSetting("Content:", "")
	return page.Write(os.Stdout)
}

// If path starts with /, it's a URL path. Else it's a file path relative
// to the site source directory.
func cliPage(c *cli.Context, site *Site) (page Page, err error) {
	path := "/"
	if c.NArg() > 0 {
		path = c.Args().Get(0)
	}
	if strings.HasPrefix(path, "/") {
		page = site.Paths[path]
		if page == nil {
			err = &os.PathError{Op: "render", Path: path, Err: errors.New("the site does not include a file with this URL path")}
		}
	} else {
		page = site.FindPageByFilePath(path)
		if page == nil {
			err = &os.PathError{Op: "render", Path: path, Err: errors.New("no such file")}
		}
	}
	return
}

// FindPageByFilePath returns a Page or nil, referenced by relative path.
func (s *Site) FindPageByFilePath(path string) Page {
	for _, p := range s.Paths {
		if p.Path() == path {
			return p
		}
	}
	return nil
}

// Load the site specified at destination into the site global, and print the common banner settings.
func loadSite(source, destination string) (*Site, error) {
	site, err := NewSiteFromDirectory(source)
	if err != nil {
		return nil, err
	}
	if destination != "" {
		site.Destination = destination
	}
	if site.ConfigFile != nil {
		printPathSetting(configurationFileLabel, *site.ConfigFile)
	} else {
		printSetting(configurationFileLabel, "none")

	}
	printPathSetting("Source:", site.Source)
	return site, site.ReadFiles()
}

// Given a subcommand function, load the site and then call the subcommand.
func loadSiteAndRun(siteLoader func() (*Site, error), cmd func(*cli.Context, *Site) error) func(*cli.Context) error {
	return func(c *cli.Context) error {
		site, err := siteLoader()
		if err != nil {
			return cli.NewExitError(err, 1)
		}
		if cmd(c, site); err != nil {
			return cli.NewExitError(err, 1)
		}
		return nil
	}
}