src/ubuntu/file-transfer.cpp - cordova-plugin-file-transfer - Git at Google

 /*
  *
  * Copyright 2013 Canonical Ltd.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
  *
 */

 #include "file-transfer.h"
 #include <cassert>

 void FileTransfer::download(int scId, int ecId, const QString& url, const QString &target, bool /*trustAllHost*/, int id, const QVariantMap &/*headers*/) {
     QSharedPointer<FileTransferRequest> request(new FileTransferRequest(_manager, scId, ecId, id, this));

     assert(_id2request.find(id) == _id2request.end());

     _id2request.insert(id, request);

     request->connect(request.data(), &FileTransferRequest::done, [&]() {
         auto it = _id2request.find(id);
         while (it != _id2request.end() && it.key() == id) {
             if (it.value().data() == request.data()) {
                 _id2request.erase(it);
                 break;
             }
             it++;
         }
     });
     request->download(url, target);
 }

 void FileTransfer::upload(int scId, int ecId, const QString &filePath, const QString& url, const QString& fileKey, const QString& fileName, const QString& mimeType,
                           const QVariantMap & params, bool /*trustAllHosts*/, bool /*chunkedMode*/, const QVariantMap &headers, int id, const QString &httpMethod) {
     QSharedPointer<FileTransferRequest> request(new FileTransferRequest(_manager, scId, ecId, id, this));

     assert(_id2request.find(id) == _id2request.end());

     _id2request.insert(id, request);

     request->connect(request.data(), &FileTransferRequest::done, [&]() {
         auto it = _id2request.find(id);
         while (it != _id2request.end() && it.key() == id) {
             if (it.value().data() == request.data()) {
                 _id2request.erase(it);
                 break;
             }
             it++;
         }
     });
     request->upload(url, filePath, fileKey, fileName, mimeType, params, headers);
 }

 void FileTransfer::abort(int scId, int ecId, int id) {
     Q_UNUSED(scId)
     Q_UNUSED(ecId)

     auto it = _id2request.find(id);
     while (it != _id2request.end() && it.key() == id) {
         (*it)->abort();
         it++;
     }
 }

 void FileTransferRequest::download(const QString& uri, const QString &target) {
     QUrl url(uri);
     QNetworkRequest request;

     if (!url.isValid()) {
         QVariantMap map;
         map.insert("code", INVALID_URL_ERR);
         map.insert("source", uri);
         map.insert("target", target);
         _plugin->cb(_ecId, map);
         emit done();
         return;
     }

     request.setUrl(url);
     if (url.password().size() || url.userName().size()) {
         QString headerData = "Basic " + (url.userName() + ":" + url.password()).toLocal8Bit().toBase64();
         request.setRawHeader("Authorization", headerData.toLocal8Bit());
     }
     _reply = QSharedPointer<QNetworkReply>(_manager.get(request));

     _reply->connect(_reply.data(), &QNetworkReply::finished, [this, target, uri]() {
         if (!_scId || _reply->error() != QNetworkReply::NoError)
             return;

         QFile res(target);
         qCritical() << target;
         if (target[0] != '/' || !res.open(QIODevice::WriteOnly)) {
             QVariantMap map;
             map.insert("code", INVALID_URL_ERR);
             map.insert("source", uri);
             map.insert("target", target);
             _plugin->cb(_ecId, map);
             emit done();
             return;
         }
         res.write(_reply->readAll());

         QFileInfo info(target);
         QVariantMap map;
         map.insert("isFile", true);
         map.insert("isDirectory", false);
         map.insert("name", info.fileName());
         map.insert("fullPath", info.absoluteFilePath());

         _plugin->cb(_scId, map);

         emit done();
     });
     _reply->connect(_reply.data(), SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(error(QNetworkReply::NetworkError)));
     _reply->connect(_reply.data(), SIGNAL(downloadProgress(qint64, qint64)), this, SLOT(progress(qint64, qint64)));
 }

 void FileTransferRequest::upload(const QString& _url, const QString& filePath, QString fileKey, QString fileName, QString mimeType, const QVariantMap &params, const QVariantMap &headers) {
     QUrl url(_url);
     QNetworkRequest request;

     if (!url.isValid()) {
         QVariantMap map;
         map.insert("code", INVALID_URL_ERR);
         map.insert("source", filePath);
         map.insert("target", _url);
         _plugin->cb(_ecId, map);
         emit done();
         return;
     }

     QFile file(filePath);
     if (filePath[0] != '/' || !file.open(QIODevice::ReadOnly)) {
         QVariantMap map;
         map.insert("code", FILE_NOT_FOUND_ERR);
         map.insert("source", filePath);
         map.insert("target", _url);
         _plugin->cb(_ecId, map);
         emit done();
         return;
     }
     QString content{file.readAll()};

     request.setUrl(url);
     if (url.password().size() || url.userName().size()) {
         QString headerData = "Basic " + (url.userName() + ":" + url.password()).toLocal8Bit().toBase64();
         request.setRawHeader("Authorization", headerData.toLocal8Bit());
     }

     for (const QString &key: headers.keys()) {
         const QString &value = headers.find(key)->toString();
         request.setRawHeader(key.toUtf8(), value.toUtf8());
     }

     QString boundary = QString("CORDOVA-QT-%1A").arg(qrand());
     while (content.contains(boundary)) {
         boundary += QString("B%1A").arg(qrand());
     }

     request.setHeader(QNetworkRequest::ContentTypeHeader, QString("multipart/form-data; boundary=") + boundary);

     fileKey.replace("\"", "");
     fileName.replace("\"", "");
     mimeType.replace("\"", "");
     QString part = "--" + boundary + "\r\n";

     part += "Content-Disposition: form-data; name=\"" + fileKey +"\"; filename=\"" + fileName + "\"\r\n";
     part += "Content-Type: " + mimeType + "\r\n\r\n";
     part += content + "\r\n";

     for (QString key: params.keys()) {
         part += "--" + boundary + "\r\n";
         part += "Content-Disposition: form-data; name=\"" +  key + "\";\r\n\r\n";
         part += params.find(key)->toString();
         part += "\r\n";
     }

     part += QString("--") + boundary + "--" + "\r\n";

     _reply = QSharedPointer<QNetworkReply>(_manager.post(request, QByteArray(part.toUtf8())));

     _reply->connect(_reply.data(), &QNetworkReply::finished, [this, content]() {
         if (_reply->error() != QNetworkReply::NoError)
             return;
         int status = 200;
         QVariant statusCode = _reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);

         if (statusCode.isValid()) {
             status = statusCode.toInt();
         }

         QVariantMap map;
         map.insert("responseCode", status);
         map.insert("response", QString(_reply->readAll()));
         map.insert("bytesSent", content.size());
         _plugin->cb(_scId, map);
         emit done();
     });
     _reply->connect(_reply.data(), SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(error(QNetworkReply::NetworkError)));
     _reply->connect(_reply.data(), SIGNAL(uploadProgress(qint64, qint64)), this, SLOT(progress(qint64, qint64)));
 }

 void FileTransferRequest::abort() {
     QVariantMap map;
     map.insert("code", ABORT_ERR);
     _plugin->cb(_ecId, map);
     _scId = 0;
     emit done();
 }

 void FileTransferRequest::error(QNetworkReply::NetworkError code) {
     Q_UNUSED(code);

     int status = 404;
     QVariant statusCode = _reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);

     if (statusCode.isValid()) {
         status = statusCode.toInt();
     }

     QVariantMap map;
     map.insert("http_status", status);
     map.insert("body", QString(_reply->readAll()));
     map.insert("code", CONNECTION_ERR);
     _plugin->cb(_ecId, map);
     emit done();
 }

 void FileTransferRequest::progress(qint64 bytesReceived, qint64 bytesTotal) {
     QVariantMap map;
     map.insert("lengthComputable", true);
     map.insert("total", bytesTotal);
     map.insert("loaded", bytesReceived);

     if (bytesReceived && bytesTotal && _scId)
         _plugin->callbackWithoutRemove(_scId, CordovaInternal::format(map));
 }
	/*
	*
	* Copyright 2013 Canonical Ltd.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing,
	* software distributed under the License is distributed on an
	* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
	* KIND, either express or implied. See the License for the
	* specific language governing permissions and limitations
	* under the License.
	*
	*/

	#include "file-transfer.h"
	#include <cassert>

	void FileTransfer::download(int scId, int ecId, const QString& url, const QString &target, bool /trustAllHost/, int id, const QVariantMap &/headers/) {
	QSharedPointer<FileTransferRequest> request(new FileTransferRequest(_manager, scId, ecId, id, this));

	assert(_id2request.find(id) == _id2request.end());

	_id2request.insert(id, request);

	request->connect(request.data(), &FileTransferRequest::done, [&]() {
	auto it = _id2request.find(id);
	while (it != _id2request.end() && it.key() == id) {
	if (it.value().data() == request.data()) {
	_id2request.erase(it);
	break;
	}
	it++;
	}
	});
	request->download(url, target);
	}

	void FileTransfer::upload(int scId, int ecId, const QString &filePath, const QString& url, const QString& fileKey, const QString& fileName, const QString& mimeType,
	const QVariantMap & params, bool /trustAllHosts/, bool /chunkedMode/, const QVariantMap &headers, int id, const QString &httpMethod) {
	QSharedPointer<FileTransferRequest> request(new FileTransferRequest(_manager, scId, ecId, id, this));

	assert(_id2request.find(id) == _id2request.end());

	_id2request.insert(id, request);

	request->connect(request.data(), &FileTransferRequest::done, [&]() {
	auto it = _id2request.find(id);
	while (it != _id2request.end() && it.key() == id) {
	if (it.value().data() == request.data()) {
	_id2request.erase(it);
	break;
	}
	it++;
	}
	});
	request->upload(url, filePath, fileKey, fileName, mimeType, params, headers);
	}

	void FileTransfer::abort(int scId, int ecId, int id) {
	Q_UNUSED(scId)
	Q_UNUSED(ecId)

	auto it = _id2request.find(id);
	while (it != _id2request.end() && it.key() == id) {
	(*it)->abort();
	it++;
	}
	}

	void FileTransferRequest::download(const QString& uri, const QString &target) {
	QUrl url(uri);
	QNetworkRequest request;

	if (!url.isValid()) {
	QVariantMap map;
	map.insert("code", INVALID_URL_ERR);
	map.insert("source", uri);
	map.insert("target", target);
	_plugin->cb(_ecId, map);
	emit done();
	return;
	}

	request.setUrl(url);
	if (url.password().size() \|\| url.userName().size()) {
	QString headerData = "Basic " + (url.userName() + ":" + url.password()).toLocal8Bit().toBase64();
	request.setRawHeader("Authorization", headerData.toLocal8Bit());
	}
	_reply = QSharedPointer<QNetworkReply>(_manager.get(request));

	_reply->connect(_reply.data(), &QNetworkReply::finished, [this, target, uri]() {
	if (!_scId \|\| _reply->error() != QNetworkReply::NoError)
	return;

	QFile res(target);
	qCritical() << target;
	if (target[0] != '/' \|\| !res.open(QIODevice::WriteOnly)) {
	QVariantMap map;
	map.insert("code", INVALID_URL_ERR);
	map.insert("source", uri);
	map.insert("target", target);
	_plugin->cb(_ecId, map);
	emit done();
	return;
	}
	res.write(_reply->readAll());

	QFileInfo info(target);
	QVariantMap map;
	map.insert("isFile", true);
	map.insert("isDirectory", false);
	map.insert("name", info.fileName());
	map.insert("fullPath", info.absoluteFilePath());

	_plugin->cb(_scId, map);

	emit done();
	});
	_reply->connect(_reply.data(), SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(error(QNetworkReply::NetworkError)));
	_reply->connect(_reply.data(), SIGNAL(downloadProgress(qint64, qint64)), this, SLOT(progress(qint64, qint64)));
	}

	void FileTransferRequest::upload(const QString& _url, const QString& filePath, QString fileKey, QString fileName, QString mimeType, const QVariantMap &params, const QVariantMap &headers) {
	QUrl url(_url);
	QNetworkRequest request;

	if (!url.isValid()) {
	QVariantMap map;
	map.insert("code", INVALID_URL_ERR);
	map.insert("source", filePath);
	map.insert("target", _url);
	_plugin->cb(_ecId, map);
	emit done();
	return;
	}

	QFile file(filePath);
	if (filePath[0] != '/' \|\| !file.open(QIODevice::ReadOnly)) {
	QVariantMap map;
	map.insert("code", FILE_NOT_FOUND_ERR);
	map.insert("source", filePath);
	map.insert("target", _url);
	_plugin->cb(_ecId, map);
	emit done();
	return;
	}
	QString content{file.readAll()};

	request.setUrl(url);
	if (url.password().size() \|\| url.userName().size()) {
	QString headerData = "Basic " + (url.userName() + ":" + url.password()).toLocal8Bit().toBase64();
	request.setRawHeader("Authorization", headerData.toLocal8Bit());
	}

	for (const QString &key: headers.keys()) {
	const QString &value = headers.find(key)->toString();
	request.setRawHeader(key.toUtf8(), value.toUtf8());
	}

	QString boundary = QString("CORDOVA-QT-%1A").arg(qrand());
	while (content.contains(boundary)) {
	boundary += QString("B%1A").arg(qrand());
	}

	request.setHeader(QNetworkRequest::ContentTypeHeader, QString("multipart/form-data; boundary=") + boundary);

	fileKey.replace("\"", "");
	fileName.replace("\"", "");
	mimeType.replace("\"", "");
	QString part = "--" + boundary + "\r\n";

	part += "Content-Disposition: form-data; name=\"" + fileKey +"\"; filename=\"" + fileName + "\"\r\n";
	part += "Content-Type: " + mimeType + "\r\n\r\n";
	part += content + "\r\n";

	for (QString key: params.keys()) {
	part += "--" + boundary + "\r\n";
	part += "Content-Disposition: form-data; name=\"" + key + "\";\r\n\r\n";
	part += params.find(key)->toString();
	part += "\r\n";
	}

	part += QString("--") + boundary + "--" + "\r\n";

	_reply = QSharedPointer<QNetworkReply>(_manager.post(request, QByteArray(part.toUtf8())));

	_reply->connect(_reply.data(), &QNetworkReply::finished, [this, content]() {
	if (_reply->error() != QNetworkReply::NoError)
	return;
	int status = 200;
	QVariant statusCode = _reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);

	if (statusCode.isValid()) {
	status = statusCode.toInt();
	}

	QVariantMap map;
	map.insert("responseCode", status);
	map.insert("response", QString(_reply->readAll()));
	map.insert("bytesSent", content.size());
	_plugin->cb(_scId, map);
	emit done();
	});
	_reply->connect(_reply.data(), SIGNAL(error(QNetworkReply::NetworkError)), this, SLOT(error(QNetworkReply::NetworkError)));
	_reply->connect(_reply.data(), SIGNAL(uploadProgress(qint64, qint64)), this, SLOT(progress(qint64, qint64)));
	}

	void FileTransferRequest::abort() {
	QVariantMap map;
	map.insert("code", ABORT_ERR);
	_plugin->cb(_ecId, map);
	_scId = 0;
	emit done();
	}

	void FileTransferRequest::error(QNetworkReply::NetworkError code) {
	Q_UNUSED(code);

	int status = 404;
	QVariant statusCode = _reply->attribute(QNetworkRequest::HttpStatusCodeAttribute);

	if (statusCode.isValid()) {
	status = statusCode.toInt();
	}

	QVariantMap map;
	map.insert("http_status", status);
	map.insert("body", QString(_reply->readAll()));
	map.insert("code", CONNECTION_ERR);
	_plugin->cb(_ecId, map);
	emit done();
	}

	void FileTransferRequest::progress(qint64 bytesReceived, qint64 bytesTotal) {
	QVariantMap map;
	map.insert("lengthComputable", true);
	map.insert("total", bytesTotal);
	map.insert("loaded", bytesReceived);

	if (bytesReceived && bytesTotal && _scId)
	_plugin->callbackWithoutRemove(_scId, CordovaInternal::format(map));
	}