123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427 |
- <?php
- declare(strict_types=1);
- namespace OC\Notification;
- use OC\AppFramework\Bootstrap\Coordinator;
- use OCP\ICache;
- use OCP\ICacheFactory;
- use OCP\IUserManager;
- use OCP\Notification\AlreadyProcessedException;
- use OCP\Notification\IApp;
- use OCP\Notification\IDeferrableApp;
- use OCP\Notification\IDismissableNotifier;
- use OCP\Notification\IManager;
- use OCP\Notification\INotification;
- use OCP\Notification\INotifier;
- use OCP\RichObjectStrings\IValidator;
- use OCP\Support\Subscription\IRegistry;
- use Psr\Container\ContainerExceptionInterface;
- use Psr\Log\LoggerInterface;
- class Manager implements IManager {
-
- protected $validator;
-
- private $userManager;
-
- protected $cache;
-
- protected $subscription;
-
- protected $logger;
-
- private $coordinator;
-
- protected $apps;
-
- protected $appClasses;
-
- protected $notifiers;
-
- protected $notifierClasses;
-
- protected $preparingPushNotification;
-
- protected $deferPushing;
-
- private $parsedRegistrationContext;
- public function __construct(IValidator $validator,
- IUserManager $userManager,
- ICacheFactory $cacheFactory,
- IRegistry $subscription,
- LoggerInterface $logger,
- Coordinator $coordinator) {
- $this->validator = $validator;
- $this->userManager = $userManager;
- $this->cache = $cacheFactory->createDistributed('notifications');
- $this->subscription = $subscription;
- $this->logger = $logger;
- $this->coordinator = $coordinator;
- $this->apps = [];
- $this->notifiers = [];
- $this->appClasses = [];
- $this->notifierClasses = [];
- $this->preparingPushNotification = false;
- $this->deferPushing = false;
- $this->parsedRegistrationContext = false;
- }
-
- public function registerApp(string $appClass): void {
- $this->appClasses[] = $appClass;
- }
-
- public function registerNotifier(\Closure $service, \Closure $info) {
- $infoData = $info();
- $exception = new \InvalidArgumentException(
- 'Notifier ' . $infoData['name'] . ' (id: ' . $infoData['id'] . ') is not considered because it is using the old way to register.'
- );
- $this->logger->error($exception->getMessage(), ['exception' => $exception]);
- }
-
- public function registerNotifierService(string $notifierService): void {
- $this->notifierClasses[] = $notifierService;
- }
-
- protected function getApps(): array {
- if (empty($this->appClasses)) {
- return $this->apps;
- }
- foreach ($this->appClasses as $appClass) {
- try {
- $app = \OC::$server->get($appClass);
- } catch (ContainerExceptionInterface $e) {
- $this->logger->error('Failed to load notification app class: ' . $appClass, [
- 'exception' => $e,
- 'app' => 'notifications',
- ]);
- continue;
- }
- if (!($app instanceof IApp)) {
- $this->logger->error('Notification app class ' . $appClass . ' is not implementing ' . IApp::class, [
- 'app' => 'notifications',
- ]);
- continue;
- }
- $this->apps[] = $app;
- }
- $this->appClasses = [];
- return $this->apps;
- }
-
- public function getNotifiers(): array {
- if (!$this->parsedRegistrationContext) {
- $notifierServices = $this->coordinator->getRegistrationContext()->getNotifierServices();
- foreach ($notifierServices as $notifierService) {
- try {
- $notifier = \OC::$server->get($notifierService->getService());
- } catch (ContainerExceptionInterface $e) {
- $this->logger->error('Failed to load notification notifier class: ' . $notifierService->getService(), [
- 'exception' => $e,
- 'app' => 'notifications',
- ]);
- continue;
- }
- if (!($notifier instanceof INotifier)) {
- $this->logger->error('Notification notifier class ' . $notifierService->getService() . ' is not implementing ' . INotifier::class, [
- 'app' => 'notifications',
- ]);
- continue;
- }
- $this->notifiers[] = $notifier;
- }
- $this->parsedRegistrationContext = true;
- }
- if (empty($this->notifierClasses)) {
- return $this->notifiers;
- }
- foreach ($this->notifierClasses as $notifierClass) {
- try {
- $notifier = \OC::$server->get($notifierClass);
- } catch (ContainerExceptionInterface $e) {
- $this->logger->error('Failed to load notification notifier class: ' . $notifierClass, [
- 'exception' => $e,
- 'app' => 'notifications',
- ]);
- continue;
- }
- if (!($notifier instanceof INotifier)) {
- $this->logger->error('Notification notifier class ' . $notifierClass . ' is not implementing ' . INotifier::class, [
- 'app' => 'notifications',
- ]);
- continue;
- }
- $this->notifiers[] = $notifier;
- }
- $this->notifierClasses = [];
- return $this->notifiers;
- }
-
- public function createNotification(): INotification {
- return new Notification($this->validator);
- }
-
- public function hasNotifiers(): bool {
- return !empty($this->notifiers) || !empty($this->notifierClasses);
- }
-
- public function setPreparingPushNotification(bool $preparingPushNotification): void {
- $this->preparingPushNotification = $preparingPushNotification;
- }
-
- public function isPreparingPushNotification(): bool {
- return $this->preparingPushNotification;
- }
-
- public function defer(): bool {
- $alreadyDeferring = $this->deferPushing;
- $this->deferPushing = true;
- $apps = $this->getApps();
- foreach ($apps as $app) {
- if ($app instanceof IDeferrableApp) {
- $app->defer();
- }
- }
- return !$alreadyDeferring;
- }
-
- public function flush(): void {
- $apps = $this->getApps();
- foreach ($apps as $app) {
- if (!$app instanceof IDeferrableApp) {
- continue;
- }
- try {
- $app->flush();
- } catch (\InvalidArgumentException $e) {
- }
- }
- $this->deferPushing = false;
- }
-
- public function isFairUseOfFreePushService(): bool {
- $pushAllowed = $this->cache->get('push_fair_use');
- if ($pushAllowed === null) {
-
- $isFairUse = $this->subscription->delegateHasValidSubscription() || $this->userManager->countSeenUsers() < 1000;
- $pushAllowed = $isFairUse ? 'yes' : 'no';
- $this->cache->set('push_fair_use', $pushAllowed, 3600);
- }
- return $pushAllowed === 'yes';
- }
-
- public function notify(INotification $notification): void {
- if (!$notification->isValid()) {
- throw new \InvalidArgumentException('The given notification is invalid');
- }
- $apps = $this->getApps();
- foreach ($apps as $app) {
- try {
- $app->notify($notification);
- } catch (\InvalidArgumentException $e) {
- }
- }
- }
-
- public function getID(): string {
- return 'core';
- }
-
- public function getName(): string {
- return 'core';
- }
-
- public function prepare(INotification $notification, string $languageCode): INotification {
- $notifiers = $this->getNotifiers();
- foreach ($notifiers as $notifier) {
- try {
- $notification = $notifier->prepare($notification, $languageCode);
- } catch (\InvalidArgumentException $e) {
- continue;
- } catch (AlreadyProcessedException $e) {
- $this->markProcessed($notification);
- throw new \InvalidArgumentException('The given notification has been processed');
- }
- if (!$notification->isValidParsed()) {
- throw new \InvalidArgumentException('The given notification has not been handled');
- }
- }
- if (!$notification->isValidParsed()) {
- throw new \InvalidArgumentException('The given notification has not been handled');
- }
- return $notification;
- }
-
- public function markProcessed(INotification $notification): void {
- $apps = $this->getApps();
- foreach ($apps as $app) {
- $app->markProcessed($notification);
- }
- }
-
- public function getCount(INotification $notification): int {
- $apps = $this->getApps();
- $count = 0;
- foreach ($apps as $app) {
- $count += $app->getCount($notification);
- }
- return $count;
- }
- public function dismissNotification(INotification $notification): void {
- $notifiers = $this->getNotifiers();
- foreach ($notifiers as $notifier) {
- if ($notifier instanceof IDismissableNotifier) {
- try {
- $notifier->dismissNotification($notification);
- } catch (\InvalidArgumentException $e) {
- continue;
- }
- }
- }
- }
- }
|