<?php
/**
 * Invoice Ninja (https://invoiceninja.com).
 *
 * @link https://github.com/invoiceninja/invoiceninja source repository
 *
 * @copyright Copyright (c) 2020. Invoice Ninja LLC (https://invoiceninja.com)
 *
 * @license https://opensource.org/licenses/AAL
 */

namespace App\Utils\Traits;

/**
 * Class UserSettings.
 */
trait UserSettings
{
    /**
     * @param string $entity
     * @return \stdClass
     */
    public function getEntity(string $entity) : \stdClass
    {
        return $this->settings()->{$entity};
    }

    /**
     * @param string $entity
     * @return \stdClass
     */
    public function getColumnVisibility(string $entity) : \stdClass
    {
        return $this->settings()->{class_basename($entity)}->datatable->column_visibility;
    }
}