149 lines
4.6 KiB
PHP
149 lines
4.6 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
namespace OCA\BetterPhotos\Db;
|
|
|
|
use OCA\BetterPhotos\AppInfo\Application;
|
|
use OCP\Files\File;
|
|
use OCP\IDBConnection;
|
|
|
|
class Util {
|
|
protected IDBConnection $connection;
|
|
|
|
public function __construct(IDBConnection $connection) {
|
|
$this->connection = $connection;
|
|
}
|
|
|
|
public static function getDateTaken($file) {
|
|
// Attempt to read exif data
|
|
if (in_array($file->getMimeType(), Application::IMAGE_MIMES)) {
|
|
$exif = exif_read_data($file->fopen('rb'));
|
|
$dt = $exif['DateTimeOriginal'];
|
|
if ($dt) {
|
|
$dt = \DateTime::createFromFormat('Y:m:d H:i:s', $dt);
|
|
if ($dt) {
|
|
return $dt->getTimestamp();
|
|
}
|
|
}
|
|
}
|
|
|
|
// Fall back to creation time
|
|
$dateTaken = $file->getCreationTime();
|
|
|
|
// Fall back to modification time
|
|
if ($dateTaken == 0) {
|
|
$dateTaken = $file->getMtime();
|
|
}
|
|
return $dateTaken;
|
|
}
|
|
|
|
public function processFile(string $user, File $file): void {
|
|
$mime = $file->getMimeType();
|
|
if (!in_array($mime, Application::IMAGE_MIMES) && !in_array($mime, Application::VIDEO_MIMES)) {
|
|
return;
|
|
}
|
|
|
|
// Get parameters
|
|
$fileId = $file->getId();
|
|
$dateTaken = $this->getDateTaken($file);
|
|
$dayId = floor($dateTaken / 86400);
|
|
|
|
// Get existing entry
|
|
$sql = 'SELECT * FROM oc_betterphotos WHERE
|
|
user_id = ? AND file_id = ?';
|
|
$res = $this->connection->executeQuery($sql, [
|
|
$user, $fileId,
|
|
]);
|
|
$erow = $res->fetch();
|
|
$exists = (bool)$erow;
|
|
|
|
// Insert or update file
|
|
if ($exists) {
|
|
$sql = 'UPDATE oc_betterphotos SET
|
|
day_id = ?, date_taken = ?
|
|
WHERE user_id = ? AND file_id = ?';
|
|
} else {
|
|
$sql = 'INSERT
|
|
INTO oc_betterphotos (day_id, date_taken, user_id, file_id)
|
|
VALUES (?, ?, ?, ?)';
|
|
}
|
|
$res = $this->connection->executeStatement($sql, [
|
|
$dayId, $dateTaken, $user, $fileId,
|
|
]);
|
|
|
|
// Change of day
|
|
$dayChange = ($exists && intval($erow['day_id']) != $dayId);
|
|
|
|
// Update day table
|
|
if (!$exists || $dayChange) {
|
|
$sql = 'INSERT
|
|
INTO oc_betterphotos_day (user_id, day_id, count)
|
|
VALUES (?, ?, 1)
|
|
ON DUPLICATE KEY
|
|
UPDATE count = count + 1';
|
|
$this->connection->executeStatement($sql, [
|
|
$user, $dayId,
|
|
]);
|
|
|
|
if ($dayChange) {
|
|
$sql = 'UPDATE oc_betterphotos_day SET
|
|
count = count - 1
|
|
WHERE user_id = ? AND day_id = ?';
|
|
$this->connection->executeStatement($sql, [
|
|
$user, $erow['day_id'],
|
|
], [
|
|
\PDO::PARAM_STR, \PDO::PARAM_INT,
|
|
]);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function deleteFile(File $file) {
|
|
$sql = 'DELETE
|
|
FROM oc_betterphotos
|
|
WHERE file_id = ?
|
|
RETURNING *';
|
|
$res = $this->connection->executeQuery($sql, [$file->getId()], [\PDO::PARAM_INT]);
|
|
$rows = $res->fetchAll();
|
|
|
|
foreach ($rows as $row) {
|
|
$dayId = $row['day_id'];
|
|
$userId = $row['user_id'];
|
|
$sql = 'UPDATE oc_betterphotos_day
|
|
SET count = count - 1
|
|
WHERE user_id = ? AND day_id = ?';
|
|
$this->connection->executeStatement($sql, [$userId, $dayId], [
|
|
\PDO::PARAM_STR, \PDO::PARAM_INT,
|
|
]);
|
|
}
|
|
}
|
|
|
|
public function getDays(
|
|
string $user,
|
|
): array {
|
|
$qb = $this->connection->getQueryBuilder();
|
|
$qb->select('day_id', 'count')
|
|
->from('betterphotos_day')
|
|
->where($qb->expr()->eq('user_id', $qb->createNamedParameter($user)))
|
|
->orderBy('day_id', 'DESC');
|
|
$result = $qb->executeQuery();
|
|
$rows = $result->fetchAll();
|
|
return $rows;
|
|
}
|
|
|
|
public function getDay(
|
|
string $user,
|
|
int $dayId,
|
|
): array {
|
|
$sql = 'SELECT file_id, oc_filecache.etag
|
|
FROM oc_betterphotos
|
|
LEFT JOIN oc_filecache
|
|
ON oc_filecache.fileid = oc_betterphotos.file_id
|
|
WHERE user_id = ? AND day_id = ?
|
|
ORDER BY date_taken DESC';
|
|
$rows = $this->connection->executeQuery($sql, [$user, $dayId], [
|
|
\PDO::PARAM_STR, \PDO::PARAM_INT,
|
|
]);
|
|
return $rows->fetchAll();
|
|
}
|
|
} |