From f994a182d37d163a9e2850c4ae399c1ac116ecbc Mon Sep 17 00:00:00 2001
From: Jonathan Treffler <mail@jonathan-treffler.de>
Date: Tue, 23 Jul 2024 02:05:38 +0200
Subject: [PATCH] add entity, mapper and service for subscriptions_webpush

---
 lib/Db/Subscription.php                    |  2 -
 lib/Db/WebPushSubscription.php             | 23 ++++++++
 lib/Db/WebPushSubscriptionMapper.php       | 49 +++++++++++++++++
 lib/Errors/WebPushSubscriptionNotFound.php |  6 ++
 lib/Service/WebPushSubscriptionService.php | 64 ++++++++++++++++++++++
 5 files changed, 142 insertions(+), 2 deletions(-)
 create mode 100644 lib/Db/WebPushSubscription.php
 create mode 100644 lib/Db/WebPushSubscriptionMapper.php
 create mode 100644 lib/Errors/WebPushSubscriptionNotFound.php
 create mode 100644 lib/Service/WebPushSubscriptionService.php

diff --git a/lib/Db/Subscription.php b/lib/Db/Subscription.php
index 7670c66..b136915 100644
--- a/lib/Db/Subscription.php
+++ b/lib/Db/Subscription.php
@@ -10,7 +10,6 @@ class Subscription extends Entity implements JsonSerializable {
 	protected $userId;
 	protected $collectionName;
 	protected $transport;
-	protected $data;
 	protected $creationTimestamp;
 	protected $expirationTimestamp;
 
@@ -25,7 +24,6 @@ class Subscription extends Entity implements JsonSerializable {
 			'userId' => $this->userId,
 			'collectionName' => $this->collectionName,
 			'transport' => $this->transport,
-			'data' => $this->data,
 			'creationTimestamp' => $this->creationTimestamp,
 			'expirationTimestamp' => $this->expirationTimestamp
 		];
diff --git a/lib/Db/WebPushSubscription.php b/lib/Db/WebPushSubscription.php
new file mode 100644
index 0000000..3a818a6
--- /dev/null
+++ b/lib/Db/WebPushSubscription.php
@@ -0,0 +1,23 @@
+<?php
+
+namespace OCA\DavPush\Db;
+
+use JsonSerializable;
+
+use OCP\AppFramework\Db\Entity;
+
+class WebPushSubscription extends Entity implements JsonSerializable {
+	protected $subscriptionId;
+	protected $pushResource;
+
+	public function __construct() {
+		$this->addType('subscriptionId','integer');
+	}
+
+	public function jsonSerialize(): array {
+		return [
+			'subscriptionId' => $this->subscriptionId,
+			'pushResource' => $this->pushResource,
+		];
+	}
+}
\ No newline at end of file
diff --git a/lib/Db/WebPushSubscriptionMapper.php b/lib/Db/WebPushSubscriptionMapper.php
new file mode 100644
index 0000000..be848ad
--- /dev/null
+++ b/lib/Db/WebPushSubscriptionMapper.php
@@ -0,0 +1,49 @@
+<?php
+
+namespace OCA\DavPush\Db;
+
+use OCP\AppFramework\Db\DoesNotExistException;
+use OCP\AppFramework\Db\Entity;
+use OCP\AppFramework\Db\QBMapper;
+use OCP\DB\QueryBuilder\IQueryBuilder;
+use OCP\IDBConnection;
+
+class WebPushSubscriptionMapper extends QBMapper {	
+	public const TABLENAME = 'dav_push_subscriptions_webpush';
+
+	public function __construct(IDBConnection $db) {
+		parent::__construct($db, self::TABLENAME, WebPushSubscription::class);
+	}
+
+	/**
+	 * @param int $subscriptionId
+	 * @return Entity|WebPushSubscription
+	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException
+	 * @throws DoesNotExistException
+	 */
+	public function findBySubscriptionId(int $subscriptionId): Subscription {
+		/* @var $qb IQueryBuilder */
+		$qb = $this->db->getQueryBuilder();
+		$qb->select('*')
+			->from(self::TABLENAME)
+			->where($qb->expr()->eq('subscription_id', $qb->createNamedParameter($subscriptionId, IQueryBuilder::PARAM_INT)));
+		
+		return $this->findEntity($qb);
+	}
+
+    /**
+	 * @param string $pushResource
+	 * @return Entity|WebPushSubscription
+	 * @throws \OCP\AppFramework\Db\MultipleObjectsReturnedException
+	 * @throws DoesNotExistException
+	 */
+	public function findByPushResource(string $pushResource): Subscription {
+		/* @var $qb IQueryBuilder */
+		$qb = $this->db->getQueryBuilder();
+		$qb->select('*')
+			->from(self::TABLENAME)
+			->where($qb->expr()->eq('push_resource', $qb->createNamedParameter($pushResource)));
+		
+		return $this->findEntity($qb);
+	}
+}
\ No newline at end of file
diff --git a/lib/Errors/WebPushSubscriptionNotFound.php b/lib/Errors/WebPushSubscriptionNotFound.php
new file mode 100644
index 0000000..2f7865d
--- /dev/null
+++ b/lib/Errors/WebPushSubscriptionNotFound.php
@@ -0,0 +1,6 @@
+<?php
+
+namespace OCA\DavPush\Errors;
+
+class WebPushSubscriptionNotFound extends NotFoundException {
+}
\ No newline at end of file
diff --git a/lib/Service/WebPushSubscriptionService.php b/lib/Service/WebPushSubscriptionService.php
new file mode 100644
index 0000000..2d84c1d
--- /dev/null
+++ b/lib/Service/WebPushSubscriptionService.php
@@ -0,0 +1,64 @@
+<?php
+
+namespace OCA\DavPush\Service;
+
+use Exception;
+
+use OCP\AppFramework\Db\DoesNotExistException;
+use OCP\AppFramework\Db\MultipleObjectsReturnedException;
+
+use OCA\DavPush\Errors\WebPushSubscriptionNotFound;
+
+use OCA\DavPush\Db\WebPushSubscription;
+use OCA\DavPush\Db\WebPushSubscriptionMapper;
+
+class WebPushSubscriptionService {
+	public function __construct(
+		private WebPushSubscriptionMapper $mapper
+	) {
+	}
+
+	private function handleException(Exception $e): void {
+		if ($e instanceof DoesNotExistException ||
+			$e instanceof MultipleObjectsReturnedException) {
+			throw new WebPushSubscriptionNotFound($e->getMessage());
+		} else {
+			throw $e;
+		}
+	}
+
+    public function findBySubscriptionId(int $subscriptionId): ?WebPushSubscription {
+        try {
+			return $this->mapper->findBySubscriptionId($subscriptionId);
+		} catch (Exception $e) {
+			$this->handleException($e);
+		}
+    }
+
+    public function findByPushResource(string $pushResource): ?WebPushSubscription {
+        try {
+			return $this->mapper->findByPushResource($pushResource);
+		} catch (Exception $e) {
+			$this->handleException($e);
+		}
+    }
+
+	public function create(int $subscriptionId, string $pushResource): WebPushSubscription {
+		$webPushSubscription = new WebPushSubscription();
+		$webPushSubscription->setSubscriptionId($subscriptionId);
+		$webPushSubscription->setPushResource($pushResource);
+		$webPushSubscription = $this->mapper->insert($webPushSubscription);
+
+		return $webPushSubscription;
+	}
+
+	public function deleteBySubscriptionId(int $subscriptionId): ?WebPushSubscription {
+		try {
+			$webPushSubscription = $this->mapper->findBySubscriptionId($subscriptionId);
+			$this->mapper->delete($webPushSubscription);
+			return $webPushSubscription;
+		} catch (Exception $e) {
+			$this->handleException($e);
+		}
+	}
+}
\ No newline at end of file