1 files changed, 83 insertions, 0 deletions
diff --git a/apps/dav/lib/migration/calendaradapter.php b/apps/dav/lib/migration/calendaradapter.php
new file mode 100644
index 00000000000..39faf2cec37
--- /dev/null
+++ b/apps/dav/lib/migration/calendaradapter.php
@@ -0,0 +1,83 @@
+<?php
+
+namespace OCA\Dav\Migration;
+
+use OCP\IDBConnection;
+
+class CalendarAdapter {
+
+	/** @var \OCP\IDBConnection */
+	protected $dbConnection;
+
+	/** @var string */
+	private $sourceCalendarTable;
+
+	/** @var string */
+	private $sourceCalObjTable;
+
+	/**
+	 * @param IDBConnection $dbConnection
+	 * @param string $sourceCalendarTable
+	 * @param string $sourceCalObjTable
+	 */
+	function __construct(IDBConnection $dbConnection,
+						 $sourceCalendarTable = 'clndr_calendars',
+						 $sourceCalObjTable = 'clndr_objects') {
+		$this->dbConnection = $dbConnection;
+		$this->sourceCalendarTable = $sourceCalendarTable;
+		$this->sourceCalObjTable = $sourceCalObjTable;
+	}
+
+	/**
+	 * @param string $user
+	 * @param \Closure $callBack
+	 */
+	public function foreachCalendar($user, \Closure $callBack) {
+		// get all calendars of that user
+		$query = $this->dbConnection->getQueryBuilder();
+		$stmt = $query->select('*')->from($this->sourceCalendarTable)
+			->where($query->expr()->eq('userid', $query->createNamedParameter($user)))
+			->execute();
+
+		while($row = $stmt->fetch()) {
+			$callBack($row);
+		}
+	}
+
+	public function setup() {
+		if (!$this->dbConnection->tableExists($this->sourceCalendarTable)) {
+			throw new \DomainException('Calendar tables are missing. Nothing to do.');
+		}
+	}
+
+	/**
+	 * @param int $calendarId
+	 * @param \Closure $callBack
+	 */
+	public function foreachCalendarObject($calendarId, \Closure $callBack) {
+		$query = $this->dbConnection->getQueryBuilder();
+		$stmt = $query->select('*')->from($this->sourceCalObjTable)
+			->where($query->expr()->eq('calendarid', $query->createNamedParameter($calendarId)))
+			->execute();
+
+		while($row = $stmt->fetch()) {
+			$callBack($row);
+		}
+	}
+
+	/**
+	 * @param int $addressBookId
+	 * @return array
+	 */
+	public function getShares($addressBookId) {
+		$query = $this->dbConnection->getQueryBuilder();
+		$shares = $query->select('*')->from('share')
+			->where($query->expr()->eq('item_source', $query->createNamedParameter($addressBookId)))
+			->andWhere($query->expr()->eq('item_type', $query->expr()->literal('calendar')))
+			->andWhere($query->expr()->in('share_type', [ $query->expr()->literal(0), $query->expr()->literal(1)]))
+			->execute()
+			->fetchAll();
+
+		return $shares;
+	}
+}