1 files changed, 118 insertions, 0 deletions
diff --git a/apps/dav/lib/migration/migrateaddressbooks.php b/apps/dav/lib/migration/migrateaddressbooks.php
new file mode 100644
index 00000000000..3bc255571e6
--- /dev/null
+++ b/apps/dav/lib/migration/migrateaddressbooks.php
@@ -0,0 +1,118 @@
+<?php
+
+namespace OCA\Dav\Migration;
+
+use OCA\DAV\CardDAV\AddressBook;
+use OCA\DAV\CardDAV\CardDavBackend;
+use OCP\IConfig;
+use OCP\IDBConnection;
+use OCP\ILogger;
+use Sabre\CardDAV\Plugin;
+use Symfony\Component\Console\Command\Command;
+use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+
+class MigrateAddressbooks {
+
+	/** @var \OCP\IDBConnection */
+	protected $dbConnection;
+
+	/** @var CardDavBackend */
+	private $backend;
+
+	/**
+	 * @param IDBConnection $dbConnection
+	 * @param IConfig $config
+	 * @param ILogger $logger
+	 */
+	function __construct(IDBConnection $dbConnection,
+						 CardDavBackend $backend
+	) {
+		$this->dbConnection = $dbConnection;
+		$this->backend = $backend;
+	}
+
+	private function verifyPreconditions() {
+		if (!$this->dbConnection->tableExists('contacts_addressbooks')) {
+			throw new \DomainException('Contacts tables are missing. Nothing to do.');
+		}
+	}
+
+	/**
+	 * @param string $user
+	 */
+	public function migrateForUser($user) {
+		// get all addressbooks of that user
+		$query = $this->dbConnection->getQueryBuilder();
+		$books = $query->select()->from('contacts_addressbooks')
+			->where($query->expr()->eq('user', $query->createNamedParameter($user)))
+			->execute()
+			->fetchAll();
+
+		$principal = "principals/users/$user";
+		foreach($books as $book) {
+
+			$knownBooks = $this->backend->getAddressBooksByUri($principal, $book['uri']);
+			if (!is_null($knownBooks)) {
+				continue;
+			}
+
+			$newId = $this->backend->createAddressBook($principal, $book['uri'], [
+				'{DAV:}displayname' => $book['displayname'],
+				'{' . Plugin::NS_CARDDAV . '}addressbook-description' => $book['description']
+			]);
+
+			$this->migrateBook($book['id'], $newId);
+			$this->migrateShares($book['id'], $newId);
+		}
+	}
+
+	public function setup() {
+		$this->verifyPreconditions();
+	}
+
+	/**
+	 * @param int $addressBookId
+	 * @param int $newAddressBookId
+	 */
+	private function migrateBook($addressBookId, $newAddressBookId) {
+		$query = $this->dbConnection->getQueryBuilder();
+		$cards = $query->select()->from('contacts_cards')
+			->where($query->expr()->eq('addressbookid', $query->createNamedParameter($addressBookId)))
+			->execute()
+			->fetchAll();
+
+		foreach ($cards as $card) {
+			$this->backend->createCard($newAddressBookId, $card['uri'], $card['carddata']);
+		}
+	}
+
+	/**
+	 * @param int $addressBookId
+	 * @param int $newAddressBookId
+	 */
+	private function migrateShares($addressBookId, $newAddressBookId) {
+		$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('addressbook')))
+			->andWhere($query->expr()->in('share_type', [ $query->expr()->literal(0), $query->expr()->literal(1)]))
+			->execute()
+			->fetchAll();
+
+		$add = array_map(function($s) {
+			$prefix = 'principal:principals/users/';
+			if ($s['share_type'] === 1) {
+				$prefix = 'principal:principals/groups/';
+			}
+			return [
+				'href' => $prefix . $s['share_with']
+			];
+		}, $shares);
+
+		$newAddressBook = $this->backend->getAddressBookById($newAddressBookId);
+		$book = new AddressBook($this->backend, $newAddressBook);
+		$this->backend->updateShares($book, $add, []);
+	}
+}