1
0
mirror of https://github.com/owncloudarchive/contacts.git synced 2024-12-13 23:49:04 +01:00
OwncloudContactsOfficial/lib/controller/importcontroller.php
2014-04-18 02:00:17 +02:00

316 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,
OCP\ICache;
/**
* 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;
}
}