dnadesign / silverstripe-elemental

src/GraphQL/ReadOneAreaResolver.php

<?php

namespace DNADesign\Elemental\GraphQL;

use DNADesign\Elemental\Models\ElementalArea;
use Exception;
use GraphQL\Type\Definition\ResolveInfo;
use InvalidArgumentException;
use SilverStripe\GraphQL\OperationResolver;

class ReadOneAreaResolver implements OperationResolver
{
    public function resolve($object, array $args, $context, ResolveInfo $info)
    {
        $area = ElementalArea::get()->byID($args['ID']);

        if (!$area) {

$area is of type SilverStripe\ORM\DataObject, thus it always evaluated to true.

            throw new InvalidArgumentException('Could not find elemental area matching ID ' . $args['ID']);
        }

        if (!$area->canView($context['currentUser'])) {
            throw new Exception('Current user cannot view element areas');
        }

        return $area;
    }
}