140 lines
4.1 KiB
PHP
140 lines
4.1 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
/**
|
|
* @copyright Copyright (c) 2021, Joas Schilling <coding@schilljs.com>
|
|
*
|
|
* @author Joas Schilling <coding@schilljs.com>
|
|
*
|
|
* @license AGPL-3.0-or-later
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
namespace OCA\Notifications\Model;
|
|
|
|
use OCP\AppFramework\Db\DoesNotExistException;
|
|
use OCP\AppFramework\Db\MultipleObjectsReturnedException;
|
|
use OCP\AppFramework\Db\QBMapper;
|
|
use OCP\DB\Exception as DBException;
|
|
use OCP\DB\QueryBuilder\IQueryBuilder;
|
|
use OCP\IDBConnection;
|
|
|
|
/**
|
|
* @template-extends QBMapper<Settings>
|
|
*
|
|
* @method Settings mapRowToEntity(array $row)
|
|
* @method Settings findEntity(IQueryBuilder $query)
|
|
* @method Settings[] findEntities(IQueryBuilder $query)
|
|
*/
|
|
class SettingsMapper extends QBMapper {
|
|
public function __construct(IDBConnection $db) {
|
|
parent::__construct($db, 'notifications_settings', Settings::class);
|
|
}
|
|
|
|
/**
|
|
* @param string $userId
|
|
* @return Settings
|
|
* @throws DBException
|
|
* @throws MultipleObjectsReturnedException
|
|
* @throws DoesNotExistException
|
|
*/
|
|
public function getSettingsByUser(string $userId): Settings {
|
|
$query = $this->db->getQueryBuilder();
|
|
|
|
$query->select('*')
|
|
->from($this->getTableName())
|
|
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)));
|
|
|
|
return $this->findEntity($query);
|
|
}
|
|
|
|
/**
|
|
* @param string $userId
|
|
* @throws DBException
|
|
*/
|
|
public function deleteSettingsByUser(string $userId): void {
|
|
$query = $this->db->getQueryBuilder();
|
|
|
|
$query->delete($this->getTableName())
|
|
->where($query->expr()->eq('user_id', $query->createNamedParameter($userId)));
|
|
|
|
$query->executeStatement();
|
|
}
|
|
|
|
public function setBatchSettingForUser(string $userId, int $batchSetting): void {
|
|
try {
|
|
$settings = $this->getSettingsByUser($userId);
|
|
} catch (DoesNotExistException $e) {
|
|
$settings = new Settings();
|
|
$settings->setUserId($userId);
|
|
/** @var Settings $settings */
|
|
$settings = $this->insert($settings);
|
|
}
|
|
|
|
if ($batchSetting === Settings::EMAIL_SEND_WEEKLY) {
|
|
$batchTime = 3600 * 24 * 7;
|
|
} elseif ($batchSetting === Settings::EMAIL_SEND_DAILY) {
|
|
$batchTime = 3600 * 24;
|
|
} elseif ($batchSetting === Settings::EMAIL_SEND_3HOURLY) {
|
|
$batchTime = 3600 * 3;
|
|
} elseif ($batchSetting === Settings::EMAIL_SEND_HOURLY) {
|
|
$batchTime = 3600;
|
|
} else {
|
|
$batchTime = 0; // Off
|
|
}
|
|
|
|
$settings->setBatchTime($batchTime);
|
|
if ($batchTime === 0) {
|
|
// When mails are Off, we don't set a "next send time" so it can be
|
|
// skipped in the background job.
|
|
$settings->setNextSendTime(0);
|
|
} else {
|
|
// This will automatically heal on the first run of the background job.
|
|
// We are just setting it to 1, so it's checked soon in case
|
|
// the time is now shorter and should trigger already.
|
|
$settings->setNextSendTime(1);
|
|
}
|
|
$this->update($settings);
|
|
}
|
|
|
|
/**
|
|
* @param int $limit
|
|
* @return Settings[]
|
|
* @throws DBException
|
|
*/
|
|
public function getUsersByNextSendTime(int $limit): array {
|
|
$query = $this->db->getQueryBuilder();
|
|
|
|
$query->select('*')
|
|
->from($this->getTableName())
|
|
->where($query->expr()->gt('next_send_time', $query->createNamedParameter(0)))
|
|
->orderBy('next_send_time', 'ASC')
|
|
->setMaxResults($limit);
|
|
|
|
return $this->findEntities($query);
|
|
}
|
|
|
|
public function createSettingsFromRow(array $row): Settings {
|
|
return $this->mapRowToEntity([
|
|
'id' => $row['id'],
|
|
'user_id' => (string) $row['user_id'],
|
|
'batch_time' => (int) $row['batch_time'],
|
|
'last_send_id' => (int) $row['last_send_id'],
|
|
'next_send_time' => (int) $row['next_send_time'],
|
|
]);
|
|
}
|
|
}
|