////  AssetMappingsScriptingInterface.cpp//  libraries/script-engine/src////  Created by Ryan Huffman on 2016-03-09.//  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 "AssetMappingsScriptingInterface.h" #include <QtScript/QScriptEngine>#include <QtCore/QFile> #include <AssetRequest.h>#include <AssetUpload.h>#include <MappingRequest.h>#include <NetworkLogging.h>#include <OffscreenUi.h> AssetMappingsScriptingInterface::AssetMappingsScriptingInterface() {    _proxyModel.setSourceModel(&_assetMappingModel);    _proxyModel.setSortRole(Qt::DisplayRole);    _proxyModel.setDynamicSortFilter(true);    _proxyModel.sort(0);} QString AssetMappingsScriptingInterface::getErrorString(int errorCode) const {    switch (errorCode) {        case MappingRequest::NoError:            return "No error";        case MappingRequest::NotFound:            return "Asset not found";        case MappingRequest::NetworkError:            return "Unable to communicate with Asset Server";        case MappingRequest::PermissionDenied:            return "Permission denied";        case MappingRequest::InvalidPath:            return "Path is invalid";        case MappingRequest::InvalidHash:            return "Hash is invalid";        case MappingRequest::UnknownError:            return "Asset Server internal error";        default:            return QString();    }} void AssetMappingsScriptingInterface::setMapping(QString path, QString hash, QJSValue callback) {    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createSetMappingRequest(path, hash);     connect(request, &SetMappingRequest::finished, this, [this, callback](SetMappingRequest* request) mutable {        if (callback.isCallable()) {            QJSValueList args { uint8_t(request->getError()) };            callback.call(args);        }         request->deleteLater();    });     request->start();} void AssetMappingsScriptingInterface::getMapping(QString path, QJSValue callback) {    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createGetMappingRequest(path);     connect(request, &GetMappingRequest::finished, this, [this, callback](GetMappingRequest* request) mutable {        if (callback.isCallable()) {            QJSValueList args { uint8_t(request->getError()) };            callback.call(args);        }         request->deleteLater();    });     request->start();} void AssetMappingsScriptingInterface::uploadFile(QString path, QString mapping, QJSValue callback) {    auto offscreenUi = DependencyManager::get<OffscreenUi>();    auto result = offscreenUi->inputDialog(OffscreenUi::ICON_NONE, "Enter asset path:", "", mapping);    if (!result.isValid()) {        return;    }    mapping = result.toString();     // Check for override    if (isKnownMapping(mapping)) {        auto message = "The following file already exists:\n" + path + "\nDo you want to override it?";        auto button = offscreenUi->messageBox(OffscreenUi::ICON_QUESTION, "", message,                                              QMessageBox::Yes | QMessageBox::No, QMessageBox::No);        if (button == QMessageBox::No) {            return;        }    }     auto upload = DependencyManager::get<AssetClient>()->createUpload(path);    QObject::connect(upload, &AssetUpload::finished, this, [=](AssetUpload* upload, const QString& hash) mutable {        if (upload->getError() != AssetUpload::NoError) {            if (callback.isCallable()) {                QJSValueList args { uint8_t(upload->getError()) };                callback.call(args);            }        } else {            setMapping(mapping, hash, callback);        }         upload->deleteLater();    });     upload->start();} void AssetMappingsScriptingInterface::deleteMappings(QStringList paths, QJSValue callback) {    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createDeleteMappingsRequest(paths);     connect(request, &DeleteMappingsRequest::finished, this, [this, callback](DeleteMappingsRequest* request) mutable {        if (callback.isCallable()) {            QJSValueList args { uint8_t(request->getError()) };            callback.call(args);        }         request->deleteLater();    });     request->start();} void AssetMappingsScriptingInterface::getAllMappings(QJSValue callback) {    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createGetAllMappingsRequest();     connect(request, &GetAllMappingsRequest::finished, this, [this, callback](GetAllMappingsRequest* request) mutable {        auto mappings = request->getMappings();        auto map = callback.engine()->newObject();         for (auto& kv : mappings ) {            map.setProperty(kv.first, kv.second);        }         if (callback.isCallable()) {            QJSValueList args { uint8_t(request->getError()) };            callback.call(args);        }         request->deleteLater();    });     request->start();} void AssetMappingsScriptingInterface::renameMapping(QString oldPath, QString newPath, QJSValue callback) {    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createRenameMappingRequest(oldPath, newPath);     connect(request, &RenameMappingRequest::finished, this, [this, callback](RenameMappingRequest* request) mutable {        if (callback.isCallable()) {            QJSValueList args { uint8_t(request->getError()) };            callback.call(args);        }         request->deleteLater();    });     request->start();}  AssetMappingItem::AssetMappingItem(const QString& name, const QString& fullPath, bool isFolder) :    name(name),    fullPath(fullPath),    isFolder(isFolder){ } static int assetMappingModelMetatypeId = qRegisterMetaType<AssetMappingModel*>("AssetMappingModel*"); void AssetMappingModel::refresh() {    qDebug() << "Refreshing asset mapping model";    auto assetClient = DependencyManager::get<AssetClient>();    auto request = assetClient->createGetAllMappingsRequest();     connect(request, &GetAllMappingsRequest::finished, this, [this](GetAllMappingsRequest* request) mutable {        if (request->getError() == MappingRequest::NoError) {            auto mappings = request->getMappings();            auto existingPaths = _pathToItemMap.keys();            for (auto& mapping : mappings) {                auto& path = mapping.first;                auto parts = path.split("/");                auto length = parts.length();                 existingPaths.removeOne(mapping.first);                 QString fullPath = "/";                 QStandardItem* lastItem = nullptr;                 // start index at 1 to avoid empty string from leading slash                for (int i = 1; i < length; ++i) {                    fullPath += (i == 1 ? "" : "/") + parts[i];                     auto it = _pathToItemMap.find(fullPath);                    if (it == _pathToItemMap.end()) {                        qDebug() << "prefix not found: " << fullPath;                        auto item = new QStandardItem(parts[i]);                        bool isFolder = i < length - 1;                        item->setData(isFolder ? fullPath + "/" : fullPath, Qt::UserRole);                        item->setData(isFolder, Qt::UserRole + 1);                        item->setData(parts[i], Qt::UserRole + 2);                        item->setData("atp:" + fullPath, Qt::UserRole + 3);                        if (lastItem) {                            lastItem->setChild(lastItem->rowCount(), 0, item);                        } else {                            appendRow(item);                        }                         lastItem = item;                        _pathToItemMap[fullPath] = lastItem;                    }                    else {                        lastItem = it.value();                    }                }                 Q_ASSERT(fullPath == path);            }             // Remove folders from list            auto it = existingPaths.begin();            while (it != existingPaths.end()) {                auto item = _pathToItemMap[*it];                if (item && item->data(Qt::UserRole + 1).toBool()) {                    it = existingPaths.erase(it);                } else {                    ++it;                }            }             for (auto& path : existingPaths) {                Q_ASSERT(_pathToItemMap.contains(path));                qDebug() << "removing existing: " << path;                 auto item = _pathToItemMap[path];                 while (item) {                    // During each iteration, delete item                    QStandardItem* nextItem = nullptr;                     auto parent = item->parent();                    if (parent) {                        parent->removeRow(item->row());                        if (parent->rowCount() > 0) {                            // The parent still contains children, set the nextItem to null so we stop processing                            nextItem = nullptr;                        } else {                            nextItem = parent;                        }                    } else {                        removeRow(item->row());                    }                     _pathToItemMap.remove(path);                    //delete item;                     item = nextItem;                }                //removeitem->index();            }        } else {            emit errorGettingMappings(static_cast<int>(request->getError()));        }    });     request->start();}