nextcloud/apps/dav/lib/migration/addressbookadapter.php

88 lines
2.4 KiB
PHP

<?php
namespace OCA\Dav\Migration;
use OCP\IDBConnection;
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 AddressBookAdapter {
/** @var \OCP\IDBConnection */
protected $dbConnection;
/** @var string */
private $sourceBookTable;
/** @var string */
private $sourceCardsTable;
/**
* @param IDBConnection $dbConnection
* @param string $sourceBookTable
* @param string $sourceCardsTable
*/
function __construct(IDBConnection $dbConnection,
$sourceBookTable = 'contacts_addressbooks',
$sourceCardsTable = 'contacts_cards') {
$this->dbConnection = $dbConnection;
$this->sourceBookTable = $sourceBookTable;
$this->sourceCardsTable = $sourceCardsTable;
}
/**
* @param string $user
* @param \Closure $callBack
*/
public function foreachBook($user, \Closure $callBack) {
// get all addressbooks of that user
$query = $this->dbConnection->getQueryBuilder();
$stmt = $query->select()->from($this->sourceBookTable)
->where($query->expr()->eq('user', $query->createNamedParameter($user)))
->execute();
while($row = $stmt->fetch()) {
$callBack($row);
}
}
public function setup() {
if (!$this->dbConnection->tableExists($this->sourceBookTable)) {
throw new \DomainException('Contacts tables are missing. Nothing to do.');
}
}
/**
* @param int $addressBookId
* @param \Closure $callBack
*/
public function foreachCard($addressBookId, \Closure $callBack) {
$query = $this->dbConnection->getQueryBuilder();
$stmt = $query->select()->from($this->sourceCardsTable)
->where($query->expr()->eq('addressbookid', $query->createNamedParameter($addressBookId)))
->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('addressbook')))
->andWhere($query->expr()->in('share_type', [ $query->expr()->literal(0), $query->expr()->literal(1)]))
->execute()
->fetchAll();
return $shares;
}
}