hugolib: Simplify Prev/Next
authorBjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>
Mon, 12 Mar 2018 21:13:25 +0000 (22:13 +0100)
committerBjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>
Mon, 12 Mar 2018 21:13:25 +0000 (22:13 +0100)
hugolib/pagesPrevNext.go

index bb474c499020ccf21bbed0aee7e8557ce0e07d3d..947a49b85816d2f95312cbf79eeec29f9f6af390 100644 (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]
                }
        }