openvk/Web/Models/Repositories/Albums.php

<?php declare(strict_types=1);
namespace openvk\Web\Models\Repositories;
use openvk\Web\Models\Entities\Album;
use openvk\Web\Models\Entities\Club;
use openvk\Web\Models\Entities\User;
use Nette\Database\Table\ActiveRow;
use Chandler\Database\DatabaseConnection;

class Albums
{
    private $context;
    private $albums;
    
    function __construct()
    {
        $this->context = DatabaseConnection::i()->getContext();
        $this->albums  = $this->context->table("albums");
    }
    
    private function toAlbum(?ActiveRow $ar): ?Album
    {
        return is_null($ar) ? NULL : new Album($ar);
    }
    
    private function getSpecialConditions(int $id, int $type): array
    {
        return [
            "name"  => "[/!\\ DO NOT EDIT: INTERNAL NAME ASSIGNMENT IS ACTIVE]",
            "owner" => $id,
            "special_type" => $type,
        ];
    }
    
    function get(int $id): ?Album
    {
        return $this->toAlbum($this->albums->get($id));
    }
    
    function getUserAlbums(User $user, int $page = 1, ?int $perPage = NULL): \Traversable
    {
        $perPage = $perPage ?? OPENVK_DEFAULT_PER_PAGE;
        $albums  = $this->albums->where("owner", $user->getId())->where("special_type", 0)->where("deleted", false);
        foreach($albums->page($page, $perPage) as $album)
            yield new Album($album);
    }
    
    function getUserAlbumsCount(User $user): int
    {
        $albums = $this->albums->where("owner", $user->getId())->where("special_type", 0)->where("deleted", false);
        return sizeof($albums);
    }
    
    function getClubAlbums(Club $club, int $page = 1, ?int $perPage = NULL): \Traversable
    {
        $perPage = $perPage ?? OPENVK_DEFAULT_PER_PAGE;
        foreach($this->albums->where("owner", $club->getId() * -1)->where("deleted", false)->page($page, $perPage) as $album)
            yield new Album($album);
    }
    
    function getClubAlbumsCount(Club $club): int
    {
        return sizeof($this->albums->where("owner", $club->getId() * -1)->where("deleted", false));
    }
    
    function getAvatarAlbumById(int $id, int $regTime): Album
    {
        $data  = $this->getSpecialConditions($id, 16);
        $album = $this->albums->where([
            "owner" => $id,
            "special_type" => 16,
        ])->fetch();
        if(!$album) {
            $album = new Album;
            $album->setName("[!!! internal album]");
            $album->setOwner($id);
            $album->setSpecial_Type(16);
            $album->setCreated($regTime);
            $album->save();
            
            return $album;
        }
        
        return new Album($album);
    }
    
    function getUserAvatarAlbum(User $user): Album
    {
        return $this->getAvatarAlbumById($user->getId(), $user->getRegistrationTime()->timestamp());
    }
    
    function getClubAvatarAlbum(Club $club): Album
    {
        return $this->getAvatarAlbumById($club->getId() * -1, 0);
    }
    
    function getUserWallAlbum(User $user): Album
    {
        $data  = $this->getSpecialConditions($user->getId(), 32);
        $album = $this->albums->where([
            "owner" => $user->getId(),
            "special_type" => 32,
        ])->fetch();
        if(!$album) {
            $album = new Album;
            $album->setName("[!!! internal album]");
            $album->setOwner($user->getId());
            $album->setSpecial_Type(32);
            $album->setCreated($user->getRegistrationTime()->timestamp());
            $album->save();
            
            return $album;
        }
        
        return new Album($album);
    }
}