memories/lib/Db/Util.php

121 lines
3.7 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, bool $update): 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);
// Insert or update file
// todo: update dateTaken and dayId if needed
$sql = 'INSERT IGNORE
INTO oc_betterphotos (user_id, file_id, date_taken, day_id)
VALUES (?, ?, ?, ?)';
$res = $this->connection->executeStatement($sql, [
$user, $fileId, $dateTaken, $dayId,
]);
// Update day table
if ($res === 1) {
$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,
]);
}
}
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();
}
}