136 lines · 4.0 KB · C++
RawDownload
////  FileScriptingInterface.cpp//  libraries/script-engine/src////  Created by Elisa Lupin-Jimenez on 6/28/16.//  Copyright 2016 High Fidelity, Inc.////  Distributed under the Apache License, Version 2.0.//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html// #include <QTemporaryDir>#include <QDir>#include <QFile>#include <QDebug>#include <QBuffer>#include <QTextCodec>#include <QIODevice>#include <QUrl>#include <QByteArray>#include <QString>#include <QFileInfo>#include <quazip5/quazip.h>#include <quazip5/JlCompress.h> #include "FileScriptingInterface.h"#include "ResourceManager.h"#include "ScriptEngineLogging.h"  FileScriptingInterface::FileScriptingInterface(QObject* parent) : QObject(parent) {    // nothing for now} void FileScriptingInterface::runUnzip(QString path, QUrl url, bool autoAdd) {    qCDebug(scriptengine) << "Url that was downloaded: " + url.toString();    qCDebug(scriptengine) << "Path where download is saved: " + path;    QString fileName = "/" + path.section("/", -1);    QString tempDir = path;    tempDir.remove(fileName);    qCDebug(scriptengine) << "Temporary directory at: " + tempDir;    if (!isTempDir(tempDir)) {        qCDebug(scriptengine) << "Temporary directory mismatch; risk of losing files";        return;    }     QString file = unzipFile(path, tempDir);    QString filename = QUrl::fromLocalFile(file).toString();    if (file != "") {        qCDebug(scriptengine) << "File to upload: " + filename;    } else {        qCDebug(scriptengine) << "Unzip failed";    }    emit unzipResult(path, filename, autoAdd);} // fix to check that we are only referring to a temporary directorybool FileScriptingInterface::isTempDir(QString tempDir) {    QString folderName = "/" + tempDir.section("/", -1);    QString tempContainer = tempDir;    tempContainer.remove(folderName);    QTemporaryDir test;    QString testDir = test.path();    folderName = "/" + testDir.section("/", -1);    QString testContainer = testDir;    testContainer.remove(folderName);    return (testContainer == tempContainer);} QString FileScriptingInterface::getTempDir() {    QTemporaryDir dir;    dir.setAutoRemove(false);    return dir.path();    // do something to delete this temp dir later} QString FileScriptingInterface::convertUrlToPath(QUrl url) {    QString newUrl;    QString oldUrl = url.toString();    newUrl = oldUrl.section("filename=", 1, 1);    qCDebug(scriptengine) << "Filename should be: " + newUrl;    return newUrl;} // this function is not in usevoid FileScriptingInterface::downloadZip(QString path, const QString link) {    QUrl url = QUrl(link);    auto request = ResourceManager::createResourceRequest(nullptr, url);    connect(request, &ResourceRequest::finished, this, [this, path]{        unzipFile(path, ""); // so intellisense isn't mad    });    request->send();} QString FileScriptingInterface::unzipFile(QString path, QString tempDir) {     QDir dir(path);    QString dirName = dir.path();    QString target = tempDir + "/model_repo";    QStringList list = JlCompress::extractDir(dirName, target);     qCDebug(scriptengine) << list;     if (!list.isEmpty()) {        return list.front();    } else {        qCDebug(scriptengine) << "Extraction failed";        return "";    } } // this function is not in usevoid FileScriptingInterface::recursiveFileScan(QFileInfo file, QString* dirName) {    /*if (!file.isDir()) {        qCDebug(scriptengine) << "Regular file logged: " + file.fileName();        return;    }*/    QFileInfoList files;     if (file.fileName().contains(".zip")) {        qCDebug(scriptengine) << "Extracting archive: " + file.fileName();        JlCompress::extractDir(file.fileName());    }    files = file.dir().entryInfoList();     /*if (files.empty()) {        files = JlCompress::getFileList(file.fileName());    }*/     foreach (QFileInfo file, files) {        qCDebug(scriptengine) << "Looking into file: " + file.fileName();        recursiveFileScan(file, dirName);    }    return;}