<?php

class ImageNavigation extends DataExtension {

    /**
     * @param $direction
     * @return bool|DataObject
     */
    public function NavigationLink($direction){

        switch($direction){
            case 'next':
                $sort = 'Sort:GreaterThan';
                break;
            case 'prev':
                $sort = 'Sort:LessThan';
                break;
            default:
                return false;
        }
        $page = PortfolioPage::get()->filter(array(
            'ParentID' => $this->owner->ParentID,
            $sort => $this->owner->Sort
        ))->sort('Sort ASC')->first();

        return $page;
    }

    /**
     * @return ArrayData|bool
     */
    public function ImageNavigation(){
        if($next = $this->NavigationLink('next')){
            $page = $next;
        }else{
            if($prev = $this->NavigationLink('prev')){
                $page = $prev;
            }else{
                return false;
            }
        }
        return new ArrayData(array(
            'Link' => $page->Link(),
            'MenuTitle' => $page->MenuTitle,
            'Title' => $page->Title,
            'Image' => $page->FeaturedImage()
        ));
    }

}