mirror of
https://github.com/owncloudarchive/contacts.git
synced 2024-11-29 11:24:11 +01:00
315 lines
9.4 KiB
PHP
315 lines
9.4 KiB
PHP
<?php
|
|
/**
|
|
* @author Thomas Tanghus
|
|
* @copyright 2013-2014 Thomas Tanghus (thomas@tanghus.net)
|
|
*
|
|
* This file is licensed under the Affero General Public License version 3 or
|
|
* later.
|
|
* See the COPYING-README file.
|
|
*/
|
|
|
|
namespace OCA\Contacts\Controller;
|
|
|
|
use OCA\Contacts\App,
|
|
OCA\Contacts\JSONResponse,
|
|
OCA\Contacts\Controller,
|
|
Sabre\VObject,
|
|
OCA\Contacts\VObject\VCard as MyVCard,
|
|
OCA\Contacts\ImportManager,
|
|
OCP\IRequest;
|
|
|
|
/**
|
|
* Controller importing contacts
|
|
*/
|
|
class ImportController extends Controller {
|
|
|
|
public function __construct($appName, IRequest $request, App $app, ICache $cache) {
|
|
parent::__construct($appName, $request, $app);
|
|
$this->cache = $cache;
|
|
}
|
|
|
|
/**
|
|
* @NoAdminRequired
|
|
*/
|
|
public function upload() {
|
|
$request = $this->request;
|
|
$params = $this->request->urlParams;
|
|
$addressBookId = $params['addressBookId'];
|
|
$format = $params['importType'];
|
|
$response = new JSONResponse();
|
|
|
|
$view = \OCP\Files::getStorage('contacts');
|
|
if(!$view->file_exists('imports')) {
|
|
$view->mkdir('imports');
|
|
}
|
|
|
|
if (!isset($request->files['file'])) {
|
|
$response->bailOut(App::$l10n->t('No file was uploaded. Unknown error'));
|
|
return $response;
|
|
}
|
|
|
|
$file=$request->files['file'];
|
|
|
|
if($file['error'] !== UPLOAD_ERR_OK) {
|
|
$error = $file['error'];
|
|
$errors = array(
|
|
UPLOAD_ERR_OK => App::$l10n->t("There is no error, the file uploaded with success"),
|
|
UPLOAD_ERR_INI_SIZE => App::$l10n->t("The uploaded file exceeds the upload_max_filesize directive in php.ini")
|
|
.ini_get('upload_max_filesize'),
|
|
UPLOAD_ERR_FORM_SIZE => App::$l10n->t("The uploaded file exceeds the MAX_FILE_SIZE directive that was specified in the HTML form"),
|
|
UPLOAD_ERR_PARTIAL => App::$l10n->t("The uploaded file was only partially uploaded"),
|
|
UPLOAD_ERR_NO_FILE => App::$l10n->t("No file was uploaded"),
|
|
UPLOAD_ERR_NO_TMP_DIR => App::$l10n->t('Missing a temporary folder'),
|
|
UPLOAD_ERR_CANT_WRITE => App::$l10n->t('Failed to write to disk'),
|
|
);
|
|
$response->bailOut($errors[$error]);
|
|
return $response;
|
|
}
|
|
|
|
$maxUploadFilesize = \OCP\Util::maxUploadFilesize('/');
|
|
$maxHumanFilesize = \OCP\Util::humanFileSize($maxUploadFilesize);
|
|
|
|
$totalSize = $file['size'];
|
|
if ($maxUploadFilesize >= 0 and $totalSize > $maxUploadFilesize) {
|
|
$response->bailOut(App::$l10n->t('Not enough storage available'));
|
|
return $response;
|
|
}
|
|
|
|
$tmpname = $file['tmp_name'];
|
|
$filename = strtr($file['name'], array('/' => '', "\\" => ''));
|
|
if(is_uploaded_file($tmpname)) {
|
|
if(\OC\Files\Filesystem::isFileBlacklisted($filename)) {
|
|
$response->bailOut(App::$l10n->t('Attempt to upload blacklisted file:') . $filename);
|
|
return $response;
|
|
}
|
|
$content = file_get_contents($tmpname);
|
|
$proxyStatus = \OC_FileProxy::$enabled;
|
|
\OC_FileProxy::$enabled = false;
|
|
if($view->file_put_contents('/imports/'.$filename, $content)) {
|
|
\OC_FileProxy::$enabled = $proxyStatus;
|
|
$progresskey = 'contacts-import-' . rand();
|
|
$response->setParams(
|
|
array(
|
|
'filename'=>$filename,
|
|
'progresskey' => $progresskey,
|
|
'backend' => $params['backend'],
|
|
'addressBookId' => $params['addressBookId'],
|
|
'importType' => $format
|
|
)
|
|
);
|
|
} else {
|
|
\OC_FileProxy::$enabled = $proxyStatus;
|
|
$response->bailOut(App::$l10n->t('Error uploading contacts to storage.'));
|
|
return $response;
|
|
}
|
|
} else {
|
|
$response->bailOut('Temporary file: \''.$tmpname.'\' has gone AWOL?');
|
|
return $response;
|
|
}
|
|
return $response;
|
|
}
|
|
|
|
/**
|
|
* @NoAdminRequired
|
|
*/
|
|
public function prepare() {
|
|
$request = $this->request;
|
|
$params = $this->request->urlParams;
|
|
$addressBookId = $params['addressBookId'];
|
|
$format = $params['importType'];
|
|
$response = new JSONResponse();
|
|
$filename = $request->post['filename'];
|
|
$path = $request->post['path'];
|
|
|
|
$view = \OCP\Files::getStorage('contacts');
|
|
if(!$view->file_exists('imports')) {
|
|
$view->mkdir('imports');
|
|
}
|
|
|
|
$proxyStatus = \OC_FileProxy::$enabled;
|
|
\OC_FileProxy::$enabled = false;
|
|
$content = \OC_Filesystem::file_get_contents($path . '/' . $filename);
|
|
//$content = file_get_contents('oc://' . $path . '/' . $filename);
|
|
if($view->file_put_contents('/imports/' . $filename, $content)) {
|
|
\OC_FileProxy::$enabled = $proxyStatus;
|
|
$progresskey = 'contacts-import-' . rand();
|
|
$response->setParams(
|
|
array(
|
|
'filename'=>$filename,
|
|
'progresskey' => $progresskey,
|
|
'backend' => $params['backend'],
|
|
'addressBookId' => $params['addressBookId'],
|
|
'importType' => $params['importType']
|
|
)
|
|
);
|
|
} else {
|
|
\OC_FileProxy::$enabled = $proxyStatus;
|
|
$response->bailOut(App::$l10n->t('Error moving file to imports folder.'));
|
|
}
|
|
return $response;
|
|
}
|
|
|
|
/**
|
|
* @NoAdminRequired
|
|
*/
|
|
public function start() {
|
|
$request = $this->request;
|
|
$response = new JSONResponse();
|
|
$params = $this->request->urlParams;
|
|
$app = new App(\OCP\User::getUser());
|
|
$addressBookId = $params['addressBookId'];
|
|
$format = $params['importType'];
|
|
|
|
$addressBook = $app->getAddressBook($params['backend'], $addressBookId);
|
|
if(!$addressBook->hasPermission(\OCP\PERMISSION_CREATE)) {
|
|
$response->setStatus('403');
|
|
$response->bailOut(App::$l10n->t('You do not have permissions to import into this address book.'));
|
|
return $response;
|
|
}
|
|
|
|
$filename = isset($request->post['filename']) ? $request->post['filename'] : null;
|
|
$progresskey = isset($request->post['progresskey']) ? $request->post['progresskey'] : null;
|
|
|
|
if(is_null($filename)) {
|
|
$response->bailOut(App::$l10n->t('File name missing from request.'));
|
|
return $response;
|
|
}
|
|
|
|
if(is_null($progresskey)) {
|
|
$response->bailOut(App::$l10n->t('Progress key missing from request.'));
|
|
return $response;
|
|
}
|
|
|
|
$filename = strtr($filename, array('/' => '', "\\" => ''));
|
|
if(\OC\Files\Filesystem::isFileBlacklisted($filename)) {
|
|
$response->bailOut(App::$l10n->t('Attempt to access blacklisted file:') . $filename);
|
|
return $response;
|
|
}
|
|
$view = \OCP\Files::getStorage('contacts');
|
|
$proxyStatus = \OC_FileProxy::$enabled;
|
|
\OC_FileProxy::$enabled = false;
|
|
$file = $view->file_get_contents('/imports/' . $filename);
|
|
\OC_FileProxy::$enabled = $proxyStatus;
|
|
|
|
$writeProgress = function($pct, $total) use ($progresskey) {
|
|
$this->cache->set($progresskey, $pct, 300);
|
|
$this->cache->set($progresskey.'_total', $total, 300);
|
|
};
|
|
|
|
$cleanup = function() use ($view, $filename, $progresskey, $response) {
|
|
if(!$view->unlink('/imports/' . $filename)) {
|
|
$response->debug('Unable to unlink /imports/' . $filename);
|
|
}
|
|
$this->cache->remove($progresskey);
|
|
$this->cache->remove($progresskey.'_total');
|
|
};
|
|
|
|
$importManager = new ImportManager();
|
|
|
|
$formatList = $importManager->getTypes();
|
|
|
|
$found = false;
|
|
$parts = array();
|
|
foreach ($formatList as $formatName => $formatDisplayName) {
|
|
if ($formatName == $format) {
|
|
$parts = $importManager->importFile($view->getLocalFile('/imports/' . $filename), $formatName);
|
|
$found = true;
|
|
}
|
|
}
|
|
|
|
if (!$found) {
|
|
// detect file type
|
|
$mostLikelyName = "";
|
|
$mostLikelyValue = 0;
|
|
$probability = $importManager->detectFileType($view->getLocalFile('/imports/' . $filename));
|
|
foreach ($probability as $probName => $probValue) {
|
|
if ($probValue > $mostLikelyValue) {
|
|
$mostLikelyName = $probName;
|
|
$mostLikelyValue = $probValue;
|
|
}
|
|
}
|
|
|
|
if ($mostLikelyValue > 0) {
|
|
// found one (most likely...)
|
|
$parts = $importManager->importFile($view->getLocalFile('/imports/' . $filename), $mostLikelyName);
|
|
}
|
|
}
|
|
|
|
if ($parts) {
|
|
//import the contacts
|
|
$imported = 0;
|
|
$failed = 0;
|
|
$processed = 0;
|
|
$total = count($parts);
|
|
|
|
foreach($parts as $part) {
|
|
/**
|
|
* TODO
|
|
* - Check if a contact with identical UID exists.
|
|
* - If so, fetch that contact and call $contact->mergeFromVCard($part);
|
|
* - Increment $updated var (not present yet.)
|
|
* - continue
|
|
*/
|
|
try {
|
|
$id = $addressBook->addChild($part);
|
|
if($id) {
|
|
$imported++;
|
|
$favourites = $part->select('X-FAVOURITES');
|
|
foreach ($favourites as $favourite) {
|
|
if ($favourite->getValue() == 'yes') {
|
|
$tagMgr = $this->server->getTagManager()->load('contact');
|
|
$tagMgr->addToFavorites($id);
|
|
}
|
|
}
|
|
} else {
|
|
$failed++;
|
|
}
|
|
} catch (\Exception $e) {
|
|
$response->debug('Error importing vcard: ' . $e->getMessage() . $nl . $part->serialize());
|
|
$failed++;
|
|
}
|
|
$processed++;
|
|
$writeProgress($processed, $total);
|
|
}
|
|
} else {
|
|
$imported = 0;
|
|
$failed = 0;
|
|
$processed = 0;
|
|
$total = 0;
|
|
}
|
|
$cleanup();
|
|
//done the import
|
|
sleep(3); // Give client side a chance to read the progress.
|
|
$response->setParams(
|
|
array(
|
|
'backend' => $params['backend'],
|
|
'addressBookId' => $params['addressBookId'],
|
|
'importType' => $params['importType'],
|
|
'imported' => $imported,
|
|
'count' => $processed,
|
|
'total' => $total,
|
|
'failed' => $failed,
|
|
)
|
|
);
|
|
return $response;
|
|
}
|
|
|
|
/**
|
|
* @NoAdminRequired
|
|
*/
|
|
public function status() {
|
|
$request = $this->request;
|
|
$response = new JSONResponse();
|
|
|
|
$progresskey = isset($request->get['progresskey']) ? $request->get['progresskey'] : null;
|
|
if(is_null($progresskey)) {
|
|
$response->bailOut(App::$l10n->t('Progress key missing from request.'));
|
|
return $response;
|
|
}
|
|
|
|
error_log("progresskey: ".$this->cache->get($progresskey)." total: ".$this->cache->get($progresskey.'_total') );
|
|
$response->setParams(array('progress' => $this->cache->get($progresskey), 'total' => $this->cache->get($progresskey.'_total') ));
|
|
return $response;
|
|
}
|
|
}
|