305 lines
10 KiB
PHP
305 lines
10 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace OCA\Memories\Db;
|
|
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IDBConnection;
|
|
|
|
trait TimelineQueryAlbums
|
|
{
|
|
protected IDBConnection $connection;
|
|
|
|
/** Transform only for album */
|
|
public function transformAlbumFilter(IQueryBuilder &$query, string $uid, string $albumId)
|
|
{
|
|
// Get album object
|
|
$album = $this->getAlbumIfAllowed($uid, $albumId);
|
|
|
|
// Check permission
|
|
if (null === $album) {
|
|
throw new \Exception("Album {$albumId} not found");
|
|
}
|
|
|
|
// WHERE these are items with this album
|
|
$query->innerJoin('m', 'photos_albums_files', 'paf', $query->expr()->andX(
|
|
$query->expr()->eq('paf.album_id', $query->createNamedParameter($album['album_id'])),
|
|
$query->expr()->eq('paf.file_id', 'm.fileid'),
|
|
));
|
|
}
|
|
|
|
/** Get list of albums */
|
|
public function getAlbums(string $uid, bool $shared = false)
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
|
|
// SELECT everything from albums
|
|
$count = $query->func()->count($query->createFunction('DISTINCT m.fileid'), 'count');
|
|
$query->select('pa.*', $count)->from('photos_albums', 'pa');
|
|
|
|
if ($shared) {
|
|
$ids = $this->getSelfCollaborators($uid);
|
|
$query->innerJoin('pa', $this->collaboratorsTable(), 'pc', $query->expr()->andX(
|
|
$query->expr()->eq('pa.album_id', 'pc.album_id'),
|
|
$query->expr()->in('pc.collaborator_id', $query->createNamedParameter($ids, IQueryBuilder::PARAM_STR_ARRAY)),
|
|
));
|
|
} else {
|
|
$query->where(
|
|
$query->expr()->eq('user', $query->createNamedParameter($uid)),
|
|
);
|
|
}
|
|
|
|
// WHERE these are items with this album
|
|
$query->leftJoin('pa', 'photos_albums_files', 'paf', $query->expr()->andX(
|
|
$query->expr()->eq('paf.album_id', 'pa.album_id'),
|
|
));
|
|
|
|
// WHERE these items are memories indexed photos
|
|
$query->leftJoin('paf', 'memories', 'm', $query->expr()->eq('m.fileid', 'paf.file_id'));
|
|
|
|
// WHERE these photos are in the filecache
|
|
$query->leftJoin('m', 'filecache', 'f', $query->expr()->eq('m.fileid', 'f.fileid'));
|
|
|
|
// GROUP and ORDER by
|
|
$query->groupBy('pa.album_id');
|
|
$query->orderBy('pa.created', 'DESC');
|
|
$query->addOrderBy('pa.album_id', 'DESC'); // tie-breaker
|
|
|
|
// FETCH all albums
|
|
$albums = $query->executeQuery()->fetchAll();
|
|
|
|
// Post process
|
|
foreach ($albums as &$row) {
|
|
$row['album_id'] = (int) $row['album_id'];
|
|
$row['created'] = (int) $row['created'];
|
|
$row['last_added_photo'] = (int) $row['last_added_photo'];
|
|
}
|
|
|
|
return $albums;
|
|
}
|
|
|
|
/**
|
|
* Convert days response to months response.
|
|
* The dayId is used to group the days into months.
|
|
*/
|
|
public function daysToMonths(array &$days)
|
|
{
|
|
$months = [];
|
|
foreach ($days as &$day) {
|
|
$dayId = $day['dayid'];
|
|
$time = $dayId * 86400;
|
|
$monthid = strtotime(date('Ym', $time).'01') / 86400;
|
|
|
|
if (empty($months) || $months[\count($months) - 1]['dayid'] !== $monthid) {
|
|
$months[] = [
|
|
'dayid' => $monthid,
|
|
'count' => 0,
|
|
];
|
|
}
|
|
|
|
$months[\count($months) - 1]['count'] += $day['count'];
|
|
}
|
|
|
|
return $months;
|
|
}
|
|
|
|
/** Convert list of month IDs to list of dayIds */
|
|
public function monthIdToDayIds(int $monthId)
|
|
{
|
|
$dayIds = [];
|
|
$firstDay = (int) $monthId;
|
|
$lastDay = strtotime(date('Ymt', $firstDay * 86400)) / 86400;
|
|
for ($i = $firstDay; $i <= $lastDay; ++$i) {
|
|
$dayIds[] = (string) $i;
|
|
}
|
|
|
|
return $dayIds;
|
|
}
|
|
|
|
/**
|
|
* Check if an album has a file.
|
|
*
|
|
* @return bool|string owner of file
|
|
*/
|
|
public function albumHasFile(int $albumId, int $fileId)
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('owner')->from('photos_albums_files')->where(
|
|
$query->expr()->andX(
|
|
$query->expr()->eq('file_id', $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT)),
|
|
$query->expr()->eq('album_id', $query->createNamedParameter($albumId, IQueryBuilder::PARAM_INT)),
|
|
)
|
|
);
|
|
|
|
return $query->executeQuery()->fetchOne();
|
|
}
|
|
|
|
/**
|
|
* Check if a file belongs to a user through an album.
|
|
*
|
|
* @return bool|string owner of file
|
|
*/
|
|
public function albumHasUserFile(string $uid, int $fileId)
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('paf.owner')->from('photos_albums_files', 'paf')->where(
|
|
$query->expr()->andX(
|
|
$query->expr()->eq('paf.file_id', $query->createNamedParameter($fileId, IQueryBuilder::PARAM_INT)),
|
|
$query->expr()->orX(
|
|
$query->expr()->eq('pa.album_id', 'paf.album_id'),
|
|
$query->expr()->eq('pc.album_id', 'paf.album_id'),
|
|
),
|
|
)
|
|
);
|
|
|
|
// Check if user-owned album or shared album
|
|
$query->leftJoin('paf', 'photos_albums', 'pa', $query->expr()->andX(
|
|
$query->expr()->eq('pa.album_id', 'paf.album_id'),
|
|
$query->expr()->eq('pa.user', $query->createNamedParameter($uid)),
|
|
));
|
|
|
|
// Join to shared album
|
|
$ids = $this->getSelfCollaborators($uid);
|
|
$query->leftJoin('paf', $this->collaboratorsTable(), 'pc', $query->expr()->andX(
|
|
$query->expr()->eq('pc.album_id', 'paf.album_id'),
|
|
$query->expr()->in('pc.collaborator_id', $query->createNamedParameter($ids, IQueryBuilder::PARAM_STR_ARRAY)),
|
|
));
|
|
|
|
return $query->executeQuery()->fetchOne();
|
|
}
|
|
|
|
/**
|
|
* Get album if allowed. Also check if album is shared with user.
|
|
*
|
|
* @param string $uid UID of CURRENT user
|
|
* @param string $albumId $user/$name where $user is the OWNER of the album
|
|
*/
|
|
public function getAlbumIfAllowed(string $uid, string $albumId)
|
|
{
|
|
$album = null;
|
|
|
|
// Split name and uid
|
|
$parts = explode('/', $albumId);
|
|
if (2 === \count($parts)) {
|
|
$albumUid = $parts[0];
|
|
$albumName = $parts[1];
|
|
|
|
// Check if owner
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('*')->from('photos_albums')->where(
|
|
$query->expr()->andX(
|
|
$query->expr()->eq('name', $query->createNamedParameter($albumName)),
|
|
$query->expr()->eq('user', $query->createNamedParameter($albumUid)),
|
|
)
|
|
);
|
|
$album = $query->executeQuery()->fetch();
|
|
}
|
|
|
|
// Album not found: it could be a link token at best
|
|
if (!$album) {
|
|
return $this->getAlbumByLink($albumId);
|
|
}
|
|
|
|
// Check if user is owner
|
|
if ($albumUid === $uid) {
|
|
return $album;
|
|
}
|
|
|
|
// Check in collaborators instead
|
|
$albumNumId = (int) $album['album_id'];
|
|
if ($this->userIsAlbumCollaborator($uid, $albumNumId)) {
|
|
return $album;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/**
|
|
* Check if user is a collaborator by numeric ID.
|
|
* Also checks if a group is a collaborator.
|
|
* Does not check if the user is the owner.
|
|
*
|
|
* @param string $uid User ID
|
|
* @param int $albumId Album ID (numeric)
|
|
*/
|
|
public function userIsAlbumCollaborator(string $uid, int $albumId): bool
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
$ids = $this->getSelfCollaborators($uid);
|
|
$query->select('album_id')->from($this->collaboratorsTable())->where(
|
|
$query->expr()->andX(
|
|
$query->expr()->eq('album_id', $query->createNamedParameter($albumId, IQueryBuilder::PARAM_INT)),
|
|
$query->expr()->in('collaborator_id', $query->createNamedParameter($ids, IQueryBuilder::PARAM_STR_ARRAY)),
|
|
)
|
|
);
|
|
|
|
return false !== $query->executeQuery()->fetchOne();
|
|
}
|
|
|
|
/**
|
|
* Get album object by token.
|
|
* Returns false if album link does not exist.
|
|
*/
|
|
public function getAlbumByLink(string $token)
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('*')->from('photos_albums', 'pa')
|
|
->innerJoin('pa', $this->collaboratorsTable(), 'pc', $query->expr()->andX(
|
|
$query->expr()->eq('pc.album_id', 'pa.album_id'),
|
|
$query->expr()->eq('collaborator_id', $query->createNamedParameter($token)),
|
|
$query->expr()->eq('collaborator_type', $query->createNamedParameter(3)), // = TYPE_LINK
|
|
))
|
|
;
|
|
|
|
return $query->executeQuery()->fetch() ?: null;
|
|
}
|
|
|
|
/**
|
|
* Get full list of fileIds in album.
|
|
*/
|
|
public function getAlbumFiles(int $albumId)
|
|
{
|
|
$query = $this->connection->getQueryBuilder();
|
|
$query->select('file_id')->from('photos_albums_files', 'paf')->where(
|
|
$query->expr()->eq('album_id', $query->createNamedParameter($albumId, IQueryBuilder::PARAM_INT))
|
|
);
|
|
$query->innerJoin('paf', 'filecache', 'fc', $query->expr()->eq('fc.fileid', 'paf.file_id'));
|
|
|
|
$fileIds = [];
|
|
$result = $query->executeQuery();
|
|
while ($row = $result->fetch()) {
|
|
$fileIds[] = (int) $row['file_id'];
|
|
}
|
|
|
|
return $fileIds;
|
|
}
|
|
|
|
/** Get list of collaborator ids including user id and groups */
|
|
private function getSelfCollaborators(string $uid)
|
|
{
|
|
// Get groups for the user
|
|
$groupManager = \OC::$server->get(\OCP\IGroupManager::class);
|
|
$user = \OC::$server->get(\OCP\IUserManager::class)->get($uid);
|
|
$groups = $groupManager->getUserGroupIds($user);
|
|
|
|
// And albums shared with user
|
|
$groups[] = $uid;
|
|
|
|
return $groups;
|
|
}
|
|
|
|
/** Get the name of the collaborators table */
|
|
private function collaboratorsTable()
|
|
{
|
|
// https://github.com/nextcloud/photos/commit/20e3e61ad577014e5f092a292c90a8476f630355
|
|
$appManager = \OC::$server->get(\OCP\App\IAppManager::class);
|
|
$photosVersion = $appManager->getAppVersion('photos');
|
|
if (version_compare($photosVersion, '2.0.1', '>=')) {
|
|
return 'photos_albums_collabs';
|
|
}
|
|
|
|
return 'photos_collaborators';
|
|
}
|
|
}
|