hugolib: Simplify Prev/Next

This commit is contained in:
Bjørn Erik Pedersen 2018-03-12 22:13:25 +01:00
parent 0dbf79c2f8
commit 79dd7cb31a
No known key found for this signature in database
GPG key ID: 330E6E2BD4859D8F

View file

@ -16,8 +16,9 @@ package hugolib
// Prev returns the previous page reletive to the given page.
func (p Pages) Prev(cur *Page) *Page {
for x, c := range p {
if c.UniqueID() == cur.UniqueID() {
if c.Eq(cur) {
if x == 0 {
// TODO(bep) consider return nil here to get it line with the other Prevs
return p[len(p)-1]
}
return p[x-1]
@ -29,10 +30,11 @@ func (p Pages) Prev(cur *Page) *Page {
// Next returns the next page reletive to the given page.
func (p Pages) Next(cur *Page) *Page {
for x, c := range p {
if c.UniqueID() == cur.UniqueID() {
if c.Eq(cur) {
if x < len(p)-1 {
return p[x+1]
}
// TODO(bep) consider return nil here to get it line with the other Nexts
return p[0]
}
}