hugo/commands/list.go

// Copyright 2019 The Hugo Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package commands

import (
	"encoding/csv"
	"os"
	"path/filepath"
	"time"

	"github.com/gohugoio/hugo/hugolib"
	"github.com/gohugoio/hugo/resources/resource"
	"github.com/spf13/cobra"
	jww "github.com/spf13/jwalterweatherman"
)

var _ cmder = (*listCmd)(nil)

type listCmd struct {
	hugoBuilderCommon
	*baseCmd
}

func newListCmd() *listCmd {
	cc := &listCmd{}

	cc.baseCmd = newBaseCmd(&cobra.Command{
		Use:   "list",
		Short: "Listing out various types of content",
		Long: `Listing out various types of content.

List requires a subcommand, e.g. ` + "`hugo list drafts`.",
		RunE: nil,
	})

	cc.cmd.AddCommand(
		&cobra.Command{
			Use:   "drafts",
			Short: "List all drafts",
			Long:  `List all of the drafts in your content directory.`,
			RunE: func(cmd *cobra.Command, args []string) error {
				cfgInit := func(c *commandeer) error {
					c.Set("buildDrafts", true)
					return nil
				}
				c, err := initializeConfig(true, false, &cc.hugoBuilderCommon, cc, cfgInit)
				if err != nil {
					return err
				}

				sites, err := hugolib.NewHugoSites(*c.DepsCfg)

				if err != nil {
					return newSystemError("Error creating sites", err)
				}

				if err := sites.Build(hugolib.BuildCfg{SkipRender: true}); err != nil {
					return newSystemError("Error Processing Source Content", err)
				}

				for _, p := range sites.Pages() {
					if p.IsDraft() {
						jww.FEEDBACK.Println(filepath.Join(p.File().Dir(), p.File().LogicalName()))
					}

				}

				return nil

			},
		},
		&cobra.Command{
			Use:   "future",
			Short: "List all posts dated in the future",
			Long: `List all of the posts in your content directory which will be
posted in the future.`,
			RunE: func(cmd *cobra.Command, args []string) error {
				cfgInit := func(c *commandeer) error {
					c.Set("buildFuture", true)
					return nil
				}
				c, err := initializeConfig(true, false, &cc.hugoBuilderCommon, cc, cfgInit)
				if err != nil {
					return err
				}

				sites, err := hugolib.NewHugoSites(*c.DepsCfg)

				if err != nil {
					return newSystemError("Error creating sites", err)
				}

				if err := sites.Build(hugolib.BuildCfg{SkipRender: true}); err != nil {
					return newSystemError("Error Processing Source Content", err)
				}

				writer := csv.NewWriter(os.Stdout)
				defer writer.Flush()

				for _, p := range sites.Pages() {
					if resource.IsFuture(p) {
						err := writer.Write([]string{filepath.Join(p.File().Dir(), p.File().LogicalName()), p.PublishDate().Format(time.RFC3339)})
						if err != nil {
							return newSystemError("Error writing future posts to stdout", err)
						}
					}
				}

				return nil

			},
		},
		&cobra.Command{
			Use:   "expired",
			Short: "List all posts already expired",
			Long: `List all of the posts in your content directory which has already
expired.`,
			RunE: func(cmd *cobra.Command, args []string) error {
				cfgInit := func(c *commandeer) error {
					c.Set("buildExpired", true)
					return nil
				}
				c, err := initializeConfig(true, false, &cc.hugoBuilderCommon, cc, cfgInit)
				if err != nil {
					return err
				}

				sites, err := hugolib.NewHugoSites(*c.DepsCfg)

				if err != nil {
					return newSystemError("Error creating sites", err)
				}

				if err := sites.Build(hugolib.BuildCfg{SkipRender: true}); err != nil {
					return newSystemError("Error Processing Source Content", err)
				}

				writer := csv.NewWriter(os.Stdout)
				defer writer.Flush()

				for _, p := range sites.Pages() {
					if resource.IsExpired(p) {
						err := writer.Write([]string{filepath.Join(p.File().Dir(), p.File().LogicalName()), p.ExpiryDate().Format(time.RFC3339)})
						if err != nil {
							return newSystemError("Error writing expired posts to stdout", err)
						}

					}
				}

				return nil

			},
		},
	)

	cc.cmd.PersistentFlags().StringVarP(&cc.source, "source", "s", "", "filesystem path to read files relative from")
	cc.cmd.PersistentFlags().SetAnnotation("source", cobra.BashCompSubdirsInDir, []string{})

	return cc
}