Skip to content

Commit

Permalink
Make GroupByParamDate work with string params
Browse files Browse the repository at this point in the history
Fixes #3983
  • Loading branch information
bep committed Jun 19, 2020
1 parent 82abca3 commit 6ff435a
Show file tree
Hide file tree
Showing 2 changed files with 47 additions and 8 deletions.
35 changes: 27 additions & 8 deletions resources/page/pagegroup.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ import (
"strings"
"time"

"github.com/spf13/cast"

"github.com/gohugoio/hugo/common/collections"
"github.com/gohugoio/hugo/compare"

Expand Down Expand Up @@ -225,6 +227,10 @@ func (p Pages) groupByDateField(sorter func(p Pages) Pages, formatter func(p Pag
sp = sp.Reverse()
}

if sp == nil {
return nil, nil
}

date := formatter(sp[0].(Page))
var r []PageGroup
r = append(r, PageGroup{Key: date, Pages: make(Pages, 0)})
Expand Down Expand Up @@ -303,23 +309,36 @@ func (p Pages) GroupByLastmod(format string, order ...string) (PagesGroup, error
// Valid values for order is asc, desc, rev and reverse.
// For valid format strings, see https://golang.org/pkg/time/#Time.Format
func (p Pages) GroupByParamDate(key string, format string, order ...string) (PagesGroup, error) {
sorter := func(p Pages) Pages {
// Cache the dates.
dates := make(map[Page]time.Time)

sorter := func(pages Pages) Pages {
var r Pages
for _, e := range p {
param := resource.GetParamToLower(e, key)
if _, ok := param.(time.Time); ok {
r = append(r, e)

for _, p := range pages {
param := resource.GetParamToLower(p, key)
var t time.Time

if param != nil {
var ok bool
if t, ok = param.(time.Time); !ok {
// Probably a string. Try to convert it to time.Time.
t = cast.ToTime(param)
}
}

dates[p] = t
r = append(r, p)
}

pdate := func(p1, p2 Page) bool {
p1p, p2p := p1.(Page), p2.(Page)
return resource.GetParamToLower(p1p, key).(time.Time).Unix() < resource.GetParamToLower(p2p, key).(time.Time).Unix()
return dates[p1].Unix() < dates[p2].Unix()
}
pageBy(pdate).Sort(r)
return r
}
formatter := func(p Page) string {
return resource.GetParamToLower(p, key).(time.Time).Format(format)
return dates[p].Format(format)
}
return p.groupByDateField(sorter, formatter, order...)
}
Expand Down
20 changes: 20 additions & 0 deletions resources/page/pagegroup_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@ func preparePageGroupTestPages(t *testing.T) Pages {
p.lastMod = cast.ToTime(src.date).AddDate(3, 0, 0)
p.params["custom_param"] = src.param
p.params["custom_date"] = cast.ToTime(src.date)
p.params["custom_string_date"] = src.date
pages = append(pages, p)
}
return pages
Expand Down Expand Up @@ -379,6 +380,25 @@ func TestGroupByParamDate(t *testing.T) {
}
}

// https://github.com/gohugoio/hugo/issues/3983
func TestGroupByParamDateWithStringParams(t *testing.T) {
t.Parallel()
pages := preparePageGroupTestPages(t)
expect := PagesGroup{
{Key: "2012-04", Pages: Pages{pages[4], pages[2], pages[0]}},
{Key: "2012-03", Pages: Pages{pages[3]}},
{Key: "2012-01", Pages: Pages{pages[1]}},
}

groups, err := pages.GroupByParamDate("custom_string_date", "2006-01")
if err != nil {
t.Fatalf("Unable to make PagesGroup array: %s", err)
}
if !reflect.DeepEqual(groups, expect) {
t.Errorf("PagesGroup has unexpected groups. It should be %#v, got %#v", expect, groups)
}
}

func TestGroupByLastmod(t *testing.T) {
t.Parallel()
pages := preparePageGroupTestPages(t)
Expand Down

0 comments on commit 6ff435a

Please sign in to comment.