Add automatic update

Author: tchapi

src/Controller/DAVController.php

@@ -4,9 +4,11 @@
 
 use App\Entity\Principal;
 use App\Entity\User;
+use App\Plugins\BirthdayCalendarPlugin;
 use App\Plugins\DavisIMipPlugin;
 use App\Plugins\PublicAwareDAVACLPlugin;
 use App\Services\BasicAuth;
+use App\Services\BirthdayService;
 use App\Services\IMAPAuth;
 use App\Services\LDAPAuth;
 use Doctrine\ORM\EntityManagerInterface;
@@ -100,6 +102,11 @@ class DAVController extends AbstractController
      */
     protected $mailer;
 
+    /**
+     * @var BirthdayService
+     */
+    protected $birthdayService;
+
     /**
      * Base URI of the server.
      *
@@ -142,7 +149,7 @@ class DAVController extends AbstractController
      */
     protected $server;
 
-    public function __construct(MailerInterface $mailer, BasicAuth $basicAuthBackend, IMAPAuth $IMAPAuthBackend, LDAPAuth $LDAPAuthBackend, UrlGeneratorInterface $router, EntityManagerInterface $entityManager, LoggerInterface $logger, string $publicDir, bool $calDAVEnabled = true, bool $cardDAVEnabled = true, bool $webDAVEnabled = false, bool $publicCalendarsEnabled = true, ?string $inviteAddress = null, ?string $authMethod = null, ?string $authRealm = null, ?string $webdavPublicDir = null, ?string $webdavHomesDir = null, ?string $webdavTmpDir = null)
+    public function __construct(MailerInterface $mailer, BasicAuth $basicAuthBackend, IMAPAuth $IMAPAuthBackend, LDAPAuth $LDAPAuthBackend, UrlGeneratorInterface $router, EntityManagerInterface $entityManager, LoggerInterface $logger, BirthdayService $birthdayService, string $publicDir, bool $calDAVEnabled = true, bool $cardDAVEnabled = true, bool $webDAVEnabled = false, bool $publicCalendarsEnabled = true, ?string $inviteAddress = null, ?string $authMethod = null, ?string $authRealm = null, ?string $webdavPublicDir = null, ?string $webdavHomesDir = null, ?string $webdavTmpDir = null)
     {
         $this->publicDir = $publicDir;
 
@@ -159,6 +166,7 @@ public function __construct(MailerInterface $mailer, BasicAuth $basicAuthBackend
         $this->em = $entityManager;
         $this->logger = $logger;
         $this->mailer = $mailer;
+        $this->birthdayService = $birthdayService;
         $this->baseUri = $router->generate('dav', ['path' => '']);
 
         $this->basicAuthBackend = $basicAuthBackend;
@@ -272,6 +280,10 @@ private function initServer(string $authMethod, string $authRealm = User::DEFAUL
             $this->server->addPlugin(new \Sabre\CardDAV\VCFExportPlugin());
         }
 
+        if ($this->cardDAVEnabled && $this->calDAVEnabled) {
+            $this->server->addPlugin(new BirthdayCalendarPlugin($this->birthdayService));
+        }
+
         // WebDAV plugins
         if ($this->webDAVEnabled && $this->webdavTmpDir && $this->webdavPublicDir) {
             if (!is_dir($this->webdavTmpDir) || !is_dir($this->webdavPublicDir)) {

src/Plugins/BirthdayCalendarPlugin.php

@@ -0,0 +1,90 @@
+<?php
+
+namespace App\Plugins;
+
+use App\Services\BirthdayService;
+use Sabre\CardDAV;
+use Sabre\DAV;
+
+class BirthdayCalendarPlugin extends DAV\ServerPlugin
+{
+    /**
+     * @var BirthdayService
+     */
+    protected $birthdayService;
+
+    /**
+     * @var DAV\Server
+     */
+    protected $server;
+
+    public function __construct(BirthdayService $birthdayService)
+    {
+        $this->birthdayService = $birthdayService;
+    }
+
+    public function initialize(DAV\Server $server)
+    {
+        $this->server = $server;
+
+        // Hook into card creation
+        $server->on('afterCreateFile', [$this, 'afterCardCreate']);
+
+        // Hook into card updates
+        $server->on('afterWriteContent', [$this, 'afterCardUpdate']);
+
+        // Hook into card deletion
+        // Note: The node no longer exists at afterCardDelete so we
+        // use beforeCardDelete for simplicity
+        $server->on('beforeUnbind', [$this, 'beforeCardDelete']);
+    }
+
+    private function resyncCurrentPrincipal()
+    {
+        $authPlugin = $this->server->getPlugin('auth');
+
+        if (!$authPlugin) {
+            return null;
+        }
+
+        $principal = $authPlugin->getCurrentPrincipal();
+
+        if ($principal) {
+            $this->birthdayService->syncPrincipal($principal);
+        }
+    }
+
+    public function afterCardCreate($path, DAV\ICollection $parentNode)
+    {
+        if (!$parentNode instanceof CardDAV\IAddressBook) {
+            return;
+        }
+
+        $principal = $this->resyncCurrentPrincipal();
+    }
+
+    public function afterCardUpdate($path, DAV\IFile $node)
+    {
+        if (!$node instanceof CardDAV\ICard) {
+            return;
+        }
+
+        $principal = $this->resyncCurrentPrincipal();
+    }
+
+    public function beforeCardDelete($path)
+    {
+        $node = $this->server->tree->getNodeForPath($path);
+
+        if (!$node instanceof CardDAV\ICard) {
+            return;
+        }
+
+        $principal = $this->resyncCurrentPrincipal();
+    }
+
+    public function getPluginName(): string
+    {
+        return 'birthday-calendar';
+    }
+}

src/Services/BirthdayService.php

@@ -239,9 +239,9 @@ public function buildDataFromContact(string $cardData): ?VCalendar
         return $vCal;
     }
 
-    public function resetForUser(string $username): void
+    public function resetForPrincipal(string $principal): void
     {
-        $calendarInstance = $this->doctrine->getRepository(CalendarInstance::class)->findOneBy(['principalUri' => Principal::PREFIX.$username, 'uri' => Constants::BIRTHDAY_CALENDAR_URI]);
+        $calendarInstance = $this->doctrine->getRepository(CalendarInstance::class)->findOneBy(['principalUri' => $principal, 'uri' => Constants::BIRTHDAY_CALENDAR_URI]);
 
         if (!$calendarInstance) {
             return; // The user's birthday calendar doesn't exist, no need to purge it
@@ -259,19 +259,24 @@ public function resetForUser(string $username): void
 
     public function syncUser(string $username): void
     {
-        if (!$this->shouldBirthdayCalendarExist(Principal::PREFIX.$username)) {
-            $this->deleteBirthdayCalendar(Principal::PREFIX.$username);
+        $this->syncPrincipal(Principal::PREFIX.$username);
+    }
+
+    public function syncPrincipal(string $principal): void
+    {
+        if (!$this->shouldBirthdayCalendarExist($principal)) {
+            $this->deleteBirthdayCalendar($principal);
 
             return;
         }
 
-        $calendarInstance = $this->ensureBirthdayCalendarExists(Principal::PREFIX.$username);
+        $calendarInstance = $this->ensureBirthdayCalendarExists($principal);
 
         // Reset the calendar
-        $this->resetForUser($username);
+        $this->resetForPrincipal($principal);
 
         // Get all address books that should be included and iterate
-        $addressbooks = $this->doctrine->getRepository(AddressBook::class)->findBy(['principalUri' => Principal::PREFIX.$username, 'includedInBirthdayCalendar' => true]);
+        $addressbooks = $this->doctrine->getRepository(AddressBook::class)->findBy(['principalUri' => $principal, 'includedInBirthdayCalendar' => true]);
         foreach ($addressbooks as $book) {
             $cards = $this->doctrine->getRepository(Card::class)->findByAddressBook($book);