hugolib: Simplify Prev/Next

gohugoio · Mar 12, 2018 · 79dd7cb · 79dd7cb
1 parent 0dbf79c
commit 79dd7cb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hugolib/pagesPrevNext.go b/hugolib/pagesPrevNext.go
@@ -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]
 		}
 	}