Repository\Styles.php

<?php

namespace apathy\UsernameStyles\Repository;

use XF\Mvc\Entity\Finder;
use XF\Mvc\Entity\Repository;

class Styles extends Repository
{
    public function findUsernameStyle()
    {
        $visitor = \XF::visitor();
	$user_id = $visitor['user_id'];

        $finder = $this->finder('apathy\UsernameStyles:Username');
        $finder
            ->where('user_id', '=', $user_id);

        return $finder;
    }

    public function findPostbitStyle()
    {
        $visitor = \XF::visitor();
	$user_id = $visitor['user_id'];

        $finder = $this->finder('apathy\UsernameStyles:Postbit');
        $finder
            ->where('user_id', '=', $user_id);

        return $finder;
    }
}